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"
15fe883422SPeter Collingbourne #include "CGCUDARuntime.h"
165d865c32SJohn McCall #include "CGCXXABI.h"
1791bbb554SDevang Patel #include "CGDebugInfo.h"
183a02247dSChandler Carruth #include "CGObjCRuntime.h"
19a8e7df36SMark Lacey #include "clang/CodeGen/CGFunctionInfo.h"
203a02247dSChandler Carruth #include "clang/Frontend/CodeGenOptions.h"
21c80ceea9SChandler Carruth #include "llvm/IR/CallSite.h"
22ffd5551bSChandler Carruth #include "llvm/IR/Intrinsics.h"
23bbe277c4SAnders Carlsson 
2459486a2dSAnders Carlsson using namespace clang;
2559486a2dSAnders Carlsson using namespace CodeGen;
2659486a2dSAnders Carlsson 
2727da15baSAnders Carlsson RValue CodeGenFunction::EmitCXXMemberCall(const CXXMethodDecl *MD,
28e30752c9SRichard Smith                                           SourceLocation CallLoc,
2927da15baSAnders Carlsson                                           llvm::Value *Callee,
3027da15baSAnders Carlsson                                           ReturnValueSlot ReturnValue,
3127da15baSAnders Carlsson                                           llvm::Value *This,
32ee6bc533STimur Iskhodzhanov                                           llvm::Value *ImplicitParam,
33ee6bc533STimur Iskhodzhanov                                           QualType ImplicitParamTy,
3427da15baSAnders Carlsson                                           CallExpr::const_arg_iterator ArgBeg,
3527da15baSAnders Carlsson                                           CallExpr::const_arg_iterator ArgEnd) {
3627da15baSAnders Carlsson   assert(MD->isInstance() &&
3727da15baSAnders Carlsson          "Trying to emit a member call expr on a static method!");
3827da15baSAnders Carlsson 
3969d0d262SRichard Smith   // C++11 [class.mfct.non-static]p2:
4069d0d262SRichard Smith   //   If a non-static member function of a class X is called for an object that
4169d0d262SRichard Smith   //   is not of type X, or of a type derived from X, the behavior is undefined.
424d3110afSRichard Smith   EmitTypeCheck(isa<CXXConstructorDecl>(MD) ? TCK_ConstructorCall
434d3110afSRichard Smith                                             : TCK_MemberCall,
444d3110afSRichard Smith                 CallLoc, This, getContext().getRecordType(MD->getParent()));
4569d0d262SRichard Smith 
4627da15baSAnders Carlsson   CallArgList Args;
4727da15baSAnders Carlsson 
4827da15baSAnders Carlsson   // Push the this ptr.
4943dca6a8SEli Friedman   Args.add(RValue::get(This), MD->getThisType(getContext()));
5027da15baSAnders Carlsson 
51ee6bc533STimur Iskhodzhanov   // If there is an implicit parameter (e.g. VTT), emit it.
52ee6bc533STimur Iskhodzhanov   if (ImplicitParam) {
53ee6bc533STimur Iskhodzhanov     Args.add(RValue::get(ImplicitParam), ImplicitParamTy);
54e36a6b3eSAnders Carlsson   }
55e36a6b3eSAnders Carlsson 
56a729c62bSJohn McCall   const FunctionProtoType *FPT = MD->getType()->castAs<FunctionProtoType>();
57a729c62bSJohn McCall   RequiredArgs required = RequiredArgs::forPrototypePlus(FPT, Args.size());
58a729c62bSJohn McCall 
59a729c62bSJohn McCall   // And the rest of the call args.
6027da15baSAnders Carlsson   EmitCallArgs(Args, FPT, ArgBeg, ArgEnd);
6127da15baSAnders Carlsson 
628dda7b27SJohn McCall   return EmitCall(CGM.getTypes().arrangeCXXMethodCall(Args, FPT, required),
63c50c27ccSRafael Espindola                   Callee, ReturnValue, Args, MD);
6427da15baSAnders Carlsson }
6527da15baSAnders Carlsson 
663b33c4ecSRafael Espindola static CXXRecordDecl *getCXXRecord(const Expr *E) {
673b33c4ecSRafael Espindola   QualType T = E->getType();
683b33c4ecSRafael Espindola   if (const PointerType *PTy = T->getAs<PointerType>())
693b33c4ecSRafael Espindola     T = PTy->getPointeeType();
703b33c4ecSRafael Espindola   const RecordType *Ty = T->castAs<RecordType>();
713b33c4ecSRafael Espindola   return cast<CXXRecordDecl>(Ty->getDecl());
723b33c4ecSRafael Espindola }
733b33c4ecSRafael Espindola 
7464225794SFrancois Pichet // Note: This function also emit constructor calls to support a MSVC
7564225794SFrancois Pichet // extensions allowing explicit constructor function call.
7627da15baSAnders Carlsson RValue CodeGenFunction::EmitCXXMemberCallExpr(const CXXMemberCallExpr *CE,
7727da15baSAnders Carlsson                                               ReturnValueSlot ReturnValue) {
782d2e8707SJohn McCall   const Expr *callee = CE->getCallee()->IgnoreParens();
792d2e8707SJohn McCall 
802d2e8707SJohn McCall   if (isa<BinaryOperator>(callee))
8127da15baSAnders Carlsson     return EmitCXXMemberPointerCallExpr(CE, ReturnValue);
8227da15baSAnders Carlsson 
832d2e8707SJohn McCall   const MemberExpr *ME = cast<MemberExpr>(callee);
8427da15baSAnders Carlsson   const CXXMethodDecl *MD = cast<CXXMethodDecl>(ME->getMemberDecl());
8527da15baSAnders Carlsson 
8627da15baSAnders Carlsson   if (MD->isStatic()) {
8727da15baSAnders Carlsson     // The method is static, emit it as we would a regular call.
8827da15baSAnders Carlsson     llvm::Value *Callee = CGM.GetAddrOfFunction(MD);
89*70b9c01bSAlexey Samsonov     return EmitCall(getContext().getPointerType(MD->getType()), Callee, CE,
90*70b9c01bSAlexey Samsonov                     ReturnValue);
9127da15baSAnders Carlsson   }
9227da15baSAnders Carlsson 
930d635f53SJohn McCall   // Compute the object pointer.
94ecbe2e97SRafael Espindola   const Expr *Base = ME->getBase();
95ecbe2e97SRafael Espindola   bool CanUseVirtualCall = MD->isVirtual() && !ME->hasQualifier();
96ecbe2e97SRafael Espindola 
978a13c418SCraig Topper   const CXXMethodDecl *DevirtualizedMethod = nullptr;
987463ed7cSBenjamin Kramer   if (CanUseVirtualCall && CanDevirtualizeMemberFunctionCall(Base, MD)) {
993b33c4ecSRafael Espindola     const CXXRecordDecl *BestDynamicDecl = Base->getBestDynamicClassType();
1003b33c4ecSRafael Espindola     DevirtualizedMethod = MD->getCorrespondingMethodInClass(BestDynamicDecl);
1013b33c4ecSRafael Espindola     assert(DevirtualizedMethod);
1023b33c4ecSRafael Espindola     const CXXRecordDecl *DevirtualizedClass = DevirtualizedMethod->getParent();
1033b33c4ecSRafael Espindola     const Expr *Inner = Base->ignoreParenBaseCasts();
1043b33c4ecSRafael Espindola     if (getCXXRecord(Inner) == DevirtualizedClass)
1053b33c4ecSRafael Espindola       // If the class of the Inner expression is where the dynamic method
1063b33c4ecSRafael Espindola       // is defined, build the this pointer from it.
1073b33c4ecSRafael Espindola       Base = Inner;
1083b33c4ecSRafael Espindola     else if (getCXXRecord(Base) != DevirtualizedClass) {
1093b33c4ecSRafael Espindola       // If the method is defined in a class that is not the best dynamic
1103b33c4ecSRafael Espindola       // one or the one of the full expression, we would have to build
1113b33c4ecSRafael Espindola       // a derived-to-base cast to compute the correct this pointer, but
1123b33c4ecSRafael Espindola       // we don't have support for that yet, so do a virtual call.
1138a13c418SCraig Topper       DevirtualizedMethod = nullptr;
1143b33c4ecSRafael Espindola     }
115b27564afSRafael Espindola     // If the return types are not the same, this might be a case where more
116b27564afSRafael Espindola     // code needs to run to compensate for it. For example, the derived
117b27564afSRafael Espindola     // method might return a type that inherits form from the return
118b27564afSRafael Espindola     // type of MD and has a prefix.
119b27564afSRafael Espindola     // For now we just avoid devirtualizing these covariant cases.
120b27564afSRafael Espindola     if (DevirtualizedMethod &&
121314cc81bSAlp Toker         DevirtualizedMethod->getReturnType().getCanonicalType() !=
122314cc81bSAlp Toker             MD->getReturnType().getCanonicalType())
1238a13c418SCraig Topper       DevirtualizedMethod = nullptr;
1243b33c4ecSRafael Espindola   }
125ecbe2e97SRafael Espindola 
12627da15baSAnders Carlsson   llvm::Value *This;
12727da15baSAnders Carlsson   if (ME->isArrow())
1283b33c4ecSRafael Espindola     This = EmitScalarExpr(Base);
129f93ac894SFariborz Jahanian   else
1303b33c4ecSRafael Espindola     This = EmitLValue(Base).getAddress();
131ecbe2e97SRafael Espindola 
13227da15baSAnders Carlsson 
1330d635f53SJohn McCall   if (MD->isTrivial()) {
1348a13c418SCraig Topper     if (isa<CXXDestructorDecl>(MD)) return RValue::get(nullptr);
13564225794SFrancois Pichet     if (isa<CXXConstructorDecl>(MD) &&
13664225794SFrancois Pichet         cast<CXXConstructorDecl>(MD)->isDefaultConstructor())
1378a13c418SCraig Topper       return RValue::get(nullptr);
1380d635f53SJohn McCall 
13922653bacSSebastian Redl     if (MD->isCopyAssignmentOperator() || MD->isMoveAssignmentOperator()) {
14022653bacSSebastian Redl       // We don't like to generate the trivial copy/move assignment operator
14122653bacSSebastian Redl       // when it isn't necessary; just produce the proper effect here.
14227da15baSAnders Carlsson       llvm::Value *RHS = EmitLValue(*CE->arg_begin()).getAddress();
1431ca66919SBenjamin Kramer       EmitAggregateAssign(This, RHS, CE->getType());
14427da15baSAnders Carlsson       return RValue::get(This);
14527da15baSAnders Carlsson     }
14627da15baSAnders Carlsson 
14764225794SFrancois Pichet     if (isa<CXXConstructorDecl>(MD) &&
14822653bacSSebastian Redl         cast<CXXConstructorDecl>(MD)->isCopyOrMoveConstructor()) {
14922653bacSSebastian Redl       // Trivial move and copy ctor are the same.
15064225794SFrancois Pichet       llvm::Value *RHS = EmitLValue(*CE->arg_begin()).getAddress();
15164225794SFrancois Pichet       EmitSynthesizedCXXCopyCtorCall(cast<CXXConstructorDecl>(MD), This, RHS,
15264225794SFrancois Pichet                                      CE->arg_begin(), CE->arg_end());
15364225794SFrancois Pichet       return RValue::get(This);
15464225794SFrancois Pichet     }
15564225794SFrancois Pichet     llvm_unreachable("unknown trivial member function");
15664225794SFrancois Pichet   }
15764225794SFrancois Pichet 
1580d635f53SJohn McCall   // Compute the function type we're calling.
159ade60977SEli Friedman   const CXXMethodDecl *CalleeDecl = DevirtualizedMethod ? DevirtualizedMethod : MD;
1608a13c418SCraig Topper   const CGFunctionInfo *FInfo = nullptr;
161ade60977SEli Friedman   if (const CXXDestructorDecl *Dtor = dyn_cast<CXXDestructorDecl>(CalleeDecl))
162ade60977SEli Friedman     FInfo = &CGM.getTypes().arrangeCXXDestructor(Dtor,
16364225794SFrancois Pichet                                                  Dtor_Complete);
164ade60977SEli Friedman   else if (const CXXConstructorDecl *Ctor = dyn_cast<CXXConstructorDecl>(CalleeDecl))
165ade60977SEli Friedman     FInfo = &CGM.getTypes().arrangeCXXConstructorDeclaration(Ctor,
16664225794SFrancois Pichet                                                              Ctor_Complete);
16764225794SFrancois Pichet   else
168ade60977SEli Friedman     FInfo = &CGM.getTypes().arrangeCXXMethodDeclaration(CalleeDecl);
1690d635f53SJohn McCall 
170e7de47efSReid Kleckner   llvm::FunctionType *Ty = CGM.getTypes().GetFunctionType(*FInfo);
1710d635f53SJohn McCall 
17227da15baSAnders Carlsson   // C++ [class.virtual]p12:
17327da15baSAnders Carlsson   //   Explicit qualification with the scope operator (5.1) suppresses the
17427da15baSAnders Carlsson   //   virtual call mechanism.
17527da15baSAnders Carlsson   //
17627da15baSAnders Carlsson   // We also don't emit a virtual call if the base expression has a record type
17727da15baSAnders Carlsson   // because then we know what the type is.
1783b33c4ecSRafael Espindola   bool UseVirtualCall = CanUseVirtualCall && !DevirtualizedMethod;
17919cee187SStephen Lin   llvm::Value *Callee;
1809dc6eef7SStephen Lin 
1810d635f53SJohn McCall   if (const CXXDestructorDecl *Dtor = dyn_cast<CXXDestructorDecl>(MD)) {
18219cee187SStephen Lin     assert(CE->arg_begin() == CE->arg_end() &&
1839dc6eef7SStephen Lin            "Destructor shouldn't have explicit parameters");
1849dc6eef7SStephen Lin     assert(ReturnValue.isNull() && "Destructor shouldn't have return value");
1859dc6eef7SStephen Lin     if (UseVirtualCall) {
1869dc6eef7SStephen Lin       CGM.getCXXABI().EmitVirtualDestructorCall(*this, Dtor, Dtor_Complete,
1879dc6eef7SStephen Lin                                                 CE->getExprLoc(), This);
18827da15baSAnders Carlsson     } else {
1899c6890a7SRichard Smith       if (getLangOpts().AppleKext &&
190265c325eSFariborz Jahanian           MD->isVirtual() &&
191265c325eSFariborz Jahanian           ME->hasQualifier())
1927f6f81baSFariborz Jahanian         Callee = BuildAppleKextVirtualCall(MD, ME->getQualifier(), Ty);
1933b33c4ecSRafael Espindola       else if (!DevirtualizedMethod)
194e7de47efSReid Kleckner         Callee = CGM.GetAddrOfCXXDestructor(Dtor, Dtor_Complete, FInfo, Ty);
19549e860b2SRafael Espindola       else {
1963b33c4ecSRafael Espindola         const CXXDestructorDecl *DDtor =
1973b33c4ecSRafael Espindola           cast<CXXDestructorDecl>(DevirtualizedMethod);
19849e860b2SRafael Espindola         Callee = CGM.GetAddrOfFunction(GlobalDecl(DDtor, Dtor_Complete), Ty);
19949e860b2SRafael Espindola       }
2009dc6eef7SStephen Lin       EmitCXXMemberCall(MD, CE->getExprLoc(), Callee, ReturnValue, This,
2018a13c418SCraig Topper                         /*ImplicitParam=*/nullptr, QualType(), nullptr,nullptr);
20227da15baSAnders Carlsson     }
2038a13c418SCraig Topper     return RValue::get(nullptr);
2049dc6eef7SStephen Lin   }
2059dc6eef7SStephen Lin 
2069dc6eef7SStephen Lin   if (const CXXConstructorDecl *Ctor = dyn_cast<CXXConstructorDecl>(MD)) {
20764225794SFrancois Pichet     Callee = CGM.GetAddrOfFunction(GlobalDecl(Ctor, Ctor_Complete), Ty);
2080d635f53SJohn McCall   } else if (UseVirtualCall) {
20988fd439aSTimur Iskhodzhanov     Callee = CGM.getCXXABI().getVirtualFunctionPointer(*this, MD, This, Ty);
21027da15baSAnders Carlsson   } else {
2119c6890a7SRichard Smith     if (getLangOpts().AppleKext &&
2129f9438b3SFariborz Jahanian         MD->isVirtual() &&
213252a47f6SFariborz Jahanian         ME->hasQualifier())
2147f6f81baSFariborz Jahanian       Callee = BuildAppleKextVirtualCall(MD, ME->getQualifier(), Ty);
2153b33c4ecSRafael Espindola     else if (!DevirtualizedMethod)
216727a771aSRafael Espindola       Callee = CGM.GetAddrOfFunction(MD, Ty);
21749e860b2SRafael Espindola     else {
2183b33c4ecSRafael Espindola       Callee = CGM.GetAddrOfFunction(DevirtualizedMethod, Ty);
21949e860b2SRafael Espindola     }
22027da15baSAnders Carlsson   }
22127da15baSAnders Carlsson 
222f1749427STimur Iskhodzhanov   if (MD->isVirtual()) {
223f1749427STimur Iskhodzhanov     This = CGM.getCXXABI().adjustThisArgumentForVirtualFunctionCall(
224f1749427STimur Iskhodzhanov         *this, MD, This, UseVirtualCall);
225f1749427STimur Iskhodzhanov   }
22688fd439aSTimur Iskhodzhanov 
227e30752c9SRichard Smith   return EmitCXXMemberCall(MD, CE->getExprLoc(), Callee, ReturnValue, This,
2288a13c418SCraig Topper                            /*ImplicitParam=*/nullptr, QualType(),
229ee6bc533STimur Iskhodzhanov                            CE->arg_begin(), CE->arg_end());
23027da15baSAnders Carlsson }
23127da15baSAnders Carlsson 
23227da15baSAnders Carlsson RValue
23327da15baSAnders Carlsson CodeGenFunction::EmitCXXMemberPointerCallExpr(const CXXMemberCallExpr *E,
23427da15baSAnders Carlsson                                               ReturnValueSlot ReturnValue) {
23527da15baSAnders Carlsson   const BinaryOperator *BO =
23627da15baSAnders Carlsson       cast<BinaryOperator>(E->getCallee()->IgnoreParens());
23727da15baSAnders Carlsson   const Expr *BaseExpr = BO->getLHS();
23827da15baSAnders Carlsson   const Expr *MemFnExpr = BO->getRHS();
23927da15baSAnders Carlsson 
24027da15baSAnders Carlsson   const MemberPointerType *MPT =
2410009fcc3SJohn McCall     MemFnExpr->getType()->castAs<MemberPointerType>();
242475999dcSJohn McCall 
24327da15baSAnders Carlsson   const FunctionProtoType *FPT =
2440009fcc3SJohn McCall     MPT->getPointeeType()->castAs<FunctionProtoType>();
24527da15baSAnders Carlsson   const CXXRecordDecl *RD =
24627da15baSAnders Carlsson     cast<CXXRecordDecl>(MPT->getClass()->getAs<RecordType>()->getDecl());
24727da15baSAnders Carlsson 
24827da15baSAnders Carlsson   // Get the member function pointer.
249a1dee530SJohn McCall   llvm::Value *MemFnPtr = EmitScalarExpr(MemFnExpr);
25027da15baSAnders Carlsson 
25127da15baSAnders Carlsson   // Emit the 'this' pointer.
25227da15baSAnders Carlsson   llvm::Value *This;
25327da15baSAnders Carlsson 
254e302792bSJohn McCall   if (BO->getOpcode() == BO_PtrMemI)
25527da15baSAnders Carlsson     This = EmitScalarExpr(BaseExpr);
25627da15baSAnders Carlsson   else
25727da15baSAnders Carlsson     This = EmitLValue(BaseExpr).getAddress();
25827da15baSAnders Carlsson 
259e30752c9SRichard Smith   EmitTypeCheck(TCK_MemberCall, E->getExprLoc(), This,
260e30752c9SRichard Smith                 QualType(MPT->getClass(), 0));
26169d0d262SRichard Smith 
262475999dcSJohn McCall   // Ask the ABI to load the callee.  Note that This is modified.
263475999dcSJohn McCall   llvm::Value *Callee =
2642b0d66dfSDavid Majnemer     CGM.getCXXABI().EmitLoadOfMemberFunctionPointer(*this, BO, This, MemFnPtr, MPT);
26527da15baSAnders Carlsson 
26627da15baSAnders Carlsson   CallArgList Args;
26727da15baSAnders Carlsson 
26827da15baSAnders Carlsson   QualType ThisType =
26927da15baSAnders Carlsson     getContext().getPointerType(getContext().getTagDeclType(RD));
27027da15baSAnders Carlsson 
27127da15baSAnders Carlsson   // Push the this ptr.
27243dca6a8SEli Friedman   Args.add(RValue::get(This), ThisType);
27327da15baSAnders Carlsson 
2748dda7b27SJohn McCall   RequiredArgs required = RequiredArgs::forPrototypePlus(FPT, 1);
2758dda7b27SJohn McCall 
27627da15baSAnders Carlsson   // And the rest of the call args
27727da15baSAnders Carlsson   EmitCallArgs(Args, FPT, E->arg_begin(), E->arg_end());
2785fa40c3bSNick Lewycky   return EmitCall(CGM.getTypes().arrangeCXXMethodCall(Args, FPT, required),
2795fa40c3bSNick Lewycky                   Callee, ReturnValue, Args);
28027da15baSAnders Carlsson }
28127da15baSAnders Carlsson 
28227da15baSAnders Carlsson RValue
28327da15baSAnders Carlsson CodeGenFunction::EmitCXXOperatorMemberCallExpr(const CXXOperatorCallExpr *E,
28427da15baSAnders Carlsson                                                const CXXMethodDecl *MD,
28527da15baSAnders Carlsson                                                ReturnValueSlot ReturnValue) {
28627da15baSAnders Carlsson   assert(MD->isInstance() &&
28727da15baSAnders Carlsson          "Trying to emit a member call expr on a static method!");
288e26a872bSJohn McCall   LValue LV = EmitLValue(E->getArg(0));
289e26a872bSJohn McCall   llvm::Value *This = LV.getAddress();
290e26a872bSJohn McCall 
291146b8e9aSDouglas Gregor   if ((MD->isCopyAssignmentOperator() || MD->isMoveAssignmentOperator()) &&
292146b8e9aSDouglas Gregor       MD->isTrivial()) {
29327da15baSAnders Carlsson     llvm::Value *Src = EmitLValue(E->getArg(1)).getAddress();
29427da15baSAnders Carlsson     QualType Ty = E->getType();
2951ca66919SBenjamin Kramer     EmitAggregateAssign(This, Src, Ty);
29627da15baSAnders Carlsson     return RValue::get(This);
29727da15baSAnders Carlsson   }
29827da15baSAnders Carlsson 
299c36783e8SAnders Carlsson   llvm::Value *Callee = EmitCXXOperatorMemberCallee(E, MD, This);
300e30752c9SRichard Smith   return EmitCXXMemberCall(MD, E->getExprLoc(), Callee, ReturnValue, This,
3018a13c418SCraig Topper                            /*ImplicitParam=*/nullptr, QualType(),
302ee6bc533STimur Iskhodzhanov                            E->arg_begin() + 1, E->arg_end());
30327da15baSAnders Carlsson }
30427da15baSAnders Carlsson 
305fe883422SPeter Collingbourne RValue CodeGenFunction::EmitCUDAKernelCallExpr(const CUDAKernelCallExpr *E,
306fe883422SPeter Collingbourne                                                ReturnValueSlot ReturnValue) {
307fe883422SPeter Collingbourne   return CGM.getCUDARuntime().EmitCUDAKernelCallExpr(*this, E, ReturnValue);
308fe883422SPeter Collingbourne }
309fe883422SPeter Collingbourne 
310fde961dbSEli Friedman static void EmitNullBaseClassInitialization(CodeGenFunction &CGF,
311fde961dbSEli Friedman                                             llvm::Value *DestPtr,
312fde961dbSEli Friedman                                             const CXXRecordDecl *Base) {
313fde961dbSEli Friedman   if (Base->isEmpty())
314fde961dbSEli Friedman     return;
315fde961dbSEli Friedman 
316fde961dbSEli Friedman   DestPtr = CGF.EmitCastToVoidPtr(DestPtr);
317fde961dbSEli Friedman 
318fde961dbSEli Friedman   const ASTRecordLayout &Layout = CGF.getContext().getASTRecordLayout(Base);
319fde961dbSEli Friedman   CharUnits Size = Layout.getNonVirtualSize();
320d640d7d9SWarren Hunt   CharUnits Align = Layout.getNonVirtualAlignment();
321fde961dbSEli Friedman 
322fde961dbSEli Friedman   llvm::Value *SizeVal = CGF.CGM.getSize(Size);
323fde961dbSEli Friedman 
324fde961dbSEli Friedman   // If the type contains a pointer to data member we can't memset it to zero.
325fde961dbSEli Friedman   // Instead, create a null constant and copy it to the destination.
326fde961dbSEli Friedman   // TODO: there are other patterns besides zero that we can usefully memset,
327fde961dbSEli Friedman   // like -1, which happens to be the pattern used by member-pointers.
328fde961dbSEli Friedman   // TODO: isZeroInitializable can be over-conservative in the case where a
329fde961dbSEli Friedman   // virtual base contains a member pointer.
330fde961dbSEli Friedman   if (!CGF.CGM.getTypes().isZeroInitializable(Base)) {
331fde961dbSEli Friedman     llvm::Constant *NullConstant = CGF.CGM.EmitNullConstantForBase(Base);
332fde961dbSEli Friedman 
333fde961dbSEli Friedman     llvm::GlobalVariable *NullVariable =
334fde961dbSEli Friedman       new llvm::GlobalVariable(CGF.CGM.getModule(), NullConstant->getType(),
335fde961dbSEli Friedman                                /*isConstant=*/true,
336fde961dbSEli Friedman                                llvm::GlobalVariable::PrivateLinkage,
337fde961dbSEli Friedman                                NullConstant, Twine());
338fde961dbSEli Friedman     NullVariable->setAlignment(Align.getQuantity());
339fde961dbSEli Friedman     llvm::Value *SrcPtr = CGF.EmitCastToVoidPtr(NullVariable);
340fde961dbSEli Friedman 
341fde961dbSEli Friedman     // Get and call the appropriate llvm.memcpy overload.
342fde961dbSEli Friedman     CGF.Builder.CreateMemCpy(DestPtr, SrcPtr, SizeVal, Align.getQuantity());
343fde961dbSEli Friedman     return;
344fde961dbSEli Friedman   }
345fde961dbSEli Friedman 
346fde961dbSEli Friedman   // Otherwise, just memset the whole thing to zero.  This is legal
347fde961dbSEli Friedman   // because in LLVM, all default initializers (other than the ones we just
348fde961dbSEli Friedman   // handled above) are guaranteed to have a bit pattern of all zeros.
349fde961dbSEli Friedman   CGF.Builder.CreateMemSet(DestPtr, CGF.Builder.getInt8(0), SizeVal,
350fde961dbSEli Friedman                            Align.getQuantity());
351fde961dbSEli Friedman }
352fde961dbSEli Friedman 
35327da15baSAnders Carlsson void
3547a626f63SJohn McCall CodeGenFunction::EmitCXXConstructExpr(const CXXConstructExpr *E,
3557a626f63SJohn McCall                                       AggValueSlot Dest) {
3567a626f63SJohn McCall   assert(!Dest.isIgnored() && "Must have a destination!");
35727da15baSAnders Carlsson   const CXXConstructorDecl *CD = E->getConstructor();
358630c76efSDouglas Gregor 
359630c76efSDouglas Gregor   // If we require zero initialization before (or instead of) calling the
360630c76efSDouglas Gregor   // constructor, as can be the case with a non-user-provided default
36103535265SArgyrios Kyrtzidis   // constructor, emit the zero initialization now, unless destination is
36203535265SArgyrios Kyrtzidis   // already zeroed.
363fde961dbSEli Friedman   if (E->requiresZeroInitialization() && !Dest.isZeroed()) {
364fde961dbSEli Friedman     switch (E->getConstructionKind()) {
365fde961dbSEli Friedman     case CXXConstructExpr::CK_Delegating:
366fde961dbSEli Friedman     case CXXConstructExpr::CK_Complete:
3677a626f63SJohn McCall       EmitNullInitialization(Dest.getAddr(), E->getType());
368fde961dbSEli Friedman       break;
369fde961dbSEli Friedman     case CXXConstructExpr::CK_VirtualBase:
370fde961dbSEli Friedman     case CXXConstructExpr::CK_NonVirtualBase:
371fde961dbSEli Friedman       EmitNullBaseClassInitialization(*this, Dest.getAddr(), CD->getParent());
372fde961dbSEli Friedman       break;
373fde961dbSEli Friedman     }
374fde961dbSEli Friedman   }
375630c76efSDouglas Gregor 
376630c76efSDouglas Gregor   // If this is a call to a trivial default constructor, do nothing.
377630c76efSDouglas Gregor   if (CD->isTrivial() && CD->isDefaultConstructor())
37827da15baSAnders Carlsson     return;
379630c76efSDouglas Gregor 
3808ea46b66SJohn McCall   // Elide the constructor if we're constructing from a temporary.
3818ea46b66SJohn McCall   // The temporary check is required because Sema sets this on NRVO
3828ea46b66SJohn McCall   // returns.
3839c6890a7SRichard Smith   if (getLangOpts().ElideConstructors && E->isElidable()) {
3848ea46b66SJohn McCall     assert(getContext().hasSameUnqualifiedType(E->getType(),
3858ea46b66SJohn McCall                                                E->getArg(0)->getType()));
3867a626f63SJohn McCall     if (E->getArg(0)->isTemporaryObject(getContext(), CD->getParent())) {
3877a626f63SJohn McCall       EmitAggExpr(E->getArg(0), Dest);
38827da15baSAnders Carlsson       return;
38927da15baSAnders Carlsson     }
390222cf0efSDouglas Gregor   }
391630c76efSDouglas Gregor 
392f677a8e9SJohn McCall   if (const ConstantArrayType *arrayType
393f677a8e9SJohn McCall         = getContext().getAsConstantArrayType(E->getType())) {
394*70b9c01bSAlexey Samsonov     EmitCXXAggrConstructorCall(CD, arrayType, Dest.getAddr(), E);
395f677a8e9SJohn McCall   } else {
396bceca20aSCameron Esfahani     CXXCtorType Type = Ctor_Complete;
397271c3681SAlexis Hunt     bool ForVirtualBase = false;
39861535005SDouglas Gregor     bool Delegating = false;
399271c3681SAlexis Hunt 
400271c3681SAlexis Hunt     switch (E->getConstructionKind()) {
401271c3681SAlexis Hunt      case CXXConstructExpr::CK_Delegating:
40261bc1737SAlexis Hunt       // We should be emitting a constructor; GlobalDecl will assert this
40361bc1737SAlexis Hunt       Type = CurGD.getCtorType();
40461535005SDouglas Gregor       Delegating = true;
405271c3681SAlexis Hunt       break;
40661bc1737SAlexis Hunt 
407271c3681SAlexis Hunt      case CXXConstructExpr::CK_Complete:
408271c3681SAlexis Hunt       Type = Ctor_Complete;
409271c3681SAlexis Hunt       break;
410271c3681SAlexis Hunt 
411271c3681SAlexis Hunt      case CXXConstructExpr::CK_VirtualBase:
412271c3681SAlexis Hunt       ForVirtualBase = true;
413271c3681SAlexis Hunt       // fall-through
414271c3681SAlexis Hunt 
415271c3681SAlexis Hunt      case CXXConstructExpr::CK_NonVirtualBase:
416271c3681SAlexis Hunt       Type = Ctor_Base;
417271c3681SAlexis Hunt     }
418e11f9ce9SAnders Carlsson 
41927da15baSAnders Carlsson     // Call the constructor.
42061535005SDouglas Gregor     EmitCXXConstructorCall(CD, Type, ForVirtualBase, Delegating, Dest.getAddr(),
421*70b9c01bSAlexey Samsonov                            E);
42227da15baSAnders Carlsson   }
423e11f9ce9SAnders Carlsson }
42427da15baSAnders Carlsson 
425e988bdacSFariborz Jahanian void
426e988bdacSFariborz Jahanian CodeGenFunction::EmitSynthesizedCXXCopyCtor(llvm::Value *Dest,
427e988bdacSFariborz Jahanian                                             llvm::Value *Src,
42850198098SFariborz Jahanian                                             const Expr *Exp) {
4295d413781SJohn McCall   if (const ExprWithCleanups *E = dyn_cast<ExprWithCleanups>(Exp))
430e988bdacSFariborz Jahanian     Exp = E->getSubExpr();
431e988bdacSFariborz Jahanian   assert(isa<CXXConstructExpr>(Exp) &&
432e988bdacSFariborz Jahanian          "EmitSynthesizedCXXCopyCtor - unknown copy ctor expr");
433e988bdacSFariborz Jahanian   const CXXConstructExpr* E = cast<CXXConstructExpr>(Exp);
434e988bdacSFariborz Jahanian   const CXXConstructorDecl *CD = E->getConstructor();
435e988bdacSFariborz Jahanian   RunCleanupsScope Scope(*this);
436e988bdacSFariborz Jahanian 
437e988bdacSFariborz Jahanian   // If we require zero initialization before (or instead of) calling the
438e988bdacSFariborz Jahanian   // constructor, as can be the case with a non-user-provided default
439e988bdacSFariborz Jahanian   // constructor, emit the zero initialization now.
440e988bdacSFariborz Jahanian   // FIXME. Do I still need this for a copy ctor synthesis?
441e988bdacSFariborz Jahanian   if (E->requiresZeroInitialization())
442e988bdacSFariborz Jahanian     EmitNullInitialization(Dest, E->getType());
443e988bdacSFariborz Jahanian 
44499da11cfSChandler Carruth   assert(!getContext().getAsConstantArrayType(E->getType())
44599da11cfSChandler Carruth          && "EmitSynthesizedCXXCopyCtor - Copied-in Array");
4465fa40c3bSNick Lewycky   EmitSynthesizedCXXCopyCtorCall(CD, Dest, Src, E->arg_begin(), E->arg_end());
447e988bdacSFariborz Jahanian }
448e988bdacSFariborz Jahanian 
4498ed55a54SJohn McCall static CharUnits CalculateCookiePadding(CodeGenFunction &CGF,
4508ed55a54SJohn McCall                                         const CXXNewExpr *E) {
45121122cf6SAnders Carlsson   if (!E->isArray())
4523eb55cfeSKen Dyck     return CharUnits::Zero();
45321122cf6SAnders Carlsson 
4547ec4b434SJohn McCall   // No cookie is required if the operator new[] being used is the
4557ec4b434SJohn McCall   // reserved placement operator new[].
4567ec4b434SJohn McCall   if (E->getOperatorNew()->isReservedGlobalPlacementOperator())
4573eb55cfeSKen Dyck     return CharUnits::Zero();
458399f499fSAnders Carlsson 
459284c48ffSJohn McCall   return CGF.CGM.getCXXABI().GetArrayCookieSize(E);
46059486a2dSAnders Carlsson }
46159486a2dSAnders Carlsson 
462036f2f6bSJohn McCall static llvm::Value *EmitCXXNewAllocSize(CodeGenFunction &CGF,
463036f2f6bSJohn McCall                                         const CXXNewExpr *e,
464f862eb6aSSebastian Redl                                         unsigned minElements,
465036f2f6bSJohn McCall                                         llvm::Value *&numElements,
466036f2f6bSJohn McCall                                         llvm::Value *&sizeWithoutCookie) {
467036f2f6bSJohn McCall   QualType type = e->getAllocatedType();
46859486a2dSAnders Carlsson 
469036f2f6bSJohn McCall   if (!e->isArray()) {
470036f2f6bSJohn McCall     CharUnits typeSize = CGF.getContext().getTypeSizeInChars(type);
471036f2f6bSJohn McCall     sizeWithoutCookie
472036f2f6bSJohn McCall       = llvm::ConstantInt::get(CGF.SizeTy, typeSize.getQuantity());
473036f2f6bSJohn McCall     return sizeWithoutCookie;
47405fc5be3SDouglas Gregor   }
47559486a2dSAnders Carlsson 
476036f2f6bSJohn McCall   // The width of size_t.
477036f2f6bSJohn McCall   unsigned sizeWidth = CGF.SizeTy->getBitWidth();
478036f2f6bSJohn McCall 
4798ed55a54SJohn McCall   // Figure out the cookie size.
480036f2f6bSJohn McCall   llvm::APInt cookieSize(sizeWidth,
481036f2f6bSJohn McCall                          CalculateCookiePadding(CGF, e).getQuantity());
4828ed55a54SJohn McCall 
48359486a2dSAnders Carlsson   // Emit the array size expression.
4847648fb46SArgyrios Kyrtzidis   // We multiply the size of all dimensions for NumElements.
4857648fb46SArgyrios Kyrtzidis   // e.g for 'int[2][3]', ElemType is 'int' and NumElements is 6.
486036f2f6bSJohn McCall   numElements = CGF.EmitScalarExpr(e->getArraySize());
487036f2f6bSJohn McCall   assert(isa<llvm::IntegerType>(numElements->getType()));
4888ed55a54SJohn McCall 
489036f2f6bSJohn McCall   // The number of elements can be have an arbitrary integer type;
490036f2f6bSJohn McCall   // essentially, we need to multiply it by a constant factor, add a
491036f2f6bSJohn McCall   // cookie size, and verify that the result is representable as a
492036f2f6bSJohn McCall   // size_t.  That's just a gloss, though, and it's wrong in one
493036f2f6bSJohn McCall   // important way: if the count is negative, it's an error even if
494036f2f6bSJohn McCall   // the cookie size would bring the total size >= 0.
4956ab2fa8fSDouglas Gregor   bool isSigned
4966ab2fa8fSDouglas Gregor     = e->getArraySize()->getType()->isSignedIntegerOrEnumerationType();
4972192fe50SChris Lattner   llvm::IntegerType *numElementsType
498036f2f6bSJohn McCall     = cast<llvm::IntegerType>(numElements->getType());
499036f2f6bSJohn McCall   unsigned numElementsWidth = numElementsType->getBitWidth();
500036f2f6bSJohn McCall 
501036f2f6bSJohn McCall   // Compute the constant factor.
502036f2f6bSJohn McCall   llvm::APInt arraySizeMultiplier(sizeWidth, 1);
5037648fb46SArgyrios Kyrtzidis   while (const ConstantArrayType *CAT
504036f2f6bSJohn McCall              = CGF.getContext().getAsConstantArrayType(type)) {
505036f2f6bSJohn McCall     type = CAT->getElementType();
506036f2f6bSJohn McCall     arraySizeMultiplier *= CAT->getSize();
5077648fb46SArgyrios Kyrtzidis   }
50859486a2dSAnders Carlsson 
509036f2f6bSJohn McCall   CharUnits typeSize = CGF.getContext().getTypeSizeInChars(type);
510036f2f6bSJohn McCall   llvm::APInt typeSizeMultiplier(sizeWidth, typeSize.getQuantity());
511036f2f6bSJohn McCall   typeSizeMultiplier *= arraySizeMultiplier;
512036f2f6bSJohn McCall 
513036f2f6bSJohn McCall   // This will be a size_t.
514036f2f6bSJohn McCall   llvm::Value *size;
51532ac583dSChris Lattner 
51632ac583dSChris Lattner   // If someone is doing 'new int[42]' there is no need to do a dynamic check.
51732ac583dSChris Lattner   // Don't bloat the -O0 code.
518036f2f6bSJohn McCall   if (llvm::ConstantInt *numElementsC =
519036f2f6bSJohn McCall         dyn_cast<llvm::ConstantInt>(numElements)) {
520036f2f6bSJohn McCall     const llvm::APInt &count = numElementsC->getValue();
52132ac583dSChris Lattner 
522036f2f6bSJohn McCall     bool hasAnyOverflow = false;
52332ac583dSChris Lattner 
524036f2f6bSJohn McCall     // If 'count' was a negative number, it's an overflow.
525036f2f6bSJohn McCall     if (isSigned && count.isNegative())
526036f2f6bSJohn McCall       hasAnyOverflow = true;
5278ed55a54SJohn McCall 
528036f2f6bSJohn McCall     // We want to do all this arithmetic in size_t.  If numElements is
529036f2f6bSJohn McCall     // wider than that, check whether it's already too big, and if so,
530036f2f6bSJohn McCall     // overflow.
531036f2f6bSJohn McCall     else if (numElementsWidth > sizeWidth &&
532036f2f6bSJohn McCall              numElementsWidth - sizeWidth > count.countLeadingZeros())
533036f2f6bSJohn McCall       hasAnyOverflow = true;
534036f2f6bSJohn McCall 
535036f2f6bSJohn McCall     // Okay, compute a count at the right width.
536036f2f6bSJohn McCall     llvm::APInt adjustedCount = count.zextOrTrunc(sizeWidth);
537036f2f6bSJohn McCall 
538f862eb6aSSebastian Redl     // If there is a brace-initializer, we cannot allocate fewer elements than
539f862eb6aSSebastian Redl     // there are initializers. If we do, that's treated like an overflow.
540f862eb6aSSebastian Redl     if (adjustedCount.ult(minElements))
541f862eb6aSSebastian Redl       hasAnyOverflow = true;
542f862eb6aSSebastian Redl 
543036f2f6bSJohn McCall     // Scale numElements by that.  This might overflow, but we don't
544036f2f6bSJohn McCall     // care because it only overflows if allocationSize does, too, and
545036f2f6bSJohn McCall     // if that overflows then we shouldn't use this.
546036f2f6bSJohn McCall     numElements = llvm::ConstantInt::get(CGF.SizeTy,
547036f2f6bSJohn McCall                                          adjustedCount * arraySizeMultiplier);
548036f2f6bSJohn McCall 
549036f2f6bSJohn McCall     // Compute the size before cookie, and track whether it overflowed.
550036f2f6bSJohn McCall     bool overflow;
551036f2f6bSJohn McCall     llvm::APInt allocationSize
552036f2f6bSJohn McCall       = adjustedCount.umul_ov(typeSizeMultiplier, overflow);
553036f2f6bSJohn McCall     hasAnyOverflow |= overflow;
554036f2f6bSJohn McCall 
555036f2f6bSJohn McCall     // Add in the cookie, and check whether it's overflowed.
556036f2f6bSJohn McCall     if (cookieSize != 0) {
557036f2f6bSJohn McCall       // Save the current size without a cookie.  This shouldn't be
558036f2f6bSJohn McCall       // used if there was overflow.
559036f2f6bSJohn McCall       sizeWithoutCookie = llvm::ConstantInt::get(CGF.SizeTy, allocationSize);
560036f2f6bSJohn McCall 
561036f2f6bSJohn McCall       allocationSize = allocationSize.uadd_ov(cookieSize, overflow);
562036f2f6bSJohn McCall       hasAnyOverflow |= overflow;
5638ed55a54SJohn McCall     }
5648ed55a54SJohn McCall 
565036f2f6bSJohn McCall     // On overflow, produce a -1 so operator new will fail.
566036f2f6bSJohn McCall     if (hasAnyOverflow) {
567036f2f6bSJohn McCall       size = llvm::Constant::getAllOnesValue(CGF.SizeTy);
56832ac583dSChris Lattner     } else {
569036f2f6bSJohn McCall       size = llvm::ConstantInt::get(CGF.SizeTy, allocationSize);
57032ac583dSChris Lattner     }
57132ac583dSChris Lattner 
572036f2f6bSJohn McCall   // Otherwise, we might need to use the overflow intrinsics.
5738ed55a54SJohn McCall   } else {
574f862eb6aSSebastian Redl     // There are up to five conditions we need to test for:
575036f2f6bSJohn McCall     // 1) if isSigned, we need to check whether numElements is negative;
576036f2f6bSJohn McCall     // 2) if numElementsWidth > sizeWidth, we need to check whether
577036f2f6bSJohn McCall     //   numElements is larger than something representable in size_t;
578f862eb6aSSebastian Redl     // 3) if minElements > 0, we need to check whether numElements is smaller
579f862eb6aSSebastian Redl     //    than that.
580f862eb6aSSebastian Redl     // 4) we need to compute
581036f2f6bSJohn McCall     //      sizeWithoutCookie := numElements * typeSizeMultiplier
582036f2f6bSJohn McCall     //    and check whether it overflows; and
583f862eb6aSSebastian Redl     // 5) if we need a cookie, we need to compute
584036f2f6bSJohn McCall     //      size := sizeWithoutCookie + cookieSize
585036f2f6bSJohn McCall     //    and check whether it overflows.
5868ed55a54SJohn McCall 
5878a13c418SCraig Topper     llvm::Value *hasOverflow = nullptr;
5888ed55a54SJohn McCall 
589036f2f6bSJohn McCall     // If numElementsWidth > sizeWidth, then one way or another, we're
590036f2f6bSJohn McCall     // going to have to do a comparison for (2), and this happens to
591036f2f6bSJohn McCall     // take care of (1), too.
592036f2f6bSJohn McCall     if (numElementsWidth > sizeWidth) {
593036f2f6bSJohn McCall       llvm::APInt threshold(numElementsWidth, 1);
594036f2f6bSJohn McCall       threshold <<= sizeWidth;
5958ed55a54SJohn McCall 
596036f2f6bSJohn McCall       llvm::Value *thresholdV
597036f2f6bSJohn McCall         = llvm::ConstantInt::get(numElementsType, threshold);
598036f2f6bSJohn McCall 
599036f2f6bSJohn McCall       hasOverflow = CGF.Builder.CreateICmpUGE(numElements, thresholdV);
600036f2f6bSJohn McCall       numElements = CGF.Builder.CreateTrunc(numElements, CGF.SizeTy);
601036f2f6bSJohn McCall 
602036f2f6bSJohn McCall     // Otherwise, if we're signed, we want to sext up to size_t.
603036f2f6bSJohn McCall     } else if (isSigned) {
604036f2f6bSJohn McCall       if (numElementsWidth < sizeWidth)
605036f2f6bSJohn McCall         numElements = CGF.Builder.CreateSExt(numElements, CGF.SizeTy);
606036f2f6bSJohn McCall 
607036f2f6bSJohn McCall       // If there's a non-1 type size multiplier, then we can do the
608036f2f6bSJohn McCall       // signedness check at the same time as we do the multiply
609036f2f6bSJohn McCall       // because a negative number times anything will cause an
610f862eb6aSSebastian Redl       // unsigned overflow.  Otherwise, we have to do it here. But at least
611f862eb6aSSebastian Redl       // in this case, we can subsume the >= minElements check.
612036f2f6bSJohn McCall       if (typeSizeMultiplier == 1)
613036f2f6bSJohn McCall         hasOverflow = CGF.Builder.CreateICmpSLT(numElements,
614f862eb6aSSebastian Redl                               llvm::ConstantInt::get(CGF.SizeTy, minElements));
615036f2f6bSJohn McCall 
616036f2f6bSJohn McCall     // Otherwise, zext up to size_t if necessary.
617036f2f6bSJohn McCall     } else if (numElementsWidth < sizeWidth) {
618036f2f6bSJohn McCall       numElements = CGF.Builder.CreateZExt(numElements, CGF.SizeTy);
619036f2f6bSJohn McCall     }
620036f2f6bSJohn McCall 
621036f2f6bSJohn McCall     assert(numElements->getType() == CGF.SizeTy);
622036f2f6bSJohn McCall 
623f862eb6aSSebastian Redl     if (minElements) {
624f862eb6aSSebastian Redl       // Don't allow allocation of fewer elements than we have initializers.
625f862eb6aSSebastian Redl       if (!hasOverflow) {
626f862eb6aSSebastian Redl         hasOverflow = CGF.Builder.CreateICmpULT(numElements,
627f862eb6aSSebastian Redl                               llvm::ConstantInt::get(CGF.SizeTy, minElements));
628f862eb6aSSebastian Redl       } else if (numElementsWidth > sizeWidth) {
629f862eb6aSSebastian Redl         // The other existing overflow subsumes this check.
630f862eb6aSSebastian Redl         // We do an unsigned comparison, since any signed value < -1 is
631f862eb6aSSebastian Redl         // taken care of either above or below.
632f862eb6aSSebastian Redl         hasOverflow = CGF.Builder.CreateOr(hasOverflow,
633f862eb6aSSebastian Redl                           CGF.Builder.CreateICmpULT(numElements,
634f862eb6aSSebastian Redl                               llvm::ConstantInt::get(CGF.SizeTy, minElements)));
635f862eb6aSSebastian Redl       }
636f862eb6aSSebastian Redl     }
637f862eb6aSSebastian Redl 
638036f2f6bSJohn McCall     size = numElements;
639036f2f6bSJohn McCall 
640036f2f6bSJohn McCall     // Multiply by the type size if necessary.  This multiplier
641036f2f6bSJohn McCall     // includes all the factors for nested arrays.
6428ed55a54SJohn McCall     //
643036f2f6bSJohn McCall     // This step also causes numElements to be scaled up by the
644036f2f6bSJohn McCall     // nested-array factor if necessary.  Overflow on this computation
645036f2f6bSJohn McCall     // can be ignored because the result shouldn't be used if
646036f2f6bSJohn McCall     // allocation fails.
647036f2f6bSJohn McCall     if (typeSizeMultiplier != 1) {
648036f2f6bSJohn McCall       llvm::Value *umul_with_overflow
6498d375cefSBenjamin Kramer         = CGF.CGM.getIntrinsic(llvm::Intrinsic::umul_with_overflow, CGF.SizeTy);
6508ed55a54SJohn McCall 
651036f2f6bSJohn McCall       llvm::Value *tsmV =
652036f2f6bSJohn McCall         llvm::ConstantInt::get(CGF.SizeTy, typeSizeMultiplier);
653036f2f6bSJohn McCall       llvm::Value *result =
654036f2f6bSJohn McCall         CGF.Builder.CreateCall2(umul_with_overflow, size, tsmV);
6558ed55a54SJohn McCall 
656036f2f6bSJohn McCall       llvm::Value *overflowed = CGF.Builder.CreateExtractValue(result, 1);
657036f2f6bSJohn McCall       if (hasOverflow)
658036f2f6bSJohn McCall         hasOverflow = CGF.Builder.CreateOr(hasOverflow, overflowed);
6598ed55a54SJohn McCall       else
660036f2f6bSJohn McCall         hasOverflow = overflowed;
66159486a2dSAnders Carlsson 
662036f2f6bSJohn McCall       size = CGF.Builder.CreateExtractValue(result, 0);
663036f2f6bSJohn McCall 
664036f2f6bSJohn McCall       // Also scale up numElements by the array size multiplier.
665036f2f6bSJohn McCall       if (arraySizeMultiplier != 1) {
666036f2f6bSJohn McCall         // If the base element type size is 1, then we can re-use the
667036f2f6bSJohn McCall         // multiply we just did.
668036f2f6bSJohn McCall         if (typeSize.isOne()) {
669036f2f6bSJohn McCall           assert(arraySizeMultiplier == typeSizeMultiplier);
670036f2f6bSJohn McCall           numElements = size;
671036f2f6bSJohn McCall 
672036f2f6bSJohn McCall         // Otherwise we need a separate multiply.
673036f2f6bSJohn McCall         } else {
674036f2f6bSJohn McCall           llvm::Value *asmV =
675036f2f6bSJohn McCall             llvm::ConstantInt::get(CGF.SizeTy, arraySizeMultiplier);
676036f2f6bSJohn McCall           numElements = CGF.Builder.CreateMul(numElements, asmV);
677036f2f6bSJohn McCall         }
678036f2f6bSJohn McCall       }
679036f2f6bSJohn McCall     } else {
680036f2f6bSJohn McCall       // numElements doesn't need to be scaled.
681036f2f6bSJohn McCall       assert(arraySizeMultiplier == 1);
682036f2f6bSJohn McCall     }
683036f2f6bSJohn McCall 
684036f2f6bSJohn McCall     // Add in the cookie size if necessary.
685036f2f6bSJohn McCall     if (cookieSize != 0) {
686036f2f6bSJohn McCall       sizeWithoutCookie = size;
687036f2f6bSJohn McCall 
688036f2f6bSJohn McCall       llvm::Value *uadd_with_overflow
6898d375cefSBenjamin Kramer         = CGF.CGM.getIntrinsic(llvm::Intrinsic::uadd_with_overflow, CGF.SizeTy);
690036f2f6bSJohn McCall 
691036f2f6bSJohn McCall       llvm::Value *cookieSizeV = llvm::ConstantInt::get(CGF.SizeTy, cookieSize);
692036f2f6bSJohn McCall       llvm::Value *result =
693036f2f6bSJohn McCall         CGF.Builder.CreateCall2(uadd_with_overflow, size, cookieSizeV);
694036f2f6bSJohn McCall 
695036f2f6bSJohn McCall       llvm::Value *overflowed = CGF.Builder.CreateExtractValue(result, 1);
696036f2f6bSJohn McCall       if (hasOverflow)
697036f2f6bSJohn McCall         hasOverflow = CGF.Builder.CreateOr(hasOverflow, overflowed);
698036f2f6bSJohn McCall       else
699036f2f6bSJohn McCall         hasOverflow = overflowed;
700036f2f6bSJohn McCall 
701036f2f6bSJohn McCall       size = CGF.Builder.CreateExtractValue(result, 0);
702036f2f6bSJohn McCall     }
703036f2f6bSJohn McCall 
704036f2f6bSJohn McCall     // If we had any possibility of dynamic overflow, make a select to
705036f2f6bSJohn McCall     // overwrite 'size' with an all-ones value, which should cause
706036f2f6bSJohn McCall     // operator new to throw.
707036f2f6bSJohn McCall     if (hasOverflow)
708036f2f6bSJohn McCall       size = CGF.Builder.CreateSelect(hasOverflow,
709036f2f6bSJohn McCall                                  llvm::Constant::getAllOnesValue(CGF.SizeTy),
710036f2f6bSJohn McCall                                       size);
711036f2f6bSJohn McCall   }
712036f2f6bSJohn McCall 
713036f2f6bSJohn McCall   if (cookieSize == 0)
714036f2f6bSJohn McCall     sizeWithoutCookie = size;
715036f2f6bSJohn McCall   else
716036f2f6bSJohn McCall     assert(sizeWithoutCookie && "didn't set sizeWithoutCookie?");
717036f2f6bSJohn McCall 
718036f2f6bSJohn McCall   return size;
71959486a2dSAnders Carlsson }
72059486a2dSAnders Carlsson 
721f862eb6aSSebastian Redl static void StoreAnyExprIntoOneUnit(CodeGenFunction &CGF, const Expr *Init,
722f862eb6aSSebastian Redl                                     QualType AllocType, llvm::Value *NewPtr) {
7231c96bc5dSRichard Smith   // FIXME: Refactor with EmitExprAsInit.
72438cd36dbSEli Friedman   CharUnits Alignment = CGF.getContext().getTypeAlignInChars(AllocType);
72547fb9508SJohn McCall   switch (CGF.getEvaluationKind(AllocType)) {
72647fb9508SJohn McCall   case TEK_Scalar:
7278a13c418SCraig Topper     CGF.EmitScalarInit(Init, nullptr, CGF.MakeAddrLValue(NewPtr, AllocType,
728a0544d6fSEli Friedman                                                          Alignment),
7291553b190SJohn McCall                        false);
73047fb9508SJohn McCall     return;
73147fb9508SJohn McCall   case TEK_Complex:
73247fb9508SJohn McCall     CGF.EmitComplexExprIntoLValue(Init, CGF.MakeAddrLValue(NewPtr, AllocType,
73347fb9508SJohn McCall                                                            Alignment),
73447fb9508SJohn McCall                                   /*isInit*/ true);
73547fb9508SJohn McCall     return;
73647fb9508SJohn McCall   case TEK_Aggregate: {
7377a626f63SJohn McCall     AggValueSlot Slot
738c1d85b93SEli Friedman       = AggValueSlot::forAddr(NewPtr, Alignment, AllocType.getQualifiers(),
7398d6fc958SJohn McCall                               AggValueSlot::IsDestructed,
74046759f4fSJohn McCall                               AggValueSlot::DoesNotNeedGCBarriers,
741615ed1a3SChad Rosier                               AggValueSlot::IsNotAliased);
7427a626f63SJohn McCall     CGF.EmitAggExpr(Init, Slot);
74347fb9508SJohn McCall     return;
7447a626f63SJohn McCall   }
745d5202e09SFariborz Jahanian   }
74647fb9508SJohn McCall   llvm_unreachable("bad evaluation kind");
74747fb9508SJohn McCall }
748d5202e09SFariborz Jahanian 
749d5202e09SFariborz Jahanian void
750d5202e09SFariborz Jahanian CodeGenFunction::EmitNewArrayInitializer(const CXXNewExpr *E,
75106a67e2cSRichard Smith                                          QualType ElementType,
75206a67e2cSRichard Smith                                          llvm::Value *BeginPtr,
75306a67e2cSRichard Smith                                          llvm::Value *NumElements,
75406a67e2cSRichard Smith                                          llvm::Value *AllocSizeWithoutCookie) {
75506a67e2cSRichard Smith   // If we have a type with trivial initialization and no initializer,
75606a67e2cSRichard Smith   // there's nothing to do.
7576047f07eSSebastian Redl   if (!E->hasInitializer())
75806a67e2cSRichard Smith     return;
759b66b08efSFariborz Jahanian 
76006a67e2cSRichard Smith   llvm::Value *CurPtr = BeginPtr;
761d5202e09SFariborz Jahanian 
76206a67e2cSRichard Smith   unsigned InitListElements = 0;
763f862eb6aSSebastian Redl 
764f862eb6aSSebastian Redl   const Expr *Init = E->getInitializer();
76506a67e2cSRichard Smith   llvm::AllocaInst *EndOfInit = nullptr;
76606a67e2cSRichard Smith   QualType::DestructionKind DtorKind = ElementType.isDestructedType();
76706a67e2cSRichard Smith   EHScopeStack::stable_iterator Cleanup;
76806a67e2cSRichard Smith   llvm::Instruction *CleanupDominator = nullptr;
7691c96bc5dSRichard Smith 
770f862eb6aSSebastian Redl   // If the initializer is an initializer list, first do the explicit elements.
771f862eb6aSSebastian Redl   if (const InitListExpr *ILE = dyn_cast<InitListExpr>(Init)) {
77206a67e2cSRichard Smith     InitListElements = ILE->getNumInits();
773f62290a1SChad Rosier 
7741c96bc5dSRichard Smith     // If this is a multi-dimensional array new, we will initialize multiple
7751c96bc5dSRichard Smith     // elements with each init list element.
7761c96bc5dSRichard Smith     QualType AllocType = E->getAllocatedType();
7771c96bc5dSRichard Smith     if (const ConstantArrayType *CAT = dyn_cast_or_null<ConstantArrayType>(
7781c96bc5dSRichard Smith             AllocType->getAsArrayTypeUnsafe())) {
77906a67e2cSRichard Smith       unsigned AS = CurPtr->getType()->getPointerAddressSpace();
7801c96bc5dSRichard Smith       llvm::Type *AllocPtrTy = ConvertTypeForMem(AllocType)->getPointerTo(AS);
78106a67e2cSRichard Smith       CurPtr = Builder.CreateBitCast(CurPtr, AllocPtrTy);
78206a67e2cSRichard Smith       InitListElements *= getContext().getConstantArrayElementCount(CAT);
7831c96bc5dSRichard Smith     }
7841c96bc5dSRichard Smith 
78506a67e2cSRichard Smith     // Enter a partial-destruction Cleanup if necessary.
78606a67e2cSRichard Smith     if (needsEHCleanup(DtorKind)) {
78706a67e2cSRichard Smith       // In principle we could tell the Cleanup where we are more
788f62290a1SChad Rosier       // directly, but the control flow can get so varied here that it
789f62290a1SChad Rosier       // would actually be quite complex.  Therefore we go through an
790f62290a1SChad Rosier       // alloca.
79106a67e2cSRichard Smith       EndOfInit = CreateTempAlloca(BeginPtr->getType(), "array.init.end");
79206a67e2cSRichard Smith       CleanupDominator = Builder.CreateStore(BeginPtr, EndOfInit);
79306a67e2cSRichard Smith       pushIrregularPartialArrayCleanup(BeginPtr, EndOfInit, ElementType,
79406a67e2cSRichard Smith                                        getDestroyer(DtorKind));
79506a67e2cSRichard Smith       Cleanup = EHStack.stable_begin();
796f62290a1SChad Rosier     }
797f62290a1SChad Rosier 
798f862eb6aSSebastian Redl     for (unsigned i = 0, e = ILE->getNumInits(); i != e; ++i) {
799f62290a1SChad Rosier       // Tell the cleanup that it needs to destroy up to this
800f62290a1SChad Rosier       // element.  TODO: some of these stores can be trivially
801f62290a1SChad Rosier       // observed to be unnecessary.
80206a67e2cSRichard Smith       if (EndOfInit)
80306a67e2cSRichard Smith         Builder.CreateStore(Builder.CreateBitCast(CurPtr, BeginPtr->getType()),
80406a67e2cSRichard Smith                             EndOfInit);
80506a67e2cSRichard Smith       // FIXME: If the last initializer is an incomplete initializer list for
80606a67e2cSRichard Smith       // an array, and we have an array filler, we can fold together the two
80706a67e2cSRichard Smith       // initialization loops.
8081c96bc5dSRichard Smith       StoreAnyExprIntoOneUnit(*this, ILE->getInit(i),
80906a67e2cSRichard Smith                               ILE->getInit(i)->getType(), CurPtr);
81006a67e2cSRichard Smith       CurPtr = Builder.CreateConstInBoundsGEP1_32(CurPtr, 1, "array.exp.next");
811f862eb6aSSebastian Redl     }
812f862eb6aSSebastian Redl 
813f862eb6aSSebastian Redl     // The remaining elements are filled with the array filler expression.
814f862eb6aSSebastian Redl     Init = ILE->getArrayFiller();
8151c96bc5dSRichard Smith 
81606a67e2cSRichard Smith     // Extract the initializer for the individual array elements by pulling
81706a67e2cSRichard Smith     // out the array filler from all the nested initializer lists. This avoids
81806a67e2cSRichard Smith     // generating a nested loop for the initialization.
81906a67e2cSRichard Smith     while (Init && Init->getType()->isConstantArrayType()) {
82006a67e2cSRichard Smith       auto *SubILE = dyn_cast<InitListExpr>(Init);
82106a67e2cSRichard Smith       if (!SubILE)
82206a67e2cSRichard Smith         break;
82306a67e2cSRichard Smith       assert(SubILE->getNumInits() == 0 && "explicit inits in array filler?");
82406a67e2cSRichard Smith       Init = SubILE->getArrayFiller();
825f862eb6aSSebastian Redl     }
826f862eb6aSSebastian Redl 
82706a67e2cSRichard Smith     // Switch back to initializing one base element at a time.
82806a67e2cSRichard Smith     CurPtr = Builder.CreateBitCast(CurPtr, BeginPtr->getType());
829f62290a1SChad Rosier   }
830e6c980c4SChandler Carruth 
83106a67e2cSRichard Smith   // Attempt to perform zero-initialization using memset.
83206a67e2cSRichard Smith   auto TryMemsetInitialization = [&]() -> bool {
83306a67e2cSRichard Smith     // FIXME: If the type is a pointer-to-data-member under the Itanium ABI,
83406a67e2cSRichard Smith     // we can initialize with a memset to -1.
83506a67e2cSRichard Smith     if (!CGM.getTypes().isZeroInitializable(ElementType))
83606a67e2cSRichard Smith       return false;
837e6c980c4SChandler Carruth 
83806a67e2cSRichard Smith     // Optimization: since zero initialization will just set the memory
83906a67e2cSRichard Smith     // to all zeroes, generate a single memset to do it in one shot.
84006a67e2cSRichard Smith 
84106a67e2cSRichard Smith     // Subtract out the size of any elements we've already initialized.
84206a67e2cSRichard Smith     auto *RemainingSize = AllocSizeWithoutCookie;
84306a67e2cSRichard Smith     if (InitListElements) {
84406a67e2cSRichard Smith       // We know this can't overflow; we check this when doing the allocation.
84506a67e2cSRichard Smith       auto *InitializedSize = llvm::ConstantInt::get(
84606a67e2cSRichard Smith           RemainingSize->getType(),
84706a67e2cSRichard Smith           getContext().getTypeSizeInChars(ElementType).getQuantity() *
84806a67e2cSRichard Smith               InitListElements);
84906a67e2cSRichard Smith       RemainingSize = Builder.CreateSub(RemainingSize, InitializedSize);
85099210dc9SJohn McCall     }
851d5202e09SFariborz Jahanian 
85206a67e2cSRichard Smith     // Create the memset.
85306a67e2cSRichard Smith     CharUnits Alignment = getContext().getTypeAlignInChars(ElementType);
85406a67e2cSRichard Smith     Builder.CreateMemSet(CurPtr, Builder.getInt8(0), RemainingSize,
855705ba07eSKen Dyck                          Alignment.getQuantity(), false);
85606a67e2cSRichard Smith     return true;
85706a67e2cSRichard Smith   };
85805fc5be3SDouglas Gregor 
859454a7cdfSRichard Smith   // If all elements have already been initialized, skip any further
860454a7cdfSRichard Smith   // initialization.
861454a7cdfSRichard Smith   llvm::ConstantInt *ConstNum = dyn_cast<llvm::ConstantInt>(NumElements);
862454a7cdfSRichard Smith   if (ConstNum && ConstNum->getZExtValue() <= InitListElements) {
863454a7cdfSRichard Smith     // If there was a Cleanup, deactivate it.
864454a7cdfSRichard Smith     if (CleanupDominator)
865454a7cdfSRichard Smith       DeactivateCleanupBlock(Cleanup, CleanupDominator);
866454a7cdfSRichard Smith     return;
867454a7cdfSRichard Smith   }
868454a7cdfSRichard Smith 
869454a7cdfSRichard Smith   assert(Init && "have trailing elements to initialize but no initializer");
870454a7cdfSRichard Smith 
87106a67e2cSRichard Smith   // If this is a constructor call, try to optimize it out, and failing that
87206a67e2cSRichard Smith   // emit a single loop to initialize all remaining elements.
873454a7cdfSRichard Smith   if (const CXXConstructExpr *CCE = dyn_cast<CXXConstructExpr>(Init)) {
8746047f07eSSebastian Redl     CXXConstructorDecl *Ctor = CCE->getConstructor();
875d153103cSDouglas Gregor     if (Ctor->isTrivial()) {
87605fc5be3SDouglas Gregor       // If new expression did not specify value-initialization, then there
87705fc5be3SDouglas Gregor       // is no initialization.
8786047f07eSSebastian Redl       if (!CCE->requiresZeroInitialization() || Ctor->getParent()->isEmpty())
87905fc5be3SDouglas Gregor         return;
88005fc5be3SDouglas Gregor 
88106a67e2cSRichard Smith       if (TryMemsetInitialization())
8823a202f60SAnders Carlsson         return;
8833a202f60SAnders Carlsson     }
88405fc5be3SDouglas Gregor 
88506a67e2cSRichard Smith     // Store the new Cleanup position for irregular Cleanups.
88606a67e2cSRichard Smith     //
88706a67e2cSRichard Smith     // FIXME: Share this cleanup with the constructor call emission rather than
88806a67e2cSRichard Smith     // having it create a cleanup of its own.
88906a67e2cSRichard Smith     if (EndOfInit) Builder.CreateStore(CurPtr, EndOfInit);
89006a67e2cSRichard Smith 
89106a67e2cSRichard Smith     // Emit a constructor call loop to initialize the remaining elements.
89206a67e2cSRichard Smith     if (InitListElements)
89306a67e2cSRichard Smith       NumElements = Builder.CreateSub(
89406a67e2cSRichard Smith           NumElements,
89506a67e2cSRichard Smith           llvm::ConstantInt::get(NumElements->getType(), InitListElements));
896*70b9c01bSAlexey Samsonov     EmitCXXAggrConstructorCall(Ctor, NumElements, CurPtr, CCE,
89748ddcf2cSEli Friedman                                CCE->requiresZeroInitialization());
89805fc5be3SDouglas Gregor     return;
8996047f07eSSebastian Redl   }
90006a67e2cSRichard Smith 
90106a67e2cSRichard Smith   // If this is value-initialization, we can usually use memset.
90206a67e2cSRichard Smith   ImplicitValueInitExpr IVIE(ElementType);
903454a7cdfSRichard Smith   if (isa<ImplicitValueInitExpr>(Init)) {
90406a67e2cSRichard Smith     if (TryMemsetInitialization())
90506a67e2cSRichard Smith       return;
90606a67e2cSRichard Smith 
90706a67e2cSRichard Smith     // Switch to an ImplicitValueInitExpr for the element type. This handles
90806a67e2cSRichard Smith     // only one case: multidimensional array new of pointers to members. In
90906a67e2cSRichard Smith     // all other cases, we already have an initializer for the array element.
91006a67e2cSRichard Smith     Init = &IVIE;
91106a67e2cSRichard Smith   }
91206a67e2cSRichard Smith 
91306a67e2cSRichard Smith   // At this point we should have found an initializer for the individual
91406a67e2cSRichard Smith   // elements of the array.
91506a67e2cSRichard Smith   assert(getContext().hasSameUnqualifiedType(ElementType, Init->getType()) &&
91606a67e2cSRichard Smith          "got wrong type of element to initialize");
91706a67e2cSRichard Smith 
918454a7cdfSRichard Smith   // If we have an empty initializer list, we can usually use memset.
919454a7cdfSRichard Smith   if (auto *ILE = dyn_cast<InitListExpr>(Init))
920454a7cdfSRichard Smith     if (ILE->getNumInits() == 0 && TryMemsetInitialization())
921d5202e09SFariborz Jahanian       return;
92259486a2dSAnders Carlsson 
92306a67e2cSRichard Smith   // Create the loop blocks.
92406a67e2cSRichard Smith   llvm::BasicBlock *EntryBB = Builder.GetInsertBlock();
92506a67e2cSRichard Smith   llvm::BasicBlock *LoopBB = createBasicBlock("new.loop");
92606a67e2cSRichard Smith   llvm::BasicBlock *ContBB = createBasicBlock("new.loop.end");
92759486a2dSAnders Carlsson 
92806a67e2cSRichard Smith   // Find the end of the array, hoisted out of the loop.
92906a67e2cSRichard Smith   llvm::Value *EndPtr =
93006a67e2cSRichard Smith     Builder.CreateInBoundsGEP(BeginPtr, NumElements, "array.end");
93106a67e2cSRichard Smith 
93206a67e2cSRichard Smith   // If the number of elements isn't constant, we have to now check if there is
93306a67e2cSRichard Smith   // anything left to initialize.
93406a67e2cSRichard Smith   if (!ConstNum) {
93506a67e2cSRichard Smith     llvm::Value *IsEmpty = Builder.CreateICmpEQ(CurPtr, EndPtr,
93606a67e2cSRichard Smith                                                 "array.isempty");
93706a67e2cSRichard Smith     Builder.CreateCondBr(IsEmpty, ContBB, LoopBB);
93806a67e2cSRichard Smith   }
93906a67e2cSRichard Smith 
94006a67e2cSRichard Smith   // Enter the loop.
94106a67e2cSRichard Smith   EmitBlock(LoopBB);
94206a67e2cSRichard Smith 
94306a67e2cSRichard Smith   // Set up the current-element phi.
94406a67e2cSRichard Smith   llvm::PHINode *CurPtrPhi =
94506a67e2cSRichard Smith     Builder.CreatePHI(CurPtr->getType(), 2, "array.cur");
94606a67e2cSRichard Smith   CurPtrPhi->addIncoming(CurPtr, EntryBB);
94706a67e2cSRichard Smith   CurPtr = CurPtrPhi;
94806a67e2cSRichard Smith 
94906a67e2cSRichard Smith   // Store the new Cleanup position for irregular Cleanups.
95006a67e2cSRichard Smith   if (EndOfInit) Builder.CreateStore(CurPtr, EndOfInit);
95106a67e2cSRichard Smith 
95206a67e2cSRichard Smith   // Enter a partial-destruction Cleanup if necessary.
95306a67e2cSRichard Smith   if (!CleanupDominator && needsEHCleanup(DtorKind)) {
95406a67e2cSRichard Smith     pushRegularPartialArrayCleanup(BeginPtr, CurPtr, ElementType,
95506a67e2cSRichard Smith                                    getDestroyer(DtorKind));
95606a67e2cSRichard Smith     Cleanup = EHStack.stable_begin();
95706a67e2cSRichard Smith     CleanupDominator = Builder.CreateUnreachable();
95806a67e2cSRichard Smith   }
95906a67e2cSRichard Smith 
96006a67e2cSRichard Smith   // Emit the initializer into this element.
96106a67e2cSRichard Smith   StoreAnyExprIntoOneUnit(*this, Init, Init->getType(), CurPtr);
96206a67e2cSRichard Smith 
96306a67e2cSRichard Smith   // Leave the Cleanup if we entered one.
96406a67e2cSRichard Smith   if (CleanupDominator) {
96506a67e2cSRichard Smith     DeactivateCleanupBlock(Cleanup, CleanupDominator);
96606a67e2cSRichard Smith     CleanupDominator->eraseFromParent();
96706a67e2cSRichard Smith   }
96806a67e2cSRichard Smith 
96906a67e2cSRichard Smith   // Advance to the next element by adjusting the pointer type as necessary.
97006a67e2cSRichard Smith   llvm::Value *NextPtr =
97106a67e2cSRichard Smith       Builder.CreateConstInBoundsGEP1_32(CurPtr, 1, "array.next");
97206a67e2cSRichard Smith 
97306a67e2cSRichard Smith   // Check whether we've gotten to the end of the array and, if so,
97406a67e2cSRichard Smith   // exit the loop.
97506a67e2cSRichard Smith   llvm::Value *IsEnd = Builder.CreateICmpEQ(NextPtr, EndPtr, "array.atend");
97606a67e2cSRichard Smith   Builder.CreateCondBr(IsEnd, ContBB, LoopBB);
97706a67e2cSRichard Smith   CurPtrPhi->addIncoming(NextPtr, Builder.GetInsertBlock());
97806a67e2cSRichard Smith 
97906a67e2cSRichard Smith   EmitBlock(ContBB);
98006a67e2cSRichard Smith }
98106a67e2cSRichard Smith 
98206a67e2cSRichard Smith static void EmitNewInitializer(CodeGenFunction &CGF, const CXXNewExpr *E,
98306a67e2cSRichard Smith                                QualType ElementType,
98406a67e2cSRichard Smith                                llvm::Value *NewPtr,
98506a67e2cSRichard Smith                                llvm::Value *NumElements,
98606a67e2cSRichard Smith                                llvm::Value *AllocSizeWithoutCookie) {
98706a67e2cSRichard Smith   if (E->isArray())
98806a67e2cSRichard Smith     CGF.EmitNewArrayInitializer(E, ElementType, NewPtr, NumElements,
98906a67e2cSRichard Smith                                 AllocSizeWithoutCookie);
99006a67e2cSRichard Smith   else if (const Expr *Init = E->getInitializer())
991f862eb6aSSebastian Redl     StoreAnyExprIntoOneUnit(CGF, Init, E->getAllocatedType(), NewPtr);
99259486a2dSAnders Carlsson }
99359486a2dSAnders Carlsson 
9948d0dc31dSRichard Smith /// Emit a call to an operator new or operator delete function, as implicitly
9958d0dc31dSRichard Smith /// created by new-expressions and delete-expressions.
9968d0dc31dSRichard Smith static RValue EmitNewDeleteCall(CodeGenFunction &CGF,
9978d0dc31dSRichard Smith                                 const FunctionDecl *Callee,
9988d0dc31dSRichard Smith                                 const FunctionProtoType *CalleeType,
9998d0dc31dSRichard Smith                                 const CallArgList &Args) {
10008d0dc31dSRichard Smith   llvm::Instruction *CallOrInvoke;
10011235a8daSRichard Smith   llvm::Value *CalleeAddr = CGF.CGM.GetAddrOfFunction(Callee);
10028d0dc31dSRichard Smith   RValue RV =
10038d0dc31dSRichard Smith       CGF.EmitCall(CGF.CGM.getTypes().arrangeFreeFunctionCall(Args, CalleeType),
10041235a8daSRichard Smith                    CalleeAddr, ReturnValueSlot(), Args,
10058d0dc31dSRichard Smith                    Callee, &CallOrInvoke);
10068d0dc31dSRichard Smith 
10078d0dc31dSRichard Smith   /// C++1y [expr.new]p10:
10088d0dc31dSRichard Smith   ///   [In a new-expression,] an implementation is allowed to omit a call
10098d0dc31dSRichard Smith   ///   to a replaceable global allocation function.
10108d0dc31dSRichard Smith   ///
10118d0dc31dSRichard Smith   /// We model such elidable calls with the 'builtin' attribute.
10126956d587SRafael Espindola   llvm::Function *Fn = dyn_cast<llvm::Function>(CalleeAddr);
10131235a8daSRichard Smith   if (Callee->isReplaceableGlobalAllocationFunction() &&
10146956d587SRafael Espindola       Fn && Fn->hasFnAttribute(llvm::Attribute::NoBuiltin)) {
10158d0dc31dSRichard Smith     // FIXME: Add addAttribute to CallSite.
10168d0dc31dSRichard Smith     if (llvm::CallInst *CI = dyn_cast<llvm::CallInst>(CallOrInvoke))
10178d0dc31dSRichard Smith       CI->addAttribute(llvm::AttributeSet::FunctionIndex,
10188d0dc31dSRichard Smith                        llvm::Attribute::Builtin);
10198d0dc31dSRichard Smith     else if (llvm::InvokeInst *II = dyn_cast<llvm::InvokeInst>(CallOrInvoke))
10208d0dc31dSRichard Smith       II->addAttribute(llvm::AttributeSet::FunctionIndex,
10218d0dc31dSRichard Smith                        llvm::Attribute::Builtin);
10228d0dc31dSRichard Smith     else
10238d0dc31dSRichard Smith       llvm_unreachable("unexpected kind of call instruction");
10248d0dc31dSRichard Smith   }
10258d0dc31dSRichard Smith 
10268d0dc31dSRichard Smith   return RV;
10278d0dc31dSRichard Smith }
10288d0dc31dSRichard Smith 
1029760520bcSRichard Smith RValue CodeGenFunction::EmitBuiltinNewDeleteCall(const FunctionProtoType *Type,
1030760520bcSRichard Smith                                                  const Expr *Arg,
1031760520bcSRichard Smith                                                  bool IsDelete) {
1032760520bcSRichard Smith   CallArgList Args;
1033760520bcSRichard Smith   const Stmt *ArgS = Arg;
1034760520bcSRichard Smith   EmitCallArgs(Args, *Type->param_type_begin(),
1035760520bcSRichard Smith                ConstExprIterator(&ArgS), ConstExprIterator(&ArgS + 1));
1036760520bcSRichard Smith   // Find the allocation or deallocation function that we're calling.
1037760520bcSRichard Smith   ASTContext &Ctx = getContext();
1038760520bcSRichard Smith   DeclarationName Name = Ctx.DeclarationNames
1039760520bcSRichard Smith       .getCXXOperatorName(IsDelete ? OO_Delete : OO_New);
1040760520bcSRichard Smith   for (auto *Decl : Ctx.getTranslationUnitDecl()->lookup(Name))
1041599bed75SRichard Smith     if (auto *FD = dyn_cast<FunctionDecl>(Decl))
1042599bed75SRichard Smith       if (Ctx.hasSameType(FD->getType(), QualType(Type, 0)))
1043760520bcSRichard Smith         return EmitNewDeleteCall(*this, cast<FunctionDecl>(Decl), Type, Args);
1044760520bcSRichard Smith   llvm_unreachable("predeclared global operator new/delete is missing");
1045760520bcSRichard Smith }
1046760520bcSRichard Smith 
1047824c2f53SJohn McCall namespace {
1048824c2f53SJohn McCall   /// A cleanup to call the given 'operator delete' function upon
1049824c2f53SJohn McCall   /// abnormal exit from a new expression.
1050824c2f53SJohn McCall   class CallDeleteDuringNew : public EHScopeStack::Cleanup {
1051824c2f53SJohn McCall     size_t NumPlacementArgs;
1052824c2f53SJohn McCall     const FunctionDecl *OperatorDelete;
1053824c2f53SJohn McCall     llvm::Value *Ptr;
1054824c2f53SJohn McCall     llvm::Value *AllocSize;
1055824c2f53SJohn McCall 
1056824c2f53SJohn McCall     RValue *getPlacementArgs() { return reinterpret_cast<RValue*>(this+1); }
1057824c2f53SJohn McCall 
1058824c2f53SJohn McCall   public:
1059824c2f53SJohn McCall     static size_t getExtraSize(size_t NumPlacementArgs) {
1060824c2f53SJohn McCall       return NumPlacementArgs * sizeof(RValue);
1061824c2f53SJohn McCall     }
1062824c2f53SJohn McCall 
1063824c2f53SJohn McCall     CallDeleteDuringNew(size_t NumPlacementArgs,
1064824c2f53SJohn McCall                         const FunctionDecl *OperatorDelete,
1065824c2f53SJohn McCall                         llvm::Value *Ptr,
1066824c2f53SJohn McCall                         llvm::Value *AllocSize)
1067824c2f53SJohn McCall       : NumPlacementArgs(NumPlacementArgs), OperatorDelete(OperatorDelete),
1068824c2f53SJohn McCall         Ptr(Ptr), AllocSize(AllocSize) {}
1069824c2f53SJohn McCall 
1070824c2f53SJohn McCall     void setPlacementArg(unsigned I, RValue Arg) {
1071824c2f53SJohn McCall       assert(I < NumPlacementArgs && "index out of range");
1072824c2f53SJohn McCall       getPlacementArgs()[I] = Arg;
1073824c2f53SJohn McCall     }
1074824c2f53SJohn McCall 
10754f12f10dSCraig Topper     void Emit(CodeGenFunction &CGF, Flags flags) override {
1076824c2f53SJohn McCall       const FunctionProtoType *FPT
1077824c2f53SJohn McCall         = OperatorDelete->getType()->getAs<FunctionProtoType>();
10789cacbabdSAlp Toker       assert(FPT->getNumParams() == NumPlacementArgs + 1 ||
10799cacbabdSAlp Toker              (FPT->getNumParams() == 2 && NumPlacementArgs == 0));
1080824c2f53SJohn McCall 
1081824c2f53SJohn McCall       CallArgList DeleteArgs;
1082824c2f53SJohn McCall 
1083824c2f53SJohn McCall       // The first argument is always a void*.
10849cacbabdSAlp Toker       FunctionProtoType::param_type_iterator AI = FPT->param_type_begin();
108543dca6a8SEli Friedman       DeleteArgs.add(RValue::get(Ptr), *AI++);
1086824c2f53SJohn McCall 
1087824c2f53SJohn McCall       // A member 'operator delete' can take an extra 'size_t' argument.
10889cacbabdSAlp Toker       if (FPT->getNumParams() == NumPlacementArgs + 2)
108943dca6a8SEli Friedman         DeleteArgs.add(RValue::get(AllocSize), *AI++);
1090824c2f53SJohn McCall 
1091824c2f53SJohn McCall       // Pass the rest of the arguments, which must match exactly.
1092824c2f53SJohn McCall       for (unsigned I = 0; I != NumPlacementArgs; ++I)
109343dca6a8SEli Friedman         DeleteArgs.add(getPlacementArgs()[I], *AI++);
1094824c2f53SJohn McCall 
1095824c2f53SJohn McCall       // Call 'operator delete'.
10968d0dc31dSRichard Smith       EmitNewDeleteCall(CGF, OperatorDelete, FPT, DeleteArgs);
1097824c2f53SJohn McCall     }
1098824c2f53SJohn McCall   };
10997f9c92a9SJohn McCall 
11007f9c92a9SJohn McCall   /// A cleanup to call the given 'operator delete' function upon
11017f9c92a9SJohn McCall   /// abnormal exit from a new expression when the new expression is
11027f9c92a9SJohn McCall   /// conditional.
11037f9c92a9SJohn McCall   class CallDeleteDuringConditionalNew : public EHScopeStack::Cleanup {
11047f9c92a9SJohn McCall     size_t NumPlacementArgs;
11057f9c92a9SJohn McCall     const FunctionDecl *OperatorDelete;
1106cb5f77f0SJohn McCall     DominatingValue<RValue>::saved_type Ptr;
1107cb5f77f0SJohn McCall     DominatingValue<RValue>::saved_type AllocSize;
11087f9c92a9SJohn McCall 
1109cb5f77f0SJohn McCall     DominatingValue<RValue>::saved_type *getPlacementArgs() {
1110cb5f77f0SJohn McCall       return reinterpret_cast<DominatingValue<RValue>::saved_type*>(this+1);
11117f9c92a9SJohn McCall     }
11127f9c92a9SJohn McCall 
11137f9c92a9SJohn McCall   public:
11147f9c92a9SJohn McCall     static size_t getExtraSize(size_t NumPlacementArgs) {
1115cb5f77f0SJohn McCall       return NumPlacementArgs * sizeof(DominatingValue<RValue>::saved_type);
11167f9c92a9SJohn McCall     }
11177f9c92a9SJohn McCall 
11187f9c92a9SJohn McCall     CallDeleteDuringConditionalNew(size_t NumPlacementArgs,
11197f9c92a9SJohn McCall                                    const FunctionDecl *OperatorDelete,
1120cb5f77f0SJohn McCall                                    DominatingValue<RValue>::saved_type Ptr,
1121cb5f77f0SJohn McCall                               DominatingValue<RValue>::saved_type AllocSize)
11227f9c92a9SJohn McCall       : NumPlacementArgs(NumPlacementArgs), OperatorDelete(OperatorDelete),
11237f9c92a9SJohn McCall         Ptr(Ptr), AllocSize(AllocSize) {}
11247f9c92a9SJohn McCall 
1125cb5f77f0SJohn McCall     void setPlacementArg(unsigned I, DominatingValue<RValue>::saved_type Arg) {
11267f9c92a9SJohn McCall       assert(I < NumPlacementArgs && "index out of range");
11277f9c92a9SJohn McCall       getPlacementArgs()[I] = Arg;
11287f9c92a9SJohn McCall     }
11297f9c92a9SJohn McCall 
11304f12f10dSCraig Topper     void Emit(CodeGenFunction &CGF, Flags flags) override {
11317f9c92a9SJohn McCall       const FunctionProtoType *FPT
11327f9c92a9SJohn McCall         = OperatorDelete->getType()->getAs<FunctionProtoType>();
11339cacbabdSAlp Toker       assert(FPT->getNumParams() == NumPlacementArgs + 1 ||
11349cacbabdSAlp Toker              (FPT->getNumParams() == 2 && NumPlacementArgs == 0));
11357f9c92a9SJohn McCall 
11367f9c92a9SJohn McCall       CallArgList DeleteArgs;
11377f9c92a9SJohn McCall 
11387f9c92a9SJohn McCall       // The first argument is always a void*.
11399cacbabdSAlp Toker       FunctionProtoType::param_type_iterator AI = FPT->param_type_begin();
114043dca6a8SEli Friedman       DeleteArgs.add(Ptr.restore(CGF), *AI++);
11417f9c92a9SJohn McCall 
11427f9c92a9SJohn McCall       // A member 'operator delete' can take an extra 'size_t' argument.
11439cacbabdSAlp Toker       if (FPT->getNumParams() == NumPlacementArgs + 2) {
1144cb5f77f0SJohn McCall         RValue RV = AllocSize.restore(CGF);
114543dca6a8SEli Friedman         DeleteArgs.add(RV, *AI++);
11467f9c92a9SJohn McCall       }
11477f9c92a9SJohn McCall 
11487f9c92a9SJohn McCall       // Pass the rest of the arguments, which must match exactly.
11497f9c92a9SJohn McCall       for (unsigned I = 0; I != NumPlacementArgs; ++I) {
1150cb5f77f0SJohn McCall         RValue RV = getPlacementArgs()[I].restore(CGF);
115143dca6a8SEli Friedman         DeleteArgs.add(RV, *AI++);
11527f9c92a9SJohn McCall       }
11537f9c92a9SJohn McCall 
11547f9c92a9SJohn McCall       // Call 'operator delete'.
11558d0dc31dSRichard Smith       EmitNewDeleteCall(CGF, OperatorDelete, FPT, DeleteArgs);
11567f9c92a9SJohn McCall     }
11577f9c92a9SJohn McCall   };
11587f9c92a9SJohn McCall }
11597f9c92a9SJohn McCall 
11607f9c92a9SJohn McCall /// Enter a cleanup to call 'operator delete' if the initializer in a
11617f9c92a9SJohn McCall /// new-expression throws.
11627f9c92a9SJohn McCall static void EnterNewDeleteCleanup(CodeGenFunction &CGF,
11637f9c92a9SJohn McCall                                   const CXXNewExpr *E,
11647f9c92a9SJohn McCall                                   llvm::Value *NewPtr,
11657f9c92a9SJohn McCall                                   llvm::Value *AllocSize,
11667f9c92a9SJohn McCall                                   const CallArgList &NewArgs) {
11677f9c92a9SJohn McCall   // If we're not inside a conditional branch, then the cleanup will
11687f9c92a9SJohn McCall   // dominate and we can do the easier (and more efficient) thing.
11697f9c92a9SJohn McCall   if (!CGF.isInConditionalBranch()) {
11707f9c92a9SJohn McCall     CallDeleteDuringNew *Cleanup = CGF.EHStack
11717f9c92a9SJohn McCall       .pushCleanupWithExtra<CallDeleteDuringNew>(EHCleanup,
11727f9c92a9SJohn McCall                                                  E->getNumPlacementArgs(),
11737f9c92a9SJohn McCall                                                  E->getOperatorDelete(),
11747f9c92a9SJohn McCall                                                  NewPtr, AllocSize);
11757f9c92a9SJohn McCall     for (unsigned I = 0, N = E->getNumPlacementArgs(); I != N; ++I)
1176f4258eb4SEli Friedman       Cleanup->setPlacementArg(I, NewArgs[I+1].RV);
11777f9c92a9SJohn McCall 
11787f9c92a9SJohn McCall     return;
11797f9c92a9SJohn McCall   }
11807f9c92a9SJohn McCall 
11817f9c92a9SJohn McCall   // Otherwise, we need to save all this stuff.
1182cb5f77f0SJohn McCall   DominatingValue<RValue>::saved_type SavedNewPtr =
1183cb5f77f0SJohn McCall     DominatingValue<RValue>::save(CGF, RValue::get(NewPtr));
1184cb5f77f0SJohn McCall   DominatingValue<RValue>::saved_type SavedAllocSize =
1185cb5f77f0SJohn McCall     DominatingValue<RValue>::save(CGF, RValue::get(AllocSize));
11867f9c92a9SJohn McCall 
11877f9c92a9SJohn McCall   CallDeleteDuringConditionalNew *Cleanup = CGF.EHStack
1188f4beacd0SJohn McCall     .pushCleanupWithExtra<CallDeleteDuringConditionalNew>(EHCleanup,
11897f9c92a9SJohn McCall                                                  E->getNumPlacementArgs(),
11907f9c92a9SJohn McCall                                                  E->getOperatorDelete(),
11917f9c92a9SJohn McCall                                                  SavedNewPtr,
11927f9c92a9SJohn McCall                                                  SavedAllocSize);
11937f9c92a9SJohn McCall   for (unsigned I = 0, N = E->getNumPlacementArgs(); I != N; ++I)
1194cb5f77f0SJohn McCall     Cleanup->setPlacementArg(I,
1195f4258eb4SEli Friedman                      DominatingValue<RValue>::save(CGF, NewArgs[I+1].RV));
11967f9c92a9SJohn McCall 
1197f4beacd0SJohn McCall   CGF.initFullExprCleanup();
1198824c2f53SJohn McCall }
1199824c2f53SJohn McCall 
120059486a2dSAnders Carlsson llvm::Value *CodeGenFunction::EmitCXXNewExpr(const CXXNewExpr *E) {
120175f9498aSJohn McCall   // The element type being allocated.
120275f9498aSJohn McCall   QualType allocType = getContext().getBaseElementType(E->getAllocatedType());
12038ed55a54SJohn McCall 
120475f9498aSJohn McCall   // 1. Build a call to the allocation function.
120575f9498aSJohn McCall   FunctionDecl *allocator = E->getOperatorNew();
120675f9498aSJohn McCall   const FunctionProtoType *allocatorType =
120775f9498aSJohn McCall     allocator->getType()->castAs<FunctionProtoType>();
120859486a2dSAnders Carlsson 
120975f9498aSJohn McCall   CallArgList allocatorArgs;
121059486a2dSAnders Carlsson 
121159486a2dSAnders Carlsson   // The allocation size is the first argument.
121275f9498aSJohn McCall   QualType sizeType = getContext().getSizeType();
121359486a2dSAnders Carlsson 
1214f862eb6aSSebastian Redl   // If there is a brace-initializer, cannot allocate fewer elements than inits.
1215f862eb6aSSebastian Redl   unsigned minElements = 0;
1216f862eb6aSSebastian Redl   if (E->isArray() && E->hasInitializer()) {
1217f862eb6aSSebastian Redl     if (const InitListExpr *ILE = dyn_cast<InitListExpr>(E->getInitializer()))
1218f862eb6aSSebastian Redl       minElements = ILE->getNumInits();
1219f862eb6aSSebastian Redl   }
1220f862eb6aSSebastian Redl 
12218a13c418SCraig Topper   llvm::Value *numElements = nullptr;
12228a13c418SCraig Topper   llvm::Value *allocSizeWithoutCookie = nullptr;
122375f9498aSJohn McCall   llvm::Value *allocSize =
1224f862eb6aSSebastian Redl     EmitCXXNewAllocSize(*this, E, minElements, numElements,
1225f862eb6aSSebastian Redl                         allocSizeWithoutCookie);
122659486a2dSAnders Carlsson 
122743dca6a8SEli Friedman   allocatorArgs.add(RValue::get(allocSize), sizeType);
122859486a2dSAnders Carlsson 
122959486a2dSAnders Carlsson   // We start at 1 here because the first argument (the allocation size)
123059486a2dSAnders Carlsson   // has already been emitted.
1231739756c0SReid Kleckner   EmitCallArgs(allocatorArgs, allocatorType->isVariadic(),
12329cacbabdSAlp Toker                allocatorType->param_type_begin() + 1,
12339cacbabdSAlp Toker                allocatorType->param_type_end(), E->placement_arg_begin(),
1234739756c0SReid Kleckner                E->placement_arg_end());
123559486a2dSAnders Carlsson 
12367ec4b434SJohn McCall   // Emit the allocation call.  If the allocator is a global placement
12377ec4b434SJohn McCall   // operator, just "inline" it directly.
12387ec4b434SJohn McCall   RValue RV;
12397ec4b434SJohn McCall   if (allocator->isReservedGlobalPlacementOperator()) {
12407ec4b434SJohn McCall     assert(allocatorArgs.size() == 2);
12417ec4b434SJohn McCall     RV = allocatorArgs[1].RV;
12427ec4b434SJohn McCall     // TODO: kill any unnecessary computations done for the size
12437ec4b434SJohn McCall     // argument.
12447ec4b434SJohn McCall   } else {
12458d0dc31dSRichard Smith     RV = EmitNewDeleteCall(*this, allocator, allocatorType, allocatorArgs);
12467ec4b434SJohn McCall   }
124759486a2dSAnders Carlsson 
124875f9498aSJohn McCall   // Emit a null check on the allocation result if the allocation
124975f9498aSJohn McCall   // function is allowed to return null (because it has a non-throwing
125075f9498aSJohn McCall   // exception spec; for this part, we inline
125175f9498aSJohn McCall   // CXXNewExpr::shouldNullCheckAllocation()) and we have an
125275f9498aSJohn McCall   // interesting initializer.
125331ad754cSSebastian Redl   bool nullCheck = allocatorType->isNothrow(getContext()) &&
12546047f07eSSebastian Redl     (!allocType.isPODType(getContext()) || E->hasInitializer());
125559486a2dSAnders Carlsson 
12568a13c418SCraig Topper   llvm::BasicBlock *nullCheckBB = nullptr;
12578a13c418SCraig Topper   llvm::BasicBlock *contBB = nullptr;
125859486a2dSAnders Carlsson 
125975f9498aSJohn McCall   llvm::Value *allocation = RV.getScalarVal();
1260ea2fea2aSMicah Villmow   unsigned AS = allocation->getType()->getPointerAddressSpace();
126159486a2dSAnders Carlsson 
1262f7dcf320SJohn McCall   // The null-check means that the initializer is conditionally
1263f7dcf320SJohn McCall   // evaluated.
1264f7dcf320SJohn McCall   ConditionalEvaluation conditional(*this);
1265f7dcf320SJohn McCall 
126675f9498aSJohn McCall   if (nullCheck) {
1267f7dcf320SJohn McCall     conditional.begin(*this);
126875f9498aSJohn McCall 
126975f9498aSJohn McCall     nullCheckBB = Builder.GetInsertBlock();
127075f9498aSJohn McCall     llvm::BasicBlock *notNullBB = createBasicBlock("new.notnull");
127175f9498aSJohn McCall     contBB = createBasicBlock("new.cont");
127275f9498aSJohn McCall 
127375f9498aSJohn McCall     llvm::Value *isNull = Builder.CreateIsNull(allocation, "new.isnull");
127475f9498aSJohn McCall     Builder.CreateCondBr(isNull, contBB, notNullBB);
127575f9498aSJohn McCall     EmitBlock(notNullBB);
127659486a2dSAnders Carlsson   }
127759486a2dSAnders Carlsson 
1278824c2f53SJohn McCall   // If there's an operator delete, enter a cleanup to call it if an
1279824c2f53SJohn McCall   // exception is thrown.
128075f9498aSJohn McCall   EHScopeStack::stable_iterator operatorDeleteCleanup;
12818a13c418SCraig Topper   llvm::Instruction *cleanupDominator = nullptr;
12827ec4b434SJohn McCall   if (E->getOperatorDelete() &&
12837ec4b434SJohn McCall       !E->getOperatorDelete()->isReservedGlobalPlacementOperator()) {
128475f9498aSJohn McCall     EnterNewDeleteCleanup(*this, E, allocation, allocSize, allocatorArgs);
128575f9498aSJohn McCall     operatorDeleteCleanup = EHStack.stable_begin();
1286f4beacd0SJohn McCall     cleanupDominator = Builder.CreateUnreachable();
1287824c2f53SJohn McCall   }
1288824c2f53SJohn McCall 
1289cf9b1f65SEli Friedman   assert((allocSize == allocSizeWithoutCookie) ==
1290cf9b1f65SEli Friedman          CalculateCookiePadding(*this, E).isZero());
1291cf9b1f65SEli Friedman   if (allocSize != allocSizeWithoutCookie) {
1292cf9b1f65SEli Friedman     assert(E->isArray());
1293cf9b1f65SEli Friedman     allocation = CGM.getCXXABI().InitializeArrayCookie(*this, allocation,
1294cf9b1f65SEli Friedman                                                        numElements,
1295cf9b1f65SEli Friedman                                                        E, allocType);
1296cf9b1f65SEli Friedman   }
1297cf9b1f65SEli Friedman 
12982192fe50SChris Lattner   llvm::Type *elementPtrTy
129975f9498aSJohn McCall     = ConvertTypeForMem(allocType)->getPointerTo(AS);
130075f9498aSJohn McCall   llvm::Value *result = Builder.CreateBitCast(allocation, elementPtrTy);
1301824c2f53SJohn McCall 
130299210dc9SJohn McCall   EmitNewInitializer(*this, E, allocType, result, numElements,
130399210dc9SJohn McCall                      allocSizeWithoutCookie);
13048ed55a54SJohn McCall   if (E->isArray()) {
13058ed55a54SJohn McCall     // NewPtr is a pointer to the base element type.  If we're
13068ed55a54SJohn McCall     // allocating an array of arrays, we'll need to cast back to the
13078ed55a54SJohn McCall     // array pointer type.
13082192fe50SChris Lattner     llvm::Type *resultType = ConvertTypeForMem(E->getType());
130975f9498aSJohn McCall     if (result->getType() != resultType)
131075f9498aSJohn McCall       result = Builder.CreateBitCast(result, resultType);
131147b4629bSFariborz Jahanian   }
131259486a2dSAnders Carlsson 
1313824c2f53SJohn McCall   // Deactivate the 'operator delete' cleanup if we finished
1314824c2f53SJohn McCall   // initialization.
1315f4beacd0SJohn McCall   if (operatorDeleteCleanup.isValid()) {
1316f4beacd0SJohn McCall     DeactivateCleanupBlock(operatorDeleteCleanup, cleanupDominator);
1317f4beacd0SJohn McCall     cleanupDominator->eraseFromParent();
1318f4beacd0SJohn McCall   }
1319824c2f53SJohn McCall 
132075f9498aSJohn McCall   if (nullCheck) {
1321f7dcf320SJohn McCall     conditional.end(*this);
1322f7dcf320SJohn McCall 
132375f9498aSJohn McCall     llvm::BasicBlock *notNullBB = Builder.GetInsertBlock();
132475f9498aSJohn McCall     EmitBlock(contBB);
132559486a2dSAnders Carlsson 
132620c0f02cSJay Foad     llvm::PHINode *PHI = Builder.CreatePHI(result->getType(), 2);
132775f9498aSJohn McCall     PHI->addIncoming(result, notNullBB);
132875f9498aSJohn McCall     PHI->addIncoming(llvm::Constant::getNullValue(result->getType()),
132975f9498aSJohn McCall                      nullCheckBB);
133059486a2dSAnders Carlsson 
133175f9498aSJohn McCall     result = PHI;
133259486a2dSAnders Carlsson   }
133359486a2dSAnders Carlsson 
133475f9498aSJohn McCall   return result;
133559486a2dSAnders Carlsson }
133659486a2dSAnders Carlsson 
133759486a2dSAnders Carlsson void CodeGenFunction::EmitDeleteCall(const FunctionDecl *DeleteFD,
133859486a2dSAnders Carlsson                                      llvm::Value *Ptr,
133959486a2dSAnders Carlsson                                      QualType DeleteTy) {
13408ed55a54SJohn McCall   assert(DeleteFD->getOverloadedOperator() == OO_Delete);
13418ed55a54SJohn McCall 
134259486a2dSAnders Carlsson   const FunctionProtoType *DeleteFTy =
134359486a2dSAnders Carlsson     DeleteFD->getType()->getAs<FunctionProtoType>();
134459486a2dSAnders Carlsson 
134559486a2dSAnders Carlsson   CallArgList DeleteArgs;
134659486a2dSAnders Carlsson 
134721122cf6SAnders Carlsson   // Check if we need to pass the size to the delete operator.
13488a13c418SCraig Topper   llvm::Value *Size = nullptr;
134921122cf6SAnders Carlsson   QualType SizeTy;
13509cacbabdSAlp Toker   if (DeleteFTy->getNumParams() == 2) {
13519cacbabdSAlp Toker     SizeTy = DeleteFTy->getParamType(1);
13527df3cbebSKen Dyck     CharUnits DeleteTypeSize = getContext().getTypeSizeInChars(DeleteTy);
13537df3cbebSKen Dyck     Size = llvm::ConstantInt::get(ConvertType(SizeTy),
13547df3cbebSKen Dyck                                   DeleteTypeSize.getQuantity());
135521122cf6SAnders Carlsson   }
135621122cf6SAnders Carlsson 
13579cacbabdSAlp Toker   QualType ArgTy = DeleteFTy->getParamType(0);
135859486a2dSAnders Carlsson   llvm::Value *DeletePtr = Builder.CreateBitCast(Ptr, ConvertType(ArgTy));
135943dca6a8SEli Friedman   DeleteArgs.add(RValue::get(DeletePtr), ArgTy);
136059486a2dSAnders Carlsson 
136121122cf6SAnders Carlsson   if (Size)
136243dca6a8SEli Friedman     DeleteArgs.add(RValue::get(Size), SizeTy);
136359486a2dSAnders Carlsson 
136459486a2dSAnders Carlsson   // Emit the call to delete.
13658d0dc31dSRichard Smith   EmitNewDeleteCall(*this, DeleteFD, DeleteFTy, DeleteArgs);
136659486a2dSAnders Carlsson }
136759486a2dSAnders Carlsson 
13688ed55a54SJohn McCall namespace {
13698ed55a54SJohn McCall   /// Calls the given 'operator delete' on a single object.
13708ed55a54SJohn McCall   struct CallObjectDelete : EHScopeStack::Cleanup {
13718ed55a54SJohn McCall     llvm::Value *Ptr;
13728ed55a54SJohn McCall     const FunctionDecl *OperatorDelete;
13738ed55a54SJohn McCall     QualType ElementType;
13748ed55a54SJohn McCall 
13758ed55a54SJohn McCall     CallObjectDelete(llvm::Value *Ptr,
13768ed55a54SJohn McCall                      const FunctionDecl *OperatorDelete,
13778ed55a54SJohn McCall                      QualType ElementType)
13788ed55a54SJohn McCall       : Ptr(Ptr), OperatorDelete(OperatorDelete), ElementType(ElementType) {}
13798ed55a54SJohn McCall 
13804f12f10dSCraig Topper     void Emit(CodeGenFunction &CGF, Flags flags) override {
13818ed55a54SJohn McCall       CGF.EmitDeleteCall(OperatorDelete, Ptr, ElementType);
13828ed55a54SJohn McCall     }
13838ed55a54SJohn McCall   };
13848ed55a54SJohn McCall }
13858ed55a54SJohn McCall 
13868ed55a54SJohn McCall /// Emit the code for deleting a single object.
13878ed55a54SJohn McCall static void EmitObjectDelete(CodeGenFunction &CGF,
13888ed55a54SJohn McCall                              const FunctionDecl *OperatorDelete,
13898ed55a54SJohn McCall                              llvm::Value *Ptr,
13901c2e20d7SDouglas Gregor                              QualType ElementType,
13911c2e20d7SDouglas Gregor                              bool UseGlobalDelete) {
13928ed55a54SJohn McCall   // Find the destructor for the type, if applicable.  If the
13938ed55a54SJohn McCall   // destructor is virtual, we'll just emit the vcall and return.
13948a13c418SCraig Topper   const CXXDestructorDecl *Dtor = nullptr;
13958ed55a54SJohn McCall   if (const RecordType *RT = ElementType->getAs<RecordType>()) {
13968ed55a54SJohn McCall     CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl());
1397b23533dbSEli Friedman     if (RD->hasDefinition() && !RD->hasTrivialDestructor()) {
13988ed55a54SJohn McCall       Dtor = RD->getDestructor();
13998ed55a54SJohn McCall 
14008ed55a54SJohn McCall       if (Dtor->isVirtual()) {
14011c2e20d7SDouglas Gregor         if (UseGlobalDelete) {
14021c2e20d7SDouglas Gregor           // If we're supposed to call the global delete, make sure we do so
14031c2e20d7SDouglas Gregor           // even if the destructor throws.
140482fb8920SJohn McCall 
140582fb8920SJohn McCall           // Derive the complete-object pointer, which is what we need
140682fb8920SJohn McCall           // to pass to the deallocation function.
140782fb8920SJohn McCall           llvm::Value *completePtr =
140882fb8920SJohn McCall             CGF.CGM.getCXXABI().adjustToCompleteObject(CGF, Ptr, ElementType);
140982fb8920SJohn McCall 
14101c2e20d7SDouglas Gregor           CGF.EHStack.pushCleanup<CallObjectDelete>(NormalAndEHCleanup,
141182fb8920SJohn McCall                                                     completePtr, OperatorDelete,
14121c2e20d7SDouglas Gregor                                                     ElementType);
14131c2e20d7SDouglas Gregor         }
14141c2e20d7SDouglas Gregor 
1415e30752c9SRichard Smith         // FIXME: Provide a source location here.
1416d619711cSTimur Iskhodzhanov         CXXDtorType DtorType = UseGlobalDelete ? Dtor_Complete : Dtor_Deleting;
1417d619711cSTimur Iskhodzhanov         CGF.CGM.getCXXABI().EmitVirtualDestructorCall(CGF, Dtor, DtorType,
14189dc6eef7SStephen Lin                                                       SourceLocation(), Ptr);
14198ed55a54SJohn McCall 
14201c2e20d7SDouglas Gregor         if (UseGlobalDelete) {
14211c2e20d7SDouglas Gregor           CGF.PopCleanupBlock();
14221c2e20d7SDouglas Gregor         }
14231c2e20d7SDouglas Gregor 
14248ed55a54SJohn McCall         return;
14258ed55a54SJohn McCall       }
14268ed55a54SJohn McCall     }
14278ed55a54SJohn McCall   }
14288ed55a54SJohn McCall 
14298ed55a54SJohn McCall   // Make sure that we call delete even if the dtor throws.
1430e4df6c8dSJohn McCall   // This doesn't have to a conditional cleanup because we're going
1431e4df6c8dSJohn McCall   // to pop it off in a second.
14328ed55a54SJohn McCall   CGF.EHStack.pushCleanup<CallObjectDelete>(NormalAndEHCleanup,
14338ed55a54SJohn McCall                                             Ptr, OperatorDelete, ElementType);
14348ed55a54SJohn McCall 
14358ed55a54SJohn McCall   if (Dtor)
14368ed55a54SJohn McCall     CGF.EmitCXXDestructorCall(Dtor, Dtor_Complete,
143761535005SDouglas Gregor                               /*ForVirtualBase=*/false,
143861535005SDouglas Gregor                               /*Delegating=*/false,
143961535005SDouglas Gregor                               Ptr);
1440bbafb8a7SDavid Blaikie   else if (CGF.getLangOpts().ObjCAutoRefCount &&
144131168b07SJohn McCall            ElementType->isObjCLifetimeType()) {
144231168b07SJohn McCall     switch (ElementType.getObjCLifetime()) {
144331168b07SJohn McCall     case Qualifiers::OCL_None:
144431168b07SJohn McCall     case Qualifiers::OCL_ExplicitNone:
144531168b07SJohn McCall     case Qualifiers::OCL_Autoreleasing:
144631168b07SJohn McCall       break;
144731168b07SJohn McCall 
144831168b07SJohn McCall     case Qualifiers::OCL_Strong: {
144931168b07SJohn McCall       // Load the pointer value.
145031168b07SJohn McCall       llvm::Value *PtrValue = CGF.Builder.CreateLoad(Ptr,
145131168b07SJohn McCall                                              ElementType.isVolatileQualified());
145231168b07SJohn McCall 
1453cdda29c9SJohn McCall       CGF.EmitARCRelease(PtrValue, ARCPreciseLifetime);
145431168b07SJohn McCall       break;
145531168b07SJohn McCall     }
145631168b07SJohn McCall 
145731168b07SJohn McCall     case Qualifiers::OCL_Weak:
145831168b07SJohn McCall       CGF.EmitARCDestroyWeak(Ptr);
145931168b07SJohn McCall       break;
146031168b07SJohn McCall     }
146131168b07SJohn McCall   }
14628ed55a54SJohn McCall 
14638ed55a54SJohn McCall   CGF.PopCleanupBlock();
14648ed55a54SJohn McCall }
14658ed55a54SJohn McCall 
14668ed55a54SJohn McCall namespace {
14678ed55a54SJohn McCall   /// Calls the given 'operator delete' on an array of objects.
14688ed55a54SJohn McCall   struct CallArrayDelete : EHScopeStack::Cleanup {
14698ed55a54SJohn McCall     llvm::Value *Ptr;
14708ed55a54SJohn McCall     const FunctionDecl *OperatorDelete;
14718ed55a54SJohn McCall     llvm::Value *NumElements;
14728ed55a54SJohn McCall     QualType ElementType;
14738ed55a54SJohn McCall     CharUnits CookieSize;
14748ed55a54SJohn McCall 
14758ed55a54SJohn McCall     CallArrayDelete(llvm::Value *Ptr,
14768ed55a54SJohn McCall                     const FunctionDecl *OperatorDelete,
14778ed55a54SJohn McCall                     llvm::Value *NumElements,
14788ed55a54SJohn McCall                     QualType ElementType,
14798ed55a54SJohn McCall                     CharUnits CookieSize)
14808ed55a54SJohn McCall       : Ptr(Ptr), OperatorDelete(OperatorDelete), NumElements(NumElements),
14818ed55a54SJohn McCall         ElementType(ElementType), CookieSize(CookieSize) {}
14828ed55a54SJohn McCall 
14834f12f10dSCraig Topper     void Emit(CodeGenFunction &CGF, Flags flags) override {
14848ed55a54SJohn McCall       const FunctionProtoType *DeleteFTy =
14858ed55a54SJohn McCall         OperatorDelete->getType()->getAs<FunctionProtoType>();
14869cacbabdSAlp Toker       assert(DeleteFTy->getNumParams() == 1 || DeleteFTy->getNumParams() == 2);
14878ed55a54SJohn McCall 
14888ed55a54SJohn McCall       CallArgList Args;
14898ed55a54SJohn McCall 
14908ed55a54SJohn McCall       // Pass the pointer as the first argument.
14919cacbabdSAlp Toker       QualType VoidPtrTy = DeleteFTy->getParamType(0);
14928ed55a54SJohn McCall       llvm::Value *DeletePtr
14938ed55a54SJohn McCall         = CGF.Builder.CreateBitCast(Ptr, CGF.ConvertType(VoidPtrTy));
149443dca6a8SEli Friedman       Args.add(RValue::get(DeletePtr), VoidPtrTy);
14958ed55a54SJohn McCall 
14968ed55a54SJohn McCall       // Pass the original requested size as the second argument.
14979cacbabdSAlp Toker       if (DeleteFTy->getNumParams() == 2) {
14989cacbabdSAlp Toker         QualType size_t = DeleteFTy->getParamType(1);
14992192fe50SChris Lattner         llvm::IntegerType *SizeTy
15008ed55a54SJohn McCall           = cast<llvm::IntegerType>(CGF.ConvertType(size_t));
15018ed55a54SJohn McCall 
15028ed55a54SJohn McCall         CharUnits ElementTypeSize =
15038ed55a54SJohn McCall           CGF.CGM.getContext().getTypeSizeInChars(ElementType);
15048ed55a54SJohn McCall 
15058ed55a54SJohn McCall         // The size of an element, multiplied by the number of elements.
15068ed55a54SJohn McCall         llvm::Value *Size
15078ed55a54SJohn McCall           = llvm::ConstantInt::get(SizeTy, ElementTypeSize.getQuantity());
15088ed55a54SJohn McCall         Size = CGF.Builder.CreateMul(Size, NumElements);
15098ed55a54SJohn McCall 
15108ed55a54SJohn McCall         // Plus the size of the cookie if applicable.
15118ed55a54SJohn McCall         if (!CookieSize.isZero()) {
15128ed55a54SJohn McCall           llvm::Value *CookieSizeV
15138ed55a54SJohn McCall             = llvm::ConstantInt::get(SizeTy, CookieSize.getQuantity());
15148ed55a54SJohn McCall           Size = CGF.Builder.CreateAdd(Size, CookieSizeV);
15158ed55a54SJohn McCall         }
15168ed55a54SJohn McCall 
151743dca6a8SEli Friedman         Args.add(RValue::get(Size), size_t);
15188ed55a54SJohn McCall       }
15198ed55a54SJohn McCall 
15208ed55a54SJohn McCall       // Emit the call to delete.
15218d0dc31dSRichard Smith       EmitNewDeleteCall(CGF, OperatorDelete, DeleteFTy, Args);
15228ed55a54SJohn McCall     }
15238ed55a54SJohn McCall   };
15248ed55a54SJohn McCall }
15258ed55a54SJohn McCall 
15268ed55a54SJohn McCall /// Emit the code for deleting an array of objects.
15278ed55a54SJohn McCall static void EmitArrayDelete(CodeGenFunction &CGF,
1528284c48ffSJohn McCall                             const CXXDeleteExpr *E,
1529ca2c56f2SJohn McCall                             llvm::Value *deletedPtr,
1530ca2c56f2SJohn McCall                             QualType elementType) {
15318a13c418SCraig Topper   llvm::Value *numElements = nullptr;
15328a13c418SCraig Topper   llvm::Value *allocatedPtr = nullptr;
1533ca2c56f2SJohn McCall   CharUnits cookieSize;
1534ca2c56f2SJohn McCall   CGF.CGM.getCXXABI().ReadArrayCookie(CGF, deletedPtr, E, elementType,
1535ca2c56f2SJohn McCall                                       numElements, allocatedPtr, cookieSize);
15368ed55a54SJohn McCall 
1537ca2c56f2SJohn McCall   assert(allocatedPtr && "ReadArrayCookie didn't set allocated pointer");
15388ed55a54SJohn McCall 
15398ed55a54SJohn McCall   // Make sure that we call delete even if one of the dtors throws.
1540ca2c56f2SJohn McCall   const FunctionDecl *operatorDelete = E->getOperatorDelete();
15418ed55a54SJohn McCall   CGF.EHStack.pushCleanup<CallArrayDelete>(NormalAndEHCleanup,
1542ca2c56f2SJohn McCall                                            allocatedPtr, operatorDelete,
1543ca2c56f2SJohn McCall                                            numElements, elementType,
1544ca2c56f2SJohn McCall                                            cookieSize);
15458ed55a54SJohn McCall 
1546ca2c56f2SJohn McCall   // Destroy the elements.
1547ca2c56f2SJohn McCall   if (QualType::DestructionKind dtorKind = elementType.isDestructedType()) {
1548ca2c56f2SJohn McCall     assert(numElements && "no element count for a type with a destructor!");
154931168b07SJohn McCall 
1550ca2c56f2SJohn McCall     llvm::Value *arrayEnd =
1551ca2c56f2SJohn McCall       CGF.Builder.CreateInBoundsGEP(deletedPtr, numElements, "delete.end");
155297eab0a2SJohn McCall 
155397eab0a2SJohn McCall     // Note that it is legal to allocate a zero-length array, and we
155497eab0a2SJohn McCall     // can never fold the check away because the length should always
155597eab0a2SJohn McCall     // come from a cookie.
1556ca2c56f2SJohn McCall     CGF.emitArrayDestroy(deletedPtr, arrayEnd, elementType,
1557ca2c56f2SJohn McCall                          CGF.getDestroyer(dtorKind),
155897eab0a2SJohn McCall                          /*checkZeroLength*/ true,
1559ca2c56f2SJohn McCall                          CGF.needsEHCleanup(dtorKind));
15608ed55a54SJohn McCall   }
15618ed55a54SJohn McCall 
1562ca2c56f2SJohn McCall   // Pop the cleanup block.
15638ed55a54SJohn McCall   CGF.PopCleanupBlock();
15648ed55a54SJohn McCall }
15658ed55a54SJohn McCall 
156659486a2dSAnders Carlsson void CodeGenFunction::EmitCXXDeleteExpr(const CXXDeleteExpr *E) {
156759486a2dSAnders Carlsson   const Expr *Arg = E->getArgument();
156859486a2dSAnders Carlsson   llvm::Value *Ptr = EmitScalarExpr(Arg);
156959486a2dSAnders Carlsson 
157059486a2dSAnders Carlsson   // Null check the pointer.
157159486a2dSAnders Carlsson   llvm::BasicBlock *DeleteNotNull = createBasicBlock("delete.notnull");
157259486a2dSAnders Carlsson   llvm::BasicBlock *DeleteEnd = createBasicBlock("delete.end");
157359486a2dSAnders Carlsson 
157498981b10SAnders Carlsson   llvm::Value *IsNull = Builder.CreateIsNull(Ptr, "isnull");
157559486a2dSAnders Carlsson 
157659486a2dSAnders Carlsson   Builder.CreateCondBr(IsNull, DeleteEnd, DeleteNotNull);
157759486a2dSAnders Carlsson   EmitBlock(DeleteNotNull);
157859486a2dSAnders Carlsson 
15798ed55a54SJohn McCall   // We might be deleting a pointer to array.  If so, GEP down to the
15808ed55a54SJohn McCall   // first non-array element.
15818ed55a54SJohn McCall   // (this assumes that A(*)[3][7] is converted to [3 x [7 x %A]]*)
15828ed55a54SJohn McCall   QualType DeleteTy = Arg->getType()->getAs<PointerType>()->getPointeeType();
15838ed55a54SJohn McCall   if (DeleteTy->isConstantArrayType()) {
15848ed55a54SJohn McCall     llvm::Value *Zero = Builder.getInt32(0);
15850e62c1ccSChris Lattner     SmallVector<llvm::Value*,8> GEP;
158659486a2dSAnders Carlsson 
15878ed55a54SJohn McCall     GEP.push_back(Zero); // point at the outermost array
15888ed55a54SJohn McCall 
15898ed55a54SJohn McCall     // For each layer of array type we're pointing at:
15908ed55a54SJohn McCall     while (const ConstantArrayType *Arr
15918ed55a54SJohn McCall              = getContext().getAsConstantArrayType(DeleteTy)) {
15928ed55a54SJohn McCall       // 1. Unpeel the array type.
15938ed55a54SJohn McCall       DeleteTy = Arr->getElementType();
15948ed55a54SJohn McCall 
15958ed55a54SJohn McCall       // 2. GEP to the first element of the array.
15968ed55a54SJohn McCall       GEP.push_back(Zero);
15978ed55a54SJohn McCall     }
15988ed55a54SJohn McCall 
1599040dd82fSJay Foad     Ptr = Builder.CreateInBoundsGEP(Ptr, GEP, "del.first");
16008ed55a54SJohn McCall   }
16018ed55a54SJohn McCall 
160204f36218SDouglas Gregor   assert(ConvertTypeForMem(DeleteTy) ==
160304f36218SDouglas Gregor          cast<llvm::PointerType>(Ptr->getType())->getElementType());
16048ed55a54SJohn McCall 
160559486a2dSAnders Carlsson   if (E->isArrayForm()) {
1606284c48ffSJohn McCall     EmitArrayDelete(*this, E, Ptr, DeleteTy);
16078ed55a54SJohn McCall   } else {
16081c2e20d7SDouglas Gregor     EmitObjectDelete(*this, E->getOperatorDelete(), Ptr, DeleteTy,
16091c2e20d7SDouglas Gregor                      E->isGlobalDelete());
161059486a2dSAnders Carlsson   }
161159486a2dSAnders Carlsson 
161259486a2dSAnders Carlsson   EmitBlock(DeleteEnd);
161359486a2dSAnders Carlsson }
161459486a2dSAnders Carlsson 
16151c3d95ebSDavid Majnemer static bool isGLValueFromPointerDeref(const Expr *E) {
16161c3d95ebSDavid Majnemer   E = E->IgnoreParens();
16171c3d95ebSDavid Majnemer 
16181c3d95ebSDavid Majnemer   if (const auto *CE = dyn_cast<CastExpr>(E)) {
16191c3d95ebSDavid Majnemer     if (!CE->getSubExpr()->isGLValue())
16201c3d95ebSDavid Majnemer       return false;
16211c3d95ebSDavid Majnemer     return isGLValueFromPointerDeref(CE->getSubExpr());
16221c3d95ebSDavid Majnemer   }
16231c3d95ebSDavid Majnemer 
16241c3d95ebSDavid Majnemer   if (const auto *OVE = dyn_cast<OpaqueValueExpr>(E))
16251c3d95ebSDavid Majnemer     return isGLValueFromPointerDeref(OVE->getSourceExpr());
16261c3d95ebSDavid Majnemer 
16271c3d95ebSDavid Majnemer   if (const auto *BO = dyn_cast<BinaryOperator>(E))
16281c3d95ebSDavid Majnemer     if (BO->getOpcode() == BO_Comma)
16291c3d95ebSDavid Majnemer       return isGLValueFromPointerDeref(BO->getRHS());
16301c3d95ebSDavid Majnemer 
16311c3d95ebSDavid Majnemer   if (const auto *ACO = dyn_cast<AbstractConditionalOperator>(E))
16321c3d95ebSDavid Majnemer     return isGLValueFromPointerDeref(ACO->getTrueExpr()) ||
16331c3d95ebSDavid Majnemer            isGLValueFromPointerDeref(ACO->getFalseExpr());
16341c3d95ebSDavid Majnemer 
16351c3d95ebSDavid Majnemer   // C++11 [expr.sub]p1:
16361c3d95ebSDavid Majnemer   //   The expression E1[E2] is identical (by definition) to *((E1)+(E2))
16371c3d95ebSDavid Majnemer   if (isa<ArraySubscriptExpr>(E))
16381c3d95ebSDavid Majnemer     return true;
16391c3d95ebSDavid Majnemer 
16401c3d95ebSDavid Majnemer   if (const auto *UO = dyn_cast<UnaryOperator>(E))
16411c3d95ebSDavid Majnemer     if (UO->getOpcode() == UO_Deref)
16421c3d95ebSDavid Majnemer       return true;
16431c3d95ebSDavid Majnemer 
16441c3d95ebSDavid Majnemer   return false;
16451c3d95ebSDavid Majnemer }
16461c3d95ebSDavid Majnemer 
1647747e301eSWarren Hunt static llvm::Value *EmitTypeidFromVTable(CodeGenFunction &CGF, const Expr *E,
16482192fe50SChris Lattner                                          llvm::Type *StdTypeInfoPtrTy) {
1649940f02d2SAnders Carlsson   // Get the vtable pointer.
1650940f02d2SAnders Carlsson   llvm::Value *ThisPtr = CGF.EmitLValue(E).getAddress();
1651940f02d2SAnders Carlsson 
1652940f02d2SAnders Carlsson   // C++ [expr.typeid]p2:
1653940f02d2SAnders Carlsson   //   If the glvalue expression is obtained by applying the unary * operator to
1654940f02d2SAnders Carlsson   //   a pointer and the pointer is a null pointer value, the typeid expression
1655940f02d2SAnders Carlsson   //   throws the std::bad_typeid exception.
16561c3d95ebSDavid Majnemer   //
16571c3d95ebSDavid Majnemer   // However, this paragraph's intent is not clear.  We choose a very generous
16581c3d95ebSDavid Majnemer   // interpretation which implores us to consider comma operators, conditional
16591c3d95ebSDavid Majnemer   // operators, parentheses and other such constructs.
16601162d25cSDavid Majnemer   QualType SrcRecordTy = E->getType();
16611c3d95ebSDavid Majnemer   if (CGF.CGM.getCXXABI().shouldTypeidBeNullChecked(
16621c3d95ebSDavid Majnemer           isGLValueFromPointerDeref(E), SrcRecordTy)) {
1663940f02d2SAnders Carlsson     llvm::BasicBlock *BadTypeidBlock =
1664940f02d2SAnders Carlsson         CGF.createBasicBlock("typeid.bad_typeid");
16651162d25cSDavid Majnemer     llvm::BasicBlock *EndBlock = CGF.createBasicBlock("typeid.end");
1666940f02d2SAnders Carlsson 
1667940f02d2SAnders Carlsson     llvm::Value *IsNull = CGF.Builder.CreateIsNull(ThisPtr);
1668940f02d2SAnders Carlsson     CGF.Builder.CreateCondBr(IsNull, BadTypeidBlock, EndBlock);
1669940f02d2SAnders Carlsson 
1670940f02d2SAnders Carlsson     CGF.EmitBlock(BadTypeidBlock);
16711162d25cSDavid Majnemer     CGF.CGM.getCXXABI().EmitBadTypeidCall(CGF);
1672940f02d2SAnders Carlsson     CGF.EmitBlock(EndBlock);
1673940f02d2SAnders Carlsson   }
1674940f02d2SAnders Carlsson 
16751162d25cSDavid Majnemer   return CGF.CGM.getCXXABI().EmitTypeid(CGF, SrcRecordTy, ThisPtr,
16761162d25cSDavid Majnemer                                         StdTypeInfoPtrTy);
1677940f02d2SAnders Carlsson }
1678940f02d2SAnders Carlsson 
167959486a2dSAnders Carlsson llvm::Value *CodeGenFunction::EmitCXXTypeidExpr(const CXXTypeidExpr *E) {
16802192fe50SChris Lattner   llvm::Type *StdTypeInfoPtrTy =
1681940f02d2SAnders Carlsson     ConvertType(E->getType())->getPointerTo();
1682fd7dfeb7SAnders Carlsson 
16833f4336cbSAnders Carlsson   if (E->isTypeOperand()) {
16843f4336cbSAnders Carlsson     llvm::Constant *TypeInfo =
1685143c55eaSDavid Majnemer         CGM.GetAddrOfRTTIDescriptor(E->getTypeOperand(getContext()));
1686940f02d2SAnders Carlsson     return Builder.CreateBitCast(TypeInfo, StdTypeInfoPtrTy);
16873f4336cbSAnders Carlsson   }
1688fd7dfeb7SAnders Carlsson 
1689940f02d2SAnders Carlsson   // C++ [expr.typeid]p2:
1690940f02d2SAnders Carlsson   //   When typeid is applied to a glvalue expression whose type is a
1691940f02d2SAnders Carlsson   //   polymorphic class type, the result refers to a std::type_info object
1692940f02d2SAnders Carlsson   //   representing the type of the most derived object (that is, the dynamic
1693940f02d2SAnders Carlsson   //   type) to which the glvalue refers.
1694ef8bf436SRichard Smith   if (E->isPotentiallyEvaluated())
1695940f02d2SAnders Carlsson     return EmitTypeidFromVTable(*this, E->getExprOperand(),
1696940f02d2SAnders Carlsson                                 StdTypeInfoPtrTy);
1697940f02d2SAnders Carlsson 
1698940f02d2SAnders Carlsson   QualType OperandTy = E->getExprOperand()->getType();
1699940f02d2SAnders Carlsson   return Builder.CreateBitCast(CGM.GetAddrOfRTTIDescriptor(OperandTy),
1700940f02d2SAnders Carlsson                                StdTypeInfoPtrTy);
170159486a2dSAnders Carlsson }
170259486a2dSAnders Carlsson 
1703c1c9971cSAnders Carlsson static llvm::Value *EmitDynamicCastToNull(CodeGenFunction &CGF,
1704c1c9971cSAnders Carlsson                                           QualType DestTy) {
17052192fe50SChris Lattner   llvm::Type *DestLTy = CGF.ConvertType(DestTy);
1706c1c9971cSAnders Carlsson   if (DestTy->isPointerType())
1707c1c9971cSAnders Carlsson     return llvm::Constant::getNullValue(DestLTy);
1708c1c9971cSAnders Carlsson 
1709c1c9971cSAnders Carlsson   /// C++ [expr.dynamic.cast]p9:
1710c1c9971cSAnders Carlsson   ///   A failed cast to reference type throws std::bad_cast
17111162d25cSDavid Majnemer   if (!CGF.CGM.getCXXABI().EmitBadCastCall(CGF))
17121162d25cSDavid Majnemer     return nullptr;
1713c1c9971cSAnders Carlsson 
1714c1c9971cSAnders Carlsson   CGF.EmitBlock(CGF.createBasicBlock("dynamic_cast.end"));
1715c1c9971cSAnders Carlsson   return llvm::UndefValue::get(DestLTy);
1716c1c9971cSAnders Carlsson }
1717c1c9971cSAnders Carlsson 
1718882d790fSAnders Carlsson llvm::Value *CodeGenFunction::EmitDynamicCast(llvm::Value *Value,
171959486a2dSAnders Carlsson                                               const CXXDynamicCastExpr *DCE) {
17203f4336cbSAnders Carlsson   QualType DestTy = DCE->getTypeAsWritten();
17213f4336cbSAnders Carlsson 
1722c1c9971cSAnders Carlsson   if (DCE->isAlwaysNull())
17231162d25cSDavid Majnemer     if (llvm::Value *T = EmitDynamicCastToNull(*this, DestTy))
17241162d25cSDavid Majnemer       return T;
1725c1c9971cSAnders Carlsson 
1726c1c9971cSAnders Carlsson   QualType SrcTy = DCE->getSubExpr()->getType();
1727c1c9971cSAnders Carlsson 
17281162d25cSDavid Majnemer   // C++ [expr.dynamic.cast]p7:
17291162d25cSDavid Majnemer   //   If T is "pointer to cv void," then the result is a pointer to the most
17301162d25cSDavid Majnemer   //   derived object pointed to by v.
17311162d25cSDavid Majnemer   const PointerType *DestPTy = DestTy->getAs<PointerType>();
17321162d25cSDavid Majnemer 
17331162d25cSDavid Majnemer   bool isDynamicCastToVoid;
17341162d25cSDavid Majnemer   QualType SrcRecordTy;
17351162d25cSDavid Majnemer   QualType DestRecordTy;
17361162d25cSDavid Majnemer   if (DestPTy) {
17371162d25cSDavid Majnemer     isDynamicCastToVoid = DestPTy->getPointeeType()->isVoidType();
17381162d25cSDavid Majnemer     SrcRecordTy = SrcTy->castAs<PointerType>()->getPointeeType();
17391162d25cSDavid Majnemer     DestRecordTy = DestPTy->getPointeeType();
17401162d25cSDavid Majnemer   } else {
17411162d25cSDavid Majnemer     isDynamicCastToVoid = false;
17421162d25cSDavid Majnemer     SrcRecordTy = SrcTy;
17431162d25cSDavid Majnemer     DestRecordTy = DestTy->castAs<ReferenceType>()->getPointeeType();
17441162d25cSDavid Majnemer   }
17451162d25cSDavid Majnemer 
17461162d25cSDavid Majnemer   assert(SrcRecordTy->isRecordType() && "source type must be a record type!");
17471162d25cSDavid Majnemer 
1748882d790fSAnders Carlsson   // C++ [expr.dynamic.cast]p4:
1749882d790fSAnders Carlsson   //   If the value of v is a null pointer value in the pointer case, the result
1750882d790fSAnders Carlsson   //   is the null pointer value of type T.
17511162d25cSDavid Majnemer   bool ShouldNullCheckSrcValue =
17521162d25cSDavid Majnemer       CGM.getCXXABI().shouldDynamicCastCallBeNullChecked(SrcTy->isPointerType(),
17531162d25cSDavid Majnemer                                                          SrcRecordTy);
175459486a2dSAnders Carlsson 
17558a13c418SCraig Topper   llvm::BasicBlock *CastNull = nullptr;
17568a13c418SCraig Topper   llvm::BasicBlock *CastNotNull = nullptr;
1757882d790fSAnders Carlsson   llvm::BasicBlock *CastEnd = createBasicBlock("dynamic_cast.end");
1758fa8b4955SDouglas Gregor 
1759882d790fSAnders Carlsson   if (ShouldNullCheckSrcValue) {
1760882d790fSAnders Carlsson     CastNull = createBasicBlock("dynamic_cast.null");
1761882d790fSAnders Carlsson     CastNotNull = createBasicBlock("dynamic_cast.notnull");
1762882d790fSAnders Carlsson 
1763882d790fSAnders Carlsson     llvm::Value *IsNull = Builder.CreateIsNull(Value);
1764882d790fSAnders Carlsson     Builder.CreateCondBr(IsNull, CastNull, CastNotNull);
1765882d790fSAnders Carlsson     EmitBlock(CastNotNull);
176659486a2dSAnders Carlsson   }
176759486a2dSAnders Carlsson 
17681162d25cSDavid Majnemer   if (isDynamicCastToVoid) {
17691162d25cSDavid Majnemer     Value = CGM.getCXXABI().EmitDynamicCastToVoid(*this, Value, SrcRecordTy,
17701162d25cSDavid Majnemer                                                   DestTy);
17711162d25cSDavid Majnemer   } else {
17721162d25cSDavid Majnemer     assert(DestRecordTy->isRecordType() &&
17731162d25cSDavid Majnemer            "destination type must be a record type!");
17741162d25cSDavid Majnemer     Value = CGM.getCXXABI().EmitDynamicCastCall(*this, Value, SrcRecordTy,
17751162d25cSDavid Majnemer                                                 DestTy, DestRecordTy, CastEnd);
17761162d25cSDavid Majnemer   }
17773f4336cbSAnders Carlsson 
1778882d790fSAnders Carlsson   if (ShouldNullCheckSrcValue) {
1779882d790fSAnders Carlsson     EmitBranch(CastEnd);
178059486a2dSAnders Carlsson 
1781882d790fSAnders Carlsson     EmitBlock(CastNull);
1782882d790fSAnders Carlsson     EmitBranch(CastEnd);
178359486a2dSAnders Carlsson   }
178459486a2dSAnders Carlsson 
1785882d790fSAnders Carlsson   EmitBlock(CastEnd);
178659486a2dSAnders Carlsson 
1787882d790fSAnders Carlsson   if (ShouldNullCheckSrcValue) {
1788882d790fSAnders Carlsson     llvm::PHINode *PHI = Builder.CreatePHI(Value->getType(), 2);
1789882d790fSAnders Carlsson     PHI->addIncoming(Value, CastNotNull);
1790882d790fSAnders Carlsson     PHI->addIncoming(llvm::Constant::getNullValue(Value->getType()), CastNull);
179159486a2dSAnders Carlsson 
1792882d790fSAnders Carlsson     Value = PHI;
179359486a2dSAnders Carlsson   }
179459486a2dSAnders Carlsson 
1795882d790fSAnders Carlsson   return Value;
179659486a2dSAnders Carlsson }
1797c370a7eeSEli Friedman 
1798c370a7eeSEli Friedman void CodeGenFunction::EmitLambdaExpr(const LambdaExpr *E, AggValueSlot Slot) {
17998631f3e8SEli Friedman   RunCleanupsScope Scope(*this);
18007f1ff600SEli Friedman   LValue SlotLV = MakeAddrLValue(Slot.getAddr(), E->getType(),
18017f1ff600SEli Friedman                                  Slot.getAlignment());
18028631f3e8SEli Friedman 
1803c370a7eeSEli Friedman   CXXRecordDecl::field_iterator CurField = E->getLambdaClass()->field_begin();
1804c370a7eeSEli Friedman   for (LambdaExpr::capture_init_iterator i = E->capture_init_begin(),
1805c370a7eeSEli Friedman                                          e = E->capture_init_end();
1806c370a7eeSEli Friedman        i != e; ++i, ++CurField) {
1807c370a7eeSEli Friedman     // Emit initialization
18087f1ff600SEli Friedman 
180940ed2973SDavid Blaikie     LValue LV = EmitLValueForFieldInitialization(SlotLV, *CurField);
18105f1a04ffSEli Friedman     ArrayRef<VarDecl *> ArrayIndexes;
18115f1a04ffSEli Friedman     if (CurField->getType()->isArrayType())
18125f1a04ffSEli Friedman       ArrayIndexes = E->getCaptureInitIndexVars(i);
181340ed2973SDavid Blaikie     EmitInitializerForField(*CurField, LV, *i, ArrayIndexes);
1814c370a7eeSEli Friedman   }
1815c370a7eeSEli Friedman }
1816