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"
193a02247dSChandler Carruth #include "clang/Frontend/CodeGenOptions.h"
20ffd5551bSChandler Carruth #include "llvm/IR/Intrinsics.h"
21bbe277c4SAnders Carlsson #include "llvm/Support/CallSite.h"
22bbe277c4SAnders Carlsson 
2359486a2dSAnders Carlsson using namespace clang;
2459486a2dSAnders Carlsson using namespace CodeGen;
2559486a2dSAnders Carlsson 
2627da15baSAnders Carlsson RValue CodeGenFunction::EmitCXXMemberCall(const CXXMethodDecl *MD,
27e30752c9SRichard Smith                                           SourceLocation CallLoc,
2827da15baSAnders Carlsson                                           llvm::Value *Callee,
2927da15baSAnders Carlsson                                           ReturnValueSlot ReturnValue,
3027da15baSAnders Carlsson                                           llvm::Value *This,
31e36a6b3eSAnders Carlsson                                           llvm::Value *VTT,
3227da15baSAnders Carlsson                                           CallExpr::const_arg_iterator ArgBeg,
3327da15baSAnders Carlsson                                           CallExpr::const_arg_iterator ArgEnd) {
3427da15baSAnders Carlsson   assert(MD->isInstance() &&
3527da15baSAnders Carlsson          "Trying to emit a member 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.
404d3110afSRichard Smith   EmitTypeCheck(isa<CXXConstructorDecl>(MD) ? TCK_ConstructorCall
414d3110afSRichard Smith                                             : TCK_MemberCall,
424d3110afSRichard Smith                 CallLoc, This, getContext().getRecordType(MD->getParent()));
4369d0d262SRichard Smith 
4427da15baSAnders Carlsson   CallArgList Args;
4527da15baSAnders Carlsson 
4627da15baSAnders Carlsson   // Push the this ptr.
4743dca6a8SEli Friedman   Args.add(RValue::get(This), MD->getThisType(getContext()));
4827da15baSAnders Carlsson 
49e36a6b3eSAnders Carlsson   // If there is a VTT parameter, emit it.
50e36a6b3eSAnders Carlsson   if (VTT) {
51e36a6b3eSAnders Carlsson     QualType T = getContext().getPointerType(getContext().VoidPtrTy);
5243dca6a8SEli Friedman     Args.add(RValue::get(VTT), T);
53e36a6b3eSAnders Carlsson   }
54e36a6b3eSAnders Carlsson 
55a729c62bSJohn McCall   const FunctionProtoType *FPT = MD->getType()->castAs<FunctionProtoType>();
56a729c62bSJohn McCall   RequiredArgs required = RequiredArgs::forPrototypePlus(FPT, Args.size());
57a729c62bSJohn McCall 
58a729c62bSJohn McCall   // And the rest of the call args.
5927da15baSAnders Carlsson   EmitCallArgs(Args, FPT, ArgBeg, ArgEnd);
6027da15baSAnders Carlsson 
618dda7b27SJohn McCall   return EmitCall(CGM.getTypes().arrangeCXXMethodCall(Args, FPT, required),
62c50c27ccSRafael Espindola                   Callee, ReturnValue, Args, MD);
6327da15baSAnders Carlsson }
6427da15baSAnders Carlsson 
65c53d9e83SAnders Carlsson // FIXME: Ideally Expr::IgnoreParenNoopCasts should do this, but it doesn't do
66c53d9e83SAnders Carlsson // quite what we want.
67c53d9e83SAnders Carlsson static const Expr *skipNoOpCastsAndParens(const Expr *E) {
68c53d9e83SAnders Carlsson   while (true) {
69c53d9e83SAnders Carlsson     if (const ParenExpr *PE = dyn_cast<ParenExpr>(E)) {
70c53d9e83SAnders Carlsson       E = PE->getSubExpr();
71c53d9e83SAnders Carlsson       continue;
72c53d9e83SAnders Carlsson     }
73c53d9e83SAnders Carlsson 
74c53d9e83SAnders Carlsson     if (const CastExpr *CE = dyn_cast<CastExpr>(E)) {
75c53d9e83SAnders Carlsson       if (CE->getCastKind() == CK_NoOp) {
76c53d9e83SAnders Carlsson         E = CE->getSubExpr();
77c53d9e83SAnders Carlsson         continue;
78c53d9e83SAnders Carlsson       }
79c53d9e83SAnders Carlsson     }
80c53d9e83SAnders Carlsson     if (const UnaryOperator *UO = dyn_cast<UnaryOperator>(E)) {
81c53d9e83SAnders Carlsson       if (UO->getOpcode() == UO_Extension) {
82c53d9e83SAnders Carlsson         E = UO->getSubExpr();
83c53d9e83SAnders Carlsson         continue;
84c53d9e83SAnders Carlsson       }
85c53d9e83SAnders Carlsson     }
86c53d9e83SAnders Carlsson     return E;
87c53d9e83SAnders Carlsson   }
88c53d9e83SAnders Carlsson }
89c53d9e83SAnders Carlsson 
9027da15baSAnders Carlsson /// canDevirtualizeMemberFunctionCalls - Checks whether virtual calls on given
9127da15baSAnders Carlsson /// expr can be devirtualized.
92252a47f6SFariborz Jahanian static bool canDevirtualizeMemberFunctionCalls(ASTContext &Context,
93252a47f6SFariborz Jahanian                                                const Expr *Base,
94a7911fa3SAnders Carlsson                                                const CXXMethodDecl *MD) {
95a7911fa3SAnders Carlsson 
961ae64c5aSAnders Carlsson   // When building with -fapple-kext, all calls must go through the vtable since
971ae64c5aSAnders Carlsson   // the kernel linker can do runtime patching of vtables.
98bbafb8a7SDavid Blaikie   if (Context.getLangOpts().AppleKext)
99252a47f6SFariborz Jahanian     return false;
100252a47f6SFariborz Jahanian 
1011ae64c5aSAnders Carlsson   // If the most derived class is marked final, we know that no subclass can
1021ae64c5aSAnders Carlsson   // override this member function and so we can devirtualize it. For example:
1031ae64c5aSAnders Carlsson   //
1041ae64c5aSAnders Carlsson   // struct A { virtual void f(); }
1051ae64c5aSAnders Carlsson   // struct B final : A { };
1061ae64c5aSAnders Carlsson   //
1071ae64c5aSAnders Carlsson   // void f(B *b) {
1081ae64c5aSAnders Carlsson   //   b->f();
1091ae64c5aSAnders Carlsson   // }
1101ae64c5aSAnders Carlsson   //
111b7f5a9c5SRafael Espindola   const CXXRecordDecl *MostDerivedClassDecl = Base->getBestDynamicClassType();
1121ae64c5aSAnders Carlsson   if (MostDerivedClassDecl->hasAttr<FinalAttr>())
1131ae64c5aSAnders Carlsson     return true;
1141ae64c5aSAnders Carlsson 
11519588aa4SAnders Carlsson   // If the member function is marked 'final', we know that it can't be
116b00c2144SAnders Carlsson   // overridden and can therefore devirtualize it.
1171eb95961SAnders Carlsson   if (MD->hasAttr<FinalAttr>())
118a7911fa3SAnders Carlsson     return true;
119a7911fa3SAnders Carlsson 
12019588aa4SAnders Carlsson   // Similarly, if the class itself is marked 'final' it can't be overridden
12119588aa4SAnders Carlsson   // and we can therefore devirtualize the member function call.
1221eb95961SAnders Carlsson   if (MD->getParent()->hasAttr<FinalAttr>())
123b00c2144SAnders Carlsson     return true;
124b00c2144SAnders Carlsson 
125c53d9e83SAnders Carlsson   Base = skipNoOpCastsAndParens(Base);
12627da15baSAnders Carlsson   if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(Base)) {
12727da15baSAnders Carlsson     if (const VarDecl *VD = dyn_cast<VarDecl>(DRE->getDecl())) {
12827da15baSAnders Carlsson       // This is a record decl. We know the type and can devirtualize it.
12927da15baSAnders Carlsson       return VD->getType()->isRecordType();
13027da15baSAnders Carlsson     }
13127da15baSAnders Carlsson 
13227da15baSAnders Carlsson     return false;
13327da15baSAnders Carlsson   }
13427da15baSAnders Carlsson 
13548c15319SRichard Smith   // We can devirtualize calls on an object accessed by a class member access
13648c15319SRichard Smith   // expression, since by C++11 [basic.life]p6 we know that it can't refer to
13748c15319SRichard Smith   // a derived class object constructed in the same location.
13848c15319SRichard Smith   if (const MemberExpr *ME = dyn_cast<MemberExpr>(Base))
13948c15319SRichard Smith     if (const ValueDecl *VD = dyn_cast<ValueDecl>(ME->getMemberDecl()))
14048c15319SRichard Smith       return VD->getType()->isRecordType();
14148c15319SRichard Smith 
14227da15baSAnders Carlsson   // We can always devirtualize calls on temporary object expressions.
143a682427eSEli Friedman   if (isa<CXXConstructExpr>(Base))
14427da15baSAnders Carlsson     return true;
14527da15baSAnders Carlsson 
14627da15baSAnders Carlsson   // And calls on bound temporaries.
14727da15baSAnders Carlsson   if (isa<CXXBindTemporaryExpr>(Base))
14827da15baSAnders Carlsson     return true;
14927da15baSAnders Carlsson 
15027da15baSAnders Carlsson   // Check if this is a call expr that returns a record type.
15127da15baSAnders Carlsson   if (const CallExpr *CE = dyn_cast<CallExpr>(Base))
15227da15baSAnders Carlsson     return CE->getCallReturnType()->isRecordType();
15327da15baSAnders Carlsson 
15427da15baSAnders Carlsson   // We can't devirtualize the call.
15527da15baSAnders Carlsson   return false;
15627da15baSAnders Carlsson }
15727da15baSAnders Carlsson 
1583b33c4ecSRafael Espindola static CXXRecordDecl *getCXXRecord(const Expr *E) {
1593b33c4ecSRafael Espindola   QualType T = E->getType();
1603b33c4ecSRafael Espindola   if (const PointerType *PTy = T->getAs<PointerType>())
1613b33c4ecSRafael Espindola     T = PTy->getPointeeType();
1623b33c4ecSRafael Espindola   const RecordType *Ty = T->castAs<RecordType>();
1633b33c4ecSRafael Espindola   return cast<CXXRecordDecl>(Ty->getDecl());
1643b33c4ecSRafael Espindola }
1653b33c4ecSRafael Espindola 
16664225794SFrancois Pichet // Note: This function also emit constructor calls to support a MSVC
16764225794SFrancois Pichet // extensions allowing explicit constructor function call.
16827da15baSAnders Carlsson RValue CodeGenFunction::EmitCXXMemberCallExpr(const CXXMemberCallExpr *CE,
16927da15baSAnders Carlsson                                               ReturnValueSlot ReturnValue) {
1702d2e8707SJohn McCall   const Expr *callee = CE->getCallee()->IgnoreParens();
1712d2e8707SJohn McCall 
1722d2e8707SJohn McCall   if (isa<BinaryOperator>(callee))
17327da15baSAnders Carlsson     return EmitCXXMemberPointerCallExpr(CE, ReturnValue);
17427da15baSAnders Carlsson 
1752d2e8707SJohn McCall   const MemberExpr *ME = cast<MemberExpr>(callee);
17627da15baSAnders Carlsson   const CXXMethodDecl *MD = cast<CXXMethodDecl>(ME->getMemberDecl());
17727da15baSAnders Carlsson 
17891bbb554SDevang Patel   CGDebugInfo *DI = getDebugInfo();
179b0eea8b5SDouglas Gregor   if (DI &&
180b0eea8b5SDouglas Gregor       CGM.getCodeGenOpts().getDebugInfo() == CodeGenOptions::LimitedDebugInfo &&
181b0eea8b5SDouglas Gregor       !isa<CallExpr>(ME->getBase())) {
18291bbb554SDevang Patel     QualType PQTy = ME->getBase()->IgnoreParenImpCasts()->getType();
18391bbb554SDevang Patel     if (const PointerType * PTy = dyn_cast<PointerType>(PQTy)) {
18491bbb554SDevang Patel       DI->getOrCreateRecordType(PTy->getPointeeType(),
18591bbb554SDevang Patel                                 MD->getParent()->getLocation());
18691bbb554SDevang Patel     }
18791bbb554SDevang Patel   }
18891bbb554SDevang Patel 
18927da15baSAnders Carlsson   if (MD->isStatic()) {
19027da15baSAnders Carlsson     // The method is static, emit it as we would a regular call.
19127da15baSAnders Carlsson     llvm::Value *Callee = CGM.GetAddrOfFunction(MD);
19227da15baSAnders Carlsson     return EmitCall(getContext().getPointerType(MD->getType()), Callee,
19327da15baSAnders Carlsson                     ReturnValue, CE->arg_begin(), CE->arg_end());
19427da15baSAnders Carlsson   }
19527da15baSAnders Carlsson 
1960d635f53SJohn McCall   // Compute the object pointer.
197ecbe2e97SRafael Espindola   const Expr *Base = ME->getBase();
198ecbe2e97SRafael Espindola   bool CanUseVirtualCall = MD->isVirtual() && !ME->hasQualifier();
199ecbe2e97SRafael Espindola 
2003b33c4ecSRafael Espindola   const CXXMethodDecl *DevirtualizedMethod = NULL;
2013b33c4ecSRafael Espindola   if (CanUseVirtualCall &&
2023b33c4ecSRafael Espindola       canDevirtualizeMemberFunctionCalls(getContext(), Base, MD)) {
2033b33c4ecSRafael Espindola     const CXXRecordDecl *BestDynamicDecl = Base->getBestDynamicClassType();
2043b33c4ecSRafael Espindola     DevirtualizedMethod = MD->getCorrespondingMethodInClass(BestDynamicDecl);
2053b33c4ecSRafael Espindola     assert(DevirtualizedMethod);
2063b33c4ecSRafael Espindola     const CXXRecordDecl *DevirtualizedClass = DevirtualizedMethod->getParent();
2073b33c4ecSRafael Espindola     const Expr *Inner = Base->ignoreParenBaseCasts();
2083b33c4ecSRafael Espindola     if (getCXXRecord(Inner) == DevirtualizedClass)
2093b33c4ecSRafael Espindola       // If the class of the Inner expression is where the dynamic method
2103b33c4ecSRafael Espindola       // is defined, build the this pointer from it.
2113b33c4ecSRafael Espindola       Base = Inner;
2123b33c4ecSRafael Espindola     else if (getCXXRecord(Base) != DevirtualizedClass) {
2133b33c4ecSRafael Espindola       // If the method is defined in a class that is not the best dynamic
2143b33c4ecSRafael Espindola       // one or the one of the full expression, we would have to build
2153b33c4ecSRafael Espindola       // a derived-to-base cast to compute the correct this pointer, but
2163b33c4ecSRafael Espindola       // we don't have support for that yet, so do a virtual call.
2173b33c4ecSRafael Espindola       DevirtualizedMethod = NULL;
2183b33c4ecSRafael Espindola     }
219b27564afSRafael Espindola     // If the return types are not the same, this might be a case where more
220b27564afSRafael Espindola     // code needs to run to compensate for it. For example, the derived
221b27564afSRafael Espindola     // method might return a type that inherits form from the return
222b27564afSRafael Espindola     // type of MD and has a prefix.
223b27564afSRafael Espindola     // For now we just avoid devirtualizing these covariant cases.
224b27564afSRafael Espindola     if (DevirtualizedMethod &&
225b27564afSRafael Espindola         DevirtualizedMethod->getResultType().getCanonicalType() !=
226b27564afSRafael Espindola         MD->getResultType().getCanonicalType())
227debc71ceSRafael Espindola       DevirtualizedMethod = NULL;
2283b33c4ecSRafael Espindola   }
229ecbe2e97SRafael Espindola 
23027da15baSAnders Carlsson   llvm::Value *This;
23127da15baSAnders Carlsson   if (ME->isArrow())
2323b33c4ecSRafael Espindola     This = EmitScalarExpr(Base);
233f93ac894SFariborz Jahanian   else
2343b33c4ecSRafael Espindola     This = EmitLValue(Base).getAddress();
235ecbe2e97SRafael Espindola 
23627da15baSAnders Carlsson 
2370d635f53SJohn McCall   if (MD->isTrivial()) {
2380d635f53SJohn McCall     if (isa<CXXDestructorDecl>(MD)) return RValue::get(0);
23964225794SFrancois Pichet     if (isa<CXXConstructorDecl>(MD) &&
24064225794SFrancois Pichet         cast<CXXConstructorDecl>(MD)->isDefaultConstructor())
24164225794SFrancois Pichet       return RValue::get(0);
2420d635f53SJohn McCall 
24322653bacSSebastian Redl     if (MD->isCopyAssignmentOperator() || MD->isMoveAssignmentOperator()) {
24422653bacSSebastian Redl       // We don't like to generate the trivial copy/move assignment operator
24522653bacSSebastian Redl       // when it isn't necessary; just produce the proper effect here.
24627da15baSAnders Carlsson       llvm::Value *RHS = EmitLValue(*CE->arg_begin()).getAddress();
2471ca66919SBenjamin Kramer       EmitAggregateAssign(This, RHS, CE->getType());
24827da15baSAnders Carlsson       return RValue::get(This);
24927da15baSAnders Carlsson     }
25027da15baSAnders Carlsson 
25164225794SFrancois Pichet     if (isa<CXXConstructorDecl>(MD) &&
25222653bacSSebastian Redl         cast<CXXConstructorDecl>(MD)->isCopyOrMoveConstructor()) {
25322653bacSSebastian Redl       // Trivial move and copy ctor are the same.
25464225794SFrancois Pichet       llvm::Value *RHS = EmitLValue(*CE->arg_begin()).getAddress();
25564225794SFrancois Pichet       EmitSynthesizedCXXCopyCtorCall(cast<CXXConstructorDecl>(MD), This, RHS,
25664225794SFrancois Pichet                                      CE->arg_begin(), CE->arg_end());
25764225794SFrancois Pichet       return RValue::get(This);
25864225794SFrancois Pichet     }
25964225794SFrancois Pichet     llvm_unreachable("unknown trivial member function");
26064225794SFrancois Pichet   }
26164225794SFrancois Pichet 
2620d635f53SJohn McCall   // Compute the function type we're calling.
263ade60977SEli Friedman   const CXXMethodDecl *CalleeDecl = DevirtualizedMethod ? DevirtualizedMethod : MD;
26464225794SFrancois Pichet   const CGFunctionInfo *FInfo = 0;
265ade60977SEli Friedman   if (const CXXDestructorDecl *Dtor = dyn_cast<CXXDestructorDecl>(CalleeDecl))
266ade60977SEli Friedman     FInfo = &CGM.getTypes().arrangeCXXDestructor(Dtor,
26764225794SFrancois Pichet                                                  Dtor_Complete);
268ade60977SEli Friedman   else if (const CXXConstructorDecl *Ctor = dyn_cast<CXXConstructorDecl>(CalleeDecl))
269ade60977SEli Friedman     FInfo = &CGM.getTypes().arrangeCXXConstructorDeclaration(Ctor,
27064225794SFrancois Pichet                                                              Ctor_Complete);
27164225794SFrancois Pichet   else
272ade60977SEli Friedman     FInfo = &CGM.getTypes().arrangeCXXMethodDeclaration(CalleeDecl);
2730d635f53SJohn McCall 
274a729c62bSJohn McCall   llvm::Type *Ty = CGM.getTypes().GetFunctionType(*FInfo);
2750d635f53SJohn McCall 
27627da15baSAnders Carlsson   // C++ [class.virtual]p12:
27727da15baSAnders Carlsson   //   Explicit qualification with the scope operator (5.1) suppresses the
27827da15baSAnders Carlsson   //   virtual call mechanism.
27927da15baSAnders Carlsson   //
28027da15baSAnders Carlsson   // We also don't emit a virtual call if the base expression has a record type
28127da15baSAnders Carlsson   // because then we know what the type is.
2823b33c4ecSRafael Espindola   bool UseVirtualCall = CanUseVirtualCall && !DevirtualizedMethod;
28349e860b2SRafael Espindola 
28427da15baSAnders Carlsson   llvm::Value *Callee;
2850d635f53SJohn McCall   if (const CXXDestructorDecl *Dtor = dyn_cast<CXXDestructorDecl>(MD)) {
2860d635f53SJohn McCall     if (UseVirtualCall) {
2870d635f53SJohn McCall       Callee = BuildVirtualCall(Dtor, Dtor_Complete, This, Ty);
28827da15baSAnders Carlsson     } else {
2899c6890a7SRichard Smith       if (getLangOpts().AppleKext &&
290265c325eSFariborz Jahanian           MD->isVirtual() &&
291265c325eSFariborz Jahanian           ME->hasQualifier())
2927f6f81baSFariborz Jahanian         Callee = BuildAppleKextVirtualCall(MD, ME->getQualifier(), Ty);
2933b33c4ecSRafael Espindola       else if (!DevirtualizedMethod)
294727a771aSRafael Espindola         Callee = CGM.GetAddrOfFunction(GlobalDecl(Dtor, Dtor_Complete), Ty);
29549e860b2SRafael Espindola       else {
2963b33c4ecSRafael Espindola         const CXXDestructorDecl *DDtor =
2973b33c4ecSRafael Espindola           cast<CXXDestructorDecl>(DevirtualizedMethod);
29849e860b2SRafael Espindola         Callee = CGM.GetAddrOfFunction(GlobalDecl(DDtor, Dtor_Complete), Ty);
29949e860b2SRafael Espindola       }
30027da15baSAnders Carlsson     }
30164225794SFrancois Pichet   } else if (const CXXConstructorDecl *Ctor =
30264225794SFrancois Pichet                dyn_cast<CXXConstructorDecl>(MD)) {
30364225794SFrancois Pichet     Callee = CGM.GetAddrOfFunction(GlobalDecl(Ctor, Ctor_Complete), Ty);
3040d635f53SJohn McCall   } else if (UseVirtualCall) {
30527da15baSAnders Carlsson       Callee = BuildVirtualCall(MD, This, Ty);
30627da15baSAnders Carlsson   } else {
3079c6890a7SRichard Smith     if (getLangOpts().AppleKext &&
3089f9438b3SFariborz Jahanian         MD->isVirtual() &&
309252a47f6SFariborz Jahanian         ME->hasQualifier())
3107f6f81baSFariborz Jahanian       Callee = BuildAppleKextVirtualCall(MD, ME->getQualifier(), Ty);
3113b33c4ecSRafael Espindola     else if (!DevirtualizedMethod)
312727a771aSRafael Espindola       Callee = CGM.GetAddrOfFunction(MD, Ty);
31349e860b2SRafael Espindola     else {
3143b33c4ecSRafael Espindola       Callee = CGM.GetAddrOfFunction(DevirtualizedMethod, Ty);
31549e860b2SRafael Espindola     }
31627da15baSAnders Carlsson   }
31727da15baSAnders Carlsson 
318e30752c9SRichard Smith   return EmitCXXMemberCall(MD, CE->getExprLoc(), Callee, ReturnValue, This,
319e30752c9SRichard Smith                            /*VTT=*/0, CE->arg_begin(), CE->arg_end());
32027da15baSAnders Carlsson }
32127da15baSAnders Carlsson 
32227da15baSAnders Carlsson RValue
32327da15baSAnders Carlsson CodeGenFunction::EmitCXXMemberPointerCallExpr(const CXXMemberCallExpr *E,
32427da15baSAnders Carlsson                                               ReturnValueSlot ReturnValue) {
32527da15baSAnders Carlsson   const BinaryOperator *BO =
32627da15baSAnders Carlsson       cast<BinaryOperator>(E->getCallee()->IgnoreParens());
32727da15baSAnders Carlsson   const Expr *BaseExpr = BO->getLHS();
32827da15baSAnders Carlsson   const Expr *MemFnExpr = BO->getRHS();
32927da15baSAnders Carlsson 
33027da15baSAnders Carlsson   const MemberPointerType *MPT =
3310009fcc3SJohn McCall     MemFnExpr->getType()->castAs<MemberPointerType>();
332475999dcSJohn McCall 
33327da15baSAnders Carlsson   const FunctionProtoType *FPT =
3340009fcc3SJohn McCall     MPT->getPointeeType()->castAs<FunctionProtoType>();
33527da15baSAnders Carlsson   const CXXRecordDecl *RD =
33627da15baSAnders Carlsson     cast<CXXRecordDecl>(MPT->getClass()->getAs<RecordType>()->getDecl());
33727da15baSAnders Carlsson 
33827da15baSAnders Carlsson   // Get the member function pointer.
339a1dee530SJohn McCall   llvm::Value *MemFnPtr = EmitScalarExpr(MemFnExpr);
34027da15baSAnders Carlsson 
34127da15baSAnders Carlsson   // Emit the 'this' pointer.
34227da15baSAnders Carlsson   llvm::Value *This;
34327da15baSAnders Carlsson 
344e302792bSJohn McCall   if (BO->getOpcode() == BO_PtrMemI)
34527da15baSAnders Carlsson     This = EmitScalarExpr(BaseExpr);
34627da15baSAnders Carlsson   else
34727da15baSAnders Carlsson     This = EmitLValue(BaseExpr).getAddress();
34827da15baSAnders Carlsson 
349e30752c9SRichard Smith   EmitTypeCheck(TCK_MemberCall, E->getExprLoc(), This,
350e30752c9SRichard Smith                 QualType(MPT->getClass(), 0));
35169d0d262SRichard Smith 
352475999dcSJohn McCall   // Ask the ABI to load the callee.  Note that This is modified.
353475999dcSJohn McCall   llvm::Value *Callee =
354ad7c5c16SJohn McCall     CGM.getCXXABI().EmitLoadOfMemberFunctionPointer(*this, This, MemFnPtr, MPT);
35527da15baSAnders Carlsson 
35627da15baSAnders Carlsson   CallArgList Args;
35727da15baSAnders Carlsson 
35827da15baSAnders Carlsson   QualType ThisType =
35927da15baSAnders Carlsson     getContext().getPointerType(getContext().getTagDeclType(RD));
36027da15baSAnders Carlsson 
36127da15baSAnders Carlsson   // Push the this ptr.
36243dca6a8SEli Friedman   Args.add(RValue::get(This), ThisType);
36327da15baSAnders Carlsson 
3648dda7b27SJohn McCall   RequiredArgs required = RequiredArgs::forPrototypePlus(FPT, 1);
3658dda7b27SJohn McCall 
36627da15baSAnders Carlsson   // And the rest of the call args
36727da15baSAnders Carlsson   EmitCallArgs(Args, FPT, E->arg_begin(), E->arg_end());
3688dda7b27SJohn McCall   return EmitCall(CGM.getTypes().arrangeCXXMethodCall(Args, FPT, required), Callee,
36999cc30c3STilmann Scheller                   ReturnValue, Args);
37027da15baSAnders Carlsson }
37127da15baSAnders Carlsson 
37227da15baSAnders Carlsson RValue
37327da15baSAnders Carlsson CodeGenFunction::EmitCXXOperatorMemberCallExpr(const CXXOperatorCallExpr *E,
37427da15baSAnders Carlsson                                                const CXXMethodDecl *MD,
37527da15baSAnders Carlsson                                                ReturnValueSlot ReturnValue) {
37627da15baSAnders Carlsson   assert(MD->isInstance() &&
37727da15baSAnders Carlsson          "Trying to emit a member call expr on a static method!");
378e26a872bSJohn McCall   LValue LV = EmitLValue(E->getArg(0));
379e26a872bSJohn McCall   llvm::Value *This = LV.getAddress();
380e26a872bSJohn McCall 
381146b8e9aSDouglas Gregor   if ((MD->isCopyAssignmentOperator() || MD->isMoveAssignmentOperator()) &&
382146b8e9aSDouglas Gregor       MD->isTrivial()) {
38327da15baSAnders Carlsson     llvm::Value *Src = EmitLValue(E->getArg(1)).getAddress();
38427da15baSAnders Carlsson     QualType Ty = E->getType();
3851ca66919SBenjamin Kramer     EmitAggregateAssign(This, Src, Ty);
38627da15baSAnders Carlsson     return RValue::get(This);
38727da15baSAnders Carlsson   }
38827da15baSAnders Carlsson 
389c36783e8SAnders Carlsson   llvm::Value *Callee = EmitCXXOperatorMemberCallee(E, MD, This);
390e30752c9SRichard Smith   return EmitCXXMemberCall(MD, E->getExprLoc(), Callee, ReturnValue, This,
391e30752c9SRichard Smith                            /*VTT=*/0, E->arg_begin() + 1, E->arg_end());
39227da15baSAnders Carlsson }
39327da15baSAnders Carlsson 
394fe883422SPeter Collingbourne RValue CodeGenFunction::EmitCUDAKernelCallExpr(const CUDAKernelCallExpr *E,
395fe883422SPeter Collingbourne                                                ReturnValueSlot ReturnValue) {
396fe883422SPeter Collingbourne   return CGM.getCUDARuntime().EmitCUDAKernelCallExpr(*this, E, ReturnValue);
397fe883422SPeter Collingbourne }
398fe883422SPeter Collingbourne 
399fde961dbSEli Friedman static void EmitNullBaseClassInitialization(CodeGenFunction &CGF,
400fde961dbSEli Friedman                                             llvm::Value *DestPtr,
401fde961dbSEli Friedman                                             const CXXRecordDecl *Base) {
402fde961dbSEli Friedman   if (Base->isEmpty())
403fde961dbSEli Friedman     return;
404fde961dbSEli Friedman 
405fde961dbSEli Friedman   DestPtr = CGF.EmitCastToVoidPtr(DestPtr);
406fde961dbSEli Friedman 
407fde961dbSEli Friedman   const ASTRecordLayout &Layout = CGF.getContext().getASTRecordLayout(Base);
408fde961dbSEli Friedman   CharUnits Size = Layout.getNonVirtualSize();
409fde961dbSEli Friedman   CharUnits Align = Layout.getNonVirtualAlign();
410fde961dbSEli Friedman 
411fde961dbSEli Friedman   llvm::Value *SizeVal = CGF.CGM.getSize(Size);
412fde961dbSEli Friedman 
413fde961dbSEli Friedman   // If the type contains a pointer to data member we can't memset it to zero.
414fde961dbSEli Friedman   // Instead, create a null constant and copy it to the destination.
415fde961dbSEli Friedman   // TODO: there are other patterns besides zero that we can usefully memset,
416fde961dbSEli Friedman   // like -1, which happens to be the pattern used by member-pointers.
417fde961dbSEli Friedman   // TODO: isZeroInitializable can be over-conservative in the case where a
418fde961dbSEli Friedman   // virtual base contains a member pointer.
419fde961dbSEli Friedman   if (!CGF.CGM.getTypes().isZeroInitializable(Base)) {
420fde961dbSEli Friedman     llvm::Constant *NullConstant = CGF.CGM.EmitNullConstantForBase(Base);
421fde961dbSEli Friedman 
422fde961dbSEli Friedman     llvm::GlobalVariable *NullVariable =
423fde961dbSEli Friedman       new llvm::GlobalVariable(CGF.CGM.getModule(), NullConstant->getType(),
424fde961dbSEli Friedman                                /*isConstant=*/true,
425fde961dbSEli Friedman                                llvm::GlobalVariable::PrivateLinkage,
426fde961dbSEli Friedman                                NullConstant, Twine());
427fde961dbSEli Friedman     NullVariable->setAlignment(Align.getQuantity());
428fde961dbSEli Friedman     llvm::Value *SrcPtr = CGF.EmitCastToVoidPtr(NullVariable);
429fde961dbSEli Friedman 
430fde961dbSEli Friedman     // Get and call the appropriate llvm.memcpy overload.
431fde961dbSEli Friedman     CGF.Builder.CreateMemCpy(DestPtr, SrcPtr, SizeVal, Align.getQuantity());
432fde961dbSEli Friedman     return;
433fde961dbSEli Friedman   }
434fde961dbSEli Friedman 
435fde961dbSEli Friedman   // Otherwise, just memset the whole thing to zero.  This is legal
436fde961dbSEli Friedman   // because in LLVM, all default initializers (other than the ones we just
437fde961dbSEli Friedman   // handled above) are guaranteed to have a bit pattern of all zeros.
438fde961dbSEli Friedman   CGF.Builder.CreateMemSet(DestPtr, CGF.Builder.getInt8(0), SizeVal,
439fde961dbSEli Friedman                            Align.getQuantity());
440fde961dbSEli Friedman }
441fde961dbSEli Friedman 
44227da15baSAnders Carlsson void
4437a626f63SJohn McCall CodeGenFunction::EmitCXXConstructExpr(const CXXConstructExpr *E,
4447a626f63SJohn McCall                                       AggValueSlot Dest) {
4457a626f63SJohn McCall   assert(!Dest.isIgnored() && "Must have a destination!");
44627da15baSAnders Carlsson   const CXXConstructorDecl *CD = E->getConstructor();
447630c76efSDouglas Gregor 
448630c76efSDouglas Gregor   // If we require zero initialization before (or instead of) calling the
449630c76efSDouglas Gregor   // constructor, as can be the case with a non-user-provided default
45003535265SArgyrios Kyrtzidis   // constructor, emit the zero initialization now, unless destination is
45103535265SArgyrios Kyrtzidis   // already zeroed.
452fde961dbSEli Friedman   if (E->requiresZeroInitialization() && !Dest.isZeroed()) {
453fde961dbSEli Friedman     switch (E->getConstructionKind()) {
454fde961dbSEli Friedman     case CXXConstructExpr::CK_Delegating:
455fde961dbSEli Friedman     case CXXConstructExpr::CK_Complete:
4567a626f63SJohn McCall       EmitNullInitialization(Dest.getAddr(), E->getType());
457fde961dbSEli Friedman       break;
458fde961dbSEli Friedman     case CXXConstructExpr::CK_VirtualBase:
459fde961dbSEli Friedman     case CXXConstructExpr::CK_NonVirtualBase:
460fde961dbSEli Friedman       EmitNullBaseClassInitialization(*this, Dest.getAddr(), CD->getParent());
461fde961dbSEli Friedman       break;
462fde961dbSEli Friedman     }
463fde961dbSEli Friedman   }
464630c76efSDouglas Gregor 
465630c76efSDouglas Gregor   // If this is a call to a trivial default constructor, do nothing.
466630c76efSDouglas Gregor   if (CD->isTrivial() && CD->isDefaultConstructor())
46727da15baSAnders Carlsson     return;
468630c76efSDouglas Gregor 
4698ea46b66SJohn McCall   // Elide the constructor if we're constructing from a temporary.
4708ea46b66SJohn McCall   // The temporary check is required because Sema sets this on NRVO
4718ea46b66SJohn McCall   // returns.
4729c6890a7SRichard Smith   if (getLangOpts().ElideConstructors && E->isElidable()) {
4738ea46b66SJohn McCall     assert(getContext().hasSameUnqualifiedType(E->getType(),
4748ea46b66SJohn McCall                                                E->getArg(0)->getType()));
4757a626f63SJohn McCall     if (E->getArg(0)->isTemporaryObject(getContext(), CD->getParent())) {
4767a626f63SJohn McCall       EmitAggExpr(E->getArg(0), Dest);
47727da15baSAnders Carlsson       return;
47827da15baSAnders Carlsson     }
479222cf0efSDouglas Gregor   }
480630c76efSDouglas Gregor 
481f677a8e9SJohn McCall   if (const ConstantArrayType *arrayType
482f677a8e9SJohn McCall         = getContext().getAsConstantArrayType(E->getType())) {
483f677a8e9SJohn McCall     EmitCXXAggrConstructorCall(CD, arrayType, Dest.getAddr(),
48427da15baSAnders Carlsson                                E->arg_begin(), E->arg_end());
485f677a8e9SJohn McCall   } else {
486bceca20aSCameron Esfahani     CXXCtorType Type = Ctor_Complete;
487271c3681SAlexis Hunt     bool ForVirtualBase = false;
48861535005SDouglas Gregor     bool Delegating = false;
489271c3681SAlexis Hunt 
490271c3681SAlexis Hunt     switch (E->getConstructionKind()) {
491271c3681SAlexis Hunt      case CXXConstructExpr::CK_Delegating:
49261bc1737SAlexis Hunt       // We should be emitting a constructor; GlobalDecl will assert this
49361bc1737SAlexis Hunt       Type = CurGD.getCtorType();
49461535005SDouglas Gregor       Delegating = true;
495271c3681SAlexis Hunt       break;
49661bc1737SAlexis Hunt 
497271c3681SAlexis Hunt      case CXXConstructExpr::CK_Complete:
498271c3681SAlexis Hunt       Type = Ctor_Complete;
499271c3681SAlexis Hunt       break;
500271c3681SAlexis Hunt 
501271c3681SAlexis Hunt      case CXXConstructExpr::CK_VirtualBase:
502271c3681SAlexis Hunt       ForVirtualBase = true;
503271c3681SAlexis Hunt       // fall-through
504271c3681SAlexis Hunt 
505271c3681SAlexis Hunt      case CXXConstructExpr::CK_NonVirtualBase:
506271c3681SAlexis Hunt       Type = Ctor_Base;
507271c3681SAlexis Hunt     }
508e11f9ce9SAnders Carlsson 
50927da15baSAnders Carlsson     // Call the constructor.
51061535005SDouglas Gregor     EmitCXXConstructorCall(CD, Type, ForVirtualBase, Delegating, Dest.getAddr(),
51127da15baSAnders Carlsson                            E->arg_begin(), E->arg_end());
51227da15baSAnders Carlsson   }
513e11f9ce9SAnders Carlsson }
51427da15baSAnders Carlsson 
515e988bdacSFariborz Jahanian void
516e988bdacSFariborz Jahanian CodeGenFunction::EmitSynthesizedCXXCopyCtor(llvm::Value *Dest,
517e988bdacSFariborz Jahanian                                             llvm::Value *Src,
51850198098SFariborz Jahanian                                             const Expr *Exp) {
5195d413781SJohn McCall   if (const ExprWithCleanups *E = dyn_cast<ExprWithCleanups>(Exp))
520e988bdacSFariborz Jahanian     Exp = E->getSubExpr();
521e988bdacSFariborz Jahanian   assert(isa<CXXConstructExpr>(Exp) &&
522e988bdacSFariborz Jahanian          "EmitSynthesizedCXXCopyCtor - unknown copy ctor expr");
523e988bdacSFariborz Jahanian   const CXXConstructExpr* E = cast<CXXConstructExpr>(Exp);
524e988bdacSFariborz Jahanian   const CXXConstructorDecl *CD = E->getConstructor();
525e988bdacSFariborz Jahanian   RunCleanupsScope Scope(*this);
526e988bdacSFariborz Jahanian 
527e988bdacSFariborz Jahanian   // If we require zero initialization before (or instead of) calling the
528e988bdacSFariborz Jahanian   // constructor, as can be the case with a non-user-provided default
529e988bdacSFariborz Jahanian   // constructor, emit the zero initialization now.
530e988bdacSFariborz Jahanian   // FIXME. Do I still need this for a copy ctor synthesis?
531e988bdacSFariborz Jahanian   if (E->requiresZeroInitialization())
532e988bdacSFariborz Jahanian     EmitNullInitialization(Dest, E->getType());
533e988bdacSFariborz Jahanian 
53499da11cfSChandler Carruth   assert(!getContext().getAsConstantArrayType(E->getType())
53599da11cfSChandler Carruth          && "EmitSynthesizedCXXCopyCtor - Copied-in Array");
536e988bdacSFariborz Jahanian   EmitSynthesizedCXXCopyCtorCall(CD, Dest, Src,
537e988bdacSFariborz Jahanian                                  E->arg_begin(), E->arg_end());
538e988bdacSFariborz Jahanian }
539e988bdacSFariborz Jahanian 
5408ed55a54SJohn McCall static CharUnits CalculateCookiePadding(CodeGenFunction &CGF,
5418ed55a54SJohn McCall                                         const CXXNewExpr *E) {
54221122cf6SAnders Carlsson   if (!E->isArray())
5433eb55cfeSKen Dyck     return CharUnits::Zero();
54421122cf6SAnders Carlsson 
5457ec4b434SJohn McCall   // No cookie is required if the operator new[] being used is the
5467ec4b434SJohn McCall   // reserved placement operator new[].
5477ec4b434SJohn McCall   if (E->getOperatorNew()->isReservedGlobalPlacementOperator())
5483eb55cfeSKen Dyck     return CharUnits::Zero();
549399f499fSAnders Carlsson 
550284c48ffSJohn McCall   return CGF.CGM.getCXXABI().GetArrayCookieSize(E);
55159486a2dSAnders Carlsson }
55259486a2dSAnders Carlsson 
553036f2f6bSJohn McCall static llvm::Value *EmitCXXNewAllocSize(CodeGenFunction &CGF,
554036f2f6bSJohn McCall                                         const CXXNewExpr *e,
555f862eb6aSSebastian Redl                                         unsigned minElements,
556036f2f6bSJohn McCall                                         llvm::Value *&numElements,
557036f2f6bSJohn McCall                                         llvm::Value *&sizeWithoutCookie) {
558036f2f6bSJohn McCall   QualType type = e->getAllocatedType();
55959486a2dSAnders Carlsson 
560036f2f6bSJohn McCall   if (!e->isArray()) {
561036f2f6bSJohn McCall     CharUnits typeSize = CGF.getContext().getTypeSizeInChars(type);
562036f2f6bSJohn McCall     sizeWithoutCookie
563036f2f6bSJohn McCall       = llvm::ConstantInt::get(CGF.SizeTy, typeSize.getQuantity());
564036f2f6bSJohn McCall     return sizeWithoutCookie;
56505fc5be3SDouglas Gregor   }
56659486a2dSAnders Carlsson 
567036f2f6bSJohn McCall   // The width of size_t.
568036f2f6bSJohn McCall   unsigned sizeWidth = CGF.SizeTy->getBitWidth();
569036f2f6bSJohn McCall 
5708ed55a54SJohn McCall   // Figure out the cookie size.
571036f2f6bSJohn McCall   llvm::APInt cookieSize(sizeWidth,
572036f2f6bSJohn McCall                          CalculateCookiePadding(CGF, e).getQuantity());
5738ed55a54SJohn McCall 
57459486a2dSAnders Carlsson   // Emit the array size expression.
5757648fb46SArgyrios Kyrtzidis   // We multiply the size of all dimensions for NumElements.
5767648fb46SArgyrios Kyrtzidis   // e.g for 'int[2][3]', ElemType is 'int' and NumElements is 6.
577036f2f6bSJohn McCall   numElements = CGF.EmitScalarExpr(e->getArraySize());
578036f2f6bSJohn McCall   assert(isa<llvm::IntegerType>(numElements->getType()));
5798ed55a54SJohn McCall 
580036f2f6bSJohn McCall   // The number of elements can be have an arbitrary integer type;
581036f2f6bSJohn McCall   // essentially, we need to multiply it by a constant factor, add a
582036f2f6bSJohn McCall   // cookie size, and verify that the result is representable as a
583036f2f6bSJohn McCall   // size_t.  That's just a gloss, though, and it's wrong in one
584036f2f6bSJohn McCall   // important way: if the count is negative, it's an error even if
585036f2f6bSJohn McCall   // the cookie size would bring the total size >= 0.
5866ab2fa8fSDouglas Gregor   bool isSigned
5876ab2fa8fSDouglas Gregor     = e->getArraySize()->getType()->isSignedIntegerOrEnumerationType();
5882192fe50SChris Lattner   llvm::IntegerType *numElementsType
589036f2f6bSJohn McCall     = cast<llvm::IntegerType>(numElements->getType());
590036f2f6bSJohn McCall   unsigned numElementsWidth = numElementsType->getBitWidth();
591036f2f6bSJohn McCall 
592036f2f6bSJohn McCall   // Compute the constant factor.
593036f2f6bSJohn McCall   llvm::APInt arraySizeMultiplier(sizeWidth, 1);
5947648fb46SArgyrios Kyrtzidis   while (const ConstantArrayType *CAT
595036f2f6bSJohn McCall              = CGF.getContext().getAsConstantArrayType(type)) {
596036f2f6bSJohn McCall     type = CAT->getElementType();
597036f2f6bSJohn McCall     arraySizeMultiplier *= CAT->getSize();
5987648fb46SArgyrios Kyrtzidis   }
59959486a2dSAnders Carlsson 
600036f2f6bSJohn McCall   CharUnits typeSize = CGF.getContext().getTypeSizeInChars(type);
601036f2f6bSJohn McCall   llvm::APInt typeSizeMultiplier(sizeWidth, typeSize.getQuantity());
602036f2f6bSJohn McCall   typeSizeMultiplier *= arraySizeMultiplier;
603036f2f6bSJohn McCall 
604036f2f6bSJohn McCall   // This will be a size_t.
605036f2f6bSJohn McCall   llvm::Value *size;
60632ac583dSChris Lattner 
60732ac583dSChris Lattner   // If someone is doing 'new int[42]' there is no need to do a dynamic check.
60832ac583dSChris Lattner   // Don't bloat the -O0 code.
609036f2f6bSJohn McCall   if (llvm::ConstantInt *numElementsC =
610036f2f6bSJohn McCall         dyn_cast<llvm::ConstantInt>(numElements)) {
611036f2f6bSJohn McCall     const llvm::APInt &count = numElementsC->getValue();
61232ac583dSChris Lattner 
613036f2f6bSJohn McCall     bool hasAnyOverflow = false;
61432ac583dSChris Lattner 
615036f2f6bSJohn McCall     // If 'count' was a negative number, it's an overflow.
616036f2f6bSJohn McCall     if (isSigned && count.isNegative())
617036f2f6bSJohn McCall       hasAnyOverflow = true;
6188ed55a54SJohn McCall 
619036f2f6bSJohn McCall     // We want to do all this arithmetic in size_t.  If numElements is
620036f2f6bSJohn McCall     // wider than that, check whether it's already too big, and if so,
621036f2f6bSJohn McCall     // overflow.
622036f2f6bSJohn McCall     else if (numElementsWidth > sizeWidth &&
623036f2f6bSJohn McCall              numElementsWidth - sizeWidth > count.countLeadingZeros())
624036f2f6bSJohn McCall       hasAnyOverflow = true;
625036f2f6bSJohn McCall 
626036f2f6bSJohn McCall     // Okay, compute a count at the right width.
627036f2f6bSJohn McCall     llvm::APInt adjustedCount = count.zextOrTrunc(sizeWidth);
628036f2f6bSJohn McCall 
629f862eb6aSSebastian Redl     // If there is a brace-initializer, we cannot allocate fewer elements than
630f862eb6aSSebastian Redl     // there are initializers. If we do, that's treated like an overflow.
631f862eb6aSSebastian Redl     if (adjustedCount.ult(minElements))
632f862eb6aSSebastian Redl       hasAnyOverflow = true;
633f862eb6aSSebastian Redl 
634036f2f6bSJohn McCall     // Scale numElements by that.  This might overflow, but we don't
635036f2f6bSJohn McCall     // care because it only overflows if allocationSize does, too, and
636036f2f6bSJohn McCall     // if that overflows then we shouldn't use this.
637036f2f6bSJohn McCall     numElements = llvm::ConstantInt::get(CGF.SizeTy,
638036f2f6bSJohn McCall                                          adjustedCount * arraySizeMultiplier);
639036f2f6bSJohn McCall 
640036f2f6bSJohn McCall     // Compute the size before cookie, and track whether it overflowed.
641036f2f6bSJohn McCall     bool overflow;
642036f2f6bSJohn McCall     llvm::APInt allocationSize
643036f2f6bSJohn McCall       = adjustedCount.umul_ov(typeSizeMultiplier, overflow);
644036f2f6bSJohn McCall     hasAnyOverflow |= overflow;
645036f2f6bSJohn McCall 
646036f2f6bSJohn McCall     // Add in the cookie, and check whether it's overflowed.
647036f2f6bSJohn McCall     if (cookieSize != 0) {
648036f2f6bSJohn McCall       // Save the current size without a cookie.  This shouldn't be
649036f2f6bSJohn McCall       // used if there was overflow.
650036f2f6bSJohn McCall       sizeWithoutCookie = llvm::ConstantInt::get(CGF.SizeTy, allocationSize);
651036f2f6bSJohn McCall 
652036f2f6bSJohn McCall       allocationSize = allocationSize.uadd_ov(cookieSize, overflow);
653036f2f6bSJohn McCall       hasAnyOverflow |= overflow;
6548ed55a54SJohn McCall     }
6558ed55a54SJohn McCall 
656036f2f6bSJohn McCall     // On overflow, produce a -1 so operator new will fail.
657036f2f6bSJohn McCall     if (hasAnyOverflow) {
658036f2f6bSJohn McCall       size = llvm::Constant::getAllOnesValue(CGF.SizeTy);
65932ac583dSChris Lattner     } else {
660036f2f6bSJohn McCall       size = llvm::ConstantInt::get(CGF.SizeTy, allocationSize);
66132ac583dSChris Lattner     }
66232ac583dSChris Lattner 
663036f2f6bSJohn McCall   // Otherwise, we might need to use the overflow intrinsics.
6648ed55a54SJohn McCall   } else {
665f862eb6aSSebastian Redl     // There are up to five conditions we need to test for:
666036f2f6bSJohn McCall     // 1) if isSigned, we need to check whether numElements is negative;
667036f2f6bSJohn McCall     // 2) if numElementsWidth > sizeWidth, we need to check whether
668036f2f6bSJohn McCall     //   numElements is larger than something representable in size_t;
669f862eb6aSSebastian Redl     // 3) if minElements > 0, we need to check whether numElements is smaller
670f862eb6aSSebastian Redl     //    than that.
671f862eb6aSSebastian Redl     // 4) we need to compute
672036f2f6bSJohn McCall     //      sizeWithoutCookie := numElements * typeSizeMultiplier
673036f2f6bSJohn McCall     //    and check whether it overflows; and
674f862eb6aSSebastian Redl     // 5) if we need a cookie, we need to compute
675036f2f6bSJohn McCall     //      size := sizeWithoutCookie + cookieSize
676036f2f6bSJohn McCall     //    and check whether it overflows.
6778ed55a54SJohn McCall 
678036f2f6bSJohn McCall     llvm::Value *hasOverflow = 0;
6798ed55a54SJohn McCall 
680036f2f6bSJohn McCall     // If numElementsWidth > sizeWidth, then one way or another, we're
681036f2f6bSJohn McCall     // going to have to do a comparison for (2), and this happens to
682036f2f6bSJohn McCall     // take care of (1), too.
683036f2f6bSJohn McCall     if (numElementsWidth > sizeWidth) {
684036f2f6bSJohn McCall       llvm::APInt threshold(numElementsWidth, 1);
685036f2f6bSJohn McCall       threshold <<= sizeWidth;
6868ed55a54SJohn McCall 
687036f2f6bSJohn McCall       llvm::Value *thresholdV
688036f2f6bSJohn McCall         = llvm::ConstantInt::get(numElementsType, threshold);
689036f2f6bSJohn McCall 
690036f2f6bSJohn McCall       hasOverflow = CGF.Builder.CreateICmpUGE(numElements, thresholdV);
691036f2f6bSJohn McCall       numElements = CGF.Builder.CreateTrunc(numElements, CGF.SizeTy);
692036f2f6bSJohn McCall 
693036f2f6bSJohn McCall     // Otherwise, if we're signed, we want to sext up to size_t.
694036f2f6bSJohn McCall     } else if (isSigned) {
695036f2f6bSJohn McCall       if (numElementsWidth < sizeWidth)
696036f2f6bSJohn McCall         numElements = CGF.Builder.CreateSExt(numElements, CGF.SizeTy);
697036f2f6bSJohn McCall 
698036f2f6bSJohn McCall       // If there's a non-1 type size multiplier, then we can do the
699036f2f6bSJohn McCall       // signedness check at the same time as we do the multiply
700036f2f6bSJohn McCall       // because a negative number times anything will cause an
701f862eb6aSSebastian Redl       // unsigned overflow.  Otherwise, we have to do it here. But at least
702f862eb6aSSebastian Redl       // in this case, we can subsume the >= minElements check.
703036f2f6bSJohn McCall       if (typeSizeMultiplier == 1)
704036f2f6bSJohn McCall         hasOverflow = CGF.Builder.CreateICmpSLT(numElements,
705f862eb6aSSebastian Redl                               llvm::ConstantInt::get(CGF.SizeTy, minElements));
706036f2f6bSJohn McCall 
707036f2f6bSJohn McCall     // Otherwise, zext up to size_t if necessary.
708036f2f6bSJohn McCall     } else if (numElementsWidth < sizeWidth) {
709036f2f6bSJohn McCall       numElements = CGF.Builder.CreateZExt(numElements, CGF.SizeTy);
710036f2f6bSJohn McCall     }
711036f2f6bSJohn McCall 
712036f2f6bSJohn McCall     assert(numElements->getType() == CGF.SizeTy);
713036f2f6bSJohn McCall 
714f862eb6aSSebastian Redl     if (minElements) {
715f862eb6aSSebastian Redl       // Don't allow allocation of fewer elements than we have initializers.
716f862eb6aSSebastian Redl       if (!hasOverflow) {
717f862eb6aSSebastian Redl         hasOverflow = CGF.Builder.CreateICmpULT(numElements,
718f862eb6aSSebastian Redl                               llvm::ConstantInt::get(CGF.SizeTy, minElements));
719f862eb6aSSebastian Redl       } else if (numElementsWidth > sizeWidth) {
720f862eb6aSSebastian Redl         // The other existing overflow subsumes this check.
721f862eb6aSSebastian Redl         // We do an unsigned comparison, since any signed value < -1 is
722f862eb6aSSebastian Redl         // taken care of either above or below.
723f862eb6aSSebastian Redl         hasOverflow = CGF.Builder.CreateOr(hasOverflow,
724f862eb6aSSebastian Redl                           CGF.Builder.CreateICmpULT(numElements,
725f862eb6aSSebastian Redl                               llvm::ConstantInt::get(CGF.SizeTy, minElements)));
726f862eb6aSSebastian Redl       }
727f862eb6aSSebastian Redl     }
728f862eb6aSSebastian Redl 
729036f2f6bSJohn McCall     size = numElements;
730036f2f6bSJohn McCall 
731036f2f6bSJohn McCall     // Multiply by the type size if necessary.  This multiplier
732036f2f6bSJohn McCall     // includes all the factors for nested arrays.
7338ed55a54SJohn McCall     //
734036f2f6bSJohn McCall     // This step also causes numElements to be scaled up by the
735036f2f6bSJohn McCall     // nested-array factor if necessary.  Overflow on this computation
736036f2f6bSJohn McCall     // can be ignored because the result shouldn't be used if
737036f2f6bSJohn McCall     // allocation fails.
738036f2f6bSJohn McCall     if (typeSizeMultiplier != 1) {
739036f2f6bSJohn McCall       llvm::Value *umul_with_overflow
7408d375cefSBenjamin Kramer         = CGF.CGM.getIntrinsic(llvm::Intrinsic::umul_with_overflow, CGF.SizeTy);
7418ed55a54SJohn McCall 
742036f2f6bSJohn McCall       llvm::Value *tsmV =
743036f2f6bSJohn McCall         llvm::ConstantInt::get(CGF.SizeTy, typeSizeMultiplier);
744036f2f6bSJohn McCall       llvm::Value *result =
745036f2f6bSJohn McCall         CGF.Builder.CreateCall2(umul_with_overflow, size, tsmV);
7468ed55a54SJohn McCall 
747036f2f6bSJohn McCall       llvm::Value *overflowed = CGF.Builder.CreateExtractValue(result, 1);
748036f2f6bSJohn McCall       if (hasOverflow)
749036f2f6bSJohn McCall         hasOverflow = CGF.Builder.CreateOr(hasOverflow, overflowed);
7508ed55a54SJohn McCall       else
751036f2f6bSJohn McCall         hasOverflow = overflowed;
75259486a2dSAnders Carlsson 
753036f2f6bSJohn McCall       size = CGF.Builder.CreateExtractValue(result, 0);
754036f2f6bSJohn McCall 
755036f2f6bSJohn McCall       // Also scale up numElements by the array size multiplier.
756036f2f6bSJohn McCall       if (arraySizeMultiplier != 1) {
757036f2f6bSJohn McCall         // If the base element type size is 1, then we can re-use the
758036f2f6bSJohn McCall         // multiply we just did.
759036f2f6bSJohn McCall         if (typeSize.isOne()) {
760036f2f6bSJohn McCall           assert(arraySizeMultiplier == typeSizeMultiplier);
761036f2f6bSJohn McCall           numElements = size;
762036f2f6bSJohn McCall 
763036f2f6bSJohn McCall         // Otherwise we need a separate multiply.
764036f2f6bSJohn McCall         } else {
765036f2f6bSJohn McCall           llvm::Value *asmV =
766036f2f6bSJohn McCall             llvm::ConstantInt::get(CGF.SizeTy, arraySizeMultiplier);
767036f2f6bSJohn McCall           numElements = CGF.Builder.CreateMul(numElements, asmV);
768036f2f6bSJohn McCall         }
769036f2f6bSJohn McCall       }
770036f2f6bSJohn McCall     } else {
771036f2f6bSJohn McCall       // numElements doesn't need to be scaled.
772036f2f6bSJohn McCall       assert(arraySizeMultiplier == 1);
773036f2f6bSJohn McCall     }
774036f2f6bSJohn McCall 
775036f2f6bSJohn McCall     // Add in the cookie size if necessary.
776036f2f6bSJohn McCall     if (cookieSize != 0) {
777036f2f6bSJohn McCall       sizeWithoutCookie = size;
778036f2f6bSJohn McCall 
779036f2f6bSJohn McCall       llvm::Value *uadd_with_overflow
7808d375cefSBenjamin Kramer         = CGF.CGM.getIntrinsic(llvm::Intrinsic::uadd_with_overflow, CGF.SizeTy);
781036f2f6bSJohn McCall 
782036f2f6bSJohn McCall       llvm::Value *cookieSizeV = llvm::ConstantInt::get(CGF.SizeTy, cookieSize);
783036f2f6bSJohn McCall       llvm::Value *result =
784036f2f6bSJohn McCall         CGF.Builder.CreateCall2(uadd_with_overflow, size, cookieSizeV);
785036f2f6bSJohn McCall 
786036f2f6bSJohn McCall       llvm::Value *overflowed = CGF.Builder.CreateExtractValue(result, 1);
787036f2f6bSJohn McCall       if (hasOverflow)
788036f2f6bSJohn McCall         hasOverflow = CGF.Builder.CreateOr(hasOverflow, overflowed);
789036f2f6bSJohn McCall       else
790036f2f6bSJohn McCall         hasOverflow = overflowed;
791036f2f6bSJohn McCall 
792036f2f6bSJohn McCall       size = CGF.Builder.CreateExtractValue(result, 0);
793036f2f6bSJohn McCall     }
794036f2f6bSJohn McCall 
795036f2f6bSJohn McCall     // If we had any possibility of dynamic overflow, make a select to
796036f2f6bSJohn McCall     // overwrite 'size' with an all-ones value, which should cause
797036f2f6bSJohn McCall     // operator new to throw.
798036f2f6bSJohn McCall     if (hasOverflow)
799036f2f6bSJohn McCall       size = CGF.Builder.CreateSelect(hasOverflow,
800036f2f6bSJohn McCall                                  llvm::Constant::getAllOnesValue(CGF.SizeTy),
801036f2f6bSJohn McCall                                       size);
802036f2f6bSJohn McCall   }
803036f2f6bSJohn McCall 
804036f2f6bSJohn McCall   if (cookieSize == 0)
805036f2f6bSJohn McCall     sizeWithoutCookie = size;
806036f2f6bSJohn McCall   else
807036f2f6bSJohn McCall     assert(sizeWithoutCookie && "didn't set sizeWithoutCookie?");
808036f2f6bSJohn McCall 
809036f2f6bSJohn McCall   return size;
81059486a2dSAnders Carlsson }
81159486a2dSAnders Carlsson 
812f862eb6aSSebastian Redl static void StoreAnyExprIntoOneUnit(CodeGenFunction &CGF, const Expr *Init,
813f862eb6aSSebastian Redl                                     QualType AllocType, llvm::Value *NewPtr) {
814d5202e09SFariborz Jahanian 
81538cd36dbSEli Friedman   CharUnits Alignment = CGF.getContext().getTypeAlignInChars(AllocType);
816d5202e09SFariborz Jahanian   if (!CGF.hasAggregateLLVMType(AllocType))
81738cd36dbSEli Friedman     CGF.EmitScalarInit(Init, 0, CGF.MakeAddrLValue(NewPtr, AllocType,
818a0544d6fSEli Friedman                                                    Alignment),
8191553b190SJohn McCall                        false);
820d5202e09SFariborz Jahanian   else if (AllocType->isAnyComplexType())
821d5202e09SFariborz Jahanian     CGF.EmitComplexExprIntoAddr(Init, NewPtr,
822d5202e09SFariborz Jahanian                                 AllocType.isVolatileQualified());
8237a626f63SJohn McCall   else {
8247a626f63SJohn McCall     AggValueSlot Slot
825c1d85b93SEli Friedman       = AggValueSlot::forAddr(NewPtr, Alignment, AllocType.getQualifiers(),
8268d6fc958SJohn McCall                               AggValueSlot::IsDestructed,
82746759f4fSJohn McCall                               AggValueSlot::DoesNotNeedGCBarriers,
828615ed1a3SChad Rosier                               AggValueSlot::IsNotAliased);
8297a626f63SJohn McCall     CGF.EmitAggExpr(Init, Slot);
830d026dc49SSebastian Redl 
831d026dc49SSebastian Redl     CGF.MaybeEmitStdInitializerListCleanup(NewPtr, Init);
8327a626f63SJohn McCall   }
833d5202e09SFariborz Jahanian }
834d5202e09SFariborz Jahanian 
835d5202e09SFariborz Jahanian void
836d5202e09SFariborz Jahanian CodeGenFunction::EmitNewArrayInitializer(const CXXNewExpr *E,
83799210dc9SJohn McCall                                          QualType elementType,
83899210dc9SJohn McCall                                          llvm::Value *beginPtr,
83999210dc9SJohn McCall                                          llvm::Value *numElements) {
8406047f07eSSebastian Redl   if (!E->hasInitializer())
8416047f07eSSebastian Redl     return; // We have a POD type.
842b66b08efSFariborz Jahanian 
843f862eb6aSSebastian Redl   llvm::Value *explicitPtr = beginPtr;
84499210dc9SJohn McCall   // Find the end of the array, hoisted out of the loop.
84599210dc9SJohn McCall   llvm::Value *endPtr =
84699210dc9SJohn McCall     Builder.CreateInBoundsGEP(beginPtr, numElements, "array.end");
847d5202e09SFariborz Jahanian 
848f862eb6aSSebastian Redl   unsigned initializerElements = 0;
849f862eb6aSSebastian Redl 
850f862eb6aSSebastian Redl   const Expr *Init = E->getInitializer();
851f62290a1SChad Rosier   llvm::AllocaInst *endOfInit = 0;
852f62290a1SChad Rosier   QualType::DestructionKind dtorKind = elementType.isDestructedType();
853f62290a1SChad Rosier   EHScopeStack::stable_iterator cleanup;
854f62290a1SChad Rosier   llvm::Instruction *cleanupDominator = 0;
855f862eb6aSSebastian Redl   // If the initializer is an initializer list, first do the explicit elements.
856f862eb6aSSebastian Redl   if (const InitListExpr *ILE = dyn_cast<InitListExpr>(Init)) {
857f862eb6aSSebastian Redl     initializerElements = ILE->getNumInits();
858f62290a1SChad Rosier 
859f62290a1SChad Rosier     // Enter a partial-destruction cleanup if necessary.
860f62290a1SChad Rosier     if (needsEHCleanup(dtorKind)) {
861f62290a1SChad Rosier       // In principle we could tell the cleanup where we are more
862f62290a1SChad Rosier       // directly, but the control flow can get so varied here that it
863f62290a1SChad Rosier       // would actually be quite complex.  Therefore we go through an
864f62290a1SChad Rosier       // alloca.
865f62290a1SChad Rosier       endOfInit = CreateTempAlloca(beginPtr->getType(), "array.endOfInit");
866f62290a1SChad Rosier       cleanupDominator = Builder.CreateStore(beginPtr, endOfInit);
867f62290a1SChad Rosier       pushIrregularPartialArrayCleanup(beginPtr, endOfInit, elementType,
868f62290a1SChad Rosier                                        getDestroyer(dtorKind));
869f62290a1SChad Rosier       cleanup = EHStack.stable_begin();
870f62290a1SChad Rosier     }
871f62290a1SChad Rosier 
872f862eb6aSSebastian Redl     for (unsigned i = 0, e = ILE->getNumInits(); i != e; ++i) {
873f62290a1SChad Rosier       // Tell the cleanup that it needs to destroy up to this
874f62290a1SChad Rosier       // element.  TODO: some of these stores can be trivially
875f62290a1SChad Rosier       // observed to be unnecessary.
876f62290a1SChad Rosier       if (endOfInit) Builder.CreateStore(explicitPtr, endOfInit);
877f862eb6aSSebastian Redl       StoreAnyExprIntoOneUnit(*this, ILE->getInit(i), elementType, explicitPtr);
878f862eb6aSSebastian Redl       explicitPtr =Builder.CreateConstGEP1_32(explicitPtr, 1, "array.exp.next");
879f862eb6aSSebastian Redl     }
880f862eb6aSSebastian Redl 
881f862eb6aSSebastian Redl     // The remaining elements are filled with the array filler expression.
882f862eb6aSSebastian Redl     Init = ILE->getArrayFiller();
883f862eb6aSSebastian Redl   }
884f862eb6aSSebastian Redl 
88599210dc9SJohn McCall   // Create the continuation block.
88699210dc9SJohn McCall   llvm::BasicBlock *contBB = createBasicBlock("new.loop.end");
887d5202e09SFariborz Jahanian 
888f862eb6aSSebastian Redl   // If the number of elements isn't constant, we have to now check if there is
889f862eb6aSSebastian Redl   // anything left to initialize.
890f862eb6aSSebastian Redl   if (llvm::ConstantInt *constNum = dyn_cast<llvm::ConstantInt>(numElements)) {
891f862eb6aSSebastian Redl     // If all elements have already been initialized, skip the whole loop.
892f62290a1SChad Rosier     if (constNum->getZExtValue() <= initializerElements) {
893f62290a1SChad Rosier       // If there was a cleanup, deactivate it.
894f62290a1SChad Rosier       if (cleanupDominator)
89576bb5cabSDmitri Gribenko         DeactivateCleanupBlock(cleanup, cleanupDominator);
896f62290a1SChad Rosier       return;
897f62290a1SChad Rosier     }
898f862eb6aSSebastian Redl   } else {
89999210dc9SJohn McCall     llvm::BasicBlock *nonEmptyBB = createBasicBlock("new.loop.nonempty");
900f862eb6aSSebastian Redl     llvm::Value *isEmpty = Builder.CreateICmpEQ(explicitPtr, endPtr,
90199210dc9SJohn McCall                                                 "array.isempty");
90299210dc9SJohn McCall     Builder.CreateCondBr(isEmpty, contBB, nonEmptyBB);
90399210dc9SJohn McCall     EmitBlock(nonEmptyBB);
90499210dc9SJohn McCall   }
905d5202e09SFariborz Jahanian 
90699210dc9SJohn McCall   // Enter the loop.
90799210dc9SJohn McCall   llvm::BasicBlock *entryBB = Builder.GetInsertBlock();
90899210dc9SJohn McCall   llvm::BasicBlock *loopBB = createBasicBlock("new.loop");
909d5202e09SFariborz Jahanian 
91099210dc9SJohn McCall   EmitBlock(loopBB);
911d5202e09SFariborz Jahanian 
91299210dc9SJohn McCall   // Set up the current-element phi.
91399210dc9SJohn McCall   llvm::PHINode *curPtr =
914f862eb6aSSebastian Redl     Builder.CreatePHI(explicitPtr->getType(), 2, "array.cur");
915f862eb6aSSebastian Redl   curPtr->addIncoming(explicitPtr, entryBB);
916d5202e09SFariborz Jahanian 
917f62290a1SChad Rosier   // Store the new cleanup position for irregular cleanups.
918f62290a1SChad Rosier   if (endOfInit) Builder.CreateStore(curPtr, endOfInit);
919f62290a1SChad Rosier 
92099210dc9SJohn McCall   // Enter a partial-destruction cleanup if necessary.
921f62290a1SChad Rosier   if (!cleanupDominator && needsEHCleanup(dtorKind)) {
92299210dc9SJohn McCall     pushRegularPartialArrayCleanup(beginPtr, curPtr, elementType,
92399210dc9SJohn McCall                                    getDestroyer(dtorKind));
92499210dc9SJohn McCall     cleanup = EHStack.stable_begin();
925f4beacd0SJohn McCall     cleanupDominator = Builder.CreateUnreachable();
92699210dc9SJohn McCall   }
927d5202e09SFariborz Jahanian 
92899210dc9SJohn McCall   // Emit the initializer into this element.
929f862eb6aSSebastian Redl   StoreAnyExprIntoOneUnit(*this, Init, E->getAllocatedType(), curPtr);
930d5202e09SFariborz Jahanian 
93199210dc9SJohn McCall   // Leave the cleanup if we entered one.
932de6a86b4SEli Friedman   if (cleanupDominator) {
933f4beacd0SJohn McCall     DeactivateCleanupBlock(cleanup, cleanupDominator);
934f4beacd0SJohn McCall     cleanupDominator->eraseFromParent();
935f4beacd0SJohn McCall   }
936d5202e09SFariborz Jahanian 
93799210dc9SJohn McCall   // Advance to the next element.
93899210dc9SJohn McCall   llvm::Value *nextPtr = Builder.CreateConstGEP1_32(curPtr, 1, "array.next");
93999210dc9SJohn McCall 
94099210dc9SJohn McCall   // Check whether we've gotten to the end of the array and, if so,
94199210dc9SJohn McCall   // exit the loop.
94299210dc9SJohn McCall   llvm::Value *isEnd = Builder.CreateICmpEQ(nextPtr, endPtr, "array.atend");
94399210dc9SJohn McCall   Builder.CreateCondBr(isEnd, contBB, loopBB);
94499210dc9SJohn McCall   curPtr->addIncoming(nextPtr, Builder.GetInsertBlock());
94599210dc9SJohn McCall 
94699210dc9SJohn McCall   EmitBlock(contBB);
947d5202e09SFariborz Jahanian }
948d5202e09SFariborz Jahanian 
94905fc5be3SDouglas Gregor static void EmitZeroMemSet(CodeGenFunction &CGF, QualType T,
95005fc5be3SDouglas Gregor                            llvm::Value *NewPtr, llvm::Value *Size) {
951ad7c5c16SJohn McCall   CGF.EmitCastToVoidPtr(NewPtr);
952705ba07eSKen Dyck   CharUnits Alignment = CGF.getContext().getTypeAlignInChars(T);
953acc6b4e2SBenjamin Kramer   CGF.Builder.CreateMemSet(NewPtr, CGF.Builder.getInt8(0), Size,
954705ba07eSKen Dyck                            Alignment.getQuantity(), false);
95505fc5be3SDouglas Gregor }
95605fc5be3SDouglas Gregor 
95759486a2dSAnders Carlsson static void EmitNewInitializer(CodeGenFunction &CGF, const CXXNewExpr *E,
95899210dc9SJohn McCall                                QualType ElementType,
95959486a2dSAnders Carlsson                                llvm::Value *NewPtr,
96005fc5be3SDouglas Gregor                                llvm::Value *NumElements,
96105fc5be3SDouglas Gregor                                llvm::Value *AllocSizeWithoutCookie) {
9626047f07eSSebastian Redl   const Expr *Init = E->getInitializer();
9633a202f60SAnders Carlsson   if (E->isArray()) {
9646047f07eSSebastian Redl     if (const CXXConstructExpr *CCE = dyn_cast_or_null<CXXConstructExpr>(Init)){
9656047f07eSSebastian Redl       CXXConstructorDecl *Ctor = CCE->getConstructor();
966d153103cSDouglas Gregor       if (Ctor->isTrivial()) {
96705fc5be3SDouglas Gregor         // If new expression did not specify value-initialization, then there
96805fc5be3SDouglas Gregor         // is no initialization.
9696047f07eSSebastian Redl         if (!CCE->requiresZeroInitialization() || Ctor->getParent()->isEmpty())
97005fc5be3SDouglas Gregor           return;
97105fc5be3SDouglas Gregor 
97299210dc9SJohn McCall         if (CGF.CGM.getTypes().isZeroInitializable(ElementType)) {
97305fc5be3SDouglas Gregor           // Optimization: since zero initialization will just set the memory
97405fc5be3SDouglas Gregor           // to all zeroes, generate a single memset to do it in one shot.
97599210dc9SJohn McCall           EmitZeroMemSet(CGF, ElementType, NewPtr, AllocSizeWithoutCookie);
9763a202f60SAnders Carlsson           return;
9773a202f60SAnders Carlsson         }
97805fc5be3SDouglas Gregor       }
97905fc5be3SDouglas Gregor 
98005fc5be3SDouglas Gregor       CGF.EmitCXXAggrConstructorCall(Ctor, NumElements, NewPtr,
9816047f07eSSebastian Redl                                      CCE->arg_begin(),  CCE->arg_end(),
98248ddcf2cSEli Friedman                                      CCE->requiresZeroInitialization());
98305fc5be3SDouglas Gregor       return;
9846047f07eSSebastian Redl     } else if (Init && isa<ImplicitValueInitExpr>(Init) &&
985de6a86b4SEli Friedman                CGF.CGM.getTypes().isZeroInitializable(ElementType)) {
98605fc5be3SDouglas Gregor       // Optimization: since zero initialization will just set the memory
98705fc5be3SDouglas Gregor       // to all zeroes, generate a single memset to do it in one shot.
98899210dc9SJohn McCall       EmitZeroMemSet(CGF, ElementType, NewPtr, AllocSizeWithoutCookie);
98905fc5be3SDouglas Gregor       return;
9906047f07eSSebastian Redl     }
99199210dc9SJohn McCall     CGF.EmitNewArrayInitializer(E, ElementType, NewPtr, NumElements);
992d5202e09SFariborz Jahanian     return;
993d040e6b2SAnders Carlsson   }
99459486a2dSAnders Carlsson 
9956047f07eSSebastian Redl   if (!Init)
996b66b08efSFariborz Jahanian     return;
99759486a2dSAnders Carlsson 
998f862eb6aSSebastian Redl   StoreAnyExprIntoOneUnit(CGF, Init, E->getAllocatedType(), NewPtr);
99959486a2dSAnders Carlsson }
100059486a2dSAnders Carlsson 
1001824c2f53SJohn McCall namespace {
1002824c2f53SJohn McCall   /// A cleanup to call the given 'operator delete' function upon
1003824c2f53SJohn McCall   /// abnormal exit from a new expression.
1004824c2f53SJohn McCall   class CallDeleteDuringNew : public EHScopeStack::Cleanup {
1005824c2f53SJohn McCall     size_t NumPlacementArgs;
1006824c2f53SJohn McCall     const FunctionDecl *OperatorDelete;
1007824c2f53SJohn McCall     llvm::Value *Ptr;
1008824c2f53SJohn McCall     llvm::Value *AllocSize;
1009824c2f53SJohn McCall 
1010824c2f53SJohn McCall     RValue *getPlacementArgs() { return reinterpret_cast<RValue*>(this+1); }
1011824c2f53SJohn McCall 
1012824c2f53SJohn McCall   public:
1013824c2f53SJohn McCall     static size_t getExtraSize(size_t NumPlacementArgs) {
1014824c2f53SJohn McCall       return NumPlacementArgs * sizeof(RValue);
1015824c2f53SJohn McCall     }
1016824c2f53SJohn McCall 
1017824c2f53SJohn McCall     CallDeleteDuringNew(size_t NumPlacementArgs,
1018824c2f53SJohn McCall                         const FunctionDecl *OperatorDelete,
1019824c2f53SJohn McCall                         llvm::Value *Ptr,
1020824c2f53SJohn McCall                         llvm::Value *AllocSize)
1021824c2f53SJohn McCall       : NumPlacementArgs(NumPlacementArgs), OperatorDelete(OperatorDelete),
1022824c2f53SJohn McCall         Ptr(Ptr), AllocSize(AllocSize) {}
1023824c2f53SJohn McCall 
1024824c2f53SJohn McCall     void setPlacementArg(unsigned I, RValue Arg) {
1025824c2f53SJohn McCall       assert(I < NumPlacementArgs && "index out of range");
1026824c2f53SJohn McCall       getPlacementArgs()[I] = Arg;
1027824c2f53SJohn McCall     }
1028824c2f53SJohn McCall 
102930317fdaSJohn McCall     void Emit(CodeGenFunction &CGF, Flags flags) {
1030824c2f53SJohn McCall       const FunctionProtoType *FPT
1031824c2f53SJohn McCall         = OperatorDelete->getType()->getAs<FunctionProtoType>();
1032824c2f53SJohn McCall       assert(FPT->getNumArgs() == NumPlacementArgs + 1 ||
1033d441b1e6SJohn McCall              (FPT->getNumArgs() == 2 && NumPlacementArgs == 0));
1034824c2f53SJohn McCall 
1035824c2f53SJohn McCall       CallArgList DeleteArgs;
1036824c2f53SJohn McCall 
1037824c2f53SJohn McCall       // The first argument is always a void*.
1038824c2f53SJohn McCall       FunctionProtoType::arg_type_iterator AI = FPT->arg_type_begin();
103943dca6a8SEli Friedman       DeleteArgs.add(RValue::get(Ptr), *AI++);
1040824c2f53SJohn McCall 
1041824c2f53SJohn McCall       // A member 'operator delete' can take an extra 'size_t' argument.
1042824c2f53SJohn McCall       if (FPT->getNumArgs() == NumPlacementArgs + 2)
104343dca6a8SEli Friedman         DeleteArgs.add(RValue::get(AllocSize), *AI++);
1044824c2f53SJohn McCall 
1045824c2f53SJohn McCall       // Pass the rest of the arguments, which must match exactly.
1046824c2f53SJohn McCall       for (unsigned I = 0; I != NumPlacementArgs; ++I)
104743dca6a8SEli Friedman         DeleteArgs.add(getPlacementArgs()[I], *AI++);
1048824c2f53SJohn McCall 
1049824c2f53SJohn McCall       // Call 'operator delete'.
10508dda7b27SJohn McCall       CGF.EmitCall(CGF.CGM.getTypes().arrangeFreeFunctionCall(DeleteArgs, FPT),
1051824c2f53SJohn McCall                    CGF.CGM.GetAddrOfFunction(OperatorDelete),
1052824c2f53SJohn McCall                    ReturnValueSlot(), DeleteArgs, OperatorDelete);
1053824c2f53SJohn McCall     }
1054824c2f53SJohn McCall   };
10557f9c92a9SJohn McCall 
10567f9c92a9SJohn McCall   /// A cleanup to call the given 'operator delete' function upon
10577f9c92a9SJohn McCall   /// abnormal exit from a new expression when the new expression is
10587f9c92a9SJohn McCall   /// conditional.
10597f9c92a9SJohn McCall   class CallDeleteDuringConditionalNew : public EHScopeStack::Cleanup {
10607f9c92a9SJohn McCall     size_t NumPlacementArgs;
10617f9c92a9SJohn McCall     const FunctionDecl *OperatorDelete;
1062cb5f77f0SJohn McCall     DominatingValue<RValue>::saved_type Ptr;
1063cb5f77f0SJohn McCall     DominatingValue<RValue>::saved_type AllocSize;
10647f9c92a9SJohn McCall 
1065cb5f77f0SJohn McCall     DominatingValue<RValue>::saved_type *getPlacementArgs() {
1066cb5f77f0SJohn McCall       return reinterpret_cast<DominatingValue<RValue>::saved_type*>(this+1);
10677f9c92a9SJohn McCall     }
10687f9c92a9SJohn McCall 
10697f9c92a9SJohn McCall   public:
10707f9c92a9SJohn McCall     static size_t getExtraSize(size_t NumPlacementArgs) {
1071cb5f77f0SJohn McCall       return NumPlacementArgs * sizeof(DominatingValue<RValue>::saved_type);
10727f9c92a9SJohn McCall     }
10737f9c92a9SJohn McCall 
10747f9c92a9SJohn McCall     CallDeleteDuringConditionalNew(size_t NumPlacementArgs,
10757f9c92a9SJohn McCall                                    const FunctionDecl *OperatorDelete,
1076cb5f77f0SJohn McCall                                    DominatingValue<RValue>::saved_type Ptr,
1077cb5f77f0SJohn McCall                               DominatingValue<RValue>::saved_type AllocSize)
10787f9c92a9SJohn McCall       : NumPlacementArgs(NumPlacementArgs), OperatorDelete(OperatorDelete),
10797f9c92a9SJohn McCall         Ptr(Ptr), AllocSize(AllocSize) {}
10807f9c92a9SJohn McCall 
1081cb5f77f0SJohn McCall     void setPlacementArg(unsigned I, DominatingValue<RValue>::saved_type Arg) {
10827f9c92a9SJohn McCall       assert(I < NumPlacementArgs && "index out of range");
10837f9c92a9SJohn McCall       getPlacementArgs()[I] = Arg;
10847f9c92a9SJohn McCall     }
10857f9c92a9SJohn McCall 
108630317fdaSJohn McCall     void Emit(CodeGenFunction &CGF, Flags flags) {
10877f9c92a9SJohn McCall       const FunctionProtoType *FPT
10887f9c92a9SJohn McCall         = OperatorDelete->getType()->getAs<FunctionProtoType>();
10897f9c92a9SJohn McCall       assert(FPT->getNumArgs() == NumPlacementArgs + 1 ||
10907f9c92a9SJohn McCall              (FPT->getNumArgs() == 2 && NumPlacementArgs == 0));
10917f9c92a9SJohn McCall 
10927f9c92a9SJohn McCall       CallArgList DeleteArgs;
10937f9c92a9SJohn McCall 
10947f9c92a9SJohn McCall       // The first argument is always a void*.
10957f9c92a9SJohn McCall       FunctionProtoType::arg_type_iterator AI = FPT->arg_type_begin();
109643dca6a8SEli Friedman       DeleteArgs.add(Ptr.restore(CGF), *AI++);
10977f9c92a9SJohn McCall 
10987f9c92a9SJohn McCall       // A member 'operator delete' can take an extra 'size_t' argument.
10997f9c92a9SJohn McCall       if (FPT->getNumArgs() == NumPlacementArgs + 2) {
1100cb5f77f0SJohn McCall         RValue RV = AllocSize.restore(CGF);
110143dca6a8SEli Friedman         DeleteArgs.add(RV, *AI++);
11027f9c92a9SJohn McCall       }
11037f9c92a9SJohn McCall 
11047f9c92a9SJohn McCall       // Pass the rest of the arguments, which must match exactly.
11057f9c92a9SJohn McCall       for (unsigned I = 0; I != NumPlacementArgs; ++I) {
1106cb5f77f0SJohn McCall         RValue RV = getPlacementArgs()[I].restore(CGF);
110743dca6a8SEli Friedman         DeleteArgs.add(RV, *AI++);
11087f9c92a9SJohn McCall       }
11097f9c92a9SJohn McCall 
11107f9c92a9SJohn McCall       // Call 'operator delete'.
11118dda7b27SJohn McCall       CGF.EmitCall(CGF.CGM.getTypes().arrangeFreeFunctionCall(DeleteArgs, FPT),
11127f9c92a9SJohn McCall                    CGF.CGM.GetAddrOfFunction(OperatorDelete),
11137f9c92a9SJohn McCall                    ReturnValueSlot(), DeleteArgs, OperatorDelete);
11147f9c92a9SJohn McCall     }
11157f9c92a9SJohn McCall   };
11167f9c92a9SJohn McCall }
11177f9c92a9SJohn McCall 
11187f9c92a9SJohn McCall /// Enter a cleanup to call 'operator delete' if the initializer in a
11197f9c92a9SJohn McCall /// new-expression throws.
11207f9c92a9SJohn McCall static void EnterNewDeleteCleanup(CodeGenFunction &CGF,
11217f9c92a9SJohn McCall                                   const CXXNewExpr *E,
11227f9c92a9SJohn McCall                                   llvm::Value *NewPtr,
11237f9c92a9SJohn McCall                                   llvm::Value *AllocSize,
11247f9c92a9SJohn McCall                                   const CallArgList &NewArgs) {
11257f9c92a9SJohn McCall   // If we're not inside a conditional branch, then the cleanup will
11267f9c92a9SJohn McCall   // dominate and we can do the easier (and more efficient) thing.
11277f9c92a9SJohn McCall   if (!CGF.isInConditionalBranch()) {
11287f9c92a9SJohn McCall     CallDeleteDuringNew *Cleanup = CGF.EHStack
11297f9c92a9SJohn McCall       .pushCleanupWithExtra<CallDeleteDuringNew>(EHCleanup,
11307f9c92a9SJohn McCall                                                  E->getNumPlacementArgs(),
11317f9c92a9SJohn McCall                                                  E->getOperatorDelete(),
11327f9c92a9SJohn McCall                                                  NewPtr, AllocSize);
11337f9c92a9SJohn McCall     for (unsigned I = 0, N = E->getNumPlacementArgs(); I != N; ++I)
1134f4258eb4SEli Friedman       Cleanup->setPlacementArg(I, NewArgs[I+1].RV);
11357f9c92a9SJohn McCall 
11367f9c92a9SJohn McCall     return;
11377f9c92a9SJohn McCall   }
11387f9c92a9SJohn McCall 
11397f9c92a9SJohn McCall   // Otherwise, we need to save all this stuff.
1140cb5f77f0SJohn McCall   DominatingValue<RValue>::saved_type SavedNewPtr =
1141cb5f77f0SJohn McCall     DominatingValue<RValue>::save(CGF, RValue::get(NewPtr));
1142cb5f77f0SJohn McCall   DominatingValue<RValue>::saved_type SavedAllocSize =
1143cb5f77f0SJohn McCall     DominatingValue<RValue>::save(CGF, RValue::get(AllocSize));
11447f9c92a9SJohn McCall 
11457f9c92a9SJohn McCall   CallDeleteDuringConditionalNew *Cleanup = CGF.EHStack
1146f4beacd0SJohn McCall     .pushCleanupWithExtra<CallDeleteDuringConditionalNew>(EHCleanup,
11477f9c92a9SJohn McCall                                                  E->getNumPlacementArgs(),
11487f9c92a9SJohn McCall                                                  E->getOperatorDelete(),
11497f9c92a9SJohn McCall                                                  SavedNewPtr,
11507f9c92a9SJohn McCall                                                  SavedAllocSize);
11517f9c92a9SJohn McCall   for (unsigned I = 0, N = E->getNumPlacementArgs(); I != N; ++I)
1152cb5f77f0SJohn McCall     Cleanup->setPlacementArg(I,
1153f4258eb4SEli Friedman                      DominatingValue<RValue>::save(CGF, NewArgs[I+1].RV));
11547f9c92a9SJohn McCall 
1155f4beacd0SJohn McCall   CGF.initFullExprCleanup();
1156824c2f53SJohn McCall }
1157824c2f53SJohn McCall 
115859486a2dSAnders Carlsson llvm::Value *CodeGenFunction::EmitCXXNewExpr(const CXXNewExpr *E) {
115975f9498aSJohn McCall   // The element type being allocated.
116075f9498aSJohn McCall   QualType allocType = getContext().getBaseElementType(E->getAllocatedType());
11618ed55a54SJohn McCall 
116275f9498aSJohn McCall   // 1. Build a call to the allocation function.
116375f9498aSJohn McCall   FunctionDecl *allocator = E->getOperatorNew();
116475f9498aSJohn McCall   const FunctionProtoType *allocatorType =
116575f9498aSJohn McCall     allocator->getType()->castAs<FunctionProtoType>();
116659486a2dSAnders Carlsson 
116775f9498aSJohn McCall   CallArgList allocatorArgs;
116859486a2dSAnders Carlsson 
116959486a2dSAnders Carlsson   // The allocation size is the first argument.
117075f9498aSJohn McCall   QualType sizeType = getContext().getSizeType();
117159486a2dSAnders Carlsson 
1172f862eb6aSSebastian Redl   // If there is a brace-initializer, cannot allocate fewer elements than inits.
1173f862eb6aSSebastian Redl   unsigned minElements = 0;
1174f862eb6aSSebastian Redl   if (E->isArray() && E->hasInitializer()) {
1175f862eb6aSSebastian Redl     if (const InitListExpr *ILE = dyn_cast<InitListExpr>(E->getInitializer()))
1176f862eb6aSSebastian Redl       minElements = ILE->getNumInits();
1177f862eb6aSSebastian Redl   }
1178f862eb6aSSebastian Redl 
117975f9498aSJohn McCall   llvm::Value *numElements = 0;
118075f9498aSJohn McCall   llvm::Value *allocSizeWithoutCookie = 0;
118175f9498aSJohn McCall   llvm::Value *allocSize =
1182f862eb6aSSebastian Redl     EmitCXXNewAllocSize(*this, E, minElements, numElements,
1183f862eb6aSSebastian Redl                         allocSizeWithoutCookie);
118459486a2dSAnders Carlsson 
118543dca6a8SEli Friedman   allocatorArgs.add(RValue::get(allocSize), sizeType);
118659486a2dSAnders Carlsson 
118759486a2dSAnders Carlsson   // Emit the rest of the arguments.
118859486a2dSAnders Carlsson   // FIXME: Ideally, this should just use EmitCallArgs.
118975f9498aSJohn McCall   CXXNewExpr::const_arg_iterator placementArg = E->placement_arg_begin();
119059486a2dSAnders Carlsson 
119159486a2dSAnders Carlsson   // First, use the types from the function type.
119259486a2dSAnders Carlsson   // We start at 1 here because the first argument (the allocation size)
119359486a2dSAnders Carlsson   // has already been emitted.
119475f9498aSJohn McCall   for (unsigned i = 1, e = allocatorType->getNumArgs(); i != e;
119575f9498aSJohn McCall        ++i, ++placementArg) {
119675f9498aSJohn McCall     QualType argType = allocatorType->getArgType(i);
119759486a2dSAnders Carlsson 
119875f9498aSJohn McCall     assert(getContext().hasSameUnqualifiedType(argType.getNonReferenceType(),
119975f9498aSJohn McCall                                                placementArg->getType()) &&
120059486a2dSAnders Carlsson            "type mismatch in call argument!");
120159486a2dSAnders Carlsson 
120232ea9694SJohn McCall     EmitCallArg(allocatorArgs, *placementArg, argType);
120359486a2dSAnders Carlsson   }
120459486a2dSAnders Carlsson 
120559486a2dSAnders Carlsson   // Either we've emitted all the call args, or we have a call to a
120659486a2dSAnders Carlsson   // variadic function.
120775f9498aSJohn McCall   assert((placementArg == E->placement_arg_end() ||
120875f9498aSJohn McCall           allocatorType->isVariadic()) &&
120975f9498aSJohn McCall          "Extra arguments to non-variadic function!");
121059486a2dSAnders Carlsson 
121159486a2dSAnders Carlsson   // If we still have any arguments, emit them using the type of the argument.
121275f9498aSJohn McCall   for (CXXNewExpr::const_arg_iterator placementArgsEnd = E->placement_arg_end();
121375f9498aSJohn McCall        placementArg != placementArgsEnd; ++placementArg) {
121432ea9694SJohn McCall     EmitCallArg(allocatorArgs, *placementArg, placementArg->getType());
121559486a2dSAnders Carlsson   }
121659486a2dSAnders Carlsson 
12177ec4b434SJohn McCall   // Emit the allocation call.  If the allocator is a global placement
12187ec4b434SJohn McCall   // operator, just "inline" it directly.
12197ec4b434SJohn McCall   RValue RV;
12207ec4b434SJohn McCall   if (allocator->isReservedGlobalPlacementOperator()) {
12217ec4b434SJohn McCall     assert(allocatorArgs.size() == 2);
12227ec4b434SJohn McCall     RV = allocatorArgs[1].RV;
12237ec4b434SJohn McCall     // TODO: kill any unnecessary computations done for the size
12247ec4b434SJohn McCall     // argument.
12257ec4b434SJohn McCall   } else {
12268dda7b27SJohn McCall     RV = EmitCall(CGM.getTypes().arrangeFreeFunctionCall(allocatorArgs,
1227a729c62bSJohn McCall                                                          allocatorType),
122875f9498aSJohn McCall                   CGM.GetAddrOfFunction(allocator), ReturnValueSlot(),
122975f9498aSJohn McCall                   allocatorArgs, allocator);
12307ec4b434SJohn McCall   }
123159486a2dSAnders Carlsson 
123275f9498aSJohn McCall   // Emit a null check on the allocation result if the allocation
123375f9498aSJohn McCall   // function is allowed to return null (because it has a non-throwing
123475f9498aSJohn McCall   // exception spec; for this part, we inline
123575f9498aSJohn McCall   // CXXNewExpr::shouldNullCheckAllocation()) and we have an
123675f9498aSJohn McCall   // interesting initializer.
123731ad754cSSebastian Redl   bool nullCheck = allocatorType->isNothrow(getContext()) &&
12386047f07eSSebastian Redl     (!allocType.isPODType(getContext()) || E->hasInitializer());
123959486a2dSAnders Carlsson 
124075f9498aSJohn McCall   llvm::BasicBlock *nullCheckBB = 0;
124175f9498aSJohn McCall   llvm::BasicBlock *contBB = 0;
124259486a2dSAnders Carlsson 
124375f9498aSJohn McCall   llvm::Value *allocation = RV.getScalarVal();
1244ea2fea2aSMicah Villmow   unsigned AS = allocation->getType()->getPointerAddressSpace();
124559486a2dSAnders Carlsson 
1246f7dcf320SJohn McCall   // The null-check means that the initializer is conditionally
1247f7dcf320SJohn McCall   // evaluated.
1248f7dcf320SJohn McCall   ConditionalEvaluation conditional(*this);
1249f7dcf320SJohn McCall 
125075f9498aSJohn McCall   if (nullCheck) {
1251f7dcf320SJohn McCall     conditional.begin(*this);
125275f9498aSJohn McCall 
125375f9498aSJohn McCall     nullCheckBB = Builder.GetInsertBlock();
125475f9498aSJohn McCall     llvm::BasicBlock *notNullBB = createBasicBlock("new.notnull");
125575f9498aSJohn McCall     contBB = createBasicBlock("new.cont");
125675f9498aSJohn McCall 
125775f9498aSJohn McCall     llvm::Value *isNull = Builder.CreateIsNull(allocation, "new.isnull");
125875f9498aSJohn McCall     Builder.CreateCondBr(isNull, contBB, notNullBB);
125975f9498aSJohn McCall     EmitBlock(notNullBB);
126059486a2dSAnders Carlsson   }
126159486a2dSAnders Carlsson 
1262824c2f53SJohn McCall   // If there's an operator delete, enter a cleanup to call it if an
1263824c2f53SJohn McCall   // exception is thrown.
126475f9498aSJohn McCall   EHScopeStack::stable_iterator operatorDeleteCleanup;
1265f4beacd0SJohn McCall   llvm::Instruction *cleanupDominator = 0;
12667ec4b434SJohn McCall   if (E->getOperatorDelete() &&
12677ec4b434SJohn McCall       !E->getOperatorDelete()->isReservedGlobalPlacementOperator()) {
126875f9498aSJohn McCall     EnterNewDeleteCleanup(*this, E, allocation, allocSize, allocatorArgs);
126975f9498aSJohn McCall     operatorDeleteCleanup = EHStack.stable_begin();
1270f4beacd0SJohn McCall     cleanupDominator = Builder.CreateUnreachable();
1271824c2f53SJohn McCall   }
1272824c2f53SJohn McCall 
1273cf9b1f65SEli Friedman   assert((allocSize == allocSizeWithoutCookie) ==
1274cf9b1f65SEli Friedman          CalculateCookiePadding(*this, E).isZero());
1275cf9b1f65SEli Friedman   if (allocSize != allocSizeWithoutCookie) {
1276cf9b1f65SEli Friedman     assert(E->isArray());
1277cf9b1f65SEli Friedman     allocation = CGM.getCXXABI().InitializeArrayCookie(*this, allocation,
1278cf9b1f65SEli Friedman                                                        numElements,
1279cf9b1f65SEli Friedman                                                        E, allocType);
1280cf9b1f65SEli Friedman   }
1281cf9b1f65SEli Friedman 
12822192fe50SChris Lattner   llvm::Type *elementPtrTy
128375f9498aSJohn McCall     = ConvertTypeForMem(allocType)->getPointerTo(AS);
128475f9498aSJohn McCall   llvm::Value *result = Builder.CreateBitCast(allocation, elementPtrTy);
1285824c2f53SJohn McCall 
128699210dc9SJohn McCall   EmitNewInitializer(*this, E, allocType, result, numElements,
128799210dc9SJohn McCall                      allocSizeWithoutCookie);
12888ed55a54SJohn McCall   if (E->isArray()) {
12898ed55a54SJohn McCall     // NewPtr is a pointer to the base element type.  If we're
12908ed55a54SJohn McCall     // allocating an array of arrays, we'll need to cast back to the
12918ed55a54SJohn McCall     // array pointer type.
12922192fe50SChris Lattner     llvm::Type *resultType = ConvertTypeForMem(E->getType());
129375f9498aSJohn McCall     if (result->getType() != resultType)
129475f9498aSJohn McCall       result = Builder.CreateBitCast(result, resultType);
129547b4629bSFariborz Jahanian   }
129659486a2dSAnders Carlsson 
1297824c2f53SJohn McCall   // Deactivate the 'operator delete' cleanup if we finished
1298824c2f53SJohn McCall   // initialization.
1299f4beacd0SJohn McCall   if (operatorDeleteCleanup.isValid()) {
1300f4beacd0SJohn McCall     DeactivateCleanupBlock(operatorDeleteCleanup, cleanupDominator);
1301f4beacd0SJohn McCall     cleanupDominator->eraseFromParent();
1302f4beacd0SJohn McCall   }
1303824c2f53SJohn McCall 
130475f9498aSJohn McCall   if (nullCheck) {
1305f7dcf320SJohn McCall     conditional.end(*this);
1306f7dcf320SJohn McCall 
130775f9498aSJohn McCall     llvm::BasicBlock *notNullBB = Builder.GetInsertBlock();
130875f9498aSJohn McCall     EmitBlock(contBB);
130959486a2dSAnders Carlsson 
131020c0f02cSJay Foad     llvm::PHINode *PHI = Builder.CreatePHI(result->getType(), 2);
131175f9498aSJohn McCall     PHI->addIncoming(result, notNullBB);
131275f9498aSJohn McCall     PHI->addIncoming(llvm::Constant::getNullValue(result->getType()),
131375f9498aSJohn McCall                      nullCheckBB);
131459486a2dSAnders Carlsson 
131575f9498aSJohn McCall     result = PHI;
131659486a2dSAnders Carlsson   }
131759486a2dSAnders Carlsson 
131875f9498aSJohn McCall   return result;
131959486a2dSAnders Carlsson }
132059486a2dSAnders Carlsson 
132159486a2dSAnders Carlsson void CodeGenFunction::EmitDeleteCall(const FunctionDecl *DeleteFD,
132259486a2dSAnders Carlsson                                      llvm::Value *Ptr,
132359486a2dSAnders Carlsson                                      QualType DeleteTy) {
13248ed55a54SJohn McCall   assert(DeleteFD->getOverloadedOperator() == OO_Delete);
13258ed55a54SJohn McCall 
132659486a2dSAnders Carlsson   const FunctionProtoType *DeleteFTy =
132759486a2dSAnders Carlsson     DeleteFD->getType()->getAs<FunctionProtoType>();
132859486a2dSAnders Carlsson 
132959486a2dSAnders Carlsson   CallArgList DeleteArgs;
133059486a2dSAnders Carlsson 
133121122cf6SAnders Carlsson   // Check if we need to pass the size to the delete operator.
133221122cf6SAnders Carlsson   llvm::Value *Size = 0;
133321122cf6SAnders Carlsson   QualType SizeTy;
133421122cf6SAnders Carlsson   if (DeleteFTy->getNumArgs() == 2) {
133521122cf6SAnders Carlsson     SizeTy = DeleteFTy->getArgType(1);
13367df3cbebSKen Dyck     CharUnits DeleteTypeSize = getContext().getTypeSizeInChars(DeleteTy);
13377df3cbebSKen Dyck     Size = llvm::ConstantInt::get(ConvertType(SizeTy),
13387df3cbebSKen Dyck                                   DeleteTypeSize.getQuantity());
133921122cf6SAnders Carlsson   }
134021122cf6SAnders Carlsson 
134159486a2dSAnders Carlsson   QualType ArgTy = DeleteFTy->getArgType(0);
134259486a2dSAnders Carlsson   llvm::Value *DeletePtr = Builder.CreateBitCast(Ptr, ConvertType(ArgTy));
134343dca6a8SEli Friedman   DeleteArgs.add(RValue::get(DeletePtr), ArgTy);
134459486a2dSAnders Carlsson 
134521122cf6SAnders Carlsson   if (Size)
134643dca6a8SEli Friedman     DeleteArgs.add(RValue::get(Size), SizeTy);
134759486a2dSAnders Carlsson 
134859486a2dSAnders Carlsson   // Emit the call to delete.
13498dda7b27SJohn McCall   EmitCall(CGM.getTypes().arrangeFreeFunctionCall(DeleteArgs, DeleteFTy),
135061a401caSAnders Carlsson            CGM.GetAddrOfFunction(DeleteFD), ReturnValueSlot(),
135159486a2dSAnders Carlsson            DeleteArgs, DeleteFD);
135259486a2dSAnders Carlsson }
135359486a2dSAnders Carlsson 
13548ed55a54SJohn McCall namespace {
13558ed55a54SJohn McCall   /// Calls the given 'operator delete' on a single object.
13568ed55a54SJohn McCall   struct CallObjectDelete : EHScopeStack::Cleanup {
13578ed55a54SJohn McCall     llvm::Value *Ptr;
13588ed55a54SJohn McCall     const FunctionDecl *OperatorDelete;
13598ed55a54SJohn McCall     QualType ElementType;
13608ed55a54SJohn McCall 
13618ed55a54SJohn McCall     CallObjectDelete(llvm::Value *Ptr,
13628ed55a54SJohn McCall                      const FunctionDecl *OperatorDelete,
13638ed55a54SJohn McCall                      QualType ElementType)
13648ed55a54SJohn McCall       : Ptr(Ptr), OperatorDelete(OperatorDelete), ElementType(ElementType) {}
13658ed55a54SJohn McCall 
136630317fdaSJohn McCall     void Emit(CodeGenFunction &CGF, Flags flags) {
13678ed55a54SJohn McCall       CGF.EmitDeleteCall(OperatorDelete, Ptr, ElementType);
13688ed55a54SJohn McCall     }
13698ed55a54SJohn McCall   };
13708ed55a54SJohn McCall }
13718ed55a54SJohn McCall 
13728ed55a54SJohn McCall /// Emit the code for deleting a single object.
13738ed55a54SJohn McCall static void EmitObjectDelete(CodeGenFunction &CGF,
13748ed55a54SJohn McCall                              const FunctionDecl *OperatorDelete,
13758ed55a54SJohn McCall                              llvm::Value *Ptr,
13761c2e20d7SDouglas Gregor                              QualType ElementType,
13771c2e20d7SDouglas Gregor                              bool UseGlobalDelete) {
13788ed55a54SJohn McCall   // Find the destructor for the type, if applicable.  If the
13798ed55a54SJohn McCall   // destructor is virtual, we'll just emit the vcall and return.
13808ed55a54SJohn McCall   const CXXDestructorDecl *Dtor = 0;
13818ed55a54SJohn McCall   if (const RecordType *RT = ElementType->getAs<RecordType>()) {
13828ed55a54SJohn McCall     CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl());
1383b23533dbSEli Friedman     if (RD->hasDefinition() && !RD->hasTrivialDestructor()) {
13848ed55a54SJohn McCall       Dtor = RD->getDestructor();
13858ed55a54SJohn McCall 
13868ed55a54SJohn McCall       if (Dtor->isVirtual()) {
13871c2e20d7SDouglas Gregor         if (UseGlobalDelete) {
13881c2e20d7SDouglas Gregor           // If we're supposed to call the global delete, make sure we do so
13891c2e20d7SDouglas Gregor           // even if the destructor throws.
139082fb8920SJohn McCall 
139182fb8920SJohn McCall           // Derive the complete-object pointer, which is what we need
139282fb8920SJohn McCall           // to pass to the deallocation function.
139382fb8920SJohn McCall           llvm::Value *completePtr =
139482fb8920SJohn McCall             CGF.CGM.getCXXABI().adjustToCompleteObject(CGF, Ptr, ElementType);
139582fb8920SJohn McCall 
13961c2e20d7SDouglas Gregor           CGF.EHStack.pushCleanup<CallObjectDelete>(NormalAndEHCleanup,
139782fb8920SJohn McCall                                                     completePtr, OperatorDelete,
13981c2e20d7SDouglas Gregor                                                     ElementType);
13991c2e20d7SDouglas Gregor         }
14001c2e20d7SDouglas Gregor 
14012192fe50SChris Lattner         llvm::Type *Ty =
1402a729c62bSJohn McCall           CGF.getTypes().GetFunctionType(
1403a729c62bSJohn McCall                          CGF.getTypes().arrangeCXXDestructor(Dtor, Dtor_Complete));
14048ed55a54SJohn McCall 
14058ed55a54SJohn McCall         llvm::Value *Callee
14061c2e20d7SDouglas Gregor           = CGF.BuildVirtualCall(Dtor,
14071c2e20d7SDouglas Gregor                                  UseGlobalDelete? Dtor_Complete : Dtor_Deleting,
14081c2e20d7SDouglas Gregor                                  Ptr, Ty);
1409e30752c9SRichard Smith         // FIXME: Provide a source location here.
1410e30752c9SRichard Smith         CGF.EmitCXXMemberCall(Dtor, SourceLocation(), Callee, ReturnValueSlot(),
1411e30752c9SRichard Smith                               Ptr, /*VTT=*/0, 0, 0);
14128ed55a54SJohn McCall 
14131c2e20d7SDouglas Gregor         if (UseGlobalDelete) {
14141c2e20d7SDouglas Gregor           CGF.PopCleanupBlock();
14151c2e20d7SDouglas Gregor         }
14161c2e20d7SDouglas Gregor 
14178ed55a54SJohn McCall         return;
14188ed55a54SJohn McCall       }
14198ed55a54SJohn McCall     }
14208ed55a54SJohn McCall   }
14218ed55a54SJohn McCall 
14228ed55a54SJohn McCall   // Make sure that we call delete even if the dtor throws.
1423e4df6c8dSJohn McCall   // This doesn't have to a conditional cleanup because we're going
1424e4df6c8dSJohn McCall   // to pop it off in a second.
14258ed55a54SJohn McCall   CGF.EHStack.pushCleanup<CallObjectDelete>(NormalAndEHCleanup,
14268ed55a54SJohn McCall                                             Ptr, OperatorDelete, ElementType);
14278ed55a54SJohn McCall 
14288ed55a54SJohn McCall   if (Dtor)
14298ed55a54SJohn McCall     CGF.EmitCXXDestructorCall(Dtor, Dtor_Complete,
143061535005SDouglas Gregor                               /*ForVirtualBase=*/false,
143161535005SDouglas Gregor                               /*Delegating=*/false,
143261535005SDouglas Gregor                               Ptr);
1433bbafb8a7SDavid Blaikie   else if (CGF.getLangOpts().ObjCAutoRefCount &&
143431168b07SJohn McCall            ElementType->isObjCLifetimeType()) {
143531168b07SJohn McCall     switch (ElementType.getObjCLifetime()) {
143631168b07SJohn McCall     case Qualifiers::OCL_None:
143731168b07SJohn McCall     case Qualifiers::OCL_ExplicitNone:
143831168b07SJohn McCall     case Qualifiers::OCL_Autoreleasing:
143931168b07SJohn McCall       break;
144031168b07SJohn McCall 
144131168b07SJohn McCall     case Qualifiers::OCL_Strong: {
144231168b07SJohn McCall       // Load the pointer value.
144331168b07SJohn McCall       llvm::Value *PtrValue = CGF.Builder.CreateLoad(Ptr,
144431168b07SJohn McCall                                              ElementType.isVolatileQualified());
144531168b07SJohn McCall 
144631168b07SJohn McCall       CGF.EmitARCRelease(PtrValue, /*precise*/ true);
144731168b07SJohn McCall       break;
144831168b07SJohn McCall     }
144931168b07SJohn McCall 
145031168b07SJohn McCall     case Qualifiers::OCL_Weak:
145131168b07SJohn McCall       CGF.EmitARCDestroyWeak(Ptr);
145231168b07SJohn McCall       break;
145331168b07SJohn McCall     }
145431168b07SJohn McCall   }
14558ed55a54SJohn McCall 
14568ed55a54SJohn McCall   CGF.PopCleanupBlock();
14578ed55a54SJohn McCall }
14588ed55a54SJohn McCall 
14598ed55a54SJohn McCall namespace {
14608ed55a54SJohn McCall   /// Calls the given 'operator delete' on an array of objects.
14618ed55a54SJohn McCall   struct CallArrayDelete : EHScopeStack::Cleanup {
14628ed55a54SJohn McCall     llvm::Value *Ptr;
14638ed55a54SJohn McCall     const FunctionDecl *OperatorDelete;
14648ed55a54SJohn McCall     llvm::Value *NumElements;
14658ed55a54SJohn McCall     QualType ElementType;
14668ed55a54SJohn McCall     CharUnits CookieSize;
14678ed55a54SJohn McCall 
14688ed55a54SJohn McCall     CallArrayDelete(llvm::Value *Ptr,
14698ed55a54SJohn McCall                     const FunctionDecl *OperatorDelete,
14708ed55a54SJohn McCall                     llvm::Value *NumElements,
14718ed55a54SJohn McCall                     QualType ElementType,
14728ed55a54SJohn McCall                     CharUnits CookieSize)
14738ed55a54SJohn McCall       : Ptr(Ptr), OperatorDelete(OperatorDelete), NumElements(NumElements),
14748ed55a54SJohn McCall         ElementType(ElementType), CookieSize(CookieSize) {}
14758ed55a54SJohn McCall 
147630317fdaSJohn McCall     void Emit(CodeGenFunction &CGF, Flags flags) {
14778ed55a54SJohn McCall       const FunctionProtoType *DeleteFTy =
14788ed55a54SJohn McCall         OperatorDelete->getType()->getAs<FunctionProtoType>();
14798ed55a54SJohn McCall       assert(DeleteFTy->getNumArgs() == 1 || DeleteFTy->getNumArgs() == 2);
14808ed55a54SJohn McCall 
14818ed55a54SJohn McCall       CallArgList Args;
14828ed55a54SJohn McCall 
14838ed55a54SJohn McCall       // Pass the pointer as the first argument.
14848ed55a54SJohn McCall       QualType VoidPtrTy = DeleteFTy->getArgType(0);
14858ed55a54SJohn McCall       llvm::Value *DeletePtr
14868ed55a54SJohn McCall         = CGF.Builder.CreateBitCast(Ptr, CGF.ConvertType(VoidPtrTy));
148743dca6a8SEli Friedman       Args.add(RValue::get(DeletePtr), VoidPtrTy);
14888ed55a54SJohn McCall 
14898ed55a54SJohn McCall       // Pass the original requested size as the second argument.
14908ed55a54SJohn McCall       if (DeleteFTy->getNumArgs() == 2) {
14918ed55a54SJohn McCall         QualType size_t = DeleteFTy->getArgType(1);
14922192fe50SChris Lattner         llvm::IntegerType *SizeTy
14938ed55a54SJohn McCall           = cast<llvm::IntegerType>(CGF.ConvertType(size_t));
14948ed55a54SJohn McCall 
14958ed55a54SJohn McCall         CharUnits ElementTypeSize =
14968ed55a54SJohn McCall           CGF.CGM.getContext().getTypeSizeInChars(ElementType);
14978ed55a54SJohn McCall 
14988ed55a54SJohn McCall         // The size of an element, multiplied by the number of elements.
14998ed55a54SJohn McCall         llvm::Value *Size
15008ed55a54SJohn McCall           = llvm::ConstantInt::get(SizeTy, ElementTypeSize.getQuantity());
15018ed55a54SJohn McCall         Size = CGF.Builder.CreateMul(Size, NumElements);
15028ed55a54SJohn McCall 
15038ed55a54SJohn McCall         // Plus the size of the cookie if applicable.
15048ed55a54SJohn McCall         if (!CookieSize.isZero()) {
15058ed55a54SJohn McCall           llvm::Value *CookieSizeV
15068ed55a54SJohn McCall             = llvm::ConstantInt::get(SizeTy, CookieSize.getQuantity());
15078ed55a54SJohn McCall           Size = CGF.Builder.CreateAdd(Size, CookieSizeV);
15088ed55a54SJohn McCall         }
15098ed55a54SJohn McCall 
151043dca6a8SEli Friedman         Args.add(RValue::get(Size), size_t);
15118ed55a54SJohn McCall       }
15128ed55a54SJohn McCall 
15138ed55a54SJohn McCall       // Emit the call to delete.
15148dda7b27SJohn McCall       CGF.EmitCall(CGF.getTypes().arrangeFreeFunctionCall(Args, DeleteFTy),
15158ed55a54SJohn McCall                    CGF.CGM.GetAddrOfFunction(OperatorDelete),
15168ed55a54SJohn McCall                    ReturnValueSlot(), Args, OperatorDelete);
15178ed55a54SJohn McCall     }
15188ed55a54SJohn McCall   };
15198ed55a54SJohn McCall }
15208ed55a54SJohn McCall 
15218ed55a54SJohn McCall /// Emit the code for deleting an array of objects.
15228ed55a54SJohn McCall static void EmitArrayDelete(CodeGenFunction &CGF,
1523284c48ffSJohn McCall                             const CXXDeleteExpr *E,
1524ca2c56f2SJohn McCall                             llvm::Value *deletedPtr,
1525ca2c56f2SJohn McCall                             QualType elementType) {
1526ca2c56f2SJohn McCall   llvm::Value *numElements = 0;
1527ca2c56f2SJohn McCall   llvm::Value *allocatedPtr = 0;
1528ca2c56f2SJohn McCall   CharUnits cookieSize;
1529ca2c56f2SJohn McCall   CGF.CGM.getCXXABI().ReadArrayCookie(CGF, deletedPtr, E, elementType,
1530ca2c56f2SJohn McCall                                       numElements, allocatedPtr, cookieSize);
15318ed55a54SJohn McCall 
1532ca2c56f2SJohn McCall   assert(allocatedPtr && "ReadArrayCookie didn't set allocated pointer");
15338ed55a54SJohn McCall 
15348ed55a54SJohn McCall   // Make sure that we call delete even if one of the dtors throws.
1535ca2c56f2SJohn McCall   const FunctionDecl *operatorDelete = E->getOperatorDelete();
15368ed55a54SJohn McCall   CGF.EHStack.pushCleanup<CallArrayDelete>(NormalAndEHCleanup,
1537ca2c56f2SJohn McCall                                            allocatedPtr, operatorDelete,
1538ca2c56f2SJohn McCall                                            numElements, elementType,
1539ca2c56f2SJohn McCall                                            cookieSize);
15408ed55a54SJohn McCall 
1541ca2c56f2SJohn McCall   // Destroy the elements.
1542ca2c56f2SJohn McCall   if (QualType::DestructionKind dtorKind = elementType.isDestructedType()) {
1543ca2c56f2SJohn McCall     assert(numElements && "no element count for a type with a destructor!");
154431168b07SJohn McCall 
1545ca2c56f2SJohn McCall     llvm::Value *arrayEnd =
1546ca2c56f2SJohn McCall       CGF.Builder.CreateInBoundsGEP(deletedPtr, numElements, "delete.end");
154797eab0a2SJohn McCall 
154897eab0a2SJohn McCall     // Note that it is legal to allocate a zero-length array, and we
154997eab0a2SJohn McCall     // can never fold the check away because the length should always
155097eab0a2SJohn McCall     // come from a cookie.
1551ca2c56f2SJohn McCall     CGF.emitArrayDestroy(deletedPtr, arrayEnd, elementType,
1552ca2c56f2SJohn McCall                          CGF.getDestroyer(dtorKind),
155397eab0a2SJohn McCall                          /*checkZeroLength*/ true,
1554ca2c56f2SJohn McCall                          CGF.needsEHCleanup(dtorKind));
15558ed55a54SJohn McCall   }
15568ed55a54SJohn McCall 
1557ca2c56f2SJohn McCall   // Pop the cleanup block.
15588ed55a54SJohn McCall   CGF.PopCleanupBlock();
15598ed55a54SJohn McCall }
15608ed55a54SJohn McCall 
156159486a2dSAnders Carlsson void CodeGenFunction::EmitCXXDeleteExpr(const CXXDeleteExpr *E) {
156259486a2dSAnders Carlsson   const Expr *Arg = E->getArgument();
156359486a2dSAnders Carlsson   llvm::Value *Ptr = EmitScalarExpr(Arg);
156459486a2dSAnders Carlsson 
156559486a2dSAnders Carlsson   // Null check the pointer.
156659486a2dSAnders Carlsson   llvm::BasicBlock *DeleteNotNull = createBasicBlock("delete.notnull");
156759486a2dSAnders Carlsson   llvm::BasicBlock *DeleteEnd = createBasicBlock("delete.end");
156859486a2dSAnders Carlsson 
156998981b10SAnders Carlsson   llvm::Value *IsNull = Builder.CreateIsNull(Ptr, "isnull");
157059486a2dSAnders Carlsson 
157159486a2dSAnders Carlsson   Builder.CreateCondBr(IsNull, DeleteEnd, DeleteNotNull);
157259486a2dSAnders Carlsson   EmitBlock(DeleteNotNull);
157359486a2dSAnders Carlsson 
15748ed55a54SJohn McCall   // We might be deleting a pointer to array.  If so, GEP down to the
15758ed55a54SJohn McCall   // first non-array element.
15768ed55a54SJohn McCall   // (this assumes that A(*)[3][7] is converted to [3 x [7 x %A]]*)
15778ed55a54SJohn McCall   QualType DeleteTy = Arg->getType()->getAs<PointerType>()->getPointeeType();
15788ed55a54SJohn McCall   if (DeleteTy->isConstantArrayType()) {
15798ed55a54SJohn McCall     llvm::Value *Zero = Builder.getInt32(0);
15800e62c1ccSChris Lattner     SmallVector<llvm::Value*,8> GEP;
158159486a2dSAnders Carlsson 
15828ed55a54SJohn McCall     GEP.push_back(Zero); // point at the outermost array
15838ed55a54SJohn McCall 
15848ed55a54SJohn McCall     // For each layer of array type we're pointing at:
15858ed55a54SJohn McCall     while (const ConstantArrayType *Arr
15868ed55a54SJohn McCall              = getContext().getAsConstantArrayType(DeleteTy)) {
15878ed55a54SJohn McCall       // 1. Unpeel the array type.
15888ed55a54SJohn McCall       DeleteTy = Arr->getElementType();
15898ed55a54SJohn McCall 
15908ed55a54SJohn McCall       // 2. GEP to the first element of the array.
15918ed55a54SJohn McCall       GEP.push_back(Zero);
15928ed55a54SJohn McCall     }
15938ed55a54SJohn McCall 
1594040dd82fSJay Foad     Ptr = Builder.CreateInBoundsGEP(Ptr, GEP, "del.first");
15958ed55a54SJohn McCall   }
15968ed55a54SJohn McCall 
159704f36218SDouglas Gregor   assert(ConvertTypeForMem(DeleteTy) ==
159804f36218SDouglas Gregor          cast<llvm::PointerType>(Ptr->getType())->getElementType());
15998ed55a54SJohn McCall 
160059486a2dSAnders Carlsson   if (E->isArrayForm()) {
1601284c48ffSJohn McCall     EmitArrayDelete(*this, E, Ptr, DeleteTy);
16028ed55a54SJohn McCall   } else {
16031c2e20d7SDouglas Gregor     EmitObjectDelete(*this, E->getOperatorDelete(), Ptr, DeleteTy,
16041c2e20d7SDouglas Gregor                      E->isGlobalDelete());
160559486a2dSAnders Carlsson   }
160659486a2dSAnders Carlsson 
160759486a2dSAnders Carlsson   EmitBlock(DeleteEnd);
160859486a2dSAnders Carlsson }
160959486a2dSAnders Carlsson 
16100c63350bSAnders Carlsson static llvm::Constant *getBadTypeidFn(CodeGenFunction &CGF) {
16110c63350bSAnders Carlsson   // void __cxa_bad_typeid();
1612ece0409aSChris Lattner   llvm::FunctionType *FTy = llvm::FunctionType::get(CGF.VoidTy, false);
16130c63350bSAnders Carlsson 
16140c63350bSAnders Carlsson   return CGF.CGM.CreateRuntimeFunction(FTy, "__cxa_bad_typeid");
16150c63350bSAnders Carlsson }
16160c63350bSAnders Carlsson 
16170c63350bSAnders Carlsson static void EmitBadTypeidCall(CodeGenFunction &CGF) {
1618bbe277c4SAnders Carlsson   llvm::Value *Fn = getBadTypeidFn(CGF);
16195bd375a6SJay Foad   CGF.EmitCallOrInvoke(Fn).setDoesNotReturn();
16200c63350bSAnders Carlsson   CGF.Builder.CreateUnreachable();
16210c63350bSAnders Carlsson }
16220c63350bSAnders Carlsson 
1623940f02d2SAnders Carlsson static llvm::Value *EmitTypeidFromVTable(CodeGenFunction &CGF,
1624940f02d2SAnders Carlsson                                          const Expr *E,
16252192fe50SChris Lattner                                          llvm::Type *StdTypeInfoPtrTy) {
1626940f02d2SAnders Carlsson   // Get the vtable pointer.
1627940f02d2SAnders Carlsson   llvm::Value *ThisPtr = CGF.EmitLValue(E).getAddress();
1628940f02d2SAnders Carlsson 
1629940f02d2SAnders Carlsson   // C++ [expr.typeid]p2:
1630940f02d2SAnders Carlsson   //   If the glvalue expression is obtained by applying the unary * operator to
1631940f02d2SAnders Carlsson   //   a pointer and the pointer is a null pointer value, the typeid expression
1632940f02d2SAnders Carlsson   //   throws the std::bad_typeid exception.
1633940f02d2SAnders Carlsson   if (const UnaryOperator *UO = dyn_cast<UnaryOperator>(E->IgnoreParens())) {
1634940f02d2SAnders Carlsson     if (UO->getOpcode() == UO_Deref) {
1635940f02d2SAnders Carlsson       llvm::BasicBlock *BadTypeidBlock =
1636940f02d2SAnders Carlsson         CGF.createBasicBlock("typeid.bad_typeid");
1637940f02d2SAnders Carlsson       llvm::BasicBlock *EndBlock =
1638940f02d2SAnders Carlsson         CGF.createBasicBlock("typeid.end");
1639940f02d2SAnders Carlsson 
1640940f02d2SAnders Carlsson       llvm::Value *IsNull = CGF.Builder.CreateIsNull(ThisPtr);
1641940f02d2SAnders Carlsson       CGF.Builder.CreateCondBr(IsNull, BadTypeidBlock, EndBlock);
1642940f02d2SAnders Carlsson 
1643940f02d2SAnders Carlsson       CGF.EmitBlock(BadTypeidBlock);
1644940f02d2SAnders Carlsson       EmitBadTypeidCall(CGF);
1645940f02d2SAnders Carlsson       CGF.EmitBlock(EndBlock);
1646940f02d2SAnders Carlsson     }
1647940f02d2SAnders Carlsson   }
1648940f02d2SAnders Carlsson 
1649940f02d2SAnders Carlsson   llvm::Value *Value = CGF.GetVTablePtr(ThisPtr,
1650940f02d2SAnders Carlsson                                         StdTypeInfoPtrTy->getPointerTo());
1651940f02d2SAnders Carlsson 
1652940f02d2SAnders Carlsson   // Load the type info.
1653940f02d2SAnders Carlsson   Value = CGF.Builder.CreateConstInBoundsGEP1_64(Value, -1ULL);
1654940f02d2SAnders Carlsson   return CGF.Builder.CreateLoad(Value);
1655940f02d2SAnders Carlsson }
1656940f02d2SAnders Carlsson 
165759486a2dSAnders Carlsson llvm::Value *CodeGenFunction::EmitCXXTypeidExpr(const CXXTypeidExpr *E) {
16582192fe50SChris Lattner   llvm::Type *StdTypeInfoPtrTy =
1659940f02d2SAnders Carlsson     ConvertType(E->getType())->getPointerTo();
1660fd7dfeb7SAnders Carlsson 
16613f4336cbSAnders Carlsson   if (E->isTypeOperand()) {
16623f4336cbSAnders Carlsson     llvm::Constant *TypeInfo =
16633f4336cbSAnders Carlsson       CGM.GetAddrOfRTTIDescriptor(E->getTypeOperand());
1664940f02d2SAnders Carlsson     return Builder.CreateBitCast(TypeInfo, StdTypeInfoPtrTy);
16653f4336cbSAnders Carlsson   }
1666fd7dfeb7SAnders Carlsson 
1667940f02d2SAnders Carlsson   // C++ [expr.typeid]p2:
1668940f02d2SAnders Carlsson   //   When typeid is applied to a glvalue expression whose type is a
1669940f02d2SAnders Carlsson   //   polymorphic class type, the result refers to a std::type_info object
1670940f02d2SAnders Carlsson   //   representing the type of the most derived object (that is, the dynamic
1671940f02d2SAnders Carlsson   //   type) to which the glvalue refers.
1672ef8bf436SRichard Smith   if (E->isPotentiallyEvaluated())
1673940f02d2SAnders Carlsson     return EmitTypeidFromVTable(*this, E->getExprOperand(),
1674940f02d2SAnders Carlsson                                 StdTypeInfoPtrTy);
1675940f02d2SAnders Carlsson 
1676940f02d2SAnders Carlsson   QualType OperandTy = E->getExprOperand()->getType();
1677940f02d2SAnders Carlsson   return Builder.CreateBitCast(CGM.GetAddrOfRTTIDescriptor(OperandTy),
1678940f02d2SAnders Carlsson                                StdTypeInfoPtrTy);
167959486a2dSAnders Carlsson }
168059486a2dSAnders Carlsson 
1681882d790fSAnders Carlsson static llvm::Constant *getDynamicCastFn(CodeGenFunction &CGF) {
1682882d790fSAnders Carlsson   // void *__dynamic_cast(const void *sub,
1683882d790fSAnders Carlsson   //                      const abi::__class_type_info *src,
1684882d790fSAnders Carlsson   //                      const abi::__class_type_info *dst,
1685882d790fSAnders Carlsson   //                      std::ptrdiff_t src2dst_offset);
1686882d790fSAnders Carlsson 
1687ece0409aSChris Lattner   llvm::Type *Int8PtrTy = CGF.Int8PtrTy;
1688a5f58b05SChris Lattner   llvm::Type *PtrDiffTy =
1689882d790fSAnders Carlsson     CGF.ConvertType(CGF.getContext().getPointerDiffType());
1690882d790fSAnders Carlsson 
1691a5f58b05SChris Lattner   llvm::Type *Args[4] = { Int8PtrTy, Int8PtrTy, Int8PtrTy, PtrDiffTy };
1692882d790fSAnders Carlsson 
1693*b5206330SBenjamin Kramer   llvm::FunctionType *FTy = llvm::FunctionType::get(Int8PtrTy, Args, false);
1694882d790fSAnders Carlsson 
1695*b5206330SBenjamin Kramer   // Mark the function as nounwind readonly.
1696*b5206330SBenjamin Kramer   llvm::Attribute::AttrKind FuncAttrs[] = { llvm::Attribute::NoUnwind,
1697*b5206330SBenjamin Kramer                                             llvm::Attribute::ReadOnly };
1698*b5206330SBenjamin Kramer   llvm::AttributeSet Attrs = llvm::AttributeSet::get(
1699*b5206330SBenjamin Kramer       CGF.getLLVMContext(), llvm::AttributeSet::FunctionIndex, FuncAttrs);
1700*b5206330SBenjamin Kramer 
1701*b5206330SBenjamin Kramer   return CGF.CGM.CreateRuntimeFunction(FTy, "__dynamic_cast", Attrs);
1702882d790fSAnders Carlsson }
1703882d790fSAnders Carlsson 
1704882d790fSAnders Carlsson static llvm::Constant *getBadCastFn(CodeGenFunction &CGF) {
1705882d790fSAnders Carlsson   // void __cxa_bad_cast();
1706ece0409aSChris Lattner   llvm::FunctionType *FTy = llvm::FunctionType::get(CGF.VoidTy, false);
1707882d790fSAnders Carlsson   return CGF.CGM.CreateRuntimeFunction(FTy, "__cxa_bad_cast");
1708882d790fSAnders Carlsson }
1709882d790fSAnders Carlsson 
1710c1c9971cSAnders Carlsson static void EmitBadCastCall(CodeGenFunction &CGF) {
1711bbe277c4SAnders Carlsson   llvm::Value *Fn = getBadCastFn(CGF);
17125bd375a6SJay Foad   CGF.EmitCallOrInvoke(Fn).setDoesNotReturn();
1713c1c9971cSAnders Carlsson   CGF.Builder.CreateUnreachable();
1714c1c9971cSAnders Carlsson }
1715c1c9971cSAnders Carlsson 
1716882d790fSAnders Carlsson static llvm::Value *
1717882d790fSAnders Carlsson EmitDynamicCastCall(CodeGenFunction &CGF, llvm::Value *Value,
1718882d790fSAnders Carlsson                     QualType SrcTy, QualType DestTy,
1719882d790fSAnders Carlsson                     llvm::BasicBlock *CastEnd) {
17202192fe50SChris Lattner   llvm::Type *PtrDiffLTy =
1721882d790fSAnders Carlsson     CGF.ConvertType(CGF.getContext().getPointerDiffType());
17222192fe50SChris Lattner   llvm::Type *DestLTy = CGF.ConvertType(DestTy);
1723882d790fSAnders Carlsson 
1724882d790fSAnders Carlsson   if (const PointerType *PTy = DestTy->getAs<PointerType>()) {
1725882d790fSAnders Carlsson     if (PTy->getPointeeType()->isVoidType()) {
1726882d790fSAnders Carlsson       // C++ [expr.dynamic.cast]p7:
1727882d790fSAnders Carlsson       //   If T is "pointer to cv void," then the result is a pointer to the
1728882d790fSAnders Carlsson       //   most derived object pointed to by v.
1729882d790fSAnders Carlsson 
1730882d790fSAnders Carlsson       // Get the vtable pointer.
1731882d790fSAnders Carlsson       llvm::Value *VTable = CGF.GetVTablePtr(Value, PtrDiffLTy->getPointerTo());
1732882d790fSAnders Carlsson 
1733882d790fSAnders Carlsson       // Get the offset-to-top from the vtable.
1734882d790fSAnders Carlsson       llvm::Value *OffsetToTop =
1735882d790fSAnders Carlsson         CGF.Builder.CreateConstInBoundsGEP1_64(VTable, -2ULL);
1736882d790fSAnders Carlsson       OffsetToTop = CGF.Builder.CreateLoad(OffsetToTop, "offset.to.top");
1737882d790fSAnders Carlsson 
1738882d790fSAnders Carlsson       // Finally, add the offset to the pointer.
1739882d790fSAnders Carlsson       Value = CGF.EmitCastToVoidPtr(Value);
1740882d790fSAnders Carlsson       Value = CGF.Builder.CreateInBoundsGEP(Value, OffsetToTop);
1741882d790fSAnders Carlsson 
1742882d790fSAnders Carlsson       return CGF.Builder.CreateBitCast(Value, DestLTy);
1743882d790fSAnders Carlsson     }
1744882d790fSAnders Carlsson   }
1745882d790fSAnders Carlsson 
1746882d790fSAnders Carlsson   QualType SrcRecordTy;
1747882d790fSAnders Carlsson   QualType DestRecordTy;
1748882d790fSAnders Carlsson 
1749882d790fSAnders Carlsson   if (const PointerType *DestPTy = DestTy->getAs<PointerType>()) {
1750882d790fSAnders Carlsson     SrcRecordTy = SrcTy->castAs<PointerType>()->getPointeeType();
1751882d790fSAnders Carlsson     DestRecordTy = DestPTy->getPointeeType();
1752882d790fSAnders Carlsson   } else {
1753882d790fSAnders Carlsson     SrcRecordTy = SrcTy;
1754882d790fSAnders Carlsson     DestRecordTy = DestTy->castAs<ReferenceType>()->getPointeeType();
1755882d790fSAnders Carlsson   }
1756882d790fSAnders Carlsson 
1757882d790fSAnders Carlsson   assert(SrcRecordTy->isRecordType() && "source type must be a record type!");
1758882d790fSAnders Carlsson   assert(DestRecordTy->isRecordType() && "dest type must be a record type!");
1759882d790fSAnders Carlsson 
1760882d790fSAnders Carlsson   llvm::Value *SrcRTTI =
1761882d790fSAnders Carlsson     CGF.CGM.GetAddrOfRTTIDescriptor(SrcRecordTy.getUnqualifiedType());
1762882d790fSAnders Carlsson   llvm::Value *DestRTTI =
1763882d790fSAnders Carlsson     CGF.CGM.GetAddrOfRTTIDescriptor(DestRecordTy.getUnqualifiedType());
1764882d790fSAnders Carlsson 
1765882d790fSAnders Carlsson   // FIXME: Actually compute a hint here.
1766882d790fSAnders Carlsson   llvm::Value *OffsetHint = llvm::ConstantInt::get(PtrDiffLTy, -1ULL);
1767882d790fSAnders Carlsson 
1768882d790fSAnders Carlsson   // Emit the call to __dynamic_cast.
1769882d790fSAnders Carlsson   Value = CGF.EmitCastToVoidPtr(Value);
1770882d790fSAnders Carlsson   Value = CGF.Builder.CreateCall4(getDynamicCastFn(CGF), Value,
1771882d790fSAnders Carlsson                                   SrcRTTI, DestRTTI, OffsetHint);
1772882d790fSAnders Carlsson   Value = CGF.Builder.CreateBitCast(Value, DestLTy);
1773882d790fSAnders Carlsson 
1774882d790fSAnders Carlsson   /// C++ [expr.dynamic.cast]p9:
1775882d790fSAnders Carlsson   ///   A failed cast to reference type throws std::bad_cast
1776882d790fSAnders Carlsson   if (DestTy->isReferenceType()) {
1777882d790fSAnders Carlsson     llvm::BasicBlock *BadCastBlock =
1778882d790fSAnders Carlsson       CGF.createBasicBlock("dynamic_cast.bad_cast");
1779882d790fSAnders Carlsson 
1780882d790fSAnders Carlsson     llvm::Value *IsNull = CGF.Builder.CreateIsNull(Value);
1781882d790fSAnders Carlsson     CGF.Builder.CreateCondBr(IsNull, BadCastBlock, CastEnd);
1782882d790fSAnders Carlsson 
1783882d790fSAnders Carlsson     CGF.EmitBlock(BadCastBlock);
1784c1c9971cSAnders Carlsson     EmitBadCastCall(CGF);
1785882d790fSAnders Carlsson   }
1786882d790fSAnders Carlsson 
1787882d790fSAnders Carlsson   return Value;
1788882d790fSAnders Carlsson }
1789882d790fSAnders Carlsson 
1790c1c9971cSAnders Carlsson static llvm::Value *EmitDynamicCastToNull(CodeGenFunction &CGF,
1791c1c9971cSAnders Carlsson                                           QualType DestTy) {
17922192fe50SChris Lattner   llvm::Type *DestLTy = CGF.ConvertType(DestTy);
1793c1c9971cSAnders Carlsson   if (DestTy->isPointerType())
1794c1c9971cSAnders Carlsson     return llvm::Constant::getNullValue(DestLTy);
1795c1c9971cSAnders Carlsson 
1796c1c9971cSAnders Carlsson   /// C++ [expr.dynamic.cast]p9:
1797c1c9971cSAnders Carlsson   ///   A failed cast to reference type throws std::bad_cast
1798c1c9971cSAnders Carlsson   EmitBadCastCall(CGF);
1799c1c9971cSAnders Carlsson 
1800c1c9971cSAnders Carlsson   CGF.EmitBlock(CGF.createBasicBlock("dynamic_cast.end"));
1801c1c9971cSAnders Carlsson   return llvm::UndefValue::get(DestLTy);
1802c1c9971cSAnders Carlsson }
1803c1c9971cSAnders Carlsson 
1804882d790fSAnders Carlsson llvm::Value *CodeGenFunction::EmitDynamicCast(llvm::Value *Value,
180559486a2dSAnders Carlsson                                               const CXXDynamicCastExpr *DCE) {
18063f4336cbSAnders Carlsson   QualType DestTy = DCE->getTypeAsWritten();
18073f4336cbSAnders Carlsson 
1808c1c9971cSAnders Carlsson   if (DCE->isAlwaysNull())
1809c1c9971cSAnders Carlsson     return EmitDynamicCastToNull(*this, DestTy);
1810c1c9971cSAnders Carlsson 
1811c1c9971cSAnders Carlsson   QualType SrcTy = DCE->getSubExpr()->getType();
1812c1c9971cSAnders Carlsson 
1813882d790fSAnders Carlsson   // C++ [expr.dynamic.cast]p4:
1814882d790fSAnders Carlsson   //   If the value of v is a null pointer value in the pointer case, the result
1815882d790fSAnders Carlsson   //   is the null pointer value of type T.
1816882d790fSAnders Carlsson   bool ShouldNullCheckSrcValue = SrcTy->isPointerType();
181759486a2dSAnders Carlsson 
1818882d790fSAnders Carlsson   llvm::BasicBlock *CastNull = 0;
1819882d790fSAnders Carlsson   llvm::BasicBlock *CastNotNull = 0;
1820882d790fSAnders Carlsson   llvm::BasicBlock *CastEnd = createBasicBlock("dynamic_cast.end");
1821fa8b4955SDouglas Gregor 
1822882d790fSAnders Carlsson   if (ShouldNullCheckSrcValue) {
1823882d790fSAnders Carlsson     CastNull = createBasicBlock("dynamic_cast.null");
1824882d790fSAnders Carlsson     CastNotNull = createBasicBlock("dynamic_cast.notnull");
1825882d790fSAnders Carlsson 
1826882d790fSAnders Carlsson     llvm::Value *IsNull = Builder.CreateIsNull(Value);
1827882d790fSAnders Carlsson     Builder.CreateCondBr(IsNull, CastNull, CastNotNull);
1828882d790fSAnders Carlsson     EmitBlock(CastNotNull);
182959486a2dSAnders Carlsson   }
183059486a2dSAnders Carlsson 
1831882d790fSAnders Carlsson   Value = EmitDynamicCastCall(*this, Value, SrcTy, DestTy, CastEnd);
18323f4336cbSAnders Carlsson 
1833882d790fSAnders Carlsson   if (ShouldNullCheckSrcValue) {
1834882d790fSAnders Carlsson     EmitBranch(CastEnd);
183559486a2dSAnders Carlsson 
1836882d790fSAnders Carlsson     EmitBlock(CastNull);
1837882d790fSAnders Carlsson     EmitBranch(CastEnd);
183859486a2dSAnders Carlsson   }
183959486a2dSAnders Carlsson 
1840882d790fSAnders Carlsson   EmitBlock(CastEnd);
184159486a2dSAnders Carlsson 
1842882d790fSAnders Carlsson   if (ShouldNullCheckSrcValue) {
1843882d790fSAnders Carlsson     llvm::PHINode *PHI = Builder.CreatePHI(Value->getType(), 2);
1844882d790fSAnders Carlsson     PHI->addIncoming(Value, CastNotNull);
1845882d790fSAnders Carlsson     PHI->addIncoming(llvm::Constant::getNullValue(Value->getType()), CastNull);
184659486a2dSAnders Carlsson 
1847882d790fSAnders Carlsson     Value = PHI;
184859486a2dSAnders Carlsson   }
184959486a2dSAnders Carlsson 
1850882d790fSAnders Carlsson   return Value;
185159486a2dSAnders Carlsson }
1852c370a7eeSEli Friedman 
1853c370a7eeSEli Friedman void CodeGenFunction::EmitLambdaExpr(const LambdaExpr *E, AggValueSlot Slot) {
18548631f3e8SEli Friedman   RunCleanupsScope Scope(*this);
18557f1ff600SEli Friedman   LValue SlotLV = MakeAddrLValue(Slot.getAddr(), E->getType(),
18567f1ff600SEli Friedman                                  Slot.getAlignment());
18578631f3e8SEli Friedman 
1858c370a7eeSEli Friedman   CXXRecordDecl::field_iterator CurField = E->getLambdaClass()->field_begin();
1859c370a7eeSEli Friedman   for (LambdaExpr::capture_init_iterator i = E->capture_init_begin(),
1860c370a7eeSEli Friedman                                          e = E->capture_init_end();
1861c370a7eeSEli Friedman        i != e; ++i, ++CurField) {
1862c370a7eeSEli Friedman     // Emit initialization
18637f1ff600SEli Friedman 
186440ed2973SDavid Blaikie     LValue LV = EmitLValueForFieldInitialization(SlotLV, *CurField);
18655f1a04ffSEli Friedman     ArrayRef<VarDecl *> ArrayIndexes;
18665f1a04ffSEli Friedman     if (CurField->getType()->isArrayType())
18675f1a04ffSEli Friedman       ArrayIndexes = E->getCaptureInitIndexVars(i);
186840ed2973SDavid Blaikie     EmitInitializerForField(*CurField, LV, *i, ArrayIndexes);
1869c370a7eeSEli Friedman   }
1870c370a7eeSEli Friedman }
1871