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"
2010a4972aSSaleem Abdulrasool #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 
27efa956ceSAlexey Samsonov static RequiredArgs
28efa956ceSAlexey Samsonov commonEmitCXXMemberOrOperatorCall(CodeGenFunction &CGF, const CXXMethodDecl *MD,
29efa956ceSAlexey Samsonov                                   llvm::Value *This, llvm::Value *ImplicitParam,
30efa956ceSAlexey Samsonov                                   QualType ImplicitParamTy, const CallExpr *CE,
31efa956ceSAlexey Samsonov                                   CallArgList &Args) {
32a5bf76bdSAlexey Samsonov   assert(CE == nullptr || isa<CXXMemberCallExpr>(CE) ||
33a5bf76bdSAlexey Samsonov          isa<CXXOperatorCallExpr>(CE));
3427da15baSAnders Carlsson   assert(MD->isInstance() &&
35a5bf76bdSAlexey Samsonov          "Trying to emit a member or operator call expr on a static method!");
3627da15baSAnders Carlsson 
3769d0d262SRichard Smith   // C++11 [class.mfct.non-static]p2:
3869d0d262SRichard Smith   //   If a non-static member function of a class X is called for an object that
3969d0d262SRichard Smith   //   is not of type X, or of a type derived from X, the behavior is undefined.
40a5bf76bdSAlexey Samsonov   SourceLocation CallLoc;
41a5bf76bdSAlexey Samsonov   if (CE)
42a5bf76bdSAlexey Samsonov     CallLoc = CE->getExprLoc();
430c0b6d9aSDavid Majnemer   CGF.EmitTypeCheck(
440c0b6d9aSDavid Majnemer       isa<CXXConstructorDecl>(MD) ? CodeGenFunction::TCK_ConstructorCall
450c0b6d9aSDavid Majnemer                                   : CodeGenFunction::TCK_MemberCall,
460c0b6d9aSDavid Majnemer       CallLoc, This, CGF.getContext().getRecordType(MD->getParent()));
4727da15baSAnders Carlsson 
4827da15baSAnders Carlsson   // Push the this ptr.
490c0b6d9aSDavid Majnemer   Args.add(RValue::get(This), MD->getThisType(CGF.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>();
57*419996ccSGeorge Burgess IV   RequiredArgs required = RequiredArgs::forPrototypePlus(FPT, Args.size(), MD);
58a729c62bSJohn McCall 
59a729c62bSJohn McCall   // And the rest of the call args.
608e1162c7SAlexey Samsonov   if (CE) {
61a5bf76bdSAlexey Samsonov     // Special case: skip first argument of CXXOperatorCall (it is "this").
628e1162c7SAlexey Samsonov     unsigned ArgsToSkip = isa<CXXOperatorCallExpr>(CE) ? 1 : 0;
63f05779e2SDavid Blaikie     CGF.EmitCallArgs(Args, FPT, drop_begin(CE->arguments(), ArgsToSkip),
648e1162c7SAlexey Samsonov                      CE->getDirectCallee());
65a5bf76bdSAlexey Samsonov   } else {
668e1162c7SAlexey Samsonov     assert(
678e1162c7SAlexey Samsonov         FPT->getNumParams() == 0 &&
688e1162c7SAlexey Samsonov         "No CallExpr specified for function with non-zero number of arguments");
69a5bf76bdSAlexey Samsonov   }
700c0b6d9aSDavid Majnemer   return required;
710c0b6d9aSDavid Majnemer }
7227da15baSAnders Carlsson 
730c0b6d9aSDavid Majnemer RValue CodeGenFunction::EmitCXXMemberOrOperatorCall(
740c0b6d9aSDavid Majnemer     const CXXMethodDecl *MD, llvm::Value *Callee, ReturnValueSlot ReturnValue,
750c0b6d9aSDavid Majnemer     llvm::Value *This, llvm::Value *ImplicitParam, QualType ImplicitParamTy,
760c0b6d9aSDavid Majnemer     const CallExpr *CE) {
770c0b6d9aSDavid Majnemer   const FunctionProtoType *FPT = MD->getType()->castAs<FunctionProtoType>();
780c0b6d9aSDavid Majnemer   CallArgList Args;
790c0b6d9aSDavid Majnemer   RequiredArgs required = commonEmitCXXMemberOrOperatorCall(
80efa956ceSAlexey Samsonov       *this, MD, This, ImplicitParam, ImplicitParamTy, CE, Args);
818dda7b27SJohn McCall   return EmitCall(CGM.getTypes().arrangeCXXMethodCall(Args, FPT, required),
82c50c27ccSRafael Espindola                   Callee, ReturnValue, Args, MD);
8327da15baSAnders Carlsson }
8427da15baSAnders Carlsson 
85ae81bbb4SAlexey Samsonov RValue CodeGenFunction::EmitCXXDestructorCall(
86ae81bbb4SAlexey Samsonov     const CXXDestructorDecl *DD, llvm::Value *Callee, llvm::Value *This,
87ae81bbb4SAlexey Samsonov     llvm::Value *ImplicitParam, QualType ImplicitParamTy, const CallExpr *CE,
88ae81bbb4SAlexey Samsonov     StructorType Type) {
890c0b6d9aSDavid Majnemer   CallArgList Args;
90ae81bbb4SAlexey Samsonov   commonEmitCXXMemberOrOperatorCall(*this, DD, This, ImplicitParam,
91efa956ceSAlexey Samsonov                                     ImplicitParamTy, CE, Args);
92ae81bbb4SAlexey Samsonov   return EmitCall(CGM.getTypes().arrangeCXXStructorDeclaration(DD, Type),
93ae81bbb4SAlexey Samsonov                   Callee, ReturnValueSlot(), Args, DD);
940c0b6d9aSDavid Majnemer }
950c0b6d9aSDavid Majnemer 
963b33c4ecSRafael Espindola static CXXRecordDecl *getCXXRecord(const Expr *E) {
973b33c4ecSRafael Espindola   QualType T = E->getType();
983b33c4ecSRafael Espindola   if (const PointerType *PTy = T->getAs<PointerType>())
993b33c4ecSRafael Espindola     T = PTy->getPointeeType();
1003b33c4ecSRafael Espindola   const RecordType *Ty = T->castAs<RecordType>();
1013b33c4ecSRafael Espindola   return cast<CXXRecordDecl>(Ty->getDecl());
1023b33c4ecSRafael Espindola }
1033b33c4ecSRafael Espindola 
10464225794SFrancois Pichet // Note: This function also emit constructor calls to support a MSVC
10564225794SFrancois Pichet // extensions allowing explicit constructor function call.
10627da15baSAnders Carlsson RValue CodeGenFunction::EmitCXXMemberCallExpr(const CXXMemberCallExpr *CE,
10727da15baSAnders Carlsson                                               ReturnValueSlot ReturnValue) {
1082d2e8707SJohn McCall   const Expr *callee = CE->getCallee()->IgnoreParens();
1092d2e8707SJohn McCall 
1102d2e8707SJohn McCall   if (isa<BinaryOperator>(callee))
11127da15baSAnders Carlsson     return EmitCXXMemberPointerCallExpr(CE, ReturnValue);
11227da15baSAnders Carlsson 
1132d2e8707SJohn McCall   const MemberExpr *ME = cast<MemberExpr>(callee);
11427da15baSAnders Carlsson   const CXXMethodDecl *MD = cast<CXXMethodDecl>(ME->getMemberDecl());
11527da15baSAnders Carlsson 
11627da15baSAnders Carlsson   if (MD->isStatic()) {
11727da15baSAnders Carlsson     // The method is static, emit it as we would a regular call.
11827da15baSAnders Carlsson     llvm::Value *Callee = CGM.GetAddrOfFunction(MD);
11970b9c01bSAlexey Samsonov     return EmitCall(getContext().getPointerType(MD->getType()), Callee, CE,
12070b9c01bSAlexey Samsonov                     ReturnValue);
12127da15baSAnders Carlsson   }
12227da15baSAnders Carlsson 
123aad4af6dSNico Weber   bool HasQualifier = ME->hasQualifier();
124aad4af6dSNico Weber   NestedNameSpecifier *Qualifier = HasQualifier ? ME->getQualifier() : nullptr;
125aad4af6dSNico Weber   bool IsArrow = ME->isArrow();
126ecbe2e97SRafael Espindola   const Expr *Base = ME->getBase();
127aad4af6dSNico Weber 
128aad4af6dSNico Weber   return EmitCXXMemberOrOperatorMemberCallExpr(
129aad4af6dSNico Weber       CE, MD, ReturnValue, HasQualifier, Qualifier, IsArrow, Base);
130aad4af6dSNico Weber }
131aad4af6dSNico Weber 
132aad4af6dSNico Weber RValue CodeGenFunction::EmitCXXMemberOrOperatorMemberCallExpr(
133aad4af6dSNico Weber     const CallExpr *CE, const CXXMethodDecl *MD, ReturnValueSlot ReturnValue,
134aad4af6dSNico Weber     bool HasQualifier, NestedNameSpecifier *Qualifier, bool IsArrow,
135aad4af6dSNico Weber     const Expr *Base) {
136aad4af6dSNico Weber   assert(isa<CXXMemberCallExpr>(CE) || isa<CXXOperatorCallExpr>(CE));
137aad4af6dSNico Weber 
138aad4af6dSNico Weber   // Compute the object pointer.
139aad4af6dSNico Weber   bool CanUseVirtualCall = MD->isVirtual() && !HasQualifier;
140ecbe2e97SRafael Espindola 
1418a13c418SCraig Topper   const CXXMethodDecl *DevirtualizedMethod = nullptr;
1427463ed7cSBenjamin Kramer   if (CanUseVirtualCall && CanDevirtualizeMemberFunctionCall(Base, MD)) {
1433b33c4ecSRafael Espindola     const CXXRecordDecl *BestDynamicDecl = Base->getBestDynamicClassType();
1443b33c4ecSRafael Espindola     DevirtualizedMethod = MD->getCorrespondingMethodInClass(BestDynamicDecl);
1453b33c4ecSRafael Espindola     assert(DevirtualizedMethod);
1463b33c4ecSRafael Espindola     const CXXRecordDecl *DevirtualizedClass = DevirtualizedMethod->getParent();
1473b33c4ecSRafael Espindola     const Expr *Inner = Base->ignoreParenBaseCasts();
1485bd68794SAlexey Bataev     if (DevirtualizedMethod->getReturnType().getCanonicalType() !=
1495bd68794SAlexey Bataev         MD->getReturnType().getCanonicalType())
1505bd68794SAlexey Bataev       // If the return types are not the same, this might be a case where more
1515bd68794SAlexey Bataev       // code needs to run to compensate for it. For example, the derived
1525bd68794SAlexey Bataev       // method might return a type that inherits form from the return
1535bd68794SAlexey Bataev       // type of MD and has a prefix.
1545bd68794SAlexey Bataev       // For now we just avoid devirtualizing these covariant cases.
1555bd68794SAlexey Bataev       DevirtualizedMethod = nullptr;
1565bd68794SAlexey Bataev     else if (getCXXRecord(Inner) == DevirtualizedClass)
1573b33c4ecSRafael Espindola       // If the class of the Inner expression is where the dynamic method
1583b33c4ecSRafael Espindola       // is defined, build the this pointer from it.
1593b33c4ecSRafael Espindola       Base = Inner;
1603b33c4ecSRafael Espindola     else if (getCXXRecord(Base) != DevirtualizedClass) {
1613b33c4ecSRafael Espindola       // If the method is defined in a class that is not the best dynamic
1623b33c4ecSRafael Espindola       // one or the one of the full expression, we would have to build
1633b33c4ecSRafael Espindola       // a derived-to-base cast to compute the correct this pointer, but
1643b33c4ecSRafael Espindola       // we don't have support for that yet, so do a virtual call.
1658a13c418SCraig Topper       DevirtualizedMethod = nullptr;
1663b33c4ecSRafael Espindola     }
1673b33c4ecSRafael Espindola   }
168ecbe2e97SRafael Espindola 
1697f416cc4SJohn McCall   Address This = Address::invalid();
170aad4af6dSNico Weber   if (IsArrow)
1717f416cc4SJohn McCall     This = EmitPointerWithAlignment(Base);
172f93ac894SFariborz Jahanian   else
1733b33c4ecSRafael Espindola     This = EmitLValue(Base).getAddress();
174ecbe2e97SRafael Espindola 
17527da15baSAnders Carlsson 
176419bd094SRichard Smith   if (MD->isTrivial() || (MD->isDefaulted() && MD->getParent()->isUnion())) {
1778a13c418SCraig Topper     if (isa<CXXDestructorDecl>(MD)) return RValue::get(nullptr);
17864225794SFrancois Pichet     if (isa<CXXConstructorDecl>(MD) &&
17964225794SFrancois Pichet         cast<CXXConstructorDecl>(MD)->isDefaultConstructor())
1808a13c418SCraig Topper       return RValue::get(nullptr);
1810d635f53SJohn McCall 
182aad4af6dSNico Weber     if (!MD->getParent()->mayInsertExtraPadding()) {
18322653bacSSebastian Redl       if (MD->isCopyAssignmentOperator() || MD->isMoveAssignmentOperator()) {
18422653bacSSebastian Redl         // We don't like to generate the trivial copy/move assignment operator
18522653bacSSebastian Redl         // when it isn't necessary; just produce the proper effect here.
186aad4af6dSNico Weber         // Special case: skip first argument of CXXOperatorCall (it is "this").
187aad4af6dSNico Weber         unsigned ArgsToSkip = isa<CXXOperatorCallExpr>(CE) ? 1 : 0;
1887f416cc4SJohn McCall         Address RHS = EmitLValue(*(CE->arg_begin() + ArgsToSkip)).getAddress();
1891ca66919SBenjamin Kramer         EmitAggregateAssign(This, RHS, CE->getType());
1907f416cc4SJohn McCall         return RValue::get(This.getPointer());
19127da15baSAnders Carlsson       }
19227da15baSAnders Carlsson 
19364225794SFrancois Pichet       if (isa<CXXConstructorDecl>(MD) &&
19422653bacSSebastian Redl           cast<CXXConstructorDecl>(MD)->isCopyOrMoveConstructor()) {
19522653bacSSebastian Redl         // Trivial move and copy ctor are the same.
196525bf650SAlexey Samsonov         assert(CE->getNumArgs() == 1 && "unexpected argcount for trivial ctor");
1977f416cc4SJohn McCall         Address RHS = EmitLValue(*CE->arg_begin()).getAddress();
198f48ee448SBenjamin Kramer         EmitAggregateCopy(This, RHS, (*CE->arg_begin())->getType());
1997f416cc4SJohn McCall         return RValue::get(This.getPointer());
20064225794SFrancois Pichet       }
20164225794SFrancois Pichet       llvm_unreachable("unknown trivial member function");
20264225794SFrancois Pichet     }
203aad4af6dSNico Weber   }
20464225794SFrancois Pichet 
2050d635f53SJohn McCall   // Compute the function type we're calling.
2063abfe958SNico Weber   const CXXMethodDecl *CalleeDecl =
2073abfe958SNico Weber       DevirtualizedMethod ? DevirtualizedMethod : MD;
2088a13c418SCraig Topper   const CGFunctionInfo *FInfo = nullptr;
2093abfe958SNico Weber   if (const auto *Dtor = dyn_cast<CXXDestructorDecl>(CalleeDecl))
2108d2a19b4SRafael Espindola     FInfo = &CGM.getTypes().arrangeCXXStructorDeclaration(
2118d2a19b4SRafael Espindola         Dtor, StructorType::Complete);
2123abfe958SNico Weber   else if (const auto *Ctor = dyn_cast<CXXConstructorDecl>(CalleeDecl))
2138d2a19b4SRafael Espindola     FInfo = &CGM.getTypes().arrangeCXXStructorDeclaration(
2148d2a19b4SRafael Espindola         Ctor, StructorType::Complete);
21564225794SFrancois Pichet   else
216ade60977SEli Friedman     FInfo = &CGM.getTypes().arrangeCXXMethodDeclaration(CalleeDecl);
2170d635f53SJohn McCall 
218e7de47efSReid Kleckner   llvm::FunctionType *Ty = CGM.getTypes().GetFunctionType(*FInfo);
2190d635f53SJohn McCall 
22027da15baSAnders Carlsson   // C++ [class.virtual]p12:
22127da15baSAnders Carlsson   //   Explicit qualification with the scope operator (5.1) suppresses the
22227da15baSAnders Carlsson   //   virtual call mechanism.
22327da15baSAnders Carlsson   //
22427da15baSAnders Carlsson   // We also don't emit a virtual call if the base expression has a record type
22527da15baSAnders Carlsson   // because then we know what the type is.
2263b33c4ecSRafael Espindola   bool UseVirtualCall = CanUseVirtualCall && !DevirtualizedMethod;
22719cee187SStephen Lin   llvm::Value *Callee;
2289dc6eef7SStephen Lin 
2290d635f53SJohn McCall   if (const CXXDestructorDecl *Dtor = dyn_cast<CXXDestructorDecl>(MD)) {
23019cee187SStephen Lin     assert(CE->arg_begin() == CE->arg_end() &&
2319dc6eef7SStephen Lin            "Destructor shouldn't have explicit parameters");
2329dc6eef7SStephen Lin     assert(ReturnValue.isNull() && "Destructor shouldn't have return value");
2339dc6eef7SStephen Lin     if (UseVirtualCall) {
234aad4af6dSNico Weber       CGM.getCXXABI().EmitVirtualDestructorCall(
235aad4af6dSNico Weber           *this, Dtor, Dtor_Complete, This, cast<CXXMemberCallExpr>(CE));
23627da15baSAnders Carlsson     } else {
237aad4af6dSNico Weber       if (getLangOpts().AppleKext && MD->isVirtual() && HasQualifier)
238aad4af6dSNico Weber         Callee = BuildAppleKextVirtualCall(MD, Qualifier, Ty);
2393b33c4ecSRafael Espindola       else if (!DevirtualizedMethod)
2401ac0ec86SRafael Espindola         Callee =
2411ac0ec86SRafael Espindola             CGM.getAddrOfCXXStructor(Dtor, StructorType::Complete, FInfo, Ty);
24249e860b2SRafael Espindola       else {
2433b33c4ecSRafael Espindola         const CXXDestructorDecl *DDtor =
2443b33c4ecSRafael Espindola           cast<CXXDestructorDecl>(DevirtualizedMethod);
24549e860b2SRafael Espindola         Callee = CGM.GetAddrOfFunction(GlobalDecl(DDtor, Dtor_Complete), Ty);
24649e860b2SRafael Espindola       }
2477f416cc4SJohn McCall       EmitCXXMemberOrOperatorCall(MD, Callee, ReturnValue, This.getPointer(),
248a5bf76bdSAlexey Samsonov                                   /*ImplicitParam=*/nullptr, QualType(), CE);
24927da15baSAnders Carlsson     }
2508a13c418SCraig Topper     return RValue::get(nullptr);
2519dc6eef7SStephen Lin   }
2529dc6eef7SStephen Lin 
2539dc6eef7SStephen Lin   if (const CXXConstructorDecl *Ctor = dyn_cast<CXXConstructorDecl>(MD)) {
25464225794SFrancois Pichet     Callee = CGM.GetAddrOfFunction(GlobalDecl(Ctor, Ctor_Complete), Ty);
2550d635f53SJohn McCall   } else if (UseVirtualCall) {
2566708c4a1SPeter Collingbourne     Callee = CGM.getCXXABI().getVirtualFunctionPointer(*this, MD, This, Ty,
2576708c4a1SPeter Collingbourne                                                        CE->getLocStart());
25827da15baSAnders Carlsson   } else {
2591a7488afSPeter Collingbourne     if (SanOpts.has(SanitizerKind::CFINVCall) &&
2601a7488afSPeter Collingbourne         MD->getParent()->isDynamicClass()) {
2614b1ac72cSPiotr Padlewski       llvm::Value *VTable = GetVTablePtr(This, Int8PtrTy, MD->getParent());
262fb532b9aSPeter Collingbourne       EmitVTablePtrCheckForCall(MD->getParent(), VTable, CFITCK_NVCall,
263fb532b9aSPeter Collingbourne                                 CE->getLocStart());
2641a7488afSPeter Collingbourne     }
2651a7488afSPeter Collingbourne 
266aad4af6dSNico Weber     if (getLangOpts().AppleKext && MD->isVirtual() && HasQualifier)
267aad4af6dSNico Weber       Callee = BuildAppleKextVirtualCall(MD, Qualifier, Ty);
2683b33c4ecSRafael Espindola     else if (!DevirtualizedMethod)
269727a771aSRafael Espindola       Callee = CGM.GetAddrOfFunction(MD, Ty);
27049e860b2SRafael Espindola     else {
2713b33c4ecSRafael Espindola       Callee = CGM.GetAddrOfFunction(DevirtualizedMethod, Ty);
27249e860b2SRafael Espindola     }
27327da15baSAnders Carlsson   }
27427da15baSAnders Carlsson 
275f1749427STimur Iskhodzhanov   if (MD->isVirtual()) {
276f1749427STimur Iskhodzhanov     This = CGM.getCXXABI().adjustThisArgumentForVirtualFunctionCall(
2774b60f30aSReid Kleckner         *this, CalleeDecl, This, UseVirtualCall);
278f1749427STimur Iskhodzhanov   }
27988fd439aSTimur Iskhodzhanov 
2807f416cc4SJohn McCall   return EmitCXXMemberOrOperatorCall(MD, Callee, ReturnValue, This.getPointer(),
281a5bf76bdSAlexey Samsonov                                      /*ImplicitParam=*/nullptr, QualType(), CE);
28227da15baSAnders Carlsson }
28327da15baSAnders Carlsson 
28427da15baSAnders Carlsson RValue
28527da15baSAnders Carlsson CodeGenFunction::EmitCXXMemberPointerCallExpr(const CXXMemberCallExpr *E,
28627da15baSAnders Carlsson                                               ReturnValueSlot ReturnValue) {
28727da15baSAnders Carlsson   const BinaryOperator *BO =
28827da15baSAnders Carlsson       cast<BinaryOperator>(E->getCallee()->IgnoreParens());
28927da15baSAnders Carlsson   const Expr *BaseExpr = BO->getLHS();
29027da15baSAnders Carlsson   const Expr *MemFnExpr = BO->getRHS();
29127da15baSAnders Carlsson 
29227da15baSAnders Carlsson   const MemberPointerType *MPT =
2930009fcc3SJohn McCall     MemFnExpr->getType()->castAs<MemberPointerType>();
294475999dcSJohn McCall 
29527da15baSAnders Carlsson   const FunctionProtoType *FPT =
2960009fcc3SJohn McCall     MPT->getPointeeType()->castAs<FunctionProtoType>();
29727da15baSAnders Carlsson   const CXXRecordDecl *RD =
29827da15baSAnders Carlsson     cast<CXXRecordDecl>(MPT->getClass()->getAs<RecordType>()->getDecl());
29927da15baSAnders Carlsson 
30027da15baSAnders Carlsson   // Get the member function pointer.
301a1dee530SJohn McCall   llvm::Value *MemFnPtr = EmitScalarExpr(MemFnExpr);
30227da15baSAnders Carlsson 
30327da15baSAnders Carlsson   // Emit the 'this' pointer.
3047f416cc4SJohn McCall   Address This = Address::invalid();
305e302792bSJohn McCall   if (BO->getOpcode() == BO_PtrMemI)
3067f416cc4SJohn McCall     This = EmitPointerWithAlignment(BaseExpr);
30727da15baSAnders Carlsson   else
30827da15baSAnders Carlsson     This = EmitLValue(BaseExpr).getAddress();
30927da15baSAnders Carlsson 
3107f416cc4SJohn McCall   EmitTypeCheck(TCK_MemberCall, E->getExprLoc(), This.getPointer(),
311e30752c9SRichard Smith                 QualType(MPT->getClass(), 0));
31269d0d262SRichard Smith 
313475999dcSJohn McCall   // Ask the ABI to load the callee.  Note that This is modified.
3147f416cc4SJohn McCall   llvm::Value *ThisPtrForCall = nullptr;
315475999dcSJohn McCall   llvm::Value *Callee =
3167f416cc4SJohn McCall     CGM.getCXXABI().EmitLoadOfMemberFunctionPointer(*this, BO, This,
3177f416cc4SJohn McCall                                              ThisPtrForCall, MemFnPtr, MPT);
31827da15baSAnders Carlsson 
31927da15baSAnders Carlsson   CallArgList Args;
32027da15baSAnders Carlsson 
32127da15baSAnders Carlsson   QualType ThisType =
32227da15baSAnders Carlsson     getContext().getPointerType(getContext().getTagDeclType(RD));
32327da15baSAnders Carlsson 
32427da15baSAnders Carlsson   // Push the this ptr.
3257f416cc4SJohn McCall   Args.add(RValue::get(ThisPtrForCall), ThisType);
32627da15baSAnders Carlsson 
327*419996ccSGeorge Burgess IV   RequiredArgs required =
328*419996ccSGeorge Burgess IV       RequiredArgs::forPrototypePlus(FPT, 1, /*FD=*/nullptr);
3298dda7b27SJohn McCall 
33027da15baSAnders Carlsson   // And the rest of the call args
331*419996ccSGeorge Burgess IV   EmitCallArgs(Args, FPT, E->arguments());
3325fa40c3bSNick Lewycky   return EmitCall(CGM.getTypes().arrangeCXXMethodCall(Args, FPT, required),
3335fa40c3bSNick Lewycky                   Callee, ReturnValue, Args);
33427da15baSAnders Carlsson }
33527da15baSAnders Carlsson 
33627da15baSAnders Carlsson RValue
33727da15baSAnders Carlsson CodeGenFunction::EmitCXXOperatorMemberCallExpr(const CXXOperatorCallExpr *E,
33827da15baSAnders Carlsson                                                const CXXMethodDecl *MD,
33927da15baSAnders Carlsson                                                ReturnValueSlot ReturnValue) {
34027da15baSAnders Carlsson   assert(MD->isInstance() &&
34127da15baSAnders Carlsson          "Trying to emit a member call expr on a static method!");
342aad4af6dSNico Weber   return EmitCXXMemberOrOperatorMemberCallExpr(
343aad4af6dSNico Weber       E, MD, ReturnValue, /*HasQualifier=*/false, /*Qualifier=*/nullptr,
344aad4af6dSNico Weber       /*IsArrow=*/false, E->getArg(0));
34527da15baSAnders Carlsson }
34627da15baSAnders Carlsson 
347fe883422SPeter Collingbourne RValue CodeGenFunction::EmitCUDAKernelCallExpr(const CUDAKernelCallExpr *E,
348fe883422SPeter Collingbourne                                                ReturnValueSlot ReturnValue) {
349fe883422SPeter Collingbourne   return CGM.getCUDARuntime().EmitCUDAKernelCallExpr(*this, E, ReturnValue);
350fe883422SPeter Collingbourne }
351fe883422SPeter Collingbourne 
352fde961dbSEli Friedman static void EmitNullBaseClassInitialization(CodeGenFunction &CGF,
3537f416cc4SJohn McCall                                             Address DestPtr,
354fde961dbSEli Friedman                                             const CXXRecordDecl *Base) {
355fde961dbSEli Friedman   if (Base->isEmpty())
356fde961dbSEli Friedman     return;
357fde961dbSEli Friedman 
3587f416cc4SJohn McCall   DestPtr = CGF.Builder.CreateElementBitCast(DestPtr, CGF.Int8Ty);
359fde961dbSEli Friedman 
360fde961dbSEli Friedman   const ASTRecordLayout &Layout = CGF.getContext().getASTRecordLayout(Base);
3618671c6e0SDavid Majnemer   CharUnits NVSize = Layout.getNonVirtualSize();
3628671c6e0SDavid Majnemer 
3638671c6e0SDavid Majnemer   // We cannot simply zero-initialize the entire base sub-object if vbptrs are
3648671c6e0SDavid Majnemer   // present, they are initialized by the most derived class before calling the
3658671c6e0SDavid Majnemer   // constructor.
3668671c6e0SDavid Majnemer   SmallVector<std::pair<CharUnits, CharUnits>, 1> Stores;
3678671c6e0SDavid Majnemer   Stores.emplace_back(CharUnits::Zero(), NVSize);
3688671c6e0SDavid Majnemer 
3698671c6e0SDavid Majnemer   // Each store is split by the existence of a vbptr.
3708671c6e0SDavid Majnemer   CharUnits VBPtrWidth = CGF.getPointerSize();
3718671c6e0SDavid Majnemer   std::vector<CharUnits> VBPtrOffsets =
3728671c6e0SDavid Majnemer       CGF.CGM.getCXXABI().getVBPtrOffsets(Base);
3738671c6e0SDavid Majnemer   for (CharUnits VBPtrOffset : VBPtrOffsets) {
3747f980d84SDavid Majnemer     // Stop before we hit any virtual base pointers located in virtual bases.
3757f980d84SDavid Majnemer     if (VBPtrOffset >= NVSize)
3767f980d84SDavid Majnemer       break;
3778671c6e0SDavid Majnemer     std::pair<CharUnits, CharUnits> LastStore = Stores.pop_back_val();
3788671c6e0SDavid Majnemer     CharUnits LastStoreOffset = LastStore.first;
3798671c6e0SDavid Majnemer     CharUnits LastStoreSize = LastStore.second;
3808671c6e0SDavid Majnemer 
3818671c6e0SDavid Majnemer     CharUnits SplitBeforeOffset = LastStoreOffset;
3828671c6e0SDavid Majnemer     CharUnits SplitBeforeSize = VBPtrOffset - SplitBeforeOffset;
3838671c6e0SDavid Majnemer     assert(!SplitBeforeSize.isNegative() && "negative store size!");
3848671c6e0SDavid Majnemer     if (!SplitBeforeSize.isZero())
3858671c6e0SDavid Majnemer       Stores.emplace_back(SplitBeforeOffset, SplitBeforeSize);
3868671c6e0SDavid Majnemer 
3878671c6e0SDavid Majnemer     CharUnits SplitAfterOffset = VBPtrOffset + VBPtrWidth;
3888671c6e0SDavid Majnemer     CharUnits SplitAfterSize = LastStoreSize - SplitAfterOffset;
3898671c6e0SDavid Majnemer     assert(!SplitAfterSize.isNegative() && "negative store size!");
3908671c6e0SDavid Majnemer     if (!SplitAfterSize.isZero())
3918671c6e0SDavid Majnemer       Stores.emplace_back(SplitAfterOffset, SplitAfterSize);
3928671c6e0SDavid Majnemer   }
393fde961dbSEli Friedman 
394fde961dbSEli Friedman   // If the type contains a pointer to data member we can't memset it to zero.
395fde961dbSEli Friedman   // Instead, create a null constant and copy it to the destination.
396fde961dbSEli Friedman   // TODO: there are other patterns besides zero that we can usefully memset,
397fde961dbSEli Friedman   // like -1, which happens to be the pattern used by member-pointers.
398fde961dbSEli Friedman   // TODO: isZeroInitializable can be over-conservative in the case where a
399fde961dbSEli Friedman   // virtual base contains a member pointer.
4008671c6e0SDavid Majnemer   llvm::Constant *NullConstantForBase = CGF.CGM.EmitNullConstantForBase(Base);
4018671c6e0SDavid Majnemer   if (!NullConstantForBase->isNullValue()) {
4028671c6e0SDavid Majnemer     llvm::GlobalVariable *NullVariable = new llvm::GlobalVariable(
4038671c6e0SDavid Majnemer         CGF.CGM.getModule(), NullConstantForBase->getType(),
4048671c6e0SDavid Majnemer         /*isConstant=*/true, llvm::GlobalVariable::PrivateLinkage,
4058671c6e0SDavid Majnemer         NullConstantForBase, Twine());
4067f416cc4SJohn McCall 
4077f416cc4SJohn McCall     CharUnits Align = std::max(Layout.getNonVirtualAlignment(),
4087f416cc4SJohn McCall                                DestPtr.getAlignment());
409fde961dbSEli Friedman     NullVariable->setAlignment(Align.getQuantity());
4107f416cc4SJohn McCall 
4117f416cc4SJohn McCall     Address SrcPtr = Address(CGF.EmitCastToVoidPtr(NullVariable), Align);
412fde961dbSEli Friedman 
413fde961dbSEli Friedman     // Get and call the appropriate llvm.memcpy overload.
4148671c6e0SDavid Majnemer     for (std::pair<CharUnits, CharUnits> Store : Stores) {
4158671c6e0SDavid Majnemer       CharUnits StoreOffset = Store.first;
4168671c6e0SDavid Majnemer       CharUnits StoreSize = Store.second;
4178671c6e0SDavid Majnemer       llvm::Value *StoreSizeVal = CGF.CGM.getSize(StoreSize);
4188671c6e0SDavid Majnemer       CGF.Builder.CreateMemCpy(
4198671c6e0SDavid Majnemer           CGF.Builder.CreateConstInBoundsByteGEP(DestPtr, StoreOffset),
4208671c6e0SDavid Majnemer           CGF.Builder.CreateConstInBoundsByteGEP(SrcPtr, StoreOffset),
4218671c6e0SDavid Majnemer           StoreSizeVal);
422fde961dbSEli Friedman     }
423fde961dbSEli Friedman 
424fde961dbSEli Friedman   // Otherwise, just memset the whole thing to zero.  This is legal
425fde961dbSEli Friedman   // because in LLVM, all default initializers (other than the ones we just
426fde961dbSEli Friedman   // handled above) are guaranteed to have a bit pattern of all zeros.
4278671c6e0SDavid Majnemer   } else {
4288671c6e0SDavid Majnemer     for (std::pair<CharUnits, CharUnits> Store : Stores) {
4298671c6e0SDavid Majnemer       CharUnits StoreOffset = Store.first;
4308671c6e0SDavid Majnemer       CharUnits StoreSize = Store.second;
4318671c6e0SDavid Majnemer       llvm::Value *StoreSizeVal = CGF.CGM.getSize(StoreSize);
4328671c6e0SDavid Majnemer       CGF.Builder.CreateMemSet(
4338671c6e0SDavid Majnemer           CGF.Builder.CreateConstInBoundsByteGEP(DestPtr, StoreOffset),
4348671c6e0SDavid Majnemer           CGF.Builder.getInt8(0), StoreSizeVal);
4358671c6e0SDavid Majnemer     }
4368671c6e0SDavid Majnemer   }
437fde961dbSEli Friedman }
438fde961dbSEli Friedman 
43927da15baSAnders Carlsson void
4407a626f63SJohn McCall CodeGenFunction::EmitCXXConstructExpr(const CXXConstructExpr *E,
4417a626f63SJohn McCall                                       AggValueSlot Dest) {
4427a626f63SJohn McCall   assert(!Dest.isIgnored() && "Must have a destination!");
44327da15baSAnders Carlsson   const CXXConstructorDecl *CD = E->getConstructor();
444630c76efSDouglas Gregor 
445630c76efSDouglas Gregor   // If we require zero initialization before (or instead of) calling the
446630c76efSDouglas Gregor   // constructor, as can be the case with a non-user-provided default
44703535265SArgyrios Kyrtzidis   // constructor, emit the zero initialization now, unless destination is
44803535265SArgyrios Kyrtzidis   // already zeroed.
449fde961dbSEli Friedman   if (E->requiresZeroInitialization() && !Dest.isZeroed()) {
450fde961dbSEli Friedman     switch (E->getConstructionKind()) {
451fde961dbSEli Friedman     case CXXConstructExpr::CK_Delegating:
452fde961dbSEli Friedman     case CXXConstructExpr::CK_Complete:
4537f416cc4SJohn McCall       EmitNullInitialization(Dest.getAddress(), E->getType());
454fde961dbSEli Friedman       break;
455fde961dbSEli Friedman     case CXXConstructExpr::CK_VirtualBase:
456fde961dbSEli Friedman     case CXXConstructExpr::CK_NonVirtualBase:
4577f416cc4SJohn McCall       EmitNullBaseClassInitialization(*this, Dest.getAddress(),
4587f416cc4SJohn McCall                                       CD->getParent());
459fde961dbSEli Friedman       break;
460fde961dbSEli Friedman     }
461fde961dbSEli Friedman   }
462630c76efSDouglas Gregor 
463630c76efSDouglas Gregor   // If this is a call to a trivial default constructor, do nothing.
464630c76efSDouglas Gregor   if (CD->isTrivial() && CD->isDefaultConstructor())
46527da15baSAnders Carlsson     return;
466630c76efSDouglas Gregor 
4678ea46b66SJohn McCall   // Elide the constructor if we're constructing from a temporary.
4688ea46b66SJohn McCall   // The temporary check is required because Sema sets this on NRVO
4698ea46b66SJohn McCall   // returns.
4709c6890a7SRichard Smith   if (getLangOpts().ElideConstructors && E->isElidable()) {
4718ea46b66SJohn McCall     assert(getContext().hasSameUnqualifiedType(E->getType(),
4728ea46b66SJohn McCall                                                E->getArg(0)->getType()));
4737a626f63SJohn McCall     if (E->getArg(0)->isTemporaryObject(getContext(), CD->getParent())) {
4747a626f63SJohn McCall       EmitAggExpr(E->getArg(0), Dest);
47527da15baSAnders Carlsson       return;
47627da15baSAnders Carlsson     }
477222cf0efSDouglas Gregor   }
478630c76efSDouglas Gregor 
479e7545b33SAlexey Bataev   if (const ArrayType *arrayType
480e7545b33SAlexey Bataev         = getContext().getAsArrayType(E->getType())) {
4817f416cc4SJohn McCall     EmitCXXAggrConstructorCall(CD, arrayType, Dest.getAddress(), E);
482f677a8e9SJohn McCall   } else {
483bceca20aSCameron Esfahani     CXXCtorType Type = Ctor_Complete;
484271c3681SAlexis Hunt     bool ForVirtualBase = false;
48561535005SDouglas Gregor     bool Delegating = false;
486271c3681SAlexis Hunt 
487271c3681SAlexis Hunt     switch (E->getConstructionKind()) {
488271c3681SAlexis Hunt      case CXXConstructExpr::CK_Delegating:
48961bc1737SAlexis Hunt       // We should be emitting a constructor; GlobalDecl will assert this
49061bc1737SAlexis Hunt       Type = CurGD.getCtorType();
49161535005SDouglas Gregor       Delegating = true;
492271c3681SAlexis Hunt       break;
49361bc1737SAlexis Hunt 
494271c3681SAlexis Hunt      case CXXConstructExpr::CK_Complete:
495271c3681SAlexis Hunt       Type = Ctor_Complete;
496271c3681SAlexis Hunt       break;
497271c3681SAlexis Hunt 
498271c3681SAlexis Hunt      case CXXConstructExpr::CK_VirtualBase:
499271c3681SAlexis Hunt       ForVirtualBase = true;
500271c3681SAlexis Hunt       // fall-through
501271c3681SAlexis Hunt 
502271c3681SAlexis Hunt      case CXXConstructExpr::CK_NonVirtualBase:
503271c3681SAlexis Hunt       Type = Ctor_Base;
504271c3681SAlexis Hunt     }
505e11f9ce9SAnders Carlsson 
50627da15baSAnders Carlsson     // Call the constructor.
5077f416cc4SJohn McCall     EmitCXXConstructorCall(CD, Type, ForVirtualBase, Delegating,
5087f416cc4SJohn McCall                            Dest.getAddress(), E);
50927da15baSAnders Carlsson   }
510e11f9ce9SAnders Carlsson }
51127da15baSAnders Carlsson 
5127f416cc4SJohn McCall void CodeGenFunction::EmitSynthesizedCXXCopyCtor(Address Dest, Address Src,
51350198098SFariborz Jahanian                                                  const Expr *Exp) {
5145d413781SJohn McCall   if (const ExprWithCleanups *E = dyn_cast<ExprWithCleanups>(Exp))
515e988bdacSFariborz Jahanian     Exp = E->getSubExpr();
516e988bdacSFariborz Jahanian   assert(isa<CXXConstructExpr>(Exp) &&
517e988bdacSFariborz Jahanian          "EmitSynthesizedCXXCopyCtor - unknown copy ctor expr");
518e988bdacSFariborz Jahanian   const CXXConstructExpr* E = cast<CXXConstructExpr>(Exp);
519e988bdacSFariborz Jahanian   const CXXConstructorDecl *CD = E->getConstructor();
520e988bdacSFariborz Jahanian   RunCleanupsScope Scope(*this);
521e988bdacSFariborz Jahanian 
522e988bdacSFariborz Jahanian   // If we require zero initialization before (or instead of) calling the
523e988bdacSFariborz Jahanian   // constructor, as can be the case with a non-user-provided default
524e988bdacSFariborz Jahanian   // constructor, emit the zero initialization now.
525e988bdacSFariborz Jahanian   // FIXME. Do I still need this for a copy ctor synthesis?
526e988bdacSFariborz Jahanian   if (E->requiresZeroInitialization())
527e988bdacSFariborz Jahanian     EmitNullInitialization(Dest, E->getType());
528e988bdacSFariborz Jahanian 
52999da11cfSChandler Carruth   assert(!getContext().getAsConstantArrayType(E->getType())
53099da11cfSChandler Carruth          && "EmitSynthesizedCXXCopyCtor - Copied-in Array");
531525bf650SAlexey Samsonov   EmitSynthesizedCXXCopyCtorCall(CD, Dest, Src, E);
532e988bdacSFariborz Jahanian }
533e988bdacSFariborz Jahanian 
5348ed55a54SJohn McCall static CharUnits CalculateCookiePadding(CodeGenFunction &CGF,
5358ed55a54SJohn McCall                                         const CXXNewExpr *E) {
53621122cf6SAnders Carlsson   if (!E->isArray())
5373eb55cfeSKen Dyck     return CharUnits::Zero();
53821122cf6SAnders Carlsson 
5397ec4b434SJohn McCall   // No cookie is required if the operator new[] being used is the
5407ec4b434SJohn McCall   // reserved placement operator new[].
5417ec4b434SJohn McCall   if (E->getOperatorNew()->isReservedGlobalPlacementOperator())
5423eb55cfeSKen Dyck     return CharUnits::Zero();
543399f499fSAnders Carlsson 
544284c48ffSJohn McCall   return CGF.CGM.getCXXABI().GetArrayCookieSize(E);
54559486a2dSAnders Carlsson }
54659486a2dSAnders Carlsson 
547036f2f6bSJohn McCall static llvm::Value *EmitCXXNewAllocSize(CodeGenFunction &CGF,
548036f2f6bSJohn McCall                                         const CXXNewExpr *e,
549f862eb6aSSebastian Redl                                         unsigned minElements,
550036f2f6bSJohn McCall                                         llvm::Value *&numElements,
551036f2f6bSJohn McCall                                         llvm::Value *&sizeWithoutCookie) {
552036f2f6bSJohn McCall   QualType type = e->getAllocatedType();
55359486a2dSAnders Carlsson 
554036f2f6bSJohn McCall   if (!e->isArray()) {
555036f2f6bSJohn McCall     CharUnits typeSize = CGF.getContext().getTypeSizeInChars(type);
556036f2f6bSJohn McCall     sizeWithoutCookie
557036f2f6bSJohn McCall       = llvm::ConstantInt::get(CGF.SizeTy, typeSize.getQuantity());
558036f2f6bSJohn McCall     return sizeWithoutCookie;
55905fc5be3SDouglas Gregor   }
56059486a2dSAnders Carlsson 
561036f2f6bSJohn McCall   // The width of size_t.
562036f2f6bSJohn McCall   unsigned sizeWidth = CGF.SizeTy->getBitWidth();
563036f2f6bSJohn McCall 
5648ed55a54SJohn McCall   // Figure out the cookie size.
565036f2f6bSJohn McCall   llvm::APInt cookieSize(sizeWidth,
566036f2f6bSJohn McCall                          CalculateCookiePadding(CGF, e).getQuantity());
5678ed55a54SJohn McCall 
56859486a2dSAnders Carlsson   // Emit the array size expression.
5697648fb46SArgyrios Kyrtzidis   // We multiply the size of all dimensions for NumElements.
5707648fb46SArgyrios Kyrtzidis   // e.g for 'int[2][3]', ElemType is 'int' and NumElements is 6.
571036f2f6bSJohn McCall   numElements = CGF.EmitScalarExpr(e->getArraySize());
572036f2f6bSJohn McCall   assert(isa<llvm::IntegerType>(numElements->getType()));
5738ed55a54SJohn McCall 
574036f2f6bSJohn McCall   // The number of elements can be have an arbitrary integer type;
575036f2f6bSJohn McCall   // essentially, we need to multiply it by a constant factor, add a
576036f2f6bSJohn McCall   // cookie size, and verify that the result is representable as a
577036f2f6bSJohn McCall   // size_t.  That's just a gloss, though, and it's wrong in one
578036f2f6bSJohn McCall   // important way: if the count is negative, it's an error even if
579036f2f6bSJohn McCall   // the cookie size would bring the total size >= 0.
5806ab2fa8fSDouglas Gregor   bool isSigned
5816ab2fa8fSDouglas Gregor     = e->getArraySize()->getType()->isSignedIntegerOrEnumerationType();
5822192fe50SChris Lattner   llvm::IntegerType *numElementsType
583036f2f6bSJohn McCall     = cast<llvm::IntegerType>(numElements->getType());
584036f2f6bSJohn McCall   unsigned numElementsWidth = numElementsType->getBitWidth();
585036f2f6bSJohn McCall 
586036f2f6bSJohn McCall   // Compute the constant factor.
587036f2f6bSJohn McCall   llvm::APInt arraySizeMultiplier(sizeWidth, 1);
5887648fb46SArgyrios Kyrtzidis   while (const ConstantArrayType *CAT
589036f2f6bSJohn McCall              = CGF.getContext().getAsConstantArrayType(type)) {
590036f2f6bSJohn McCall     type = CAT->getElementType();
591036f2f6bSJohn McCall     arraySizeMultiplier *= CAT->getSize();
5927648fb46SArgyrios Kyrtzidis   }
59359486a2dSAnders Carlsson 
594036f2f6bSJohn McCall   CharUnits typeSize = CGF.getContext().getTypeSizeInChars(type);
595036f2f6bSJohn McCall   llvm::APInt typeSizeMultiplier(sizeWidth, typeSize.getQuantity());
596036f2f6bSJohn McCall   typeSizeMultiplier *= arraySizeMultiplier;
597036f2f6bSJohn McCall 
598036f2f6bSJohn McCall   // This will be a size_t.
599036f2f6bSJohn McCall   llvm::Value *size;
60032ac583dSChris Lattner 
60132ac583dSChris Lattner   // If someone is doing 'new int[42]' there is no need to do a dynamic check.
60232ac583dSChris Lattner   // Don't bloat the -O0 code.
603036f2f6bSJohn McCall   if (llvm::ConstantInt *numElementsC =
604036f2f6bSJohn McCall         dyn_cast<llvm::ConstantInt>(numElements)) {
605036f2f6bSJohn McCall     const llvm::APInt &count = numElementsC->getValue();
60632ac583dSChris Lattner 
607036f2f6bSJohn McCall     bool hasAnyOverflow = false;
60832ac583dSChris Lattner 
609036f2f6bSJohn McCall     // If 'count' was a negative number, it's an overflow.
610036f2f6bSJohn McCall     if (isSigned && count.isNegative())
611036f2f6bSJohn McCall       hasAnyOverflow = true;
6128ed55a54SJohn McCall 
613036f2f6bSJohn McCall     // We want to do all this arithmetic in size_t.  If numElements is
614036f2f6bSJohn McCall     // wider than that, check whether it's already too big, and if so,
615036f2f6bSJohn McCall     // overflow.
616036f2f6bSJohn McCall     else if (numElementsWidth > sizeWidth &&
617036f2f6bSJohn McCall              numElementsWidth - sizeWidth > count.countLeadingZeros())
618036f2f6bSJohn McCall       hasAnyOverflow = true;
619036f2f6bSJohn McCall 
620036f2f6bSJohn McCall     // Okay, compute a count at the right width.
621036f2f6bSJohn McCall     llvm::APInt adjustedCount = count.zextOrTrunc(sizeWidth);
622036f2f6bSJohn McCall 
623f862eb6aSSebastian Redl     // If there is a brace-initializer, we cannot allocate fewer elements than
624f862eb6aSSebastian Redl     // there are initializers. If we do, that's treated like an overflow.
625f862eb6aSSebastian Redl     if (adjustedCount.ult(minElements))
626f862eb6aSSebastian Redl       hasAnyOverflow = true;
627f862eb6aSSebastian Redl 
628036f2f6bSJohn McCall     // Scale numElements by that.  This might overflow, but we don't
629036f2f6bSJohn McCall     // care because it only overflows if allocationSize does, too, and
630036f2f6bSJohn McCall     // if that overflows then we shouldn't use this.
631036f2f6bSJohn McCall     numElements = llvm::ConstantInt::get(CGF.SizeTy,
632036f2f6bSJohn McCall                                          adjustedCount * arraySizeMultiplier);
633036f2f6bSJohn McCall 
634036f2f6bSJohn McCall     // Compute the size before cookie, and track whether it overflowed.
635036f2f6bSJohn McCall     bool overflow;
636036f2f6bSJohn McCall     llvm::APInt allocationSize
637036f2f6bSJohn McCall       = adjustedCount.umul_ov(typeSizeMultiplier, overflow);
638036f2f6bSJohn McCall     hasAnyOverflow |= overflow;
639036f2f6bSJohn McCall 
640036f2f6bSJohn McCall     // Add in the cookie, and check whether it's overflowed.
641036f2f6bSJohn McCall     if (cookieSize != 0) {
642036f2f6bSJohn McCall       // Save the current size without a cookie.  This shouldn't be
643036f2f6bSJohn McCall       // used if there was overflow.
644036f2f6bSJohn McCall       sizeWithoutCookie = llvm::ConstantInt::get(CGF.SizeTy, allocationSize);
645036f2f6bSJohn McCall 
646036f2f6bSJohn McCall       allocationSize = allocationSize.uadd_ov(cookieSize, overflow);
647036f2f6bSJohn McCall       hasAnyOverflow |= overflow;
6488ed55a54SJohn McCall     }
6498ed55a54SJohn McCall 
650036f2f6bSJohn McCall     // On overflow, produce a -1 so operator new will fail.
651455f42c9SAaron Ballman     if (hasAnyOverflow) {
652455f42c9SAaron Ballman       size = llvm::Constant::getAllOnesValue(CGF.SizeTy);
653455f42c9SAaron Ballman     } else {
654036f2f6bSJohn McCall       size = llvm::ConstantInt::get(CGF.SizeTy, allocationSize);
655455f42c9SAaron Ballman     }
65632ac583dSChris Lattner 
657036f2f6bSJohn McCall   // Otherwise, we might need to use the overflow intrinsics.
6588ed55a54SJohn McCall   } else {
659f862eb6aSSebastian Redl     // There are up to five conditions we need to test for:
660036f2f6bSJohn McCall     // 1) if isSigned, we need to check whether numElements is negative;
661036f2f6bSJohn McCall     // 2) if numElementsWidth > sizeWidth, we need to check whether
662036f2f6bSJohn McCall     //   numElements is larger than something representable in size_t;
663f862eb6aSSebastian Redl     // 3) if minElements > 0, we need to check whether numElements is smaller
664f862eb6aSSebastian Redl     //    than that.
665f862eb6aSSebastian Redl     // 4) we need to compute
666036f2f6bSJohn McCall     //      sizeWithoutCookie := numElements * typeSizeMultiplier
667036f2f6bSJohn McCall     //    and check whether it overflows; and
668f862eb6aSSebastian Redl     // 5) if we need a cookie, we need to compute
669036f2f6bSJohn McCall     //      size := sizeWithoutCookie + cookieSize
670036f2f6bSJohn McCall     //    and check whether it overflows.
6718ed55a54SJohn McCall 
6728a13c418SCraig Topper     llvm::Value *hasOverflow = nullptr;
6738ed55a54SJohn McCall 
674036f2f6bSJohn McCall     // If numElementsWidth > sizeWidth, then one way or another, we're
675036f2f6bSJohn McCall     // going to have to do a comparison for (2), and this happens to
676036f2f6bSJohn McCall     // take care of (1), too.
677036f2f6bSJohn McCall     if (numElementsWidth > sizeWidth) {
678036f2f6bSJohn McCall       llvm::APInt threshold(numElementsWidth, 1);
679036f2f6bSJohn McCall       threshold <<= sizeWidth;
6808ed55a54SJohn McCall 
681036f2f6bSJohn McCall       llvm::Value *thresholdV
682036f2f6bSJohn McCall         = llvm::ConstantInt::get(numElementsType, threshold);
683036f2f6bSJohn McCall 
684036f2f6bSJohn McCall       hasOverflow = CGF.Builder.CreateICmpUGE(numElements, thresholdV);
685036f2f6bSJohn McCall       numElements = CGF.Builder.CreateTrunc(numElements, CGF.SizeTy);
686036f2f6bSJohn McCall 
687036f2f6bSJohn McCall     // Otherwise, if we're signed, we want to sext up to size_t.
688036f2f6bSJohn McCall     } else if (isSigned) {
689036f2f6bSJohn McCall       if (numElementsWidth < sizeWidth)
690036f2f6bSJohn McCall         numElements = CGF.Builder.CreateSExt(numElements, CGF.SizeTy);
691036f2f6bSJohn McCall 
692036f2f6bSJohn McCall       // If there's a non-1 type size multiplier, then we can do the
693036f2f6bSJohn McCall       // signedness check at the same time as we do the multiply
694036f2f6bSJohn McCall       // because a negative number times anything will cause an
695f862eb6aSSebastian Redl       // unsigned overflow.  Otherwise, we have to do it here. But at least
696f862eb6aSSebastian Redl       // in this case, we can subsume the >= minElements check.
697036f2f6bSJohn McCall       if (typeSizeMultiplier == 1)
698036f2f6bSJohn McCall         hasOverflow = CGF.Builder.CreateICmpSLT(numElements,
699f862eb6aSSebastian Redl                               llvm::ConstantInt::get(CGF.SizeTy, minElements));
700036f2f6bSJohn McCall 
701036f2f6bSJohn McCall     // Otherwise, zext up to size_t if necessary.
702036f2f6bSJohn McCall     } else if (numElementsWidth < sizeWidth) {
703036f2f6bSJohn McCall       numElements = CGF.Builder.CreateZExt(numElements, CGF.SizeTy);
704036f2f6bSJohn McCall     }
705036f2f6bSJohn McCall 
706036f2f6bSJohn McCall     assert(numElements->getType() == CGF.SizeTy);
707036f2f6bSJohn McCall 
708f862eb6aSSebastian Redl     if (minElements) {
709f862eb6aSSebastian Redl       // Don't allow allocation of fewer elements than we have initializers.
710f862eb6aSSebastian Redl       if (!hasOverflow) {
711f862eb6aSSebastian Redl         hasOverflow = CGF.Builder.CreateICmpULT(numElements,
712f862eb6aSSebastian Redl                               llvm::ConstantInt::get(CGF.SizeTy, minElements));
713f862eb6aSSebastian Redl       } else if (numElementsWidth > sizeWidth) {
714f862eb6aSSebastian Redl         // The other existing overflow subsumes this check.
715f862eb6aSSebastian Redl         // We do an unsigned comparison, since any signed value < -1 is
716f862eb6aSSebastian Redl         // taken care of either above or below.
717f862eb6aSSebastian Redl         hasOverflow = CGF.Builder.CreateOr(hasOverflow,
718f862eb6aSSebastian Redl                           CGF.Builder.CreateICmpULT(numElements,
719f862eb6aSSebastian Redl                               llvm::ConstantInt::get(CGF.SizeTy, minElements)));
720f862eb6aSSebastian Redl       }
721f862eb6aSSebastian Redl     }
722f862eb6aSSebastian Redl 
723036f2f6bSJohn McCall     size = numElements;
724036f2f6bSJohn McCall 
725036f2f6bSJohn McCall     // Multiply by the type size if necessary.  This multiplier
726036f2f6bSJohn McCall     // includes all the factors for nested arrays.
7278ed55a54SJohn McCall     //
728036f2f6bSJohn McCall     // This step also causes numElements to be scaled up by the
729036f2f6bSJohn McCall     // nested-array factor if necessary.  Overflow on this computation
730036f2f6bSJohn McCall     // can be ignored because the result shouldn't be used if
731036f2f6bSJohn McCall     // allocation fails.
732036f2f6bSJohn McCall     if (typeSizeMultiplier != 1) {
733036f2f6bSJohn McCall       llvm::Value *umul_with_overflow
7348d375cefSBenjamin Kramer         = CGF.CGM.getIntrinsic(llvm::Intrinsic::umul_with_overflow, CGF.SizeTy);
7358ed55a54SJohn McCall 
736036f2f6bSJohn McCall       llvm::Value *tsmV =
737036f2f6bSJohn McCall         llvm::ConstantInt::get(CGF.SizeTy, typeSizeMultiplier);
738036f2f6bSJohn McCall       llvm::Value *result =
73943f9bb73SDavid Blaikie           CGF.Builder.CreateCall(umul_with_overflow, {size, tsmV});
7408ed55a54SJohn McCall 
741036f2f6bSJohn McCall       llvm::Value *overflowed = CGF.Builder.CreateExtractValue(result, 1);
742036f2f6bSJohn McCall       if (hasOverflow)
743036f2f6bSJohn McCall         hasOverflow = CGF.Builder.CreateOr(hasOverflow, overflowed);
7448ed55a54SJohn McCall       else
745036f2f6bSJohn McCall         hasOverflow = overflowed;
74659486a2dSAnders Carlsson 
747036f2f6bSJohn McCall       size = CGF.Builder.CreateExtractValue(result, 0);
748036f2f6bSJohn McCall 
749036f2f6bSJohn McCall       // Also scale up numElements by the array size multiplier.
750036f2f6bSJohn McCall       if (arraySizeMultiplier != 1) {
751036f2f6bSJohn McCall         // If the base element type size is 1, then we can re-use the
752036f2f6bSJohn McCall         // multiply we just did.
753036f2f6bSJohn McCall         if (typeSize.isOne()) {
754036f2f6bSJohn McCall           assert(arraySizeMultiplier == typeSizeMultiplier);
755036f2f6bSJohn McCall           numElements = size;
756036f2f6bSJohn McCall 
757036f2f6bSJohn McCall         // Otherwise we need a separate multiply.
758036f2f6bSJohn McCall         } else {
759036f2f6bSJohn McCall           llvm::Value *asmV =
760036f2f6bSJohn McCall             llvm::ConstantInt::get(CGF.SizeTy, arraySizeMultiplier);
761036f2f6bSJohn McCall           numElements = CGF.Builder.CreateMul(numElements, asmV);
762036f2f6bSJohn McCall         }
763036f2f6bSJohn McCall       }
764036f2f6bSJohn McCall     } else {
765036f2f6bSJohn McCall       // numElements doesn't need to be scaled.
766036f2f6bSJohn McCall       assert(arraySizeMultiplier == 1);
767036f2f6bSJohn McCall     }
768036f2f6bSJohn McCall 
769036f2f6bSJohn McCall     // Add in the cookie size if necessary.
770036f2f6bSJohn McCall     if (cookieSize != 0) {
771036f2f6bSJohn McCall       sizeWithoutCookie = size;
772036f2f6bSJohn McCall 
773036f2f6bSJohn McCall       llvm::Value *uadd_with_overflow
7748d375cefSBenjamin Kramer         = CGF.CGM.getIntrinsic(llvm::Intrinsic::uadd_with_overflow, CGF.SizeTy);
775036f2f6bSJohn McCall 
776036f2f6bSJohn McCall       llvm::Value *cookieSizeV = llvm::ConstantInt::get(CGF.SizeTy, cookieSize);
777036f2f6bSJohn McCall       llvm::Value *result =
77843f9bb73SDavid Blaikie           CGF.Builder.CreateCall(uadd_with_overflow, {size, cookieSizeV});
779036f2f6bSJohn McCall 
780036f2f6bSJohn McCall       llvm::Value *overflowed = CGF.Builder.CreateExtractValue(result, 1);
781036f2f6bSJohn McCall       if (hasOverflow)
782036f2f6bSJohn McCall         hasOverflow = CGF.Builder.CreateOr(hasOverflow, overflowed);
783036f2f6bSJohn McCall       else
784036f2f6bSJohn McCall         hasOverflow = overflowed;
785036f2f6bSJohn McCall 
786036f2f6bSJohn McCall       size = CGF.Builder.CreateExtractValue(result, 0);
787036f2f6bSJohn McCall     }
788036f2f6bSJohn McCall 
789036f2f6bSJohn McCall     // If we had any possibility of dynamic overflow, make a select to
790036f2f6bSJohn McCall     // overwrite 'size' with an all-ones value, which should cause
791036f2f6bSJohn McCall     // operator new to throw.
792036f2f6bSJohn McCall     if (hasOverflow)
793455f42c9SAaron Ballman       size = CGF.Builder.CreateSelect(hasOverflow,
794455f42c9SAaron Ballman                                  llvm::Constant::getAllOnesValue(CGF.SizeTy),
795036f2f6bSJohn McCall                                       size);
796036f2f6bSJohn McCall   }
797036f2f6bSJohn McCall 
798036f2f6bSJohn McCall   if (cookieSize == 0)
799036f2f6bSJohn McCall     sizeWithoutCookie = size;
800036f2f6bSJohn McCall   else
801036f2f6bSJohn McCall     assert(sizeWithoutCookie && "didn't set sizeWithoutCookie?");
802036f2f6bSJohn McCall 
803036f2f6bSJohn McCall   return size;
80459486a2dSAnders Carlsson }
80559486a2dSAnders Carlsson 
806f862eb6aSSebastian Redl static void StoreAnyExprIntoOneUnit(CodeGenFunction &CGF, const Expr *Init,
8077f416cc4SJohn McCall                                     QualType AllocType, Address NewPtr) {
8081c96bc5dSRichard Smith   // FIXME: Refactor with EmitExprAsInit.
80947fb9508SJohn McCall   switch (CGF.getEvaluationKind(AllocType)) {
81047fb9508SJohn McCall   case TEK_Scalar:
811a2c1124fSDavid Blaikie     CGF.EmitScalarInit(Init, nullptr,
8127f416cc4SJohn McCall                        CGF.MakeAddrLValue(NewPtr, AllocType), false);
81347fb9508SJohn McCall     return;
81447fb9508SJohn McCall   case TEK_Complex:
8157f416cc4SJohn McCall     CGF.EmitComplexExprIntoLValue(Init, CGF.MakeAddrLValue(NewPtr, AllocType),
81647fb9508SJohn McCall                                   /*isInit*/ true);
81747fb9508SJohn McCall     return;
81847fb9508SJohn McCall   case TEK_Aggregate: {
8197a626f63SJohn McCall     AggValueSlot Slot
8207f416cc4SJohn McCall       = AggValueSlot::forAddr(NewPtr, AllocType.getQualifiers(),
8218d6fc958SJohn McCall                               AggValueSlot::IsDestructed,
82246759f4fSJohn McCall                               AggValueSlot::DoesNotNeedGCBarriers,
823615ed1a3SChad Rosier                               AggValueSlot::IsNotAliased);
8247a626f63SJohn McCall     CGF.EmitAggExpr(Init, Slot);
82547fb9508SJohn McCall     return;
8267a626f63SJohn McCall   }
827d5202e09SFariborz Jahanian   }
82847fb9508SJohn McCall   llvm_unreachable("bad evaluation kind");
82947fb9508SJohn McCall }
830d5202e09SFariborz Jahanian 
831fb901c7aSDavid Blaikie void CodeGenFunction::EmitNewArrayInitializer(
832fb901c7aSDavid Blaikie     const CXXNewExpr *E, QualType ElementType, llvm::Type *ElementTy,
8337f416cc4SJohn McCall     Address BeginPtr, llvm::Value *NumElements,
83406a67e2cSRichard Smith     llvm::Value *AllocSizeWithoutCookie) {
83506a67e2cSRichard Smith   // If we have a type with trivial initialization and no initializer,
83606a67e2cSRichard Smith   // there's nothing to do.
8376047f07eSSebastian Redl   if (!E->hasInitializer())
83806a67e2cSRichard Smith     return;
839b66b08efSFariborz Jahanian 
8407f416cc4SJohn McCall   Address CurPtr = BeginPtr;
841d5202e09SFariborz Jahanian 
84206a67e2cSRichard Smith   unsigned InitListElements = 0;
843f862eb6aSSebastian Redl 
844f862eb6aSSebastian Redl   const Expr *Init = E->getInitializer();
8457f416cc4SJohn McCall   Address EndOfInit = Address::invalid();
84606a67e2cSRichard Smith   QualType::DestructionKind DtorKind = ElementType.isDestructedType();
84706a67e2cSRichard Smith   EHScopeStack::stable_iterator Cleanup;
84806a67e2cSRichard Smith   llvm::Instruction *CleanupDominator = nullptr;
8491c96bc5dSRichard Smith 
8507f416cc4SJohn McCall   CharUnits ElementSize = getContext().getTypeSizeInChars(ElementType);
8517f416cc4SJohn McCall   CharUnits ElementAlign =
8527f416cc4SJohn McCall     BeginPtr.getAlignment().alignmentOfArrayElement(ElementSize);
8537f416cc4SJohn McCall 
854f862eb6aSSebastian Redl   // If the initializer is an initializer list, first do the explicit elements.
855f862eb6aSSebastian Redl   if (const InitListExpr *ILE = dyn_cast<InitListExpr>(Init)) {
85606a67e2cSRichard Smith     InitListElements = ILE->getNumInits();
857f62290a1SChad Rosier 
8581c96bc5dSRichard Smith     // If this is a multi-dimensional array new, we will initialize multiple
8591c96bc5dSRichard Smith     // elements with each init list element.
8601c96bc5dSRichard Smith     QualType AllocType = E->getAllocatedType();
8611c96bc5dSRichard Smith     if (const ConstantArrayType *CAT = dyn_cast_or_null<ConstantArrayType>(
8621c96bc5dSRichard Smith             AllocType->getAsArrayTypeUnsafe())) {
863fb901c7aSDavid Blaikie       ElementTy = ConvertTypeForMem(AllocType);
8647f416cc4SJohn McCall       CurPtr = Builder.CreateElementBitCast(CurPtr, ElementTy);
86506a67e2cSRichard Smith       InitListElements *= getContext().getConstantArrayElementCount(CAT);
8661c96bc5dSRichard Smith     }
8671c96bc5dSRichard Smith 
86806a67e2cSRichard Smith     // Enter a partial-destruction Cleanup if necessary.
86906a67e2cSRichard Smith     if (needsEHCleanup(DtorKind)) {
87006a67e2cSRichard Smith       // In principle we could tell the Cleanup where we are more
871f62290a1SChad Rosier       // directly, but the control flow can get so varied here that it
872f62290a1SChad Rosier       // would actually be quite complex.  Therefore we go through an
873f62290a1SChad Rosier       // alloca.
8747f416cc4SJohn McCall       EndOfInit = CreateTempAlloca(BeginPtr.getType(), getPointerAlign(),
8757f416cc4SJohn McCall                                    "array.init.end");
8767f416cc4SJohn McCall       CleanupDominator = Builder.CreateStore(BeginPtr.getPointer(), EndOfInit);
8777f416cc4SJohn McCall       pushIrregularPartialArrayCleanup(BeginPtr.getPointer(), EndOfInit,
8787f416cc4SJohn McCall                                        ElementType, ElementAlign,
87906a67e2cSRichard Smith                                        getDestroyer(DtorKind));
88006a67e2cSRichard Smith       Cleanup = EHStack.stable_begin();
881f62290a1SChad Rosier     }
882f62290a1SChad Rosier 
8837f416cc4SJohn McCall     CharUnits StartAlign = CurPtr.getAlignment();
884f862eb6aSSebastian Redl     for (unsigned i = 0, e = ILE->getNumInits(); i != e; ++i) {
885f62290a1SChad Rosier       // Tell the cleanup that it needs to destroy up to this
886f62290a1SChad Rosier       // element.  TODO: some of these stores can be trivially
887f62290a1SChad Rosier       // observed to be unnecessary.
8887f416cc4SJohn McCall       if (EndOfInit.isValid()) {
8897f416cc4SJohn McCall         auto FinishedPtr =
8907f416cc4SJohn McCall           Builder.CreateBitCast(CurPtr.getPointer(), BeginPtr.getType());
8917f416cc4SJohn McCall         Builder.CreateStore(FinishedPtr, EndOfInit);
8927f416cc4SJohn McCall       }
89306a67e2cSRichard Smith       // FIXME: If the last initializer is an incomplete initializer list for
89406a67e2cSRichard Smith       // an array, and we have an array filler, we can fold together the two
89506a67e2cSRichard Smith       // initialization loops.
8961c96bc5dSRichard Smith       StoreAnyExprIntoOneUnit(*this, ILE->getInit(i),
89706a67e2cSRichard Smith                               ILE->getInit(i)->getType(), CurPtr);
8987f416cc4SJohn McCall       CurPtr = Address(Builder.CreateInBoundsGEP(CurPtr.getPointer(),
8997f416cc4SJohn McCall                                                  Builder.getSize(1),
9007f416cc4SJohn McCall                                                  "array.exp.next"),
9017f416cc4SJohn McCall                        StartAlign.alignmentAtOffset((i + 1) * ElementSize));
902f862eb6aSSebastian Redl     }
903f862eb6aSSebastian Redl 
904f862eb6aSSebastian Redl     // The remaining elements are filled with the array filler expression.
905f862eb6aSSebastian Redl     Init = ILE->getArrayFiller();
9061c96bc5dSRichard Smith 
90706a67e2cSRichard Smith     // Extract the initializer for the individual array elements by pulling
90806a67e2cSRichard Smith     // out the array filler from all the nested initializer lists. This avoids
90906a67e2cSRichard Smith     // generating a nested loop for the initialization.
91006a67e2cSRichard Smith     while (Init && Init->getType()->isConstantArrayType()) {
91106a67e2cSRichard Smith       auto *SubILE = dyn_cast<InitListExpr>(Init);
91206a67e2cSRichard Smith       if (!SubILE)
91306a67e2cSRichard Smith         break;
91406a67e2cSRichard Smith       assert(SubILE->getNumInits() == 0 && "explicit inits in array filler?");
91506a67e2cSRichard Smith       Init = SubILE->getArrayFiller();
916f862eb6aSSebastian Redl     }
917f862eb6aSSebastian Redl 
91806a67e2cSRichard Smith     // Switch back to initializing one base element at a time.
9197f416cc4SJohn McCall     CurPtr = Builder.CreateBitCast(CurPtr, BeginPtr.getType());
920f62290a1SChad Rosier   }
921e6c980c4SChandler Carruth 
92206a67e2cSRichard Smith   // Attempt to perform zero-initialization using memset.
92306a67e2cSRichard Smith   auto TryMemsetInitialization = [&]() -> bool {
92406a67e2cSRichard Smith     // FIXME: If the type is a pointer-to-data-member under the Itanium ABI,
92506a67e2cSRichard Smith     // we can initialize with a memset to -1.
92606a67e2cSRichard Smith     if (!CGM.getTypes().isZeroInitializable(ElementType))
92706a67e2cSRichard Smith       return false;
928e6c980c4SChandler Carruth 
92906a67e2cSRichard Smith     // Optimization: since zero initialization will just set the memory
93006a67e2cSRichard Smith     // to all zeroes, generate a single memset to do it in one shot.
93106a67e2cSRichard Smith 
93206a67e2cSRichard Smith     // Subtract out the size of any elements we've already initialized.
93306a67e2cSRichard Smith     auto *RemainingSize = AllocSizeWithoutCookie;
93406a67e2cSRichard Smith     if (InitListElements) {
93506a67e2cSRichard Smith       // We know this can't overflow; we check this when doing the allocation.
93606a67e2cSRichard Smith       auto *InitializedSize = llvm::ConstantInt::get(
93706a67e2cSRichard Smith           RemainingSize->getType(),
93806a67e2cSRichard Smith           getContext().getTypeSizeInChars(ElementType).getQuantity() *
93906a67e2cSRichard Smith               InitListElements);
94006a67e2cSRichard Smith       RemainingSize = Builder.CreateSub(RemainingSize, InitializedSize);
94199210dc9SJohn McCall     }
942d5202e09SFariborz Jahanian 
94306a67e2cSRichard Smith     // Create the memset.
9447f416cc4SJohn McCall     Builder.CreateMemSet(CurPtr, Builder.getInt8(0), RemainingSize, false);
94506a67e2cSRichard Smith     return true;
94606a67e2cSRichard Smith   };
94705fc5be3SDouglas Gregor 
948454a7cdfSRichard Smith   // If all elements have already been initialized, skip any further
949454a7cdfSRichard Smith   // initialization.
950454a7cdfSRichard Smith   llvm::ConstantInt *ConstNum = dyn_cast<llvm::ConstantInt>(NumElements);
951454a7cdfSRichard Smith   if (ConstNum && ConstNum->getZExtValue() <= InitListElements) {
952454a7cdfSRichard Smith     // If there was a Cleanup, deactivate it.
953454a7cdfSRichard Smith     if (CleanupDominator)
954454a7cdfSRichard Smith       DeactivateCleanupBlock(Cleanup, CleanupDominator);
955454a7cdfSRichard Smith     return;
956454a7cdfSRichard Smith   }
957454a7cdfSRichard Smith 
958454a7cdfSRichard Smith   assert(Init && "have trailing elements to initialize but no initializer");
959454a7cdfSRichard Smith 
96006a67e2cSRichard Smith   // If this is a constructor call, try to optimize it out, and failing that
96106a67e2cSRichard Smith   // emit a single loop to initialize all remaining elements.
962454a7cdfSRichard Smith   if (const CXXConstructExpr *CCE = dyn_cast<CXXConstructExpr>(Init)) {
9636047f07eSSebastian Redl     CXXConstructorDecl *Ctor = CCE->getConstructor();
964d153103cSDouglas Gregor     if (Ctor->isTrivial()) {
96505fc5be3SDouglas Gregor       // If new expression did not specify value-initialization, then there
96605fc5be3SDouglas Gregor       // is no initialization.
9676047f07eSSebastian Redl       if (!CCE->requiresZeroInitialization() || Ctor->getParent()->isEmpty())
96805fc5be3SDouglas Gregor         return;
96905fc5be3SDouglas Gregor 
97006a67e2cSRichard Smith       if (TryMemsetInitialization())
9713a202f60SAnders Carlsson         return;
9723a202f60SAnders Carlsson     }
97305fc5be3SDouglas Gregor 
97406a67e2cSRichard Smith     // Store the new Cleanup position for irregular Cleanups.
97506a67e2cSRichard Smith     //
97606a67e2cSRichard Smith     // FIXME: Share this cleanup with the constructor call emission rather than
97706a67e2cSRichard Smith     // having it create a cleanup of its own.
9787f416cc4SJohn McCall     if (EndOfInit.isValid())
9797f416cc4SJohn McCall       Builder.CreateStore(CurPtr.getPointer(), EndOfInit);
98006a67e2cSRichard Smith 
98106a67e2cSRichard Smith     // Emit a constructor call loop to initialize the remaining elements.
98206a67e2cSRichard Smith     if (InitListElements)
98306a67e2cSRichard Smith       NumElements = Builder.CreateSub(
98406a67e2cSRichard Smith           NumElements,
98506a67e2cSRichard Smith           llvm::ConstantInt::get(NumElements->getType(), InitListElements));
98670b9c01bSAlexey Samsonov     EmitCXXAggrConstructorCall(Ctor, NumElements, CurPtr, CCE,
98748ddcf2cSEli Friedman                                CCE->requiresZeroInitialization());
98805fc5be3SDouglas Gregor     return;
9896047f07eSSebastian Redl   }
99006a67e2cSRichard Smith 
99106a67e2cSRichard Smith   // If this is value-initialization, we can usually use memset.
99206a67e2cSRichard Smith   ImplicitValueInitExpr IVIE(ElementType);
993454a7cdfSRichard Smith   if (isa<ImplicitValueInitExpr>(Init)) {
99406a67e2cSRichard Smith     if (TryMemsetInitialization())
99506a67e2cSRichard Smith       return;
99606a67e2cSRichard Smith 
99706a67e2cSRichard Smith     // Switch to an ImplicitValueInitExpr for the element type. This handles
99806a67e2cSRichard Smith     // only one case: multidimensional array new of pointers to members. In
99906a67e2cSRichard Smith     // all other cases, we already have an initializer for the array element.
100006a67e2cSRichard Smith     Init = &IVIE;
100106a67e2cSRichard Smith   }
100206a67e2cSRichard Smith 
100306a67e2cSRichard Smith   // At this point we should have found an initializer for the individual
100406a67e2cSRichard Smith   // elements of the array.
100506a67e2cSRichard Smith   assert(getContext().hasSameUnqualifiedType(ElementType, Init->getType()) &&
100606a67e2cSRichard Smith          "got wrong type of element to initialize");
100706a67e2cSRichard Smith 
1008454a7cdfSRichard Smith   // If we have an empty initializer list, we can usually use memset.
1009454a7cdfSRichard Smith   if (auto *ILE = dyn_cast<InitListExpr>(Init))
1010454a7cdfSRichard Smith     if (ILE->getNumInits() == 0 && TryMemsetInitialization())
1011d5202e09SFariborz Jahanian       return;
101259486a2dSAnders Carlsson 
1013cb77930dSYunzhong Gao   // If we have a struct whose every field is value-initialized, we can
1014cb77930dSYunzhong Gao   // usually use memset.
1015cb77930dSYunzhong Gao   if (auto *ILE = dyn_cast<InitListExpr>(Init)) {
1016cb77930dSYunzhong Gao     if (const RecordType *RType = ILE->getType()->getAs<RecordType>()) {
1017cb77930dSYunzhong Gao       if (RType->getDecl()->isStruct()) {
1018872307e2SRichard Smith         unsigned NumElements = 0;
1019872307e2SRichard Smith         if (auto *CXXRD = dyn_cast<CXXRecordDecl>(RType->getDecl()))
1020872307e2SRichard Smith           NumElements = CXXRD->getNumBases();
1021cb77930dSYunzhong Gao         for (auto *Field : RType->getDecl()->fields())
1022cb77930dSYunzhong Gao           if (!Field->isUnnamedBitfield())
1023872307e2SRichard Smith             ++NumElements;
1024872307e2SRichard Smith         // FIXME: Recurse into nested InitListExprs.
1025872307e2SRichard Smith         if (ILE->getNumInits() == NumElements)
1026cb77930dSYunzhong Gao           for (unsigned i = 0, e = ILE->getNumInits(); i != e; ++i)
1027cb77930dSYunzhong Gao             if (!isa<ImplicitValueInitExpr>(ILE->getInit(i)))
1028872307e2SRichard Smith               --NumElements;
1029872307e2SRichard Smith         if (ILE->getNumInits() == NumElements && TryMemsetInitialization())
1030cb77930dSYunzhong Gao           return;
1031cb77930dSYunzhong Gao       }
1032cb77930dSYunzhong Gao     }
1033cb77930dSYunzhong Gao   }
1034cb77930dSYunzhong Gao 
103506a67e2cSRichard Smith   // Create the loop blocks.
103606a67e2cSRichard Smith   llvm::BasicBlock *EntryBB = Builder.GetInsertBlock();
103706a67e2cSRichard Smith   llvm::BasicBlock *LoopBB = createBasicBlock("new.loop");
103806a67e2cSRichard Smith   llvm::BasicBlock *ContBB = createBasicBlock("new.loop.end");
103959486a2dSAnders Carlsson 
104006a67e2cSRichard Smith   // Find the end of the array, hoisted out of the loop.
104106a67e2cSRichard Smith   llvm::Value *EndPtr =
10427f416cc4SJohn McCall     Builder.CreateInBoundsGEP(BeginPtr.getPointer(), NumElements, "array.end");
104306a67e2cSRichard Smith 
104406a67e2cSRichard Smith   // If the number of elements isn't constant, we have to now check if there is
104506a67e2cSRichard Smith   // anything left to initialize.
104606a67e2cSRichard Smith   if (!ConstNum) {
10477f416cc4SJohn McCall     llvm::Value *IsEmpty =
10487f416cc4SJohn McCall       Builder.CreateICmpEQ(CurPtr.getPointer(), EndPtr, "array.isempty");
104906a67e2cSRichard Smith     Builder.CreateCondBr(IsEmpty, ContBB, LoopBB);
105006a67e2cSRichard Smith   }
105106a67e2cSRichard Smith 
105206a67e2cSRichard Smith   // Enter the loop.
105306a67e2cSRichard Smith   EmitBlock(LoopBB);
105406a67e2cSRichard Smith 
105506a67e2cSRichard Smith   // Set up the current-element phi.
105606a67e2cSRichard Smith   llvm::PHINode *CurPtrPhi =
10577f416cc4SJohn McCall     Builder.CreatePHI(CurPtr.getType(), 2, "array.cur");
10587f416cc4SJohn McCall   CurPtrPhi->addIncoming(CurPtr.getPointer(), EntryBB);
10597f416cc4SJohn McCall 
10607f416cc4SJohn McCall   CurPtr = Address(CurPtrPhi, ElementAlign);
106106a67e2cSRichard Smith 
106206a67e2cSRichard Smith   // Store the new Cleanup position for irregular Cleanups.
10637f416cc4SJohn McCall   if (EndOfInit.isValid())
10647f416cc4SJohn McCall     Builder.CreateStore(CurPtr.getPointer(), EndOfInit);
106506a67e2cSRichard Smith 
106606a67e2cSRichard Smith   // Enter a partial-destruction Cleanup if necessary.
106706a67e2cSRichard Smith   if (!CleanupDominator && needsEHCleanup(DtorKind)) {
10687f416cc4SJohn McCall     pushRegularPartialArrayCleanup(BeginPtr.getPointer(), CurPtr.getPointer(),
10697f416cc4SJohn McCall                                    ElementType, ElementAlign,
107006a67e2cSRichard Smith                                    getDestroyer(DtorKind));
107106a67e2cSRichard Smith     Cleanup = EHStack.stable_begin();
107206a67e2cSRichard Smith     CleanupDominator = Builder.CreateUnreachable();
107306a67e2cSRichard Smith   }
107406a67e2cSRichard Smith 
107506a67e2cSRichard Smith   // Emit the initializer into this element.
107606a67e2cSRichard Smith   StoreAnyExprIntoOneUnit(*this, Init, Init->getType(), CurPtr);
107706a67e2cSRichard Smith 
107806a67e2cSRichard Smith   // Leave the Cleanup if we entered one.
107906a67e2cSRichard Smith   if (CleanupDominator) {
108006a67e2cSRichard Smith     DeactivateCleanupBlock(Cleanup, CleanupDominator);
108106a67e2cSRichard Smith     CleanupDominator->eraseFromParent();
108206a67e2cSRichard Smith   }
108306a67e2cSRichard Smith 
108406a67e2cSRichard Smith   // Advance to the next element by adjusting the pointer type as necessary.
108506a67e2cSRichard Smith   llvm::Value *NextPtr =
10867f416cc4SJohn McCall     Builder.CreateConstInBoundsGEP1_32(ElementTy, CurPtr.getPointer(), 1,
10877f416cc4SJohn McCall                                        "array.next");
108806a67e2cSRichard Smith 
108906a67e2cSRichard Smith   // Check whether we've gotten to the end of the array and, if so,
109006a67e2cSRichard Smith   // exit the loop.
109106a67e2cSRichard Smith   llvm::Value *IsEnd = Builder.CreateICmpEQ(NextPtr, EndPtr, "array.atend");
109206a67e2cSRichard Smith   Builder.CreateCondBr(IsEnd, ContBB, LoopBB);
109306a67e2cSRichard Smith   CurPtrPhi->addIncoming(NextPtr, Builder.GetInsertBlock());
109406a67e2cSRichard Smith 
109506a67e2cSRichard Smith   EmitBlock(ContBB);
109606a67e2cSRichard Smith }
109706a67e2cSRichard Smith 
109806a67e2cSRichard Smith static void EmitNewInitializer(CodeGenFunction &CGF, const CXXNewExpr *E,
1099fb901c7aSDavid Blaikie                                QualType ElementType, llvm::Type *ElementTy,
11007f416cc4SJohn McCall                                Address NewPtr, llvm::Value *NumElements,
110106a67e2cSRichard Smith                                llvm::Value *AllocSizeWithoutCookie) {
11029b479666SDavid Blaikie   ApplyDebugLocation DL(CGF, E);
110306a67e2cSRichard Smith   if (E->isArray())
1104fb901c7aSDavid Blaikie     CGF.EmitNewArrayInitializer(E, ElementType, ElementTy, NewPtr, NumElements,
110506a67e2cSRichard Smith                                 AllocSizeWithoutCookie);
110606a67e2cSRichard Smith   else if (const Expr *Init = E->getInitializer())
110766e4197fSDavid Blaikie     StoreAnyExprIntoOneUnit(CGF, Init, E->getAllocatedType(), NewPtr);
110859486a2dSAnders Carlsson }
110959486a2dSAnders Carlsson 
11108d0dc31dSRichard Smith /// Emit a call to an operator new or operator delete function, as implicitly
11118d0dc31dSRichard Smith /// created by new-expressions and delete-expressions.
11128d0dc31dSRichard Smith static RValue EmitNewDeleteCall(CodeGenFunction &CGF,
11138d0dc31dSRichard Smith                                 const FunctionDecl *Callee,
11148d0dc31dSRichard Smith                                 const FunctionProtoType *CalleeType,
11158d0dc31dSRichard Smith                                 const CallArgList &Args) {
11168d0dc31dSRichard Smith   llvm::Instruction *CallOrInvoke;
11171235a8daSRichard Smith   llvm::Value *CalleeAddr = CGF.CGM.GetAddrOfFunction(Callee);
11188d0dc31dSRichard Smith   RValue RV =
1119f770683fSPeter Collingbourne       CGF.EmitCall(CGF.CGM.getTypes().arrangeFreeFunctionCall(
1120f770683fSPeter Collingbourne                        Args, CalleeType, /*chainCall=*/false),
1121f770683fSPeter Collingbourne                    CalleeAddr, ReturnValueSlot(), Args, Callee, &CallOrInvoke);
11228d0dc31dSRichard Smith 
11238d0dc31dSRichard Smith   /// C++1y [expr.new]p10:
11248d0dc31dSRichard Smith   ///   [In a new-expression,] an implementation is allowed to omit a call
11258d0dc31dSRichard Smith   ///   to a replaceable global allocation function.
11268d0dc31dSRichard Smith   ///
11278d0dc31dSRichard Smith   /// We model such elidable calls with the 'builtin' attribute.
11286956d587SRafael Espindola   llvm::Function *Fn = dyn_cast<llvm::Function>(CalleeAddr);
11291235a8daSRichard Smith   if (Callee->isReplaceableGlobalAllocationFunction() &&
11306956d587SRafael Espindola       Fn && Fn->hasFnAttribute(llvm::Attribute::NoBuiltin)) {
11318d0dc31dSRichard Smith     // FIXME: Add addAttribute to CallSite.
11328d0dc31dSRichard Smith     if (llvm::CallInst *CI = dyn_cast<llvm::CallInst>(CallOrInvoke))
11338d0dc31dSRichard Smith       CI->addAttribute(llvm::AttributeSet::FunctionIndex,
11348d0dc31dSRichard Smith                        llvm::Attribute::Builtin);
11358d0dc31dSRichard Smith     else if (llvm::InvokeInst *II = dyn_cast<llvm::InvokeInst>(CallOrInvoke))
11368d0dc31dSRichard Smith       II->addAttribute(llvm::AttributeSet::FunctionIndex,
11378d0dc31dSRichard Smith                        llvm::Attribute::Builtin);
11388d0dc31dSRichard Smith     else
11398d0dc31dSRichard Smith       llvm_unreachable("unexpected kind of call instruction");
11408d0dc31dSRichard Smith   }
11418d0dc31dSRichard Smith 
11428d0dc31dSRichard Smith   return RV;
11438d0dc31dSRichard Smith }
11448d0dc31dSRichard Smith 
1145760520bcSRichard Smith RValue CodeGenFunction::EmitBuiltinNewDeleteCall(const FunctionProtoType *Type,
1146760520bcSRichard Smith                                                  const Expr *Arg,
1147760520bcSRichard Smith                                                  bool IsDelete) {
1148760520bcSRichard Smith   CallArgList Args;
1149760520bcSRichard Smith   const Stmt *ArgS = Arg;
1150f05779e2SDavid Blaikie   EmitCallArgs(Args, *Type->param_type_begin(), llvm::makeArrayRef(ArgS));
1151760520bcSRichard Smith   // Find the allocation or deallocation function that we're calling.
1152760520bcSRichard Smith   ASTContext &Ctx = getContext();
1153760520bcSRichard Smith   DeclarationName Name = Ctx.DeclarationNames
1154760520bcSRichard Smith       .getCXXOperatorName(IsDelete ? OO_Delete : OO_New);
1155760520bcSRichard Smith   for (auto *Decl : Ctx.getTranslationUnitDecl()->lookup(Name))
1156599bed75SRichard Smith     if (auto *FD = dyn_cast<FunctionDecl>(Decl))
1157599bed75SRichard Smith       if (Ctx.hasSameType(FD->getType(), QualType(Type, 0)))
1158760520bcSRichard Smith         return EmitNewDeleteCall(*this, cast<FunctionDecl>(Decl), Type, Args);
1159760520bcSRichard Smith   llvm_unreachable("predeclared global operator new/delete is missing");
1160760520bcSRichard Smith }
1161760520bcSRichard Smith 
1162824c2f53SJohn McCall namespace {
1163824c2f53SJohn McCall   /// A cleanup to call the given 'operator delete' function upon
1164824c2f53SJohn McCall   /// abnormal exit from a new expression.
11657e70d680SDavid Blaikie   class CallDeleteDuringNew final : public EHScopeStack::Cleanup {
1166824c2f53SJohn McCall     size_t NumPlacementArgs;
1167824c2f53SJohn McCall     const FunctionDecl *OperatorDelete;
1168824c2f53SJohn McCall     llvm::Value *Ptr;
1169824c2f53SJohn McCall     llvm::Value *AllocSize;
1170824c2f53SJohn McCall 
1171824c2f53SJohn McCall     RValue *getPlacementArgs() { return reinterpret_cast<RValue*>(this+1); }
1172824c2f53SJohn McCall 
1173824c2f53SJohn McCall   public:
1174824c2f53SJohn McCall     static size_t getExtraSize(size_t NumPlacementArgs) {
1175824c2f53SJohn McCall       return NumPlacementArgs * sizeof(RValue);
1176824c2f53SJohn McCall     }
1177824c2f53SJohn McCall 
1178824c2f53SJohn McCall     CallDeleteDuringNew(size_t NumPlacementArgs,
1179824c2f53SJohn McCall                         const FunctionDecl *OperatorDelete,
1180824c2f53SJohn McCall                         llvm::Value *Ptr,
1181824c2f53SJohn McCall                         llvm::Value *AllocSize)
1182824c2f53SJohn McCall       : NumPlacementArgs(NumPlacementArgs), OperatorDelete(OperatorDelete),
1183824c2f53SJohn McCall         Ptr(Ptr), AllocSize(AllocSize) {}
1184824c2f53SJohn McCall 
1185824c2f53SJohn McCall     void setPlacementArg(unsigned I, RValue Arg) {
1186824c2f53SJohn McCall       assert(I < NumPlacementArgs && "index out of range");
1187824c2f53SJohn McCall       getPlacementArgs()[I] = Arg;
1188824c2f53SJohn McCall     }
1189824c2f53SJohn McCall 
11904f12f10dSCraig Topper     void Emit(CodeGenFunction &CGF, Flags flags) override {
1191824c2f53SJohn McCall       const FunctionProtoType *FPT
1192824c2f53SJohn McCall         = OperatorDelete->getType()->getAs<FunctionProtoType>();
11939cacbabdSAlp Toker       assert(FPT->getNumParams() == NumPlacementArgs + 1 ||
11949cacbabdSAlp Toker              (FPT->getNumParams() == 2 && NumPlacementArgs == 0));
1195824c2f53SJohn McCall 
1196824c2f53SJohn McCall       CallArgList DeleteArgs;
1197824c2f53SJohn McCall 
1198824c2f53SJohn McCall       // The first argument is always a void*.
11999cacbabdSAlp Toker       FunctionProtoType::param_type_iterator AI = FPT->param_type_begin();
120043dca6a8SEli Friedman       DeleteArgs.add(RValue::get(Ptr), *AI++);
1201824c2f53SJohn McCall 
1202824c2f53SJohn McCall       // A member 'operator delete' can take an extra 'size_t' argument.
12039cacbabdSAlp Toker       if (FPT->getNumParams() == NumPlacementArgs + 2)
120443dca6a8SEli Friedman         DeleteArgs.add(RValue::get(AllocSize), *AI++);
1205824c2f53SJohn McCall 
1206824c2f53SJohn McCall       // Pass the rest of the arguments, which must match exactly.
1207824c2f53SJohn McCall       for (unsigned I = 0; I != NumPlacementArgs; ++I)
120843dca6a8SEli Friedman         DeleteArgs.add(getPlacementArgs()[I], *AI++);
1209824c2f53SJohn McCall 
1210824c2f53SJohn McCall       // Call 'operator delete'.
12118d0dc31dSRichard Smith       EmitNewDeleteCall(CGF, OperatorDelete, FPT, DeleteArgs);
1212824c2f53SJohn McCall     }
1213824c2f53SJohn McCall   };
12147f9c92a9SJohn McCall 
12157f9c92a9SJohn McCall   /// A cleanup to call the given 'operator delete' function upon
12167f9c92a9SJohn McCall   /// abnormal exit from a new expression when the new expression is
12177f9c92a9SJohn McCall   /// conditional.
12187e70d680SDavid Blaikie   class CallDeleteDuringConditionalNew final : public EHScopeStack::Cleanup {
12197f9c92a9SJohn McCall     size_t NumPlacementArgs;
12207f9c92a9SJohn McCall     const FunctionDecl *OperatorDelete;
1221cb5f77f0SJohn McCall     DominatingValue<RValue>::saved_type Ptr;
1222cb5f77f0SJohn McCall     DominatingValue<RValue>::saved_type AllocSize;
12237f9c92a9SJohn McCall 
1224cb5f77f0SJohn McCall     DominatingValue<RValue>::saved_type *getPlacementArgs() {
1225cb5f77f0SJohn McCall       return reinterpret_cast<DominatingValue<RValue>::saved_type*>(this+1);
12267f9c92a9SJohn McCall     }
12277f9c92a9SJohn McCall 
12287f9c92a9SJohn McCall   public:
12297f9c92a9SJohn McCall     static size_t getExtraSize(size_t NumPlacementArgs) {
1230cb5f77f0SJohn McCall       return NumPlacementArgs * sizeof(DominatingValue<RValue>::saved_type);
12317f9c92a9SJohn McCall     }
12327f9c92a9SJohn McCall 
12337f9c92a9SJohn McCall     CallDeleteDuringConditionalNew(size_t NumPlacementArgs,
12347f9c92a9SJohn McCall                                    const FunctionDecl *OperatorDelete,
1235cb5f77f0SJohn McCall                                    DominatingValue<RValue>::saved_type Ptr,
1236cb5f77f0SJohn McCall                               DominatingValue<RValue>::saved_type AllocSize)
12377f9c92a9SJohn McCall       : NumPlacementArgs(NumPlacementArgs), OperatorDelete(OperatorDelete),
12387f9c92a9SJohn McCall         Ptr(Ptr), AllocSize(AllocSize) {}
12397f9c92a9SJohn McCall 
1240cb5f77f0SJohn McCall     void setPlacementArg(unsigned I, DominatingValue<RValue>::saved_type Arg) {
12417f9c92a9SJohn McCall       assert(I < NumPlacementArgs && "index out of range");
12427f9c92a9SJohn McCall       getPlacementArgs()[I] = Arg;
12437f9c92a9SJohn McCall     }
12447f9c92a9SJohn McCall 
12454f12f10dSCraig Topper     void Emit(CodeGenFunction &CGF, Flags flags) override {
12467f9c92a9SJohn McCall       const FunctionProtoType *FPT
12477f9c92a9SJohn McCall         = OperatorDelete->getType()->getAs<FunctionProtoType>();
12489cacbabdSAlp Toker       assert(FPT->getNumParams() == NumPlacementArgs + 1 ||
12499cacbabdSAlp Toker              (FPT->getNumParams() == 2 && NumPlacementArgs == 0));
12507f9c92a9SJohn McCall 
12517f9c92a9SJohn McCall       CallArgList DeleteArgs;
12527f9c92a9SJohn McCall 
12537f9c92a9SJohn McCall       // The first argument is always a void*.
12549cacbabdSAlp Toker       FunctionProtoType::param_type_iterator AI = FPT->param_type_begin();
125543dca6a8SEli Friedman       DeleteArgs.add(Ptr.restore(CGF), *AI++);
12567f9c92a9SJohn McCall 
12577f9c92a9SJohn McCall       // A member 'operator delete' can take an extra 'size_t' argument.
12589cacbabdSAlp Toker       if (FPT->getNumParams() == NumPlacementArgs + 2) {
1259cb5f77f0SJohn McCall         RValue RV = AllocSize.restore(CGF);
126043dca6a8SEli Friedman         DeleteArgs.add(RV, *AI++);
12617f9c92a9SJohn McCall       }
12627f9c92a9SJohn McCall 
12637f9c92a9SJohn McCall       // Pass the rest of the arguments, which must match exactly.
12647f9c92a9SJohn McCall       for (unsigned I = 0; I != NumPlacementArgs; ++I) {
1265cb5f77f0SJohn McCall         RValue RV = getPlacementArgs()[I].restore(CGF);
126643dca6a8SEli Friedman         DeleteArgs.add(RV, *AI++);
12677f9c92a9SJohn McCall       }
12687f9c92a9SJohn McCall 
12697f9c92a9SJohn McCall       // Call 'operator delete'.
12708d0dc31dSRichard Smith       EmitNewDeleteCall(CGF, OperatorDelete, FPT, DeleteArgs);
12717f9c92a9SJohn McCall     }
12727f9c92a9SJohn McCall   };
1273ab9db510SAlexander Kornienko }
12747f9c92a9SJohn McCall 
12757f9c92a9SJohn McCall /// Enter a cleanup to call 'operator delete' if the initializer in a
12767f9c92a9SJohn McCall /// new-expression throws.
12777f9c92a9SJohn McCall static void EnterNewDeleteCleanup(CodeGenFunction &CGF,
12787f9c92a9SJohn McCall                                   const CXXNewExpr *E,
12797f416cc4SJohn McCall                                   Address NewPtr,
12807f9c92a9SJohn McCall                                   llvm::Value *AllocSize,
12817f9c92a9SJohn McCall                                   const CallArgList &NewArgs) {
12827f9c92a9SJohn McCall   // If we're not inside a conditional branch, then the cleanup will
12837f9c92a9SJohn McCall   // dominate and we can do the easier (and more efficient) thing.
12847f9c92a9SJohn McCall   if (!CGF.isInConditionalBranch()) {
12857f9c92a9SJohn McCall     CallDeleteDuringNew *Cleanup = CGF.EHStack
12867f9c92a9SJohn McCall       .pushCleanupWithExtra<CallDeleteDuringNew>(EHCleanup,
12877f9c92a9SJohn McCall                                                  E->getNumPlacementArgs(),
12887f9c92a9SJohn McCall                                                  E->getOperatorDelete(),
12897f416cc4SJohn McCall                                                  NewPtr.getPointer(),
12907f416cc4SJohn McCall                                                  AllocSize);
12917f9c92a9SJohn McCall     for (unsigned I = 0, N = E->getNumPlacementArgs(); I != N; ++I)
1292f4258eb4SEli Friedman       Cleanup->setPlacementArg(I, NewArgs[I+1].RV);
12937f9c92a9SJohn McCall 
12947f9c92a9SJohn McCall     return;
12957f9c92a9SJohn McCall   }
12967f9c92a9SJohn McCall 
12977f9c92a9SJohn McCall   // Otherwise, we need to save all this stuff.
1298cb5f77f0SJohn McCall   DominatingValue<RValue>::saved_type SavedNewPtr =
12997f416cc4SJohn McCall     DominatingValue<RValue>::save(CGF, RValue::get(NewPtr.getPointer()));
1300cb5f77f0SJohn McCall   DominatingValue<RValue>::saved_type SavedAllocSize =
1301cb5f77f0SJohn McCall     DominatingValue<RValue>::save(CGF, RValue::get(AllocSize));
13027f9c92a9SJohn McCall 
13037f9c92a9SJohn McCall   CallDeleteDuringConditionalNew *Cleanup = CGF.EHStack
1304f4beacd0SJohn McCall     .pushCleanupWithExtra<CallDeleteDuringConditionalNew>(EHCleanup,
13057f9c92a9SJohn McCall                                                  E->getNumPlacementArgs(),
13067f9c92a9SJohn McCall                                                  E->getOperatorDelete(),
13077f9c92a9SJohn McCall                                                  SavedNewPtr,
13087f9c92a9SJohn McCall                                                  SavedAllocSize);
13097f9c92a9SJohn McCall   for (unsigned I = 0, N = E->getNumPlacementArgs(); I != N; ++I)
1310cb5f77f0SJohn McCall     Cleanup->setPlacementArg(I,
1311f4258eb4SEli Friedman                      DominatingValue<RValue>::save(CGF, NewArgs[I+1].RV));
13127f9c92a9SJohn McCall 
1313f4beacd0SJohn McCall   CGF.initFullExprCleanup();
1314824c2f53SJohn McCall }
1315824c2f53SJohn McCall 
131659486a2dSAnders Carlsson llvm::Value *CodeGenFunction::EmitCXXNewExpr(const CXXNewExpr *E) {
131775f9498aSJohn McCall   // The element type being allocated.
131875f9498aSJohn McCall   QualType allocType = getContext().getBaseElementType(E->getAllocatedType());
13198ed55a54SJohn McCall 
132075f9498aSJohn McCall   // 1. Build a call to the allocation function.
132175f9498aSJohn McCall   FunctionDecl *allocator = E->getOperatorNew();
132259486a2dSAnders Carlsson 
1323f862eb6aSSebastian Redl   // If there is a brace-initializer, cannot allocate fewer elements than inits.
1324f862eb6aSSebastian Redl   unsigned minElements = 0;
1325f862eb6aSSebastian Redl   if (E->isArray() && E->hasInitializer()) {
1326f862eb6aSSebastian Redl     if (const InitListExpr *ILE = dyn_cast<InitListExpr>(E->getInitializer()))
1327f862eb6aSSebastian Redl       minElements = ILE->getNumInits();
1328f862eb6aSSebastian Redl   }
1329f862eb6aSSebastian Redl 
13308a13c418SCraig Topper   llvm::Value *numElements = nullptr;
13318a13c418SCraig Topper   llvm::Value *allocSizeWithoutCookie = nullptr;
133275f9498aSJohn McCall   llvm::Value *allocSize =
1333f862eb6aSSebastian Redl     EmitCXXNewAllocSize(*this, E, minElements, numElements,
1334f862eb6aSSebastian Redl                         allocSizeWithoutCookie);
133559486a2dSAnders Carlsson 
13367f416cc4SJohn McCall   // Emit the allocation call.  If the allocator is a global placement
13377f416cc4SJohn McCall   // operator, just "inline" it directly.
13387f416cc4SJohn McCall   Address allocation = Address::invalid();
13397f416cc4SJohn McCall   CallArgList allocatorArgs;
13407f416cc4SJohn McCall   if (allocator->isReservedGlobalPlacementOperator()) {
134153dcf94dSJohn McCall     assert(E->getNumPlacementArgs() == 1);
134253dcf94dSJohn McCall     const Expr *arg = *E->placement_arguments().begin();
134353dcf94dSJohn McCall 
13447f416cc4SJohn McCall     AlignmentSource alignSource;
134553dcf94dSJohn McCall     allocation = EmitPointerWithAlignment(arg, &alignSource);
13467f416cc4SJohn McCall 
13477f416cc4SJohn McCall     // The pointer expression will, in many cases, be an opaque void*.
13487f416cc4SJohn McCall     // In these cases, discard the computed alignment and use the
13497f416cc4SJohn McCall     // formal alignment of the allocated type.
13507f416cc4SJohn McCall     if (alignSource != AlignmentSource::Decl) {
13517f416cc4SJohn McCall       allocation = Address(allocation.getPointer(),
13527f416cc4SJohn McCall                            getContext().getTypeAlignInChars(allocType));
13537f416cc4SJohn McCall     }
13547f416cc4SJohn McCall 
135553dcf94dSJohn McCall     // Set up allocatorArgs for the call to operator delete if it's not
135653dcf94dSJohn McCall     // the reserved global operator.
135753dcf94dSJohn McCall     if (E->getOperatorDelete() &&
135853dcf94dSJohn McCall         !E->getOperatorDelete()->isReservedGlobalPlacementOperator()) {
135953dcf94dSJohn McCall       allocatorArgs.add(RValue::get(allocSize), getContext().getSizeType());
136053dcf94dSJohn McCall       allocatorArgs.add(RValue::get(allocation.getPointer()), arg->getType());
136153dcf94dSJohn McCall     }
136253dcf94dSJohn McCall 
13637f416cc4SJohn McCall   } else {
13647f416cc4SJohn McCall     const FunctionProtoType *allocatorType =
13657f416cc4SJohn McCall       allocator->getType()->castAs<FunctionProtoType>();
13667f416cc4SJohn McCall 
13677f416cc4SJohn McCall     // The allocation size is the first argument.
13687f416cc4SJohn McCall     QualType sizeType = getContext().getSizeType();
136943dca6a8SEli Friedman     allocatorArgs.add(RValue::get(allocSize), sizeType);
137059486a2dSAnders Carlsson 
137159486a2dSAnders Carlsson     // We start at 1 here because the first argument (the allocation size)
137259486a2dSAnders Carlsson     // has already been emitted.
1373f05779e2SDavid Blaikie     EmitCallArgs(allocatorArgs, allocatorType, E->placement_arguments(),
1374f05779e2SDavid Blaikie                  /* CalleeDecl */ nullptr,
13758e1162c7SAlexey Samsonov                  /*ParamsToSkip*/ 1);
137659486a2dSAnders Carlsson 
13777f416cc4SJohn McCall     RValue RV =
13787f416cc4SJohn McCall       EmitNewDeleteCall(*this, allocator, allocatorType, allocatorArgs);
13797f416cc4SJohn McCall 
13807f416cc4SJohn McCall     // For now, only assume that the allocation function returns
13817f416cc4SJohn McCall     // something satisfactorily aligned for the element type, plus
13827f416cc4SJohn McCall     // the cookie if we have one.
13837f416cc4SJohn McCall     CharUnits allocationAlign =
13847f416cc4SJohn McCall       getContext().getTypeAlignInChars(allocType);
13857f416cc4SJohn McCall     if (allocSize != allocSizeWithoutCookie) {
13867f416cc4SJohn McCall       CharUnits cookieAlign = getSizeAlign(); // FIXME?
13877f416cc4SJohn McCall       allocationAlign = std::max(allocationAlign, cookieAlign);
13887f416cc4SJohn McCall     }
13897f416cc4SJohn McCall 
13907f416cc4SJohn McCall     allocation = Address(RV.getScalarVal(), allocationAlign);
13917ec4b434SJohn McCall   }
139259486a2dSAnders Carlsson 
139375f9498aSJohn McCall   // Emit a null check on the allocation result if the allocation
139475f9498aSJohn McCall   // function is allowed to return null (because it has a non-throwing
1395902a0238SRichard Smith   // exception spec or is the reserved placement new) and we have an
139675f9498aSJohn McCall   // interesting initializer.
1397902a0238SRichard Smith   bool nullCheck = E->shouldNullCheckAllocation(getContext()) &&
13986047f07eSSebastian Redl     (!allocType.isPODType(getContext()) || E->hasInitializer());
139959486a2dSAnders Carlsson 
14008a13c418SCraig Topper   llvm::BasicBlock *nullCheckBB = nullptr;
14018a13c418SCraig Topper   llvm::BasicBlock *contBB = nullptr;
140259486a2dSAnders Carlsson 
1403f7dcf320SJohn McCall   // The null-check means that the initializer is conditionally
1404f7dcf320SJohn McCall   // evaluated.
1405f7dcf320SJohn McCall   ConditionalEvaluation conditional(*this);
1406f7dcf320SJohn McCall 
140775f9498aSJohn McCall   if (nullCheck) {
1408f7dcf320SJohn McCall     conditional.begin(*this);
140975f9498aSJohn McCall 
141075f9498aSJohn McCall     nullCheckBB = Builder.GetInsertBlock();
141175f9498aSJohn McCall     llvm::BasicBlock *notNullBB = createBasicBlock("new.notnull");
141275f9498aSJohn McCall     contBB = createBasicBlock("new.cont");
141375f9498aSJohn McCall 
14147f416cc4SJohn McCall     llvm::Value *isNull =
14157f416cc4SJohn McCall       Builder.CreateIsNull(allocation.getPointer(), "new.isnull");
141675f9498aSJohn McCall     Builder.CreateCondBr(isNull, contBB, notNullBB);
141775f9498aSJohn McCall     EmitBlock(notNullBB);
141859486a2dSAnders Carlsson   }
141959486a2dSAnders Carlsson 
1420824c2f53SJohn McCall   // If there's an operator delete, enter a cleanup to call it if an
1421824c2f53SJohn McCall   // exception is thrown.
142275f9498aSJohn McCall   EHScopeStack::stable_iterator operatorDeleteCleanup;
14238a13c418SCraig Topper   llvm::Instruction *cleanupDominator = nullptr;
14247ec4b434SJohn McCall   if (E->getOperatorDelete() &&
14257ec4b434SJohn McCall       !E->getOperatorDelete()->isReservedGlobalPlacementOperator()) {
142675f9498aSJohn McCall     EnterNewDeleteCleanup(*this, E, allocation, allocSize, allocatorArgs);
142775f9498aSJohn McCall     operatorDeleteCleanup = EHStack.stable_begin();
1428f4beacd0SJohn McCall     cleanupDominator = Builder.CreateUnreachable();
1429824c2f53SJohn McCall   }
1430824c2f53SJohn McCall 
1431cf9b1f65SEli Friedman   assert((allocSize == allocSizeWithoutCookie) ==
1432cf9b1f65SEli Friedman          CalculateCookiePadding(*this, E).isZero());
1433cf9b1f65SEli Friedman   if (allocSize != allocSizeWithoutCookie) {
1434cf9b1f65SEli Friedman     assert(E->isArray());
1435cf9b1f65SEli Friedman     allocation = CGM.getCXXABI().InitializeArrayCookie(*this, allocation,
1436cf9b1f65SEli Friedman                                                        numElements,
1437cf9b1f65SEli Friedman                                                        E, allocType);
1438cf9b1f65SEli Friedman   }
1439cf9b1f65SEli Friedman 
1440fb901c7aSDavid Blaikie   llvm::Type *elementTy = ConvertTypeForMem(allocType);
14417f416cc4SJohn McCall   Address result = Builder.CreateElementBitCast(allocation, elementTy);
1442824c2f53SJohn McCall 
1443338c9d0aSPiotr Padlewski   // Passing pointer through invariant.group.barrier to avoid propagation of
1444338c9d0aSPiotr Padlewski   // vptrs information which may be included in previous type.
1445338c9d0aSPiotr Padlewski   if (CGM.getCodeGenOpts().StrictVTablePointers &&
1446338c9d0aSPiotr Padlewski       CGM.getCodeGenOpts().OptimizationLevel > 0 &&
1447338c9d0aSPiotr Padlewski       allocator->isReservedGlobalPlacementOperator())
1448338c9d0aSPiotr Padlewski     result = Address(Builder.CreateInvariantGroupBarrier(result.getPointer()),
1449338c9d0aSPiotr Padlewski                      result.getAlignment());
1450338c9d0aSPiotr Padlewski 
1451fb901c7aSDavid Blaikie   EmitNewInitializer(*this, E, allocType, elementTy, result, numElements,
145299210dc9SJohn McCall                      allocSizeWithoutCookie);
14538ed55a54SJohn McCall   if (E->isArray()) {
14548ed55a54SJohn McCall     // NewPtr is a pointer to the base element type.  If we're
14558ed55a54SJohn McCall     // allocating an array of arrays, we'll need to cast back to the
14568ed55a54SJohn McCall     // array pointer type.
14572192fe50SChris Lattner     llvm::Type *resultType = ConvertTypeForMem(E->getType());
14587f416cc4SJohn McCall     if (result.getType() != resultType)
145975f9498aSJohn McCall       result = Builder.CreateBitCast(result, resultType);
146047b4629bSFariborz Jahanian   }
146159486a2dSAnders Carlsson 
1462824c2f53SJohn McCall   // Deactivate the 'operator delete' cleanup if we finished
1463824c2f53SJohn McCall   // initialization.
1464f4beacd0SJohn McCall   if (operatorDeleteCleanup.isValid()) {
1465f4beacd0SJohn McCall     DeactivateCleanupBlock(operatorDeleteCleanup, cleanupDominator);
1466f4beacd0SJohn McCall     cleanupDominator->eraseFromParent();
1467f4beacd0SJohn McCall   }
1468824c2f53SJohn McCall 
14697f416cc4SJohn McCall   llvm::Value *resultPtr = result.getPointer();
147075f9498aSJohn McCall   if (nullCheck) {
1471f7dcf320SJohn McCall     conditional.end(*this);
1472f7dcf320SJohn McCall 
147375f9498aSJohn McCall     llvm::BasicBlock *notNullBB = Builder.GetInsertBlock();
147475f9498aSJohn McCall     EmitBlock(contBB);
147559486a2dSAnders Carlsson 
14767f416cc4SJohn McCall     llvm::PHINode *PHI = Builder.CreatePHI(resultPtr->getType(), 2);
14777f416cc4SJohn McCall     PHI->addIncoming(resultPtr, notNullBB);
14787f416cc4SJohn McCall     PHI->addIncoming(llvm::Constant::getNullValue(resultPtr->getType()),
147975f9498aSJohn McCall                      nullCheckBB);
148059486a2dSAnders Carlsson 
14817f416cc4SJohn McCall     resultPtr = PHI;
148259486a2dSAnders Carlsson   }
148359486a2dSAnders Carlsson 
14847f416cc4SJohn McCall   return resultPtr;
148559486a2dSAnders Carlsson }
148659486a2dSAnders Carlsson 
148759486a2dSAnders Carlsson void CodeGenFunction::EmitDeleteCall(const FunctionDecl *DeleteFD,
148859486a2dSAnders Carlsson                                      llvm::Value *Ptr,
148959486a2dSAnders Carlsson                                      QualType DeleteTy) {
14908ed55a54SJohn McCall   assert(DeleteFD->getOverloadedOperator() == OO_Delete);
14918ed55a54SJohn McCall 
149259486a2dSAnders Carlsson   const FunctionProtoType *DeleteFTy =
149359486a2dSAnders Carlsson     DeleteFD->getType()->getAs<FunctionProtoType>();
149459486a2dSAnders Carlsson 
149559486a2dSAnders Carlsson   CallArgList DeleteArgs;
149659486a2dSAnders Carlsson 
149721122cf6SAnders Carlsson   // Check if we need to pass the size to the delete operator.
14988a13c418SCraig Topper   llvm::Value *Size = nullptr;
149921122cf6SAnders Carlsson   QualType SizeTy;
15009cacbabdSAlp Toker   if (DeleteFTy->getNumParams() == 2) {
15019cacbabdSAlp Toker     SizeTy = DeleteFTy->getParamType(1);
15027df3cbebSKen Dyck     CharUnits DeleteTypeSize = getContext().getTypeSizeInChars(DeleteTy);
15037df3cbebSKen Dyck     Size = llvm::ConstantInt::get(ConvertType(SizeTy),
15047df3cbebSKen Dyck                                   DeleteTypeSize.getQuantity());
150521122cf6SAnders Carlsson   }
150621122cf6SAnders Carlsson 
15079cacbabdSAlp Toker   QualType ArgTy = DeleteFTy->getParamType(0);
150859486a2dSAnders Carlsson   llvm::Value *DeletePtr = Builder.CreateBitCast(Ptr, ConvertType(ArgTy));
150943dca6a8SEli Friedman   DeleteArgs.add(RValue::get(DeletePtr), ArgTy);
151059486a2dSAnders Carlsson 
151121122cf6SAnders Carlsson   if (Size)
151243dca6a8SEli Friedman     DeleteArgs.add(RValue::get(Size), SizeTy);
151359486a2dSAnders Carlsson 
151459486a2dSAnders Carlsson   // Emit the call to delete.
15158d0dc31dSRichard Smith   EmitNewDeleteCall(*this, DeleteFD, DeleteFTy, DeleteArgs);
151659486a2dSAnders Carlsson }
151759486a2dSAnders Carlsson 
15188ed55a54SJohn McCall namespace {
15198ed55a54SJohn McCall   /// Calls the given 'operator delete' on a single object.
15207e70d680SDavid Blaikie   struct CallObjectDelete final : EHScopeStack::Cleanup {
15218ed55a54SJohn McCall     llvm::Value *Ptr;
15228ed55a54SJohn McCall     const FunctionDecl *OperatorDelete;
15238ed55a54SJohn McCall     QualType ElementType;
15248ed55a54SJohn McCall 
15258ed55a54SJohn McCall     CallObjectDelete(llvm::Value *Ptr,
15268ed55a54SJohn McCall                      const FunctionDecl *OperatorDelete,
15278ed55a54SJohn McCall                      QualType ElementType)
15288ed55a54SJohn McCall       : Ptr(Ptr), OperatorDelete(OperatorDelete), ElementType(ElementType) {}
15298ed55a54SJohn McCall 
15304f12f10dSCraig Topper     void Emit(CodeGenFunction &CGF, Flags flags) override {
15318ed55a54SJohn McCall       CGF.EmitDeleteCall(OperatorDelete, Ptr, ElementType);
15328ed55a54SJohn McCall     }
15338ed55a54SJohn McCall   };
1534ab9db510SAlexander Kornienko }
15358ed55a54SJohn McCall 
15360c0b6d9aSDavid Majnemer void
15370c0b6d9aSDavid Majnemer CodeGenFunction::pushCallObjectDeleteCleanup(const FunctionDecl *OperatorDelete,
15380c0b6d9aSDavid Majnemer                                              llvm::Value *CompletePtr,
15390c0b6d9aSDavid Majnemer                                              QualType ElementType) {
15400c0b6d9aSDavid Majnemer   EHStack.pushCleanup<CallObjectDelete>(NormalAndEHCleanup, CompletePtr,
15410c0b6d9aSDavid Majnemer                                         OperatorDelete, ElementType);
15420c0b6d9aSDavid Majnemer }
15430c0b6d9aSDavid Majnemer 
15448ed55a54SJohn McCall /// Emit the code for deleting a single object.
15458ed55a54SJohn McCall static void EmitObjectDelete(CodeGenFunction &CGF,
15460868137aSDavid Majnemer                              const CXXDeleteExpr *DE,
15477f416cc4SJohn McCall                              Address Ptr,
15480868137aSDavid Majnemer                              QualType ElementType) {
15498ed55a54SJohn McCall   // Find the destructor for the type, if applicable.  If the
15508ed55a54SJohn McCall   // destructor is virtual, we'll just emit the vcall and return.
15518a13c418SCraig Topper   const CXXDestructorDecl *Dtor = nullptr;
15528ed55a54SJohn McCall   if (const RecordType *RT = ElementType->getAs<RecordType>()) {
15538ed55a54SJohn McCall     CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl());
1554b23533dbSEli Friedman     if (RD->hasDefinition() && !RD->hasTrivialDestructor()) {
15558ed55a54SJohn McCall       Dtor = RD->getDestructor();
15568ed55a54SJohn McCall 
15578ed55a54SJohn McCall       if (Dtor->isVirtual()) {
15580868137aSDavid Majnemer         CGF.CGM.getCXXABI().emitVirtualObjectDelete(CGF, DE, Ptr, ElementType,
15590868137aSDavid Majnemer                                                     Dtor);
15608ed55a54SJohn McCall         return;
15618ed55a54SJohn McCall       }
15628ed55a54SJohn McCall     }
15638ed55a54SJohn McCall   }
15648ed55a54SJohn McCall 
15658ed55a54SJohn McCall   // Make sure that we call delete even if the dtor throws.
1566e4df6c8dSJohn McCall   // This doesn't have to a conditional cleanup because we're going
1567e4df6c8dSJohn McCall   // to pop it off in a second.
15680868137aSDavid Majnemer   const FunctionDecl *OperatorDelete = DE->getOperatorDelete();
15698ed55a54SJohn McCall   CGF.EHStack.pushCleanup<CallObjectDelete>(NormalAndEHCleanup,
15707f416cc4SJohn McCall                                             Ptr.getPointer(),
15717f416cc4SJohn McCall                                             OperatorDelete, ElementType);
15728ed55a54SJohn McCall 
15738ed55a54SJohn McCall   if (Dtor)
15748ed55a54SJohn McCall     CGF.EmitCXXDestructorCall(Dtor, Dtor_Complete,
157561535005SDouglas Gregor                               /*ForVirtualBase=*/false,
157661535005SDouglas Gregor                               /*Delegating=*/false,
157761535005SDouglas Gregor                               Ptr);
1578460ce58fSJohn McCall   else if (auto Lifetime = ElementType.getObjCLifetime()) {
1579460ce58fSJohn McCall     switch (Lifetime) {
158031168b07SJohn McCall     case Qualifiers::OCL_None:
158131168b07SJohn McCall     case Qualifiers::OCL_ExplicitNone:
158231168b07SJohn McCall     case Qualifiers::OCL_Autoreleasing:
158331168b07SJohn McCall       break;
158431168b07SJohn McCall 
15857f416cc4SJohn McCall     case Qualifiers::OCL_Strong:
15867f416cc4SJohn McCall       CGF.EmitARCDestroyStrong(Ptr, ARCPreciseLifetime);
158731168b07SJohn McCall       break;
158831168b07SJohn McCall 
158931168b07SJohn McCall     case Qualifiers::OCL_Weak:
159031168b07SJohn McCall       CGF.EmitARCDestroyWeak(Ptr);
159131168b07SJohn McCall       break;
159231168b07SJohn McCall     }
159331168b07SJohn McCall   }
15948ed55a54SJohn McCall 
15958ed55a54SJohn McCall   CGF.PopCleanupBlock();
15968ed55a54SJohn McCall }
15978ed55a54SJohn McCall 
15988ed55a54SJohn McCall namespace {
15998ed55a54SJohn McCall   /// Calls the given 'operator delete' on an array of objects.
16007e70d680SDavid Blaikie   struct CallArrayDelete final : EHScopeStack::Cleanup {
16018ed55a54SJohn McCall     llvm::Value *Ptr;
16028ed55a54SJohn McCall     const FunctionDecl *OperatorDelete;
16038ed55a54SJohn McCall     llvm::Value *NumElements;
16048ed55a54SJohn McCall     QualType ElementType;
16058ed55a54SJohn McCall     CharUnits CookieSize;
16068ed55a54SJohn McCall 
16078ed55a54SJohn McCall     CallArrayDelete(llvm::Value *Ptr,
16088ed55a54SJohn McCall                     const FunctionDecl *OperatorDelete,
16098ed55a54SJohn McCall                     llvm::Value *NumElements,
16108ed55a54SJohn McCall                     QualType ElementType,
16118ed55a54SJohn McCall                     CharUnits CookieSize)
16128ed55a54SJohn McCall       : Ptr(Ptr), OperatorDelete(OperatorDelete), NumElements(NumElements),
16138ed55a54SJohn McCall         ElementType(ElementType), CookieSize(CookieSize) {}
16148ed55a54SJohn McCall 
16154f12f10dSCraig Topper     void Emit(CodeGenFunction &CGF, Flags flags) override {
16168ed55a54SJohn McCall       const FunctionProtoType *DeleteFTy =
16178ed55a54SJohn McCall         OperatorDelete->getType()->getAs<FunctionProtoType>();
16189cacbabdSAlp Toker       assert(DeleteFTy->getNumParams() == 1 || DeleteFTy->getNumParams() == 2);
16198ed55a54SJohn McCall 
16208ed55a54SJohn McCall       CallArgList Args;
16218ed55a54SJohn McCall 
16228ed55a54SJohn McCall       // Pass the pointer as the first argument.
16239cacbabdSAlp Toker       QualType VoidPtrTy = DeleteFTy->getParamType(0);
16248ed55a54SJohn McCall       llvm::Value *DeletePtr
16258ed55a54SJohn McCall         = CGF.Builder.CreateBitCast(Ptr, CGF.ConvertType(VoidPtrTy));
162643dca6a8SEli Friedman       Args.add(RValue::get(DeletePtr), VoidPtrTy);
16278ed55a54SJohn McCall 
16288ed55a54SJohn McCall       // Pass the original requested size as the second argument.
16299cacbabdSAlp Toker       if (DeleteFTy->getNumParams() == 2) {
16309cacbabdSAlp Toker         QualType size_t = DeleteFTy->getParamType(1);
16312192fe50SChris Lattner         llvm::IntegerType *SizeTy
16328ed55a54SJohn McCall           = cast<llvm::IntegerType>(CGF.ConvertType(size_t));
16338ed55a54SJohn McCall 
16348ed55a54SJohn McCall         CharUnits ElementTypeSize =
16358ed55a54SJohn McCall           CGF.CGM.getContext().getTypeSizeInChars(ElementType);
16368ed55a54SJohn McCall 
16378ed55a54SJohn McCall         // The size of an element, multiplied by the number of elements.
16388ed55a54SJohn McCall         llvm::Value *Size
16398ed55a54SJohn McCall           = llvm::ConstantInt::get(SizeTy, ElementTypeSize.getQuantity());
1640149e6031SDavid Majnemer         if (NumElements)
16418ed55a54SJohn McCall           Size = CGF.Builder.CreateMul(Size, NumElements);
16428ed55a54SJohn McCall 
16438ed55a54SJohn McCall         // Plus the size of the cookie if applicable.
16448ed55a54SJohn McCall         if (!CookieSize.isZero()) {
16458ed55a54SJohn McCall           llvm::Value *CookieSizeV
16468ed55a54SJohn McCall             = llvm::ConstantInt::get(SizeTy, CookieSize.getQuantity());
16478ed55a54SJohn McCall           Size = CGF.Builder.CreateAdd(Size, CookieSizeV);
16488ed55a54SJohn McCall         }
16498ed55a54SJohn McCall 
165043dca6a8SEli Friedman         Args.add(RValue::get(Size), size_t);
16518ed55a54SJohn McCall       }
16528ed55a54SJohn McCall 
16538ed55a54SJohn McCall       // Emit the call to delete.
16548d0dc31dSRichard Smith       EmitNewDeleteCall(CGF, OperatorDelete, DeleteFTy, Args);
16558ed55a54SJohn McCall     }
16568ed55a54SJohn McCall   };
1657ab9db510SAlexander Kornienko }
16588ed55a54SJohn McCall 
16598ed55a54SJohn McCall /// Emit the code for deleting an array of objects.
16608ed55a54SJohn McCall static void EmitArrayDelete(CodeGenFunction &CGF,
1661284c48ffSJohn McCall                             const CXXDeleteExpr *E,
16627f416cc4SJohn McCall                             Address deletedPtr,
1663ca2c56f2SJohn McCall                             QualType elementType) {
16648a13c418SCraig Topper   llvm::Value *numElements = nullptr;
16658a13c418SCraig Topper   llvm::Value *allocatedPtr = nullptr;
1666ca2c56f2SJohn McCall   CharUnits cookieSize;
1667ca2c56f2SJohn McCall   CGF.CGM.getCXXABI().ReadArrayCookie(CGF, deletedPtr, E, elementType,
1668ca2c56f2SJohn McCall                                       numElements, allocatedPtr, cookieSize);
16698ed55a54SJohn McCall 
1670ca2c56f2SJohn McCall   assert(allocatedPtr && "ReadArrayCookie didn't set allocated pointer");
16718ed55a54SJohn McCall 
16728ed55a54SJohn McCall   // Make sure that we call delete even if one of the dtors throws.
1673ca2c56f2SJohn McCall   const FunctionDecl *operatorDelete = E->getOperatorDelete();
16748ed55a54SJohn McCall   CGF.EHStack.pushCleanup<CallArrayDelete>(NormalAndEHCleanup,
1675ca2c56f2SJohn McCall                                            allocatedPtr, operatorDelete,
1676ca2c56f2SJohn McCall                                            numElements, elementType,
1677ca2c56f2SJohn McCall                                            cookieSize);
16788ed55a54SJohn McCall 
1679ca2c56f2SJohn McCall   // Destroy the elements.
1680ca2c56f2SJohn McCall   if (QualType::DestructionKind dtorKind = elementType.isDestructedType()) {
1681ca2c56f2SJohn McCall     assert(numElements && "no element count for a type with a destructor!");
168231168b07SJohn McCall 
16837f416cc4SJohn McCall     CharUnits elementSize = CGF.getContext().getTypeSizeInChars(elementType);
16847f416cc4SJohn McCall     CharUnits elementAlign =
16857f416cc4SJohn McCall       deletedPtr.getAlignment().alignmentOfArrayElement(elementSize);
16867f416cc4SJohn McCall 
16877f416cc4SJohn McCall     llvm::Value *arrayBegin = deletedPtr.getPointer();
1688ca2c56f2SJohn McCall     llvm::Value *arrayEnd =
16897f416cc4SJohn McCall       CGF.Builder.CreateInBoundsGEP(arrayBegin, numElements, "delete.end");
169097eab0a2SJohn McCall 
169197eab0a2SJohn McCall     // Note that it is legal to allocate a zero-length array, and we
169297eab0a2SJohn McCall     // can never fold the check away because the length should always
169397eab0a2SJohn McCall     // come from a cookie.
16947f416cc4SJohn McCall     CGF.emitArrayDestroy(arrayBegin, arrayEnd, elementType, elementAlign,
1695ca2c56f2SJohn McCall                          CGF.getDestroyer(dtorKind),
169697eab0a2SJohn McCall                          /*checkZeroLength*/ true,
1697ca2c56f2SJohn McCall                          CGF.needsEHCleanup(dtorKind));
16988ed55a54SJohn McCall   }
16998ed55a54SJohn McCall 
1700ca2c56f2SJohn McCall   // Pop the cleanup block.
17018ed55a54SJohn McCall   CGF.PopCleanupBlock();
17028ed55a54SJohn McCall }
17038ed55a54SJohn McCall 
170459486a2dSAnders Carlsson void CodeGenFunction::EmitCXXDeleteExpr(const CXXDeleteExpr *E) {
170559486a2dSAnders Carlsson   const Expr *Arg = E->getArgument();
17067f416cc4SJohn McCall   Address Ptr = EmitPointerWithAlignment(Arg);
170759486a2dSAnders Carlsson 
170859486a2dSAnders Carlsson   // Null check the pointer.
170959486a2dSAnders Carlsson   llvm::BasicBlock *DeleteNotNull = createBasicBlock("delete.notnull");
171059486a2dSAnders Carlsson   llvm::BasicBlock *DeleteEnd = createBasicBlock("delete.end");
171159486a2dSAnders Carlsson 
17127f416cc4SJohn McCall   llvm::Value *IsNull = Builder.CreateIsNull(Ptr.getPointer(), "isnull");
171359486a2dSAnders Carlsson 
171459486a2dSAnders Carlsson   Builder.CreateCondBr(IsNull, DeleteEnd, DeleteNotNull);
171559486a2dSAnders Carlsson   EmitBlock(DeleteNotNull);
171659486a2dSAnders Carlsson 
17178ed55a54SJohn McCall   // We might be deleting a pointer to array.  If so, GEP down to the
17188ed55a54SJohn McCall   // first non-array element.
17198ed55a54SJohn McCall   // (this assumes that A(*)[3][7] is converted to [3 x [7 x %A]]*)
17208ed55a54SJohn McCall   QualType DeleteTy = Arg->getType()->getAs<PointerType>()->getPointeeType();
17218ed55a54SJohn McCall   if (DeleteTy->isConstantArrayType()) {
17228ed55a54SJohn McCall     llvm::Value *Zero = Builder.getInt32(0);
17230e62c1ccSChris Lattner     SmallVector<llvm::Value*,8> GEP;
172459486a2dSAnders Carlsson 
17258ed55a54SJohn McCall     GEP.push_back(Zero); // point at the outermost array
17268ed55a54SJohn McCall 
17278ed55a54SJohn McCall     // For each layer of array type we're pointing at:
17288ed55a54SJohn McCall     while (const ConstantArrayType *Arr
17298ed55a54SJohn McCall              = getContext().getAsConstantArrayType(DeleteTy)) {
17308ed55a54SJohn McCall       // 1. Unpeel the array type.
17318ed55a54SJohn McCall       DeleteTy = Arr->getElementType();
17328ed55a54SJohn McCall 
17338ed55a54SJohn McCall       // 2. GEP to the first element of the array.
17348ed55a54SJohn McCall       GEP.push_back(Zero);
17358ed55a54SJohn McCall     }
17368ed55a54SJohn McCall 
17377f416cc4SJohn McCall     Ptr = Address(Builder.CreateInBoundsGEP(Ptr.getPointer(), GEP, "del.first"),
17387f416cc4SJohn McCall                   Ptr.getAlignment());
17398ed55a54SJohn McCall   }
17408ed55a54SJohn McCall 
17417f416cc4SJohn McCall   assert(ConvertTypeForMem(DeleteTy) == Ptr.getElementType());
17428ed55a54SJohn McCall 
17437270ef57SReid Kleckner   if (E->isArrayForm()) {
17447270ef57SReid Kleckner     EmitArrayDelete(*this, E, Ptr, DeleteTy);
17457270ef57SReid Kleckner   } else {
17467270ef57SReid Kleckner     EmitObjectDelete(*this, E, Ptr, DeleteTy);
17477270ef57SReid Kleckner   }
174859486a2dSAnders Carlsson 
174959486a2dSAnders Carlsson   EmitBlock(DeleteEnd);
175059486a2dSAnders Carlsson }
175159486a2dSAnders Carlsson 
17521c3d95ebSDavid Majnemer static bool isGLValueFromPointerDeref(const Expr *E) {
17531c3d95ebSDavid Majnemer   E = E->IgnoreParens();
17541c3d95ebSDavid Majnemer 
17551c3d95ebSDavid Majnemer   if (const auto *CE = dyn_cast<CastExpr>(E)) {
17561c3d95ebSDavid Majnemer     if (!CE->getSubExpr()->isGLValue())
17571c3d95ebSDavid Majnemer       return false;
17581c3d95ebSDavid Majnemer     return isGLValueFromPointerDeref(CE->getSubExpr());
17591c3d95ebSDavid Majnemer   }
17601c3d95ebSDavid Majnemer 
17611c3d95ebSDavid Majnemer   if (const auto *OVE = dyn_cast<OpaqueValueExpr>(E))
17621c3d95ebSDavid Majnemer     return isGLValueFromPointerDeref(OVE->getSourceExpr());
17631c3d95ebSDavid Majnemer 
17641c3d95ebSDavid Majnemer   if (const auto *BO = dyn_cast<BinaryOperator>(E))
17651c3d95ebSDavid Majnemer     if (BO->getOpcode() == BO_Comma)
17661c3d95ebSDavid Majnemer       return isGLValueFromPointerDeref(BO->getRHS());
17671c3d95ebSDavid Majnemer 
17681c3d95ebSDavid Majnemer   if (const auto *ACO = dyn_cast<AbstractConditionalOperator>(E))
17691c3d95ebSDavid Majnemer     return isGLValueFromPointerDeref(ACO->getTrueExpr()) ||
17701c3d95ebSDavid Majnemer            isGLValueFromPointerDeref(ACO->getFalseExpr());
17711c3d95ebSDavid Majnemer 
17721c3d95ebSDavid Majnemer   // C++11 [expr.sub]p1:
17731c3d95ebSDavid Majnemer   //   The expression E1[E2] is identical (by definition) to *((E1)+(E2))
17741c3d95ebSDavid Majnemer   if (isa<ArraySubscriptExpr>(E))
17751c3d95ebSDavid Majnemer     return true;
17761c3d95ebSDavid Majnemer 
17771c3d95ebSDavid Majnemer   if (const auto *UO = dyn_cast<UnaryOperator>(E))
17781c3d95ebSDavid Majnemer     if (UO->getOpcode() == UO_Deref)
17791c3d95ebSDavid Majnemer       return true;
17801c3d95ebSDavid Majnemer 
17811c3d95ebSDavid Majnemer   return false;
17821c3d95ebSDavid Majnemer }
17831c3d95ebSDavid Majnemer 
1784747e301eSWarren Hunt static llvm::Value *EmitTypeidFromVTable(CodeGenFunction &CGF, const Expr *E,
17852192fe50SChris Lattner                                          llvm::Type *StdTypeInfoPtrTy) {
1786940f02d2SAnders Carlsson   // Get the vtable pointer.
17877f416cc4SJohn McCall   Address ThisPtr = CGF.EmitLValue(E).getAddress();
1788940f02d2SAnders Carlsson 
1789940f02d2SAnders Carlsson   // C++ [expr.typeid]p2:
1790940f02d2SAnders Carlsson   //   If the glvalue expression is obtained by applying the unary * operator to
1791940f02d2SAnders Carlsson   //   a pointer and the pointer is a null pointer value, the typeid expression
1792940f02d2SAnders Carlsson   //   throws the std::bad_typeid exception.
17931c3d95ebSDavid Majnemer   //
17941c3d95ebSDavid Majnemer   // However, this paragraph's intent is not clear.  We choose a very generous
17951c3d95ebSDavid Majnemer   // interpretation which implores us to consider comma operators, conditional
17961c3d95ebSDavid Majnemer   // operators, parentheses and other such constructs.
17971162d25cSDavid Majnemer   QualType SrcRecordTy = E->getType();
17981c3d95ebSDavid Majnemer   if (CGF.CGM.getCXXABI().shouldTypeidBeNullChecked(
17991c3d95ebSDavid Majnemer           isGLValueFromPointerDeref(E), SrcRecordTy)) {
1800940f02d2SAnders Carlsson     llvm::BasicBlock *BadTypeidBlock =
1801940f02d2SAnders Carlsson         CGF.createBasicBlock("typeid.bad_typeid");
18021162d25cSDavid Majnemer     llvm::BasicBlock *EndBlock = CGF.createBasicBlock("typeid.end");
1803940f02d2SAnders Carlsson 
18047f416cc4SJohn McCall     llvm::Value *IsNull = CGF.Builder.CreateIsNull(ThisPtr.getPointer());
1805940f02d2SAnders Carlsson     CGF.Builder.CreateCondBr(IsNull, BadTypeidBlock, EndBlock);
1806940f02d2SAnders Carlsson 
1807940f02d2SAnders Carlsson     CGF.EmitBlock(BadTypeidBlock);
18081162d25cSDavid Majnemer     CGF.CGM.getCXXABI().EmitBadTypeidCall(CGF);
1809940f02d2SAnders Carlsson     CGF.EmitBlock(EndBlock);
1810940f02d2SAnders Carlsson   }
1811940f02d2SAnders Carlsson 
18121162d25cSDavid Majnemer   return CGF.CGM.getCXXABI().EmitTypeid(CGF, SrcRecordTy, ThisPtr,
18131162d25cSDavid Majnemer                                         StdTypeInfoPtrTy);
1814940f02d2SAnders Carlsson }
1815940f02d2SAnders Carlsson 
181659486a2dSAnders Carlsson llvm::Value *CodeGenFunction::EmitCXXTypeidExpr(const CXXTypeidExpr *E) {
18172192fe50SChris Lattner   llvm::Type *StdTypeInfoPtrTy =
1818940f02d2SAnders Carlsson     ConvertType(E->getType())->getPointerTo();
1819fd7dfeb7SAnders Carlsson 
18203f4336cbSAnders Carlsson   if (E->isTypeOperand()) {
18213f4336cbSAnders Carlsson     llvm::Constant *TypeInfo =
1822143c55eaSDavid Majnemer         CGM.GetAddrOfRTTIDescriptor(E->getTypeOperand(getContext()));
1823940f02d2SAnders Carlsson     return Builder.CreateBitCast(TypeInfo, StdTypeInfoPtrTy);
18243f4336cbSAnders Carlsson   }
1825fd7dfeb7SAnders Carlsson 
1826940f02d2SAnders Carlsson   // C++ [expr.typeid]p2:
1827940f02d2SAnders Carlsson   //   When typeid is applied to a glvalue expression whose type is a
1828940f02d2SAnders Carlsson   //   polymorphic class type, the result refers to a std::type_info object
1829940f02d2SAnders Carlsson   //   representing the type of the most derived object (that is, the dynamic
1830940f02d2SAnders Carlsson   //   type) to which the glvalue refers.
1831ef8bf436SRichard Smith   if (E->isPotentiallyEvaluated())
1832940f02d2SAnders Carlsson     return EmitTypeidFromVTable(*this, E->getExprOperand(),
1833940f02d2SAnders Carlsson                                 StdTypeInfoPtrTy);
1834940f02d2SAnders Carlsson 
1835940f02d2SAnders Carlsson   QualType OperandTy = E->getExprOperand()->getType();
1836940f02d2SAnders Carlsson   return Builder.CreateBitCast(CGM.GetAddrOfRTTIDescriptor(OperandTy),
1837940f02d2SAnders Carlsson                                StdTypeInfoPtrTy);
183859486a2dSAnders Carlsson }
183959486a2dSAnders Carlsson 
1840c1c9971cSAnders Carlsson static llvm::Value *EmitDynamicCastToNull(CodeGenFunction &CGF,
1841c1c9971cSAnders Carlsson                                           QualType DestTy) {
18422192fe50SChris Lattner   llvm::Type *DestLTy = CGF.ConvertType(DestTy);
1843c1c9971cSAnders Carlsson   if (DestTy->isPointerType())
1844c1c9971cSAnders Carlsson     return llvm::Constant::getNullValue(DestLTy);
1845c1c9971cSAnders Carlsson 
1846c1c9971cSAnders Carlsson   /// C++ [expr.dynamic.cast]p9:
1847c1c9971cSAnders Carlsson   ///   A failed cast to reference type throws std::bad_cast
18481162d25cSDavid Majnemer   if (!CGF.CGM.getCXXABI().EmitBadCastCall(CGF))
18491162d25cSDavid Majnemer     return nullptr;
1850c1c9971cSAnders Carlsson 
1851c1c9971cSAnders Carlsson   CGF.EmitBlock(CGF.createBasicBlock("dynamic_cast.end"));
1852c1c9971cSAnders Carlsson   return llvm::UndefValue::get(DestLTy);
1853c1c9971cSAnders Carlsson }
1854c1c9971cSAnders Carlsson 
18557f416cc4SJohn McCall llvm::Value *CodeGenFunction::EmitDynamicCast(Address ThisAddr,
185659486a2dSAnders Carlsson                                               const CXXDynamicCastExpr *DCE) {
18572bf9b4c0SAlexey Bataev   CGM.EmitExplicitCastExprType(DCE, this);
18583f4336cbSAnders Carlsson   QualType DestTy = DCE->getTypeAsWritten();
18593f4336cbSAnders Carlsson 
1860c1c9971cSAnders Carlsson   if (DCE->isAlwaysNull())
18611162d25cSDavid Majnemer     if (llvm::Value *T = EmitDynamicCastToNull(*this, DestTy))
18621162d25cSDavid Majnemer       return T;
1863c1c9971cSAnders Carlsson 
1864c1c9971cSAnders Carlsson   QualType SrcTy = DCE->getSubExpr()->getType();
1865c1c9971cSAnders Carlsson 
18661162d25cSDavid Majnemer   // C++ [expr.dynamic.cast]p7:
18671162d25cSDavid Majnemer   //   If T is "pointer to cv void," then the result is a pointer to the most
18681162d25cSDavid Majnemer   //   derived object pointed to by v.
18691162d25cSDavid Majnemer   const PointerType *DestPTy = DestTy->getAs<PointerType>();
18701162d25cSDavid Majnemer 
18711162d25cSDavid Majnemer   bool isDynamicCastToVoid;
18721162d25cSDavid Majnemer   QualType SrcRecordTy;
18731162d25cSDavid Majnemer   QualType DestRecordTy;
18741162d25cSDavid Majnemer   if (DestPTy) {
18751162d25cSDavid Majnemer     isDynamicCastToVoid = DestPTy->getPointeeType()->isVoidType();
18761162d25cSDavid Majnemer     SrcRecordTy = SrcTy->castAs<PointerType>()->getPointeeType();
18771162d25cSDavid Majnemer     DestRecordTy = DestPTy->getPointeeType();
18781162d25cSDavid Majnemer   } else {
18791162d25cSDavid Majnemer     isDynamicCastToVoid = false;
18801162d25cSDavid Majnemer     SrcRecordTy = SrcTy;
18811162d25cSDavid Majnemer     DestRecordTy = DestTy->castAs<ReferenceType>()->getPointeeType();
18821162d25cSDavid Majnemer   }
18831162d25cSDavid Majnemer 
18841162d25cSDavid Majnemer   assert(SrcRecordTy->isRecordType() && "source type must be a record type!");
18851162d25cSDavid Majnemer 
1886882d790fSAnders Carlsson   // C++ [expr.dynamic.cast]p4:
1887882d790fSAnders Carlsson   //   If the value of v is a null pointer value in the pointer case, the result
1888882d790fSAnders Carlsson   //   is the null pointer value of type T.
18891162d25cSDavid Majnemer   bool ShouldNullCheckSrcValue =
18901162d25cSDavid Majnemer       CGM.getCXXABI().shouldDynamicCastCallBeNullChecked(SrcTy->isPointerType(),
18911162d25cSDavid Majnemer                                                          SrcRecordTy);
189259486a2dSAnders Carlsson 
18938a13c418SCraig Topper   llvm::BasicBlock *CastNull = nullptr;
18948a13c418SCraig Topper   llvm::BasicBlock *CastNotNull = nullptr;
1895882d790fSAnders Carlsson   llvm::BasicBlock *CastEnd = createBasicBlock("dynamic_cast.end");
1896fa8b4955SDouglas Gregor 
1897882d790fSAnders Carlsson   if (ShouldNullCheckSrcValue) {
1898882d790fSAnders Carlsson     CastNull = createBasicBlock("dynamic_cast.null");
1899882d790fSAnders Carlsson     CastNotNull = createBasicBlock("dynamic_cast.notnull");
1900882d790fSAnders Carlsson 
19017f416cc4SJohn McCall     llvm::Value *IsNull = Builder.CreateIsNull(ThisAddr.getPointer());
1902882d790fSAnders Carlsson     Builder.CreateCondBr(IsNull, CastNull, CastNotNull);
1903882d790fSAnders Carlsson     EmitBlock(CastNotNull);
190459486a2dSAnders Carlsson   }
190559486a2dSAnders Carlsson 
19067f416cc4SJohn McCall   llvm::Value *Value;
19071162d25cSDavid Majnemer   if (isDynamicCastToVoid) {
19087f416cc4SJohn McCall     Value = CGM.getCXXABI().EmitDynamicCastToVoid(*this, ThisAddr, SrcRecordTy,
19091162d25cSDavid Majnemer                                                   DestTy);
19101162d25cSDavid Majnemer   } else {
19111162d25cSDavid Majnemer     assert(DestRecordTy->isRecordType() &&
19121162d25cSDavid Majnemer            "destination type must be a record type!");
19137f416cc4SJohn McCall     Value = CGM.getCXXABI().EmitDynamicCastCall(*this, ThisAddr, SrcRecordTy,
19141162d25cSDavid Majnemer                                                 DestTy, DestRecordTy, CastEnd);
191567528eaaSDavid Majnemer     CastNotNull = Builder.GetInsertBlock();
19161162d25cSDavid Majnemer   }
19173f4336cbSAnders Carlsson 
1918882d790fSAnders Carlsson   if (ShouldNullCheckSrcValue) {
1919882d790fSAnders Carlsson     EmitBranch(CastEnd);
192059486a2dSAnders Carlsson 
1921882d790fSAnders Carlsson     EmitBlock(CastNull);
1922882d790fSAnders Carlsson     EmitBranch(CastEnd);
192359486a2dSAnders Carlsson   }
192459486a2dSAnders Carlsson 
1925882d790fSAnders Carlsson   EmitBlock(CastEnd);
192659486a2dSAnders Carlsson 
1927882d790fSAnders Carlsson   if (ShouldNullCheckSrcValue) {
1928882d790fSAnders Carlsson     llvm::PHINode *PHI = Builder.CreatePHI(Value->getType(), 2);
1929882d790fSAnders Carlsson     PHI->addIncoming(Value, CastNotNull);
1930882d790fSAnders Carlsson     PHI->addIncoming(llvm::Constant::getNullValue(Value->getType()), CastNull);
193159486a2dSAnders Carlsson 
1932882d790fSAnders Carlsson     Value = PHI;
193359486a2dSAnders Carlsson   }
193459486a2dSAnders Carlsson 
1935882d790fSAnders Carlsson   return Value;
193659486a2dSAnders Carlsson }
1937c370a7eeSEli Friedman 
1938c370a7eeSEli Friedman void CodeGenFunction::EmitLambdaExpr(const LambdaExpr *E, AggValueSlot Slot) {
19398631f3e8SEli Friedman   RunCleanupsScope Scope(*this);
19407f416cc4SJohn McCall   LValue SlotLV = MakeAddrLValue(Slot.getAddress(), E->getType());
19418631f3e8SEli Friedman 
1942c370a7eeSEli Friedman   CXXRecordDecl::field_iterator CurField = E->getLambdaClass()->field_begin();
194353c7616eSJames Y Knight   for (LambdaExpr::const_capture_init_iterator i = E->capture_init_begin(),
1944c370a7eeSEli Friedman                                                e = E->capture_init_end();
1945c370a7eeSEli Friedman        i != e; ++i, ++CurField) {
1946c370a7eeSEli Friedman     // Emit initialization
194740ed2973SDavid Blaikie     LValue LV = EmitLValueForFieldInitialization(SlotLV, *CurField);
194839c81e28SAlexey Bataev     if (CurField->hasCapturedVLAType()) {
194939c81e28SAlexey Bataev       auto VAT = CurField->getCapturedVLAType();
195039c81e28SAlexey Bataev       EmitStoreThroughLValue(RValue::get(VLASizeMap[VAT->getSizeExpr()]), LV);
195139c81e28SAlexey Bataev     } else {
19525f1a04ffSEli Friedman       ArrayRef<VarDecl *> ArrayIndexes;
19535f1a04ffSEli Friedman       if (CurField->getType()->isArrayType())
19545f1a04ffSEli Friedman         ArrayIndexes = E->getCaptureInitIndexVars(i);
195540ed2973SDavid Blaikie       EmitInitializerForField(*CurField, LV, *i, ArrayIndexes);
1956c370a7eeSEli Friedman     }
1957c370a7eeSEli Friedman   }
195839c81e28SAlexey Bataev }
1959