159486a2dSAnders Carlsson //===--- CGExprCXX.cpp - Emit LLVM Code for C++ expressions ---------------===// 259486a2dSAnders Carlsson // 359486a2dSAnders Carlsson // The LLVM Compiler Infrastructure 459486a2dSAnders Carlsson // 559486a2dSAnders Carlsson // This file is distributed under the University of Illinois Open Source 659486a2dSAnders Carlsson // License. See LICENSE.TXT for details. 759486a2dSAnders Carlsson // 859486a2dSAnders Carlsson //===----------------------------------------------------------------------===// 959486a2dSAnders Carlsson // 1059486a2dSAnders Carlsson // This contains code dealing with code generation of C++ expressions 1159486a2dSAnders Carlsson // 1259486a2dSAnders Carlsson //===----------------------------------------------------------------------===// 1359486a2dSAnders Carlsson 1459486a2dSAnders Carlsson #include "CodeGenFunction.h" 155d865c32SJohn McCall #include "CGCXXABI.h" 1660d215b6SFariborz Jahanian #include "CGObjCRuntime.h" 1726008e07SChris Lattner #include "llvm/Intrinsics.h" 1859486a2dSAnders Carlsson using namespace clang; 1959486a2dSAnders Carlsson using namespace CodeGen; 2059486a2dSAnders Carlsson 2127da15baSAnders Carlsson RValue CodeGenFunction::EmitCXXMemberCall(const CXXMethodDecl *MD, 2227da15baSAnders Carlsson llvm::Value *Callee, 2327da15baSAnders Carlsson ReturnValueSlot ReturnValue, 2427da15baSAnders Carlsson llvm::Value *This, 25e36a6b3eSAnders Carlsson llvm::Value *VTT, 2627da15baSAnders Carlsson CallExpr::const_arg_iterator ArgBeg, 2727da15baSAnders Carlsson CallExpr::const_arg_iterator ArgEnd) { 2827da15baSAnders Carlsson assert(MD->isInstance() && 2927da15baSAnders Carlsson "Trying to emit a member call expr on a static method!"); 3027da15baSAnders Carlsson 3127da15baSAnders Carlsson const FunctionProtoType *FPT = MD->getType()->getAs<FunctionProtoType>(); 3227da15baSAnders Carlsson 3327da15baSAnders Carlsson CallArgList Args; 3427da15baSAnders Carlsson 3527da15baSAnders Carlsson // Push the this ptr. 3627da15baSAnders Carlsson Args.push_back(std::make_pair(RValue::get(This), 3727da15baSAnders Carlsson MD->getThisType(getContext()))); 3827da15baSAnders Carlsson 39e36a6b3eSAnders Carlsson // If there is a VTT parameter, emit it. 40e36a6b3eSAnders Carlsson if (VTT) { 41e36a6b3eSAnders Carlsson QualType T = getContext().getPointerType(getContext().VoidPtrTy); 42e36a6b3eSAnders Carlsson Args.push_back(std::make_pair(RValue::get(VTT), T)); 43e36a6b3eSAnders Carlsson } 44e36a6b3eSAnders Carlsson 4527da15baSAnders Carlsson // And the rest of the call args 4627da15baSAnders Carlsson EmitCallArgs(Args, FPT, ArgBeg, ArgEnd); 4727da15baSAnders Carlsson 48ab26cfa5SJohn McCall QualType ResultType = FPT->getResultType(); 49ab26cfa5SJohn McCall return EmitCall(CGM.getTypes().getFunctionInfo(ResultType, Args, 50c50c27ccSRafael Espindola FPT->getExtInfo()), 51c50c27ccSRafael Espindola Callee, ReturnValue, Args, MD); 5227da15baSAnders Carlsson } 5327da15baSAnders Carlsson 5427da15baSAnders Carlsson /// canDevirtualizeMemberFunctionCalls - Checks whether virtual calls on given 5527da15baSAnders Carlsson /// expr can be devirtualized. 5627da15baSAnders Carlsson static bool canDevirtualizeMemberFunctionCalls(const Expr *Base) { 5727da15baSAnders Carlsson if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(Base)) { 5827da15baSAnders Carlsson if (const VarDecl *VD = dyn_cast<VarDecl>(DRE->getDecl())) { 5927da15baSAnders Carlsson // This is a record decl. We know the type and can devirtualize it. 6027da15baSAnders Carlsson return VD->getType()->isRecordType(); 6127da15baSAnders Carlsson } 6227da15baSAnders Carlsson 6327da15baSAnders Carlsson return false; 6427da15baSAnders Carlsson } 6527da15baSAnders Carlsson 6627da15baSAnders Carlsson // We can always devirtualize calls on temporary object expressions. 67a682427eSEli Friedman if (isa<CXXConstructExpr>(Base)) 6827da15baSAnders Carlsson return true; 6927da15baSAnders Carlsson 7027da15baSAnders Carlsson // And calls on bound temporaries. 7127da15baSAnders Carlsson if (isa<CXXBindTemporaryExpr>(Base)) 7227da15baSAnders Carlsson return true; 7327da15baSAnders Carlsson 7427da15baSAnders Carlsson // Check if this is a call expr that returns a record type. 7527da15baSAnders Carlsson if (const CallExpr *CE = dyn_cast<CallExpr>(Base)) 7627da15baSAnders Carlsson return CE->getCallReturnType()->isRecordType(); 7727da15baSAnders Carlsson 7827da15baSAnders Carlsson // We can't devirtualize the call. 7927da15baSAnders Carlsson return false; 8027da15baSAnders Carlsson } 8127da15baSAnders Carlsson 8227da15baSAnders Carlsson RValue CodeGenFunction::EmitCXXMemberCallExpr(const CXXMemberCallExpr *CE, 8327da15baSAnders Carlsson ReturnValueSlot ReturnValue) { 8427da15baSAnders Carlsson if (isa<BinaryOperator>(CE->getCallee()->IgnoreParens())) 8527da15baSAnders Carlsson return EmitCXXMemberPointerCallExpr(CE, ReturnValue); 8627da15baSAnders Carlsson 8727da15baSAnders Carlsson const MemberExpr *ME = cast<MemberExpr>(CE->getCallee()->IgnoreParens()); 8827da15baSAnders Carlsson const CXXMethodDecl *MD = cast<CXXMethodDecl>(ME->getMemberDecl()); 8927da15baSAnders Carlsson 9027da15baSAnders Carlsson if (MD->isStatic()) { 9127da15baSAnders Carlsson // The method is static, emit it as we would a regular call. 9227da15baSAnders Carlsson llvm::Value *Callee = CGM.GetAddrOfFunction(MD); 9327da15baSAnders Carlsson return EmitCall(getContext().getPointerType(MD->getType()), Callee, 9427da15baSAnders Carlsson ReturnValue, CE->arg_begin(), CE->arg_end()); 9527da15baSAnders Carlsson } 9627da15baSAnders Carlsson 970d635f53SJohn McCall // Compute the object pointer. 9827da15baSAnders Carlsson llvm::Value *This; 9927da15baSAnders Carlsson if (ME->isArrow()) 10027da15baSAnders Carlsson This = EmitScalarExpr(ME->getBase()); 10127da15baSAnders Carlsson else { 10227da15baSAnders Carlsson LValue BaseLV = EmitLValue(ME->getBase()); 103f93ac894SFariborz Jahanian if (BaseLV.isPropertyRef() || BaseLV.isKVCRef()) { 104f93ac894SFariborz Jahanian QualType QT = ME->getBase()->getType(); 105f93ac894SFariborz Jahanian RValue RV = 106f93ac894SFariborz Jahanian BaseLV.isPropertyRef() ? EmitLoadOfPropertyRefLValue(BaseLV, QT) 107f93ac894SFariborz Jahanian : EmitLoadOfKVCRefLValue(BaseLV, QT); 108f93ac894SFariborz Jahanian This = RV.isScalar() ? RV.getScalarVal() : RV.getAggregateAddr(); 109f93ac894SFariborz Jahanian } 110f93ac894SFariborz Jahanian else 11127da15baSAnders Carlsson This = BaseLV.getAddress(); 11227da15baSAnders Carlsson } 11327da15baSAnders Carlsson 1140d635f53SJohn McCall if (MD->isTrivial()) { 1150d635f53SJohn McCall if (isa<CXXDestructorDecl>(MD)) return RValue::get(0); 1160d635f53SJohn McCall 1170d635f53SJohn McCall assert(MD->isCopyAssignment() && "unknown trivial member function"); 11827da15baSAnders Carlsson // We don't like to generate the trivial copy assignment operator when 11927da15baSAnders Carlsson // it isn't necessary; just produce the proper effect here. 12027da15baSAnders Carlsson llvm::Value *RHS = EmitLValue(*CE->arg_begin()).getAddress(); 12127da15baSAnders Carlsson EmitAggregateCopy(This, RHS, CE->getType()); 12227da15baSAnders Carlsson return RValue::get(This); 12327da15baSAnders Carlsson } 12427da15baSAnders Carlsson 1250d635f53SJohn McCall // Compute the function type we're calling. 1260d635f53SJohn McCall const CGFunctionInfo &FInfo = 1270d635f53SJohn McCall (isa<CXXDestructorDecl>(MD) 1280d635f53SJohn McCall ? CGM.getTypes().getFunctionInfo(cast<CXXDestructorDecl>(MD), 1290d635f53SJohn McCall Dtor_Complete) 1300d635f53SJohn McCall : CGM.getTypes().getFunctionInfo(MD)); 1310d635f53SJohn McCall 1320d635f53SJohn McCall const FunctionProtoType *FPT = MD->getType()->getAs<FunctionProtoType>(); 1330d635f53SJohn McCall const llvm::Type *Ty 1340d635f53SJohn McCall = CGM.getTypes().GetFunctionType(FInfo, FPT->isVariadic()); 1350d635f53SJohn McCall 13627da15baSAnders Carlsson // C++ [class.virtual]p12: 13727da15baSAnders Carlsson // Explicit qualification with the scope operator (5.1) suppresses the 13827da15baSAnders Carlsson // virtual call mechanism. 13927da15baSAnders Carlsson // 14027da15baSAnders Carlsson // We also don't emit a virtual call if the base expression has a record type 14127da15baSAnders Carlsson // because then we know what the type is. 1420d635f53SJohn McCall bool UseVirtualCall = MD->isVirtual() && !ME->hasQualifier() 1430d635f53SJohn McCall && !canDevirtualizeMemberFunctionCalls(ME->getBase()); 1440d635f53SJohn McCall 14527da15baSAnders Carlsson llvm::Value *Callee; 1460d635f53SJohn McCall if (const CXXDestructorDecl *Dtor = dyn_cast<CXXDestructorDecl>(MD)) { 1470d635f53SJohn McCall if (UseVirtualCall) { 1480d635f53SJohn McCall Callee = BuildVirtualCall(Dtor, Dtor_Complete, This, Ty); 14927da15baSAnders Carlsson } else { 1500d635f53SJohn McCall Callee = CGM.GetAddrOfFunction(GlobalDecl(Dtor, Dtor_Complete), Ty); 15127da15baSAnders Carlsson } 1520d635f53SJohn McCall } else if (UseVirtualCall) { 15327da15baSAnders Carlsson Callee = BuildVirtualCall(MD, This, Ty); 15427da15baSAnders Carlsson } else { 15527da15baSAnders Carlsson Callee = CGM.GetAddrOfFunction(MD, Ty); 15627da15baSAnders Carlsson } 15727da15baSAnders Carlsson 158e36a6b3eSAnders Carlsson return EmitCXXMemberCall(MD, Callee, ReturnValue, This, /*VTT=*/0, 15927da15baSAnders Carlsson CE->arg_begin(), CE->arg_end()); 16027da15baSAnders Carlsson } 16127da15baSAnders Carlsson 16227da15baSAnders Carlsson RValue 16327da15baSAnders Carlsson CodeGenFunction::EmitCXXMemberPointerCallExpr(const CXXMemberCallExpr *E, 16427da15baSAnders Carlsson ReturnValueSlot ReturnValue) { 16527da15baSAnders Carlsson const BinaryOperator *BO = 16627da15baSAnders Carlsson cast<BinaryOperator>(E->getCallee()->IgnoreParens()); 16727da15baSAnders Carlsson const Expr *BaseExpr = BO->getLHS(); 16827da15baSAnders Carlsson const Expr *MemFnExpr = BO->getRHS(); 16927da15baSAnders Carlsson 17027da15baSAnders Carlsson const MemberPointerType *MPT = 17127da15baSAnders Carlsson MemFnExpr->getType()->getAs<MemberPointerType>(); 172475999dcSJohn McCall 17327da15baSAnders Carlsson const FunctionProtoType *FPT = 17427da15baSAnders Carlsson MPT->getPointeeType()->getAs<FunctionProtoType>(); 17527da15baSAnders Carlsson const CXXRecordDecl *RD = 17627da15baSAnders Carlsson cast<CXXRecordDecl>(MPT->getClass()->getAs<RecordType>()->getDecl()); 17727da15baSAnders Carlsson 17827da15baSAnders Carlsson // Get the member function pointer. 179a1dee530SJohn McCall llvm::Value *MemFnPtr = EmitScalarExpr(MemFnExpr); 18027da15baSAnders Carlsson 18127da15baSAnders Carlsson // Emit the 'this' pointer. 18227da15baSAnders Carlsson llvm::Value *This; 18327da15baSAnders Carlsson 184e302792bSJohn McCall if (BO->getOpcode() == BO_PtrMemI) 18527da15baSAnders Carlsson This = EmitScalarExpr(BaseExpr); 18627da15baSAnders Carlsson else 18727da15baSAnders Carlsson This = EmitLValue(BaseExpr).getAddress(); 18827da15baSAnders Carlsson 189475999dcSJohn McCall // Ask the ABI to load the callee. Note that This is modified. 190475999dcSJohn McCall llvm::Value *Callee = 191475999dcSJohn McCall CGM.getCXXABI().EmitLoadOfMemberFunctionPointer(CGF, This, MemFnPtr, MPT); 19227da15baSAnders Carlsson 19327da15baSAnders Carlsson CallArgList Args; 19427da15baSAnders Carlsson 19527da15baSAnders Carlsson QualType ThisType = 19627da15baSAnders Carlsson getContext().getPointerType(getContext().getTagDeclType(RD)); 19727da15baSAnders Carlsson 19827da15baSAnders Carlsson // Push the this ptr. 19927da15baSAnders Carlsson Args.push_back(std::make_pair(RValue::get(This), ThisType)); 20027da15baSAnders Carlsson 20127da15baSAnders Carlsson // And the rest of the call args 20227da15baSAnders Carlsson EmitCallArgs(Args, FPT, E->arg_begin(), E->arg_end()); 203ab26cfa5SJohn McCall const FunctionType *BO_FPT = BO->getType()->getAs<FunctionProtoType>(); 204ab26cfa5SJohn McCall return EmitCall(CGM.getTypes().getFunctionInfo(Args, BO_FPT), Callee, 20527da15baSAnders Carlsson ReturnValue, Args); 20627da15baSAnders Carlsson } 20727da15baSAnders Carlsson 20827da15baSAnders Carlsson RValue 20927da15baSAnders Carlsson CodeGenFunction::EmitCXXOperatorMemberCallExpr(const CXXOperatorCallExpr *E, 21027da15baSAnders Carlsson const CXXMethodDecl *MD, 21127da15baSAnders Carlsson ReturnValueSlot ReturnValue) { 21227da15baSAnders Carlsson assert(MD->isInstance() && 21327da15baSAnders Carlsson "Trying to emit a member call expr on a static method!"); 21427da15baSAnders Carlsson if (MD->isCopyAssignment()) { 21527da15baSAnders Carlsson const CXXRecordDecl *ClassDecl = cast<CXXRecordDecl>(MD->getDeclContext()); 21627da15baSAnders Carlsson if (ClassDecl->hasTrivialCopyAssignment()) { 21727da15baSAnders Carlsson assert(!ClassDecl->hasUserDeclaredCopyAssignment() && 21827da15baSAnders Carlsson "EmitCXXOperatorMemberCallExpr - user declared copy assignment"); 21943a40f93SFariborz Jahanian LValue LV = EmitLValue(E->getArg(0)); 22043a40f93SFariborz Jahanian llvm::Value *This; 22161a31241SFariborz Jahanian if (LV.isPropertyRef() || LV.isKVCRef()) { 222*7a626f63SJohn McCall AggValueSlot Slot = CreateAggTemp(E->getArg(1)->getType()); 223*7a626f63SJohn McCall EmitAggExpr(E->getArg(1), Slot); 22461a31241SFariborz Jahanian if (LV.isPropertyRef()) 225*7a626f63SJohn McCall EmitObjCPropertySet(LV.getPropertyRefExpr(), Slot.asRValue()); 22661a31241SFariborz Jahanian else 227*7a626f63SJohn McCall EmitObjCPropertySet(LV.getKVCRefExpr(), Slot.asRValue()); 228e1b45a5eSFariborz Jahanian return RValue::getAggregate(0, false); 22943a40f93SFariborz Jahanian } 23043a40f93SFariborz Jahanian else 23143a40f93SFariborz Jahanian This = LV.getAddress(); 23243a40f93SFariborz Jahanian 23327da15baSAnders Carlsson llvm::Value *Src = EmitLValue(E->getArg(1)).getAddress(); 23427da15baSAnders Carlsson QualType Ty = E->getType(); 23527da15baSAnders Carlsson EmitAggregateCopy(This, Src, Ty); 23627da15baSAnders Carlsson return RValue::get(This); 23727da15baSAnders Carlsson } 23827da15baSAnders Carlsson } 23927da15baSAnders Carlsson 24027da15baSAnders Carlsson const FunctionProtoType *FPT = MD->getType()->getAs<FunctionProtoType>(); 24127da15baSAnders Carlsson const llvm::Type *Ty = 24227da15baSAnders Carlsson CGM.getTypes().GetFunctionType(CGM.getTypes().getFunctionInfo(MD), 24327da15baSAnders Carlsson FPT->isVariadic()); 244fdf474b0SFariborz Jahanian LValue LV = EmitLValue(E->getArg(0)); 245fdf474b0SFariborz Jahanian llvm::Value *This; 24661a31241SFariborz Jahanian if (LV.isPropertyRef() || LV.isKVCRef()) { 24761a31241SFariborz Jahanian QualType QT = E->getArg(0)->getType(); 24861a31241SFariborz Jahanian RValue RV = 24961a31241SFariborz Jahanian LV.isPropertyRef() ? EmitLoadOfPropertyRefLValue(LV, QT) 25061a31241SFariborz Jahanian : EmitLoadOfKVCRefLValue(LV, QT); 2516855ba2cSFariborz Jahanian assert (!RV.isScalar() && "EmitCXXOperatorMemberCallExpr"); 2526855ba2cSFariborz Jahanian This = RV.getAggregateAddr(); 253fdf474b0SFariborz Jahanian } 254fdf474b0SFariborz Jahanian else 255fdf474b0SFariborz Jahanian This = LV.getAddress(); 25627da15baSAnders Carlsson 25727da15baSAnders Carlsson llvm::Value *Callee; 25827da15baSAnders Carlsson if (MD->isVirtual() && !canDevirtualizeMemberFunctionCalls(E->getArg(0))) 25927da15baSAnders Carlsson Callee = BuildVirtualCall(MD, This, Ty); 26027da15baSAnders Carlsson else 26127da15baSAnders Carlsson Callee = CGM.GetAddrOfFunction(MD, Ty); 26227da15baSAnders Carlsson 263e36a6b3eSAnders Carlsson return EmitCXXMemberCall(MD, Callee, ReturnValue, This, /*VTT=*/0, 26427da15baSAnders Carlsson E->arg_begin() + 1, E->arg_end()); 26527da15baSAnders Carlsson } 26627da15baSAnders Carlsson 26727da15baSAnders Carlsson void 268*7a626f63SJohn McCall CodeGenFunction::EmitCXXConstructExpr(const CXXConstructExpr *E, 269*7a626f63SJohn McCall AggValueSlot Dest) { 270*7a626f63SJohn McCall assert(!Dest.isIgnored() && "Must have a destination!"); 27127da15baSAnders Carlsson const CXXConstructorDecl *CD = E->getConstructor(); 272630c76efSDouglas Gregor 273630c76efSDouglas Gregor // If we require zero initialization before (or instead of) calling the 274630c76efSDouglas Gregor // constructor, as can be the case with a non-user-provided default 275630c76efSDouglas Gregor // constructor, emit the zero initialization now. 276e3b3464dSDouglas Gregor if (E->requiresZeroInitialization()) 277*7a626f63SJohn McCall EmitNullInitialization(Dest.getAddr(), E->getType()); 278630c76efSDouglas Gregor 279630c76efSDouglas Gregor // If this is a call to a trivial default constructor, do nothing. 280630c76efSDouglas Gregor if (CD->isTrivial() && CD->isDefaultConstructor()) 28127da15baSAnders Carlsson return; 282630c76efSDouglas Gregor 28327da15baSAnders Carlsson // Code gen optimization to eliminate copy constructor and return 284222cf0efSDouglas Gregor // its first argument instead, if in fact that argument is a temporary 285222cf0efSDouglas Gregor // object. 28627da15baSAnders Carlsson if (getContext().getLangOptions().ElideConstructors && E->isElidable()) { 287*7a626f63SJohn McCall if (E->getArg(0)->isTemporaryObject(getContext(), CD->getParent())) { 288*7a626f63SJohn McCall EmitAggExpr(E->getArg(0), Dest); 28927da15baSAnders Carlsson return; 29027da15baSAnders Carlsson } 291222cf0efSDouglas Gregor } 292630c76efSDouglas Gregor 293630c76efSDouglas Gregor const ConstantArrayType *Array 294630c76efSDouglas Gregor = getContext().getAsConstantArrayType(E->getType()); 29527da15baSAnders Carlsson if (Array) { 29627da15baSAnders Carlsson QualType BaseElementTy = getContext().getBaseElementType(Array); 29727da15baSAnders Carlsson const llvm::Type *BasePtr = ConvertType(BaseElementTy); 29827da15baSAnders Carlsson BasePtr = llvm::PointerType::getUnqual(BasePtr); 29927da15baSAnders Carlsson llvm::Value *BaseAddrPtr = 300*7a626f63SJohn McCall Builder.CreateBitCast(Dest.getAddr(), BasePtr); 30127da15baSAnders Carlsson 30227da15baSAnders Carlsson EmitCXXAggrConstructorCall(CD, Array, BaseAddrPtr, 30327da15baSAnders Carlsson E->arg_begin(), E->arg_end()); 30427da15baSAnders Carlsson } 305e11f9ce9SAnders Carlsson else { 306e11f9ce9SAnders Carlsson CXXCtorType Type = 307e11f9ce9SAnders Carlsson (E->getConstructionKind() == CXXConstructExpr::CK_Complete) 308e11f9ce9SAnders Carlsson ? Ctor_Complete : Ctor_Base; 309e11f9ce9SAnders Carlsson bool ForVirtualBase = 310e11f9ce9SAnders Carlsson E->getConstructionKind() == CXXConstructExpr::CK_VirtualBase; 311e11f9ce9SAnders Carlsson 31227da15baSAnders Carlsson // Call the constructor. 313*7a626f63SJohn McCall EmitCXXConstructorCall(CD, Type, ForVirtualBase, Dest.getAddr(), 31427da15baSAnders Carlsson E->arg_begin(), E->arg_end()); 31527da15baSAnders Carlsson } 316e11f9ce9SAnders Carlsson } 31727da15baSAnders Carlsson 318aa4149a2SJohn McCall /// Check whether the given operator new[] is the global placement 319aa4149a2SJohn McCall /// operator new[]. 320aa4149a2SJohn McCall static bool IsPlacementOperatorNewArray(ASTContext &Ctx, 321aa4149a2SJohn McCall const FunctionDecl *Fn) { 322aa4149a2SJohn McCall // Must be in global scope. Note that allocation functions can't be 323aa4149a2SJohn McCall // declared in namespaces. 32450c68258SSebastian Redl if (!Fn->getDeclContext()->getRedeclContext()->isFileContext()) 325aa4149a2SJohn McCall return false; 326aa4149a2SJohn McCall 327aa4149a2SJohn McCall // Signature must be void *operator new[](size_t, void*). 328aa4149a2SJohn McCall // The size_t is common to all operator new[]s. 329aa4149a2SJohn McCall if (Fn->getNumParams() != 2) 330aa4149a2SJohn McCall return false; 331aa4149a2SJohn McCall 332aa4149a2SJohn McCall CanQualType ParamType = Ctx.getCanonicalType(Fn->getParamDecl(1)->getType()); 333aa4149a2SJohn McCall return (ParamType == Ctx.VoidPtrTy); 334aa4149a2SJohn McCall } 335aa4149a2SJohn McCall 3368ed55a54SJohn McCall static CharUnits CalculateCookiePadding(CodeGenFunction &CGF, 3378ed55a54SJohn McCall const CXXNewExpr *E) { 33821122cf6SAnders Carlsson if (!E->isArray()) 3393eb55cfeSKen Dyck return CharUnits::Zero(); 34021122cf6SAnders Carlsson 341399f499fSAnders Carlsson // No cookie is required if the new operator being used is 342399f499fSAnders Carlsson // ::operator new[](size_t, void*). 343399f499fSAnders Carlsson const FunctionDecl *OperatorNew = E->getOperatorNew(); 3448ed55a54SJohn McCall if (IsPlacementOperatorNewArray(CGF.getContext(), OperatorNew)) 3453eb55cfeSKen Dyck return CharUnits::Zero(); 346399f499fSAnders Carlsson 3478ed55a54SJohn McCall return CGF.CGM.getCXXABI().GetArrayCookieSize(E->getAllocatedType()); 34859486a2dSAnders Carlsson } 34959486a2dSAnders Carlsson 35047b4629bSFariborz Jahanian static llvm::Value *EmitCXXNewAllocSize(ASTContext &Context, 35147b4629bSFariborz Jahanian CodeGenFunction &CGF, 35259486a2dSAnders Carlsson const CXXNewExpr *E, 35305fc5be3SDouglas Gregor llvm::Value *&NumElements, 35405fc5be3SDouglas Gregor llvm::Value *&SizeWithoutCookie) { 3557648fb46SArgyrios Kyrtzidis QualType ElemType = E->getAllocatedType(); 35659486a2dSAnders Carlsson 3578ed55a54SJohn McCall const llvm::IntegerType *SizeTy = 3588ed55a54SJohn McCall cast<llvm::IntegerType>(CGF.ConvertType(CGF.getContext().getSizeType())); 3598ed55a54SJohn McCall 3607648fb46SArgyrios Kyrtzidis CharUnits TypeSize = CGF.getContext().getTypeSizeInChars(ElemType); 3618ed55a54SJohn McCall 3628ed55a54SJohn McCall if (!E->isArray()) { 36305fc5be3SDouglas Gregor SizeWithoutCookie = llvm::ConstantInt::get(SizeTy, TypeSize.getQuantity()); 36405fc5be3SDouglas Gregor return SizeWithoutCookie; 36505fc5be3SDouglas Gregor } 36659486a2dSAnders Carlsson 3678ed55a54SJohn McCall // Figure out the cookie size. 3688ed55a54SJohn McCall CharUnits CookieSize = CalculateCookiePadding(CGF, E); 3698ed55a54SJohn McCall 37059486a2dSAnders Carlsson // Emit the array size expression. 3717648fb46SArgyrios Kyrtzidis // We multiply the size of all dimensions for NumElements. 3727648fb46SArgyrios Kyrtzidis // e.g for 'int[2][3]', ElemType is 'int' and NumElements is 6. 37359486a2dSAnders Carlsson NumElements = CGF.EmitScalarExpr(E->getArraySize()); 3748ed55a54SJohn McCall assert(NumElements->getType() == SizeTy && "element count not a size_t"); 3758ed55a54SJohn McCall 3768ed55a54SJohn McCall uint64_t ArraySizeMultiplier = 1; 3777648fb46SArgyrios Kyrtzidis while (const ConstantArrayType *CAT 3787648fb46SArgyrios Kyrtzidis = CGF.getContext().getAsConstantArrayType(ElemType)) { 3797648fb46SArgyrios Kyrtzidis ElemType = CAT->getElementType(); 3808ed55a54SJohn McCall ArraySizeMultiplier *= CAT->getSize().getZExtValue(); 3817648fb46SArgyrios Kyrtzidis } 38259486a2dSAnders Carlsson 3838ed55a54SJohn McCall llvm::Value *Size; 38432ac583dSChris Lattner 38532ac583dSChris Lattner // If someone is doing 'new int[42]' there is no need to do a dynamic check. 38632ac583dSChris Lattner // Don't bloat the -O0 code. 38732ac583dSChris Lattner if (llvm::ConstantInt *NumElementsC = 38832ac583dSChris Lattner dyn_cast<llvm::ConstantInt>(NumElements)) { 38932ac583dSChris Lattner llvm::APInt NEC = NumElementsC->getValue(); 3908ed55a54SJohn McCall unsigned SizeWidth = NEC.getBitWidth(); 39132ac583dSChris Lattner 3928ed55a54SJohn McCall // Determine if there is an overflow here by doing an extended multiply. 3938ed55a54SJohn McCall NEC.zext(SizeWidth*2); 3948ed55a54SJohn McCall llvm::APInt SC(SizeWidth*2, TypeSize.getQuantity()); 39532ac583dSChris Lattner SC *= NEC; 39632ac583dSChris Lattner 3978ed55a54SJohn McCall if (!CookieSize.isZero()) { 3988ed55a54SJohn McCall // Save the current size without a cookie. We don't care if an 3998ed55a54SJohn McCall // overflow's already happened because SizeWithoutCookie isn't 4008ed55a54SJohn McCall // used if the allocator returns null or throws, as it should 4018ed55a54SJohn McCall // always do on an overflow. 4028ed55a54SJohn McCall llvm::APInt SWC = SC; 4038ed55a54SJohn McCall SWC.trunc(SizeWidth); 4048ed55a54SJohn McCall SizeWithoutCookie = llvm::ConstantInt::get(SizeTy, SWC); 4058ed55a54SJohn McCall 4068ed55a54SJohn McCall // Add the cookie size. 4078ed55a54SJohn McCall SC += llvm::APInt(SizeWidth*2, CookieSize.getQuantity()); 4088ed55a54SJohn McCall } 4098ed55a54SJohn McCall 4108ed55a54SJohn McCall if (SC.countLeadingZeros() >= SizeWidth) { 4118ed55a54SJohn McCall SC.trunc(SizeWidth); 4128ed55a54SJohn McCall Size = llvm::ConstantInt::get(SizeTy, SC); 41332ac583dSChris Lattner } else { 41432ac583dSChris Lattner // On overflow, produce a -1 so operator new throws. 4158ed55a54SJohn McCall Size = llvm::Constant::getAllOnesValue(SizeTy); 41632ac583dSChris Lattner } 41732ac583dSChris Lattner 4188ed55a54SJohn McCall // Scale NumElements while we're at it. 4198ed55a54SJohn McCall uint64_t N = NEC.getZExtValue() * ArraySizeMultiplier; 4208ed55a54SJohn McCall NumElements = llvm::ConstantInt::get(SizeTy, N); 42147b4629bSFariborz Jahanian 4228ed55a54SJohn McCall // Otherwise, we don't need to do an overflow-checked multiplication if 4238ed55a54SJohn McCall // we're multiplying by one. 4248ed55a54SJohn McCall } else if (TypeSize.isOne()) { 4258ed55a54SJohn McCall assert(ArraySizeMultiplier == 1); 426f2f38701SChris Lattner 4278ed55a54SJohn McCall Size = NumElements; 428f2f38701SChris Lattner 4298ed55a54SJohn McCall // If we need a cookie, add its size in with an overflow check. 4308ed55a54SJohn McCall // This is maybe a little paranoid. 4318ed55a54SJohn McCall if (!CookieSize.isZero()) { 43205fc5be3SDouglas Gregor SizeWithoutCookie = Size; 433f2f38701SChris Lattner 4348ed55a54SJohn McCall llvm::Value *CookieSizeV 4358ed55a54SJohn McCall = llvm::ConstantInt::get(SizeTy, CookieSize.getQuantity()); 4368ed55a54SJohn McCall 4378ed55a54SJohn McCall const llvm::Type *Types[] = { SizeTy }; 4388ed55a54SJohn McCall llvm::Value *UAddF 4398ed55a54SJohn McCall = CGF.CGM.getIntrinsic(llvm::Intrinsic::uadd_with_overflow, Types, 1); 4408ed55a54SJohn McCall llvm::Value *AddRes 4418ed55a54SJohn McCall = CGF.Builder.CreateCall2(UAddF, Size, CookieSizeV); 4428ed55a54SJohn McCall 4438ed55a54SJohn McCall Size = CGF.Builder.CreateExtractValue(AddRes, 0); 4448ed55a54SJohn McCall llvm::Value *DidOverflow = CGF.Builder.CreateExtractValue(AddRes, 1); 4458ed55a54SJohn McCall Size = CGF.Builder.CreateSelect(DidOverflow, 4468ed55a54SJohn McCall llvm::ConstantInt::get(SizeTy, -1), 4478ed55a54SJohn McCall Size); 4488ed55a54SJohn McCall } 4498ed55a54SJohn McCall 4508ed55a54SJohn McCall // Otherwise use the int.umul.with.overflow intrinsic. 4518ed55a54SJohn McCall } else { 4528ed55a54SJohn McCall llvm::Value *OutermostElementSize 4538ed55a54SJohn McCall = llvm::ConstantInt::get(SizeTy, TypeSize.getQuantity()); 4548ed55a54SJohn McCall 4558ed55a54SJohn McCall llvm::Value *NumOutermostElements = NumElements; 4568ed55a54SJohn McCall 4578ed55a54SJohn McCall // Scale NumElements by the array size multiplier. This might 4588ed55a54SJohn McCall // overflow, but only if the multiplication below also overflows, 4598ed55a54SJohn McCall // in which case this multiplication isn't used. 4608ed55a54SJohn McCall if (ArraySizeMultiplier != 1) 4618ed55a54SJohn McCall NumElements = CGF.Builder.CreateMul(NumElements, 4628ed55a54SJohn McCall llvm::ConstantInt::get(SizeTy, ArraySizeMultiplier)); 4638ed55a54SJohn McCall 4648ed55a54SJohn McCall // The requested size of the outermost array is non-constant. 4658ed55a54SJohn McCall // Multiply that by the static size of the elements of that array; 4668ed55a54SJohn McCall // on unsigned overflow, set the size to -1 to trigger an 4678ed55a54SJohn McCall // exception from the allocation routine. This is sufficient to 4688ed55a54SJohn McCall // prevent buffer overruns from the allocator returning a 4698ed55a54SJohn McCall // seemingly valid pointer to insufficient space. This idea comes 4708ed55a54SJohn McCall // originally from MSVC, and GCC has an open bug requesting 4718ed55a54SJohn McCall // similar behavior: 4728ed55a54SJohn McCall // http://gcc.gnu.org/bugzilla/show_bug.cgi?id=19351 4738ed55a54SJohn McCall // 4748ed55a54SJohn McCall // This will not be sufficient for C++0x, which requires a 4758ed55a54SJohn McCall // specific exception class (std::bad_array_new_length). 4768ed55a54SJohn McCall // That will require ABI support that has not yet been specified. 4778ed55a54SJohn McCall const llvm::Type *Types[] = { SizeTy }; 4788ed55a54SJohn McCall llvm::Value *UMulF 4798ed55a54SJohn McCall = CGF.CGM.getIntrinsic(llvm::Intrinsic::umul_with_overflow, Types, 1); 4808ed55a54SJohn McCall llvm::Value *MulRes = CGF.Builder.CreateCall2(UMulF, NumOutermostElements, 4818ed55a54SJohn McCall OutermostElementSize); 4828ed55a54SJohn McCall 4838ed55a54SJohn McCall // The overflow bit. 4848ed55a54SJohn McCall llvm::Value *DidOverflow = CGF.Builder.CreateExtractValue(MulRes, 1); 4858ed55a54SJohn McCall 4868ed55a54SJohn McCall // The result of the multiplication. 4878ed55a54SJohn McCall Size = CGF.Builder.CreateExtractValue(MulRes, 0); 4888ed55a54SJohn McCall 4898ed55a54SJohn McCall // If we have a cookie, we need to add that size in, too. 4908ed55a54SJohn McCall if (!CookieSize.isZero()) { 4918ed55a54SJohn McCall SizeWithoutCookie = Size; 4928ed55a54SJohn McCall 4938ed55a54SJohn McCall llvm::Value *CookieSizeV 4948ed55a54SJohn McCall = llvm::ConstantInt::get(SizeTy, CookieSize.getQuantity()); 4958ed55a54SJohn McCall llvm::Value *UAddF 4968ed55a54SJohn McCall = CGF.CGM.getIntrinsic(llvm::Intrinsic::uadd_with_overflow, Types, 1); 4978ed55a54SJohn McCall llvm::Value *AddRes 4988ed55a54SJohn McCall = CGF.Builder.CreateCall2(UAddF, SizeWithoutCookie, CookieSizeV); 4998ed55a54SJohn McCall 5008ed55a54SJohn McCall Size = CGF.Builder.CreateExtractValue(AddRes, 0); 5018ed55a54SJohn McCall 5028ed55a54SJohn McCall llvm::Value *AddDidOverflow = CGF.Builder.CreateExtractValue(AddRes, 1); 5038ed55a54SJohn McCall DidOverflow = CGF.Builder.CreateAnd(DidOverflow, AddDidOverflow); 5048ed55a54SJohn McCall } 5058ed55a54SJohn McCall 5068ed55a54SJohn McCall Size = CGF.Builder.CreateSelect(DidOverflow, 5078ed55a54SJohn McCall llvm::ConstantInt::get(SizeTy, -1), 5088ed55a54SJohn McCall Size); 5098ed55a54SJohn McCall } 5108ed55a54SJohn McCall 5118ed55a54SJohn McCall if (CookieSize.isZero()) 5128ed55a54SJohn McCall SizeWithoutCookie = Size; 5138ed55a54SJohn McCall else 5148ed55a54SJohn McCall assert(SizeWithoutCookie && "didn't set SizeWithoutCookie?"); 51559486a2dSAnders Carlsson 51632ac583dSChris Lattner return Size; 51759486a2dSAnders Carlsson } 51859486a2dSAnders Carlsson 519d5202e09SFariborz Jahanian static void StoreAnyExprIntoOneUnit(CodeGenFunction &CGF, const CXXNewExpr *E, 520d5202e09SFariborz Jahanian llvm::Value *NewPtr) { 521d5202e09SFariborz Jahanian 522d5202e09SFariborz Jahanian assert(E->getNumConstructorArgs() == 1 && 523d5202e09SFariborz Jahanian "Can only have one argument to initializer of POD type."); 524d5202e09SFariborz Jahanian 525d5202e09SFariborz Jahanian const Expr *Init = E->getConstructorArg(0); 526d5202e09SFariborz Jahanian QualType AllocType = E->getAllocatedType(); 527d5202e09SFariborz Jahanian 5280381634aSDaniel Dunbar unsigned Alignment = 5290381634aSDaniel Dunbar CGF.getContext().getTypeAlignInChars(AllocType).getQuantity(); 530d5202e09SFariborz Jahanian if (!CGF.hasAggregateLLVMType(AllocType)) 531d5202e09SFariborz Jahanian CGF.EmitStoreOfScalar(CGF.EmitScalarExpr(Init), NewPtr, 5320381634aSDaniel Dunbar AllocType.isVolatileQualified(), Alignment, 5330381634aSDaniel Dunbar AllocType); 534d5202e09SFariborz Jahanian else if (AllocType->isAnyComplexType()) 535d5202e09SFariborz Jahanian CGF.EmitComplexExprIntoAddr(Init, NewPtr, 536d5202e09SFariborz Jahanian AllocType.isVolatileQualified()); 537*7a626f63SJohn McCall else { 538*7a626f63SJohn McCall AggValueSlot Slot 539*7a626f63SJohn McCall = AggValueSlot::forAddr(NewPtr, AllocType.isVolatileQualified(), true); 540*7a626f63SJohn McCall CGF.EmitAggExpr(Init, Slot); 541*7a626f63SJohn McCall } 542d5202e09SFariborz Jahanian } 543d5202e09SFariborz Jahanian 544d5202e09SFariborz Jahanian void 545d5202e09SFariborz Jahanian CodeGenFunction::EmitNewArrayInitializer(const CXXNewExpr *E, 546d5202e09SFariborz Jahanian llvm::Value *NewPtr, 547d5202e09SFariborz Jahanian llvm::Value *NumElements) { 548b66b08efSFariborz Jahanian // We have a POD type. 549b66b08efSFariborz Jahanian if (E->getNumConstructorArgs() == 0) 550b66b08efSFariborz Jahanian return; 551b66b08efSFariborz Jahanian 552d5202e09SFariborz Jahanian const llvm::Type *SizeTy = ConvertType(getContext().getSizeType()); 553d5202e09SFariborz Jahanian 554d5202e09SFariborz Jahanian // Create a temporary for the loop index and initialize it with 0. 555d5202e09SFariborz Jahanian llvm::Value *IndexPtr = CreateTempAlloca(SizeTy, "loop.index"); 556d5202e09SFariborz Jahanian llvm::Value *Zero = llvm::Constant::getNullValue(SizeTy); 557d5202e09SFariborz Jahanian Builder.CreateStore(Zero, IndexPtr); 558d5202e09SFariborz Jahanian 559d5202e09SFariborz Jahanian // Start the loop with a block that tests the condition. 560d5202e09SFariborz Jahanian llvm::BasicBlock *CondBlock = createBasicBlock("for.cond"); 561d5202e09SFariborz Jahanian llvm::BasicBlock *AfterFor = createBasicBlock("for.end"); 562d5202e09SFariborz Jahanian 563d5202e09SFariborz Jahanian EmitBlock(CondBlock); 564d5202e09SFariborz Jahanian 565d5202e09SFariborz Jahanian llvm::BasicBlock *ForBody = createBasicBlock("for.body"); 566d5202e09SFariborz Jahanian 567d5202e09SFariborz Jahanian // Generate: if (loop-index < number-of-elements fall to the loop body, 568d5202e09SFariborz Jahanian // otherwise, go to the block after the for-loop. 569d5202e09SFariborz Jahanian llvm::Value *Counter = Builder.CreateLoad(IndexPtr); 570d5202e09SFariborz Jahanian llvm::Value *IsLess = Builder.CreateICmpULT(Counter, NumElements, "isless"); 571d5202e09SFariborz Jahanian // If the condition is true, execute the body. 572d5202e09SFariborz Jahanian Builder.CreateCondBr(IsLess, ForBody, AfterFor); 573d5202e09SFariborz Jahanian 574d5202e09SFariborz Jahanian EmitBlock(ForBody); 575d5202e09SFariborz Jahanian 576d5202e09SFariborz Jahanian llvm::BasicBlock *ContinueBlock = createBasicBlock("for.inc"); 577d5202e09SFariborz Jahanian // Inside the loop body, emit the constructor call on the array element. 578d5202e09SFariborz Jahanian Counter = Builder.CreateLoad(IndexPtr); 579d5202e09SFariborz Jahanian llvm::Value *Address = Builder.CreateInBoundsGEP(NewPtr, Counter, 580d5202e09SFariborz Jahanian "arrayidx"); 581d5202e09SFariborz Jahanian StoreAnyExprIntoOneUnit(*this, E, Address); 582d5202e09SFariborz Jahanian 583d5202e09SFariborz Jahanian EmitBlock(ContinueBlock); 584d5202e09SFariborz Jahanian 585d5202e09SFariborz Jahanian // Emit the increment of the loop counter. 586d5202e09SFariborz Jahanian llvm::Value *NextVal = llvm::ConstantInt::get(SizeTy, 1); 587d5202e09SFariborz Jahanian Counter = Builder.CreateLoad(IndexPtr); 588d5202e09SFariborz Jahanian NextVal = Builder.CreateAdd(Counter, NextVal, "inc"); 589d5202e09SFariborz Jahanian Builder.CreateStore(NextVal, IndexPtr); 590d5202e09SFariborz Jahanian 591d5202e09SFariborz Jahanian // Finally, branch back up to the condition for the next iteration. 592d5202e09SFariborz Jahanian EmitBranch(CondBlock); 593d5202e09SFariborz Jahanian 594d5202e09SFariborz Jahanian // Emit the fall-through block. 595d5202e09SFariborz Jahanian EmitBlock(AfterFor, true); 596d5202e09SFariborz Jahanian } 597d5202e09SFariborz Jahanian 59805fc5be3SDouglas Gregor static void EmitZeroMemSet(CodeGenFunction &CGF, QualType T, 59905fc5be3SDouglas Gregor llvm::Value *NewPtr, llvm::Value *Size) { 60005fc5be3SDouglas Gregor llvm::LLVMContext &VMContext = CGF.CGM.getLLVMContext(); 60105fc5be3SDouglas Gregor const llvm::Type *BP = llvm::Type::getInt8PtrTy(VMContext); 60205fc5be3SDouglas Gregor if (NewPtr->getType() != BP) 60305fc5be3SDouglas Gregor NewPtr = CGF.Builder.CreateBitCast(NewPtr, BP, "tmp"); 60405fc5be3SDouglas Gregor 60505fc5be3SDouglas Gregor CGF.Builder.CreateCall5(CGF.CGM.getMemSetFn(BP, CGF.IntPtrTy), NewPtr, 60605fc5be3SDouglas Gregor llvm::Constant::getNullValue(llvm::Type::getInt8Ty(VMContext)), 60705fc5be3SDouglas Gregor Size, 60805fc5be3SDouglas Gregor llvm::ConstantInt::get(CGF.Int32Ty, 60905fc5be3SDouglas Gregor CGF.getContext().getTypeAlign(T)/8), 61005fc5be3SDouglas Gregor llvm::ConstantInt::get(llvm::Type::getInt1Ty(VMContext), 61105fc5be3SDouglas Gregor 0)); 61205fc5be3SDouglas Gregor } 61305fc5be3SDouglas Gregor 61459486a2dSAnders Carlsson static void EmitNewInitializer(CodeGenFunction &CGF, const CXXNewExpr *E, 61559486a2dSAnders Carlsson llvm::Value *NewPtr, 61605fc5be3SDouglas Gregor llvm::Value *NumElements, 61705fc5be3SDouglas Gregor llvm::Value *AllocSizeWithoutCookie) { 6183a202f60SAnders Carlsson if (E->isArray()) { 619d040e6b2SAnders Carlsson if (CXXConstructorDecl *Ctor = E->getConstructor()) { 62005fc5be3SDouglas Gregor bool RequiresZeroInitialization = false; 62105fc5be3SDouglas Gregor if (Ctor->getParent()->hasTrivialConstructor()) { 62205fc5be3SDouglas Gregor // If new expression did not specify value-initialization, then there 62305fc5be3SDouglas Gregor // is no initialization. 62405fc5be3SDouglas Gregor if (!E->hasInitializer() || Ctor->getParent()->isEmpty()) 62505fc5be3SDouglas Gregor return; 62605fc5be3SDouglas Gregor 627614dbdcdSJohn McCall if (CGF.CGM.getTypes().isZeroInitializable(E->getAllocatedType())) { 62805fc5be3SDouglas Gregor // Optimization: since zero initialization will just set the memory 62905fc5be3SDouglas Gregor // to all zeroes, generate a single memset to do it in one shot. 63005fc5be3SDouglas Gregor EmitZeroMemSet(CGF, E->getAllocatedType(), NewPtr, 63105fc5be3SDouglas Gregor AllocSizeWithoutCookie); 6323a202f60SAnders Carlsson return; 6333a202f60SAnders Carlsson } 63405fc5be3SDouglas Gregor 63505fc5be3SDouglas Gregor RequiresZeroInitialization = true; 63605fc5be3SDouglas Gregor } 63705fc5be3SDouglas Gregor 63805fc5be3SDouglas Gregor CGF.EmitCXXAggrConstructorCall(Ctor, NumElements, NewPtr, 63905fc5be3SDouglas Gregor E->constructor_arg_begin(), 64005fc5be3SDouglas Gregor E->constructor_arg_end(), 64105fc5be3SDouglas Gregor RequiresZeroInitialization); 64205fc5be3SDouglas Gregor return; 64305fc5be3SDouglas Gregor } else if (E->getNumConstructorArgs() == 1 && 64405fc5be3SDouglas Gregor isa<ImplicitValueInitExpr>(E->getConstructorArg(0))) { 64505fc5be3SDouglas Gregor // Optimization: since zero initialization will just set the memory 64605fc5be3SDouglas Gregor // to all zeroes, generate a single memset to do it in one shot. 64705fc5be3SDouglas Gregor EmitZeroMemSet(CGF, E->getAllocatedType(), NewPtr, 64805fc5be3SDouglas Gregor AllocSizeWithoutCookie); 64905fc5be3SDouglas Gregor return; 65005fc5be3SDouglas Gregor } else { 651d5202e09SFariborz Jahanian CGF.EmitNewArrayInitializer(E, NewPtr, NumElements); 652d5202e09SFariborz Jahanian return; 653d040e6b2SAnders Carlsson } 654d5202e09SFariborz Jahanian } 65559486a2dSAnders Carlsson 65659486a2dSAnders Carlsson if (CXXConstructorDecl *Ctor = E->getConstructor()) { 657747eb784SDouglas Gregor // Per C++ [expr.new]p15, if we have an initializer, then we're performing 658747eb784SDouglas Gregor // direct initialization. C++ [dcl.init]p5 requires that we 659747eb784SDouglas Gregor // zero-initialize storage if there are no user-declared constructors. 660747eb784SDouglas Gregor if (E->hasInitializer() && 661747eb784SDouglas Gregor !Ctor->getParent()->hasUserDeclaredConstructor() && 662747eb784SDouglas Gregor !Ctor->getParent()->isEmpty()) 663747eb784SDouglas Gregor CGF.EmitNullInitialization(NewPtr, E->getAllocatedType()); 664747eb784SDouglas Gregor 665e11f9ce9SAnders Carlsson CGF.EmitCXXConstructorCall(Ctor, Ctor_Complete, /*ForVirtualBase=*/false, 666e11f9ce9SAnders Carlsson NewPtr, E->constructor_arg_begin(), 66759486a2dSAnders Carlsson E->constructor_arg_end()); 66859486a2dSAnders Carlsson 66959486a2dSAnders Carlsson return; 67059486a2dSAnders Carlsson } 671b66b08efSFariborz Jahanian // We have a POD type. 672b66b08efSFariborz Jahanian if (E->getNumConstructorArgs() == 0) 673b66b08efSFariborz Jahanian return; 67459486a2dSAnders Carlsson 675d5202e09SFariborz Jahanian StoreAnyExprIntoOneUnit(CGF, E, NewPtr); 67659486a2dSAnders Carlsson } 67759486a2dSAnders Carlsson 678824c2f53SJohn McCall namespace { 679824c2f53SJohn McCall /// A cleanup to call the given 'operator delete' function upon 680824c2f53SJohn McCall /// abnormal exit from a new expression. 681824c2f53SJohn McCall class CallDeleteDuringNew : public EHScopeStack::Cleanup { 682824c2f53SJohn McCall size_t NumPlacementArgs; 683824c2f53SJohn McCall const FunctionDecl *OperatorDelete; 684824c2f53SJohn McCall llvm::Value *Ptr; 685824c2f53SJohn McCall llvm::Value *AllocSize; 686824c2f53SJohn McCall 687824c2f53SJohn McCall RValue *getPlacementArgs() { return reinterpret_cast<RValue*>(this+1); } 688824c2f53SJohn McCall 689824c2f53SJohn McCall public: 690824c2f53SJohn McCall static size_t getExtraSize(size_t NumPlacementArgs) { 691824c2f53SJohn McCall return NumPlacementArgs * sizeof(RValue); 692824c2f53SJohn McCall } 693824c2f53SJohn McCall 694824c2f53SJohn McCall CallDeleteDuringNew(size_t NumPlacementArgs, 695824c2f53SJohn McCall const FunctionDecl *OperatorDelete, 696824c2f53SJohn McCall llvm::Value *Ptr, 697824c2f53SJohn McCall llvm::Value *AllocSize) 698824c2f53SJohn McCall : NumPlacementArgs(NumPlacementArgs), OperatorDelete(OperatorDelete), 699824c2f53SJohn McCall Ptr(Ptr), AllocSize(AllocSize) {} 700824c2f53SJohn McCall 701824c2f53SJohn McCall void setPlacementArg(unsigned I, RValue Arg) { 702824c2f53SJohn McCall assert(I < NumPlacementArgs && "index out of range"); 703824c2f53SJohn McCall getPlacementArgs()[I] = Arg; 704824c2f53SJohn McCall } 705824c2f53SJohn McCall 706824c2f53SJohn McCall void Emit(CodeGenFunction &CGF, bool IsForEH) { 707824c2f53SJohn McCall const FunctionProtoType *FPT 708824c2f53SJohn McCall = OperatorDelete->getType()->getAs<FunctionProtoType>(); 709824c2f53SJohn McCall assert(FPT->getNumArgs() == NumPlacementArgs + 1 || 710d441b1e6SJohn McCall (FPT->getNumArgs() == 2 && NumPlacementArgs == 0)); 711824c2f53SJohn McCall 712824c2f53SJohn McCall CallArgList DeleteArgs; 713824c2f53SJohn McCall 714824c2f53SJohn McCall // The first argument is always a void*. 715824c2f53SJohn McCall FunctionProtoType::arg_type_iterator AI = FPT->arg_type_begin(); 716824c2f53SJohn McCall DeleteArgs.push_back(std::make_pair(RValue::get(Ptr), *AI++)); 717824c2f53SJohn McCall 718824c2f53SJohn McCall // A member 'operator delete' can take an extra 'size_t' argument. 719824c2f53SJohn McCall if (FPT->getNumArgs() == NumPlacementArgs + 2) 720824c2f53SJohn McCall DeleteArgs.push_back(std::make_pair(RValue::get(AllocSize), *AI++)); 721824c2f53SJohn McCall 722824c2f53SJohn McCall // Pass the rest of the arguments, which must match exactly. 723824c2f53SJohn McCall for (unsigned I = 0; I != NumPlacementArgs; ++I) 724824c2f53SJohn McCall DeleteArgs.push_back(std::make_pair(getPlacementArgs()[I], *AI++)); 725824c2f53SJohn McCall 726824c2f53SJohn McCall // Call 'operator delete'. 727824c2f53SJohn McCall CGF.EmitCall(CGF.CGM.getTypes().getFunctionInfo(DeleteArgs, FPT), 728824c2f53SJohn McCall CGF.CGM.GetAddrOfFunction(OperatorDelete), 729824c2f53SJohn McCall ReturnValueSlot(), DeleteArgs, OperatorDelete); 730824c2f53SJohn McCall } 731824c2f53SJohn McCall }; 732824c2f53SJohn McCall } 733824c2f53SJohn McCall 73459486a2dSAnders Carlsson llvm::Value *CodeGenFunction::EmitCXXNewExpr(const CXXNewExpr *E) { 73559486a2dSAnders Carlsson QualType AllocType = E->getAllocatedType(); 7368ed55a54SJohn McCall if (AllocType->isArrayType()) 7378ed55a54SJohn McCall while (const ArrayType *AType = getContext().getAsArrayType(AllocType)) 7388ed55a54SJohn McCall AllocType = AType->getElementType(); 7398ed55a54SJohn McCall 74059486a2dSAnders Carlsson FunctionDecl *NewFD = E->getOperatorNew(); 74159486a2dSAnders Carlsson const FunctionProtoType *NewFTy = NewFD->getType()->getAs<FunctionProtoType>(); 74259486a2dSAnders Carlsson 74359486a2dSAnders Carlsson CallArgList NewArgs; 74459486a2dSAnders Carlsson 74559486a2dSAnders Carlsson // The allocation size is the first argument. 74659486a2dSAnders Carlsson QualType SizeTy = getContext().getSizeType(); 74759486a2dSAnders Carlsson 74859486a2dSAnders Carlsson llvm::Value *NumElements = 0; 74905fc5be3SDouglas Gregor llvm::Value *AllocSizeWithoutCookie = 0; 75047b4629bSFariborz Jahanian llvm::Value *AllocSize = EmitCXXNewAllocSize(getContext(), 75105fc5be3SDouglas Gregor *this, E, NumElements, 75205fc5be3SDouglas Gregor AllocSizeWithoutCookie); 75359486a2dSAnders Carlsson 75459486a2dSAnders Carlsson NewArgs.push_back(std::make_pair(RValue::get(AllocSize), SizeTy)); 75559486a2dSAnders Carlsson 75659486a2dSAnders Carlsson // Emit the rest of the arguments. 75759486a2dSAnders Carlsson // FIXME: Ideally, this should just use EmitCallArgs. 75859486a2dSAnders Carlsson CXXNewExpr::const_arg_iterator NewArg = E->placement_arg_begin(); 75959486a2dSAnders Carlsson 76059486a2dSAnders Carlsson // First, use the types from the function type. 76159486a2dSAnders Carlsson // We start at 1 here because the first argument (the allocation size) 76259486a2dSAnders Carlsson // has already been emitted. 76359486a2dSAnders Carlsson for (unsigned i = 1, e = NewFTy->getNumArgs(); i != e; ++i, ++NewArg) { 76459486a2dSAnders Carlsson QualType ArgType = NewFTy->getArgType(i); 76559486a2dSAnders Carlsson 76659486a2dSAnders Carlsson assert(getContext().getCanonicalType(ArgType.getNonReferenceType()). 76759486a2dSAnders Carlsson getTypePtr() == 76859486a2dSAnders Carlsson getContext().getCanonicalType(NewArg->getType()).getTypePtr() && 76959486a2dSAnders Carlsson "type mismatch in call argument!"); 77059486a2dSAnders Carlsson 77159486a2dSAnders Carlsson NewArgs.push_back(std::make_pair(EmitCallArg(*NewArg, ArgType), 77259486a2dSAnders Carlsson ArgType)); 77359486a2dSAnders Carlsson 77459486a2dSAnders Carlsson } 77559486a2dSAnders Carlsson 77659486a2dSAnders Carlsson // Either we've emitted all the call args, or we have a call to a 77759486a2dSAnders Carlsson // variadic function. 77859486a2dSAnders Carlsson assert((NewArg == E->placement_arg_end() || NewFTy->isVariadic()) && 77959486a2dSAnders Carlsson "Extra arguments in non-variadic function!"); 78059486a2dSAnders Carlsson 78159486a2dSAnders Carlsson // If we still have any arguments, emit them using the type of the argument. 78259486a2dSAnders Carlsson for (CXXNewExpr::const_arg_iterator NewArgEnd = E->placement_arg_end(); 78359486a2dSAnders Carlsson NewArg != NewArgEnd; ++NewArg) { 78459486a2dSAnders Carlsson QualType ArgType = NewArg->getType(); 78559486a2dSAnders Carlsson NewArgs.push_back(std::make_pair(EmitCallArg(*NewArg, ArgType), 78659486a2dSAnders Carlsson ArgType)); 78759486a2dSAnders Carlsson } 78859486a2dSAnders Carlsson 78959486a2dSAnders Carlsson // Emit the call to new. 79059486a2dSAnders Carlsson RValue RV = 791ab26cfa5SJohn McCall EmitCall(CGM.getTypes().getFunctionInfo(NewArgs, NewFTy), 79261a401caSAnders Carlsson CGM.GetAddrOfFunction(NewFD), ReturnValueSlot(), NewArgs, NewFD); 79359486a2dSAnders Carlsson 79459486a2dSAnders Carlsson // If an allocation function is declared with an empty exception specification 79559486a2dSAnders Carlsson // it returns null to indicate failure to allocate storage. [expr.new]p13. 79659486a2dSAnders Carlsson // (We don't need to check for null when there's no new initializer and 79759486a2dSAnders Carlsson // we're allocating a POD type). 79859486a2dSAnders Carlsson bool NullCheckResult = NewFTy->hasEmptyExceptionSpec() && 79959486a2dSAnders Carlsson !(AllocType->isPODType() && !E->hasInitializer()); 80059486a2dSAnders Carlsson 8018ed55a54SJohn McCall llvm::BasicBlock *NullCheckSource = 0; 80259486a2dSAnders Carlsson llvm::BasicBlock *NewNotNull = 0; 80359486a2dSAnders Carlsson llvm::BasicBlock *NewEnd = 0; 80459486a2dSAnders Carlsson 80559486a2dSAnders Carlsson llvm::Value *NewPtr = RV.getScalarVal(); 8068ed55a54SJohn McCall unsigned AS = cast<llvm::PointerType>(NewPtr->getType())->getAddressSpace(); 80759486a2dSAnders Carlsson 80859486a2dSAnders Carlsson if (NullCheckResult) { 8098ed55a54SJohn McCall NullCheckSource = Builder.GetInsertBlock(); 81059486a2dSAnders Carlsson NewNotNull = createBasicBlock("new.notnull"); 81159486a2dSAnders Carlsson NewEnd = createBasicBlock("new.end"); 81259486a2dSAnders Carlsson 8138ed55a54SJohn McCall llvm::Value *IsNull = Builder.CreateIsNull(NewPtr, "new.isnull"); 8148ed55a54SJohn McCall Builder.CreateCondBr(IsNull, NewEnd, NewNotNull); 81559486a2dSAnders Carlsson EmitBlock(NewNotNull); 81659486a2dSAnders Carlsson } 81759486a2dSAnders Carlsson 8188ed55a54SJohn McCall assert((AllocSize == AllocSizeWithoutCookie) == 8198ed55a54SJohn McCall CalculateCookiePadding(*this, E).isZero()); 8208ed55a54SJohn McCall if (AllocSize != AllocSizeWithoutCookie) { 8218ed55a54SJohn McCall assert(E->isArray()); 8228ed55a54SJohn McCall NewPtr = CGM.getCXXABI().InitializeArrayCookie(CGF, NewPtr, NumElements, 8238ed55a54SJohn McCall AllocType); 82459486a2dSAnders Carlsson } 82559486a2dSAnders Carlsson 826824c2f53SJohn McCall // If there's an operator delete, enter a cleanup to call it if an 827824c2f53SJohn McCall // exception is thrown. 828824c2f53SJohn McCall EHScopeStack::stable_iterator CallOperatorDelete; 829824c2f53SJohn McCall if (E->getOperatorDelete()) { 830824c2f53SJohn McCall CallDeleteDuringNew *Cleanup = CGF.EHStack 831824c2f53SJohn McCall .pushCleanupWithExtra<CallDeleteDuringNew>(EHCleanup, 832824c2f53SJohn McCall E->getNumPlacementArgs(), 833824c2f53SJohn McCall E->getOperatorDelete(), 834824c2f53SJohn McCall NewPtr, AllocSize); 835824c2f53SJohn McCall for (unsigned I = 0, N = E->getNumPlacementArgs(); I != N; ++I) 836824c2f53SJohn McCall Cleanup->setPlacementArg(I, NewArgs[I+1].first); 837824c2f53SJohn McCall CallOperatorDelete = EHStack.stable_begin(); 838824c2f53SJohn McCall } 839824c2f53SJohn McCall 840040ad500SDouglas Gregor const llvm::Type *ElementPtrTy 841040ad500SDouglas Gregor = ConvertTypeForMem(AllocType)->getPointerTo(AS); 8428ed55a54SJohn McCall NewPtr = Builder.CreateBitCast(NewPtr, ElementPtrTy); 843824c2f53SJohn McCall 8448ed55a54SJohn McCall if (E->isArray()) { 84505fc5be3SDouglas Gregor EmitNewInitializer(*this, E, NewPtr, NumElements, AllocSizeWithoutCookie); 8468ed55a54SJohn McCall 8478ed55a54SJohn McCall // NewPtr is a pointer to the base element type. If we're 8488ed55a54SJohn McCall // allocating an array of arrays, we'll need to cast back to the 8498ed55a54SJohn McCall // array pointer type. 850040ad500SDouglas Gregor const llvm::Type *ResultTy = ConvertTypeForMem(E->getType()); 8518ed55a54SJohn McCall if (NewPtr->getType() != ResultTy) 8528ed55a54SJohn McCall NewPtr = Builder.CreateBitCast(NewPtr, ResultTy); 8538ed55a54SJohn McCall } else { 85405fc5be3SDouglas Gregor EmitNewInitializer(*this, E, NewPtr, NumElements, AllocSizeWithoutCookie); 85547b4629bSFariborz Jahanian } 85659486a2dSAnders Carlsson 857824c2f53SJohn McCall // Deactivate the 'operator delete' cleanup if we finished 858824c2f53SJohn McCall // initialization. 859824c2f53SJohn McCall if (CallOperatorDelete.isValid()) 860824c2f53SJohn McCall DeactivateCleanupBlock(CallOperatorDelete); 861824c2f53SJohn McCall 86259486a2dSAnders Carlsson if (NullCheckResult) { 86359486a2dSAnders Carlsson Builder.CreateBr(NewEnd); 8648ed55a54SJohn McCall llvm::BasicBlock *NotNullSource = Builder.GetInsertBlock(); 86559486a2dSAnders Carlsson EmitBlock(NewEnd); 86659486a2dSAnders Carlsson 86759486a2dSAnders Carlsson llvm::PHINode *PHI = Builder.CreatePHI(NewPtr->getType()); 86859486a2dSAnders Carlsson PHI->reserveOperandSpace(2); 8698ed55a54SJohn McCall PHI->addIncoming(NewPtr, NotNullSource); 8708ed55a54SJohn McCall PHI->addIncoming(llvm::Constant::getNullValue(NewPtr->getType()), 8718ed55a54SJohn McCall NullCheckSource); 87259486a2dSAnders Carlsson 87359486a2dSAnders Carlsson NewPtr = PHI; 87459486a2dSAnders Carlsson } 87559486a2dSAnders Carlsson 87659486a2dSAnders Carlsson return NewPtr; 87759486a2dSAnders Carlsson } 87859486a2dSAnders Carlsson 87959486a2dSAnders Carlsson void CodeGenFunction::EmitDeleteCall(const FunctionDecl *DeleteFD, 88059486a2dSAnders Carlsson llvm::Value *Ptr, 88159486a2dSAnders Carlsson QualType DeleteTy) { 8828ed55a54SJohn McCall assert(DeleteFD->getOverloadedOperator() == OO_Delete); 8838ed55a54SJohn McCall 88459486a2dSAnders Carlsson const FunctionProtoType *DeleteFTy = 88559486a2dSAnders Carlsson DeleteFD->getType()->getAs<FunctionProtoType>(); 88659486a2dSAnders Carlsson 88759486a2dSAnders Carlsson CallArgList DeleteArgs; 88859486a2dSAnders Carlsson 88921122cf6SAnders Carlsson // Check if we need to pass the size to the delete operator. 89021122cf6SAnders Carlsson llvm::Value *Size = 0; 89121122cf6SAnders Carlsson QualType SizeTy; 89221122cf6SAnders Carlsson if (DeleteFTy->getNumArgs() == 2) { 89321122cf6SAnders Carlsson SizeTy = DeleteFTy->getArgType(1); 8947df3cbebSKen Dyck CharUnits DeleteTypeSize = getContext().getTypeSizeInChars(DeleteTy); 8957df3cbebSKen Dyck Size = llvm::ConstantInt::get(ConvertType(SizeTy), 8967df3cbebSKen Dyck DeleteTypeSize.getQuantity()); 89721122cf6SAnders Carlsson } 89821122cf6SAnders Carlsson 89959486a2dSAnders Carlsson QualType ArgTy = DeleteFTy->getArgType(0); 90059486a2dSAnders Carlsson llvm::Value *DeletePtr = Builder.CreateBitCast(Ptr, ConvertType(ArgTy)); 90159486a2dSAnders Carlsson DeleteArgs.push_back(std::make_pair(RValue::get(DeletePtr), ArgTy)); 90259486a2dSAnders Carlsson 90321122cf6SAnders Carlsson if (Size) 90459486a2dSAnders Carlsson DeleteArgs.push_back(std::make_pair(RValue::get(Size), SizeTy)); 90559486a2dSAnders Carlsson 90659486a2dSAnders Carlsson // Emit the call to delete. 907ab26cfa5SJohn McCall EmitCall(CGM.getTypes().getFunctionInfo(DeleteArgs, DeleteFTy), 90861a401caSAnders Carlsson CGM.GetAddrOfFunction(DeleteFD), ReturnValueSlot(), 90959486a2dSAnders Carlsson DeleteArgs, DeleteFD); 91059486a2dSAnders Carlsson } 91159486a2dSAnders Carlsson 9128ed55a54SJohn McCall namespace { 9138ed55a54SJohn McCall /// Calls the given 'operator delete' on a single object. 9148ed55a54SJohn McCall struct CallObjectDelete : EHScopeStack::Cleanup { 9158ed55a54SJohn McCall llvm::Value *Ptr; 9168ed55a54SJohn McCall const FunctionDecl *OperatorDelete; 9178ed55a54SJohn McCall QualType ElementType; 9188ed55a54SJohn McCall 9198ed55a54SJohn McCall CallObjectDelete(llvm::Value *Ptr, 9208ed55a54SJohn McCall const FunctionDecl *OperatorDelete, 9218ed55a54SJohn McCall QualType ElementType) 9228ed55a54SJohn McCall : Ptr(Ptr), OperatorDelete(OperatorDelete), ElementType(ElementType) {} 9238ed55a54SJohn McCall 9248ed55a54SJohn McCall void Emit(CodeGenFunction &CGF, bool IsForEH) { 9258ed55a54SJohn McCall CGF.EmitDeleteCall(OperatorDelete, Ptr, ElementType); 9268ed55a54SJohn McCall } 9278ed55a54SJohn McCall }; 9288ed55a54SJohn McCall } 9298ed55a54SJohn McCall 9308ed55a54SJohn McCall /// Emit the code for deleting a single object. 9318ed55a54SJohn McCall static void EmitObjectDelete(CodeGenFunction &CGF, 9328ed55a54SJohn McCall const FunctionDecl *OperatorDelete, 9338ed55a54SJohn McCall llvm::Value *Ptr, 9348ed55a54SJohn McCall QualType ElementType) { 9358ed55a54SJohn McCall // Find the destructor for the type, if applicable. If the 9368ed55a54SJohn McCall // destructor is virtual, we'll just emit the vcall and return. 9378ed55a54SJohn McCall const CXXDestructorDecl *Dtor = 0; 9388ed55a54SJohn McCall if (const RecordType *RT = ElementType->getAs<RecordType>()) { 9398ed55a54SJohn McCall CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl()); 9408ed55a54SJohn McCall if (!RD->hasTrivialDestructor()) { 9418ed55a54SJohn McCall Dtor = RD->getDestructor(); 9428ed55a54SJohn McCall 9438ed55a54SJohn McCall if (Dtor->isVirtual()) { 9448ed55a54SJohn McCall const llvm::Type *Ty = 9450d635f53SJohn McCall CGF.getTypes().GetFunctionType(CGF.getTypes().getFunctionInfo(Dtor, 9460d635f53SJohn McCall Dtor_Complete), 9478ed55a54SJohn McCall /*isVariadic=*/false); 9488ed55a54SJohn McCall 9498ed55a54SJohn McCall llvm::Value *Callee 9508ed55a54SJohn McCall = CGF.BuildVirtualCall(Dtor, Dtor_Deleting, Ptr, Ty); 9518ed55a54SJohn McCall CGF.EmitCXXMemberCall(Dtor, Callee, ReturnValueSlot(), Ptr, /*VTT=*/0, 9528ed55a54SJohn McCall 0, 0); 9538ed55a54SJohn McCall 9548ed55a54SJohn McCall // The dtor took care of deleting the object. 9558ed55a54SJohn McCall return; 9568ed55a54SJohn McCall } 9578ed55a54SJohn McCall } 9588ed55a54SJohn McCall } 9598ed55a54SJohn McCall 9608ed55a54SJohn McCall // Make sure that we call delete even if the dtor throws. 9618ed55a54SJohn McCall CGF.EHStack.pushCleanup<CallObjectDelete>(NormalAndEHCleanup, 9628ed55a54SJohn McCall Ptr, OperatorDelete, ElementType); 9638ed55a54SJohn McCall 9648ed55a54SJohn McCall if (Dtor) 9658ed55a54SJohn McCall CGF.EmitCXXDestructorCall(Dtor, Dtor_Complete, 9668ed55a54SJohn McCall /*ForVirtualBase=*/false, Ptr); 9678ed55a54SJohn McCall 9688ed55a54SJohn McCall CGF.PopCleanupBlock(); 9698ed55a54SJohn McCall } 9708ed55a54SJohn McCall 9718ed55a54SJohn McCall namespace { 9728ed55a54SJohn McCall /// Calls the given 'operator delete' on an array of objects. 9738ed55a54SJohn McCall struct CallArrayDelete : EHScopeStack::Cleanup { 9748ed55a54SJohn McCall llvm::Value *Ptr; 9758ed55a54SJohn McCall const FunctionDecl *OperatorDelete; 9768ed55a54SJohn McCall llvm::Value *NumElements; 9778ed55a54SJohn McCall QualType ElementType; 9788ed55a54SJohn McCall CharUnits CookieSize; 9798ed55a54SJohn McCall 9808ed55a54SJohn McCall CallArrayDelete(llvm::Value *Ptr, 9818ed55a54SJohn McCall const FunctionDecl *OperatorDelete, 9828ed55a54SJohn McCall llvm::Value *NumElements, 9838ed55a54SJohn McCall QualType ElementType, 9848ed55a54SJohn McCall CharUnits CookieSize) 9858ed55a54SJohn McCall : Ptr(Ptr), OperatorDelete(OperatorDelete), NumElements(NumElements), 9868ed55a54SJohn McCall ElementType(ElementType), CookieSize(CookieSize) {} 9878ed55a54SJohn McCall 9888ed55a54SJohn McCall void Emit(CodeGenFunction &CGF, bool IsForEH) { 9898ed55a54SJohn McCall const FunctionProtoType *DeleteFTy = 9908ed55a54SJohn McCall OperatorDelete->getType()->getAs<FunctionProtoType>(); 9918ed55a54SJohn McCall assert(DeleteFTy->getNumArgs() == 1 || DeleteFTy->getNumArgs() == 2); 9928ed55a54SJohn McCall 9938ed55a54SJohn McCall CallArgList Args; 9948ed55a54SJohn McCall 9958ed55a54SJohn McCall // Pass the pointer as the first argument. 9968ed55a54SJohn McCall QualType VoidPtrTy = DeleteFTy->getArgType(0); 9978ed55a54SJohn McCall llvm::Value *DeletePtr 9988ed55a54SJohn McCall = CGF.Builder.CreateBitCast(Ptr, CGF.ConvertType(VoidPtrTy)); 9998ed55a54SJohn McCall Args.push_back(std::make_pair(RValue::get(DeletePtr), VoidPtrTy)); 10008ed55a54SJohn McCall 10018ed55a54SJohn McCall // Pass the original requested size as the second argument. 10028ed55a54SJohn McCall if (DeleteFTy->getNumArgs() == 2) { 10038ed55a54SJohn McCall QualType size_t = DeleteFTy->getArgType(1); 10048ed55a54SJohn McCall const llvm::IntegerType *SizeTy 10058ed55a54SJohn McCall = cast<llvm::IntegerType>(CGF.ConvertType(size_t)); 10068ed55a54SJohn McCall 10078ed55a54SJohn McCall CharUnits ElementTypeSize = 10088ed55a54SJohn McCall CGF.CGM.getContext().getTypeSizeInChars(ElementType); 10098ed55a54SJohn McCall 10108ed55a54SJohn McCall // The size of an element, multiplied by the number of elements. 10118ed55a54SJohn McCall llvm::Value *Size 10128ed55a54SJohn McCall = llvm::ConstantInt::get(SizeTy, ElementTypeSize.getQuantity()); 10138ed55a54SJohn McCall Size = CGF.Builder.CreateMul(Size, NumElements); 10148ed55a54SJohn McCall 10158ed55a54SJohn McCall // Plus the size of the cookie if applicable. 10168ed55a54SJohn McCall if (!CookieSize.isZero()) { 10178ed55a54SJohn McCall llvm::Value *CookieSizeV 10188ed55a54SJohn McCall = llvm::ConstantInt::get(SizeTy, CookieSize.getQuantity()); 10198ed55a54SJohn McCall Size = CGF.Builder.CreateAdd(Size, CookieSizeV); 10208ed55a54SJohn McCall } 10218ed55a54SJohn McCall 10228ed55a54SJohn McCall Args.push_back(std::make_pair(RValue::get(Size), size_t)); 10238ed55a54SJohn McCall } 10248ed55a54SJohn McCall 10258ed55a54SJohn McCall // Emit the call to delete. 10268ed55a54SJohn McCall CGF.EmitCall(CGF.getTypes().getFunctionInfo(Args, DeleteFTy), 10278ed55a54SJohn McCall CGF.CGM.GetAddrOfFunction(OperatorDelete), 10288ed55a54SJohn McCall ReturnValueSlot(), Args, OperatorDelete); 10298ed55a54SJohn McCall } 10308ed55a54SJohn McCall }; 10318ed55a54SJohn McCall } 10328ed55a54SJohn McCall 10338ed55a54SJohn McCall /// Emit the code for deleting an array of objects. 10348ed55a54SJohn McCall static void EmitArrayDelete(CodeGenFunction &CGF, 10358ed55a54SJohn McCall const FunctionDecl *OperatorDelete, 10368ed55a54SJohn McCall llvm::Value *Ptr, 10378ed55a54SJohn McCall QualType ElementType) { 10388ed55a54SJohn McCall llvm::Value *NumElements = 0; 10398ed55a54SJohn McCall llvm::Value *AllocatedPtr = 0; 10408ed55a54SJohn McCall CharUnits CookieSize; 10418ed55a54SJohn McCall CGF.CGM.getCXXABI().ReadArrayCookie(CGF, Ptr, ElementType, 10428ed55a54SJohn McCall NumElements, AllocatedPtr, CookieSize); 10438ed55a54SJohn McCall 10448ed55a54SJohn McCall assert(AllocatedPtr && "ReadArrayCookie didn't set AllocatedPtr"); 10458ed55a54SJohn McCall 10468ed55a54SJohn McCall // Make sure that we call delete even if one of the dtors throws. 10478ed55a54SJohn McCall CGF.EHStack.pushCleanup<CallArrayDelete>(NormalAndEHCleanup, 10488ed55a54SJohn McCall AllocatedPtr, OperatorDelete, 10498ed55a54SJohn McCall NumElements, ElementType, 10508ed55a54SJohn McCall CookieSize); 10518ed55a54SJohn McCall 10528ed55a54SJohn McCall if (const CXXRecordDecl *RD = ElementType->getAsCXXRecordDecl()) { 10538ed55a54SJohn McCall if (!RD->hasTrivialDestructor()) { 10548ed55a54SJohn McCall assert(NumElements && "ReadArrayCookie didn't find element count" 10558ed55a54SJohn McCall " for a class with destructor"); 10568ed55a54SJohn McCall CGF.EmitCXXAggrDestructorCall(RD->getDestructor(), NumElements, Ptr); 10578ed55a54SJohn McCall } 10588ed55a54SJohn McCall } 10598ed55a54SJohn McCall 10608ed55a54SJohn McCall CGF.PopCleanupBlock(); 10618ed55a54SJohn McCall } 10628ed55a54SJohn McCall 106359486a2dSAnders Carlsson void CodeGenFunction::EmitCXXDeleteExpr(const CXXDeleteExpr *E) { 106459486a2dSAnders Carlsson 106559486a2dSAnders Carlsson // Get at the argument before we performed the implicit conversion 106659486a2dSAnders Carlsson // to void*. 106759486a2dSAnders Carlsson const Expr *Arg = E->getArgument(); 106859486a2dSAnders Carlsson while (const ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(Arg)) { 1069e302792bSJohn McCall if (ICE->getCastKind() != CK_UserDefinedConversion && 107059486a2dSAnders Carlsson ICE->getType()->isVoidPointerType()) 107159486a2dSAnders Carlsson Arg = ICE->getSubExpr(); 107259486a2dSAnders Carlsson else 107359486a2dSAnders Carlsson break; 107459486a2dSAnders Carlsson } 107559486a2dSAnders Carlsson 107659486a2dSAnders Carlsson llvm::Value *Ptr = EmitScalarExpr(Arg); 107759486a2dSAnders Carlsson 107859486a2dSAnders Carlsson // Null check the pointer. 107959486a2dSAnders Carlsson llvm::BasicBlock *DeleteNotNull = createBasicBlock("delete.notnull"); 108059486a2dSAnders Carlsson llvm::BasicBlock *DeleteEnd = createBasicBlock("delete.end"); 108159486a2dSAnders Carlsson 108259486a2dSAnders Carlsson llvm::Value *IsNull = 108359486a2dSAnders Carlsson Builder.CreateICmpEQ(Ptr, llvm::Constant::getNullValue(Ptr->getType()), 108459486a2dSAnders Carlsson "isnull"); 108559486a2dSAnders Carlsson 108659486a2dSAnders Carlsson Builder.CreateCondBr(IsNull, DeleteEnd, DeleteNotNull); 108759486a2dSAnders Carlsson EmitBlock(DeleteNotNull); 108859486a2dSAnders Carlsson 10898ed55a54SJohn McCall // We might be deleting a pointer to array. If so, GEP down to the 10908ed55a54SJohn McCall // first non-array element. 10918ed55a54SJohn McCall // (this assumes that A(*)[3][7] is converted to [3 x [7 x %A]]*) 10928ed55a54SJohn McCall QualType DeleteTy = Arg->getType()->getAs<PointerType>()->getPointeeType(); 10938ed55a54SJohn McCall if (DeleteTy->isConstantArrayType()) { 10948ed55a54SJohn McCall llvm::Value *Zero = Builder.getInt32(0); 10958ed55a54SJohn McCall llvm::SmallVector<llvm::Value*,8> GEP; 109659486a2dSAnders Carlsson 10978ed55a54SJohn McCall GEP.push_back(Zero); // point at the outermost array 10988ed55a54SJohn McCall 10998ed55a54SJohn McCall // For each layer of array type we're pointing at: 11008ed55a54SJohn McCall while (const ConstantArrayType *Arr 11018ed55a54SJohn McCall = getContext().getAsConstantArrayType(DeleteTy)) { 11028ed55a54SJohn McCall // 1. Unpeel the array type. 11038ed55a54SJohn McCall DeleteTy = Arr->getElementType(); 11048ed55a54SJohn McCall 11058ed55a54SJohn McCall // 2. GEP to the first element of the array. 11068ed55a54SJohn McCall GEP.push_back(Zero); 11078ed55a54SJohn McCall } 11088ed55a54SJohn McCall 11098ed55a54SJohn McCall Ptr = Builder.CreateInBoundsGEP(Ptr, GEP.begin(), GEP.end(), "del.first"); 11108ed55a54SJohn McCall } 11118ed55a54SJohn McCall 111204f36218SDouglas Gregor assert(ConvertTypeForMem(DeleteTy) == 111304f36218SDouglas Gregor cast<llvm::PointerType>(Ptr->getType())->getElementType()); 11148ed55a54SJohn McCall 111559486a2dSAnders Carlsson if (E->isArrayForm()) { 11168ed55a54SJohn McCall EmitArrayDelete(*this, E->getOperatorDelete(), Ptr, DeleteTy); 11178ed55a54SJohn McCall } else { 11188ed55a54SJohn McCall EmitObjectDelete(*this, E->getOperatorDelete(), Ptr, DeleteTy); 111959486a2dSAnders Carlsson } 112059486a2dSAnders Carlsson 112159486a2dSAnders Carlsson EmitBlock(DeleteEnd); 112259486a2dSAnders Carlsson } 112359486a2dSAnders Carlsson 112459486a2dSAnders Carlsson llvm::Value * CodeGenFunction::EmitCXXTypeidExpr(const CXXTypeidExpr *E) { 112559486a2dSAnders Carlsson QualType Ty = E->getType(); 112659486a2dSAnders Carlsson const llvm::Type *LTy = ConvertType(Ty)->getPointerTo(); 1127fd7dfeb7SAnders Carlsson 11283f4336cbSAnders Carlsson if (E->isTypeOperand()) { 11293f4336cbSAnders Carlsson llvm::Constant *TypeInfo = 11303f4336cbSAnders Carlsson CGM.GetAddrOfRTTIDescriptor(E->getTypeOperand()); 11313f4336cbSAnders Carlsson return Builder.CreateBitCast(TypeInfo, LTy); 11323f4336cbSAnders Carlsson } 1133fd7dfeb7SAnders Carlsson 113459486a2dSAnders Carlsson Expr *subE = E->getExprOperand(); 113559486a2dSAnders Carlsson Ty = subE->getType(); 113659486a2dSAnders Carlsson CanQualType CanTy = CGM.getContext().getCanonicalType(Ty); 113759486a2dSAnders Carlsson Ty = CanTy.getUnqualifiedType().getNonReferenceType(); 113859486a2dSAnders Carlsson if (const RecordType *RT = Ty->getAs<RecordType>()) { 113959486a2dSAnders Carlsson const CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl()); 114059486a2dSAnders Carlsson if (RD->isPolymorphic()) { 114159486a2dSAnders Carlsson // FIXME: if subE is an lvalue do 114259486a2dSAnders Carlsson LValue Obj = EmitLValue(subE); 114359486a2dSAnders Carlsson llvm::Value *This = Obj.getAddress(); 114459486a2dSAnders Carlsson LTy = LTy->getPointerTo()->getPointerTo(); 114559486a2dSAnders Carlsson llvm::Value *V = Builder.CreateBitCast(This, LTy); 114659486a2dSAnders Carlsson // We need to do a zero check for *p, unless it has NonNullAttr. 114759486a2dSAnders Carlsson // FIXME: PointerType->hasAttr<NonNullAttr>() 114859486a2dSAnders Carlsson bool CanBeZero = false; 114959486a2dSAnders Carlsson if (UnaryOperator *UO = dyn_cast<UnaryOperator>(subE->IgnoreParens())) 1150e302792bSJohn McCall if (UO->getOpcode() == UO_Deref) 115159486a2dSAnders Carlsson CanBeZero = true; 115259486a2dSAnders Carlsson if (CanBeZero) { 115359486a2dSAnders Carlsson llvm::BasicBlock *NonZeroBlock = createBasicBlock(); 115459486a2dSAnders Carlsson llvm::BasicBlock *ZeroBlock = createBasicBlock(); 115559486a2dSAnders Carlsson 115659486a2dSAnders Carlsson llvm::Value *Zero = llvm::Constant::getNullValue(LTy); 115759486a2dSAnders Carlsson Builder.CreateCondBr(Builder.CreateICmpNE(V, Zero), 115859486a2dSAnders Carlsson NonZeroBlock, ZeroBlock); 115959486a2dSAnders Carlsson EmitBlock(ZeroBlock); 116059486a2dSAnders Carlsson /// Call __cxa_bad_typeid 116159486a2dSAnders Carlsson const llvm::Type *ResultType = llvm::Type::getVoidTy(VMContext); 116259486a2dSAnders Carlsson const llvm::FunctionType *FTy; 116359486a2dSAnders Carlsson FTy = llvm::FunctionType::get(ResultType, false); 116459486a2dSAnders Carlsson llvm::Value *F = CGM.CreateRuntimeFunction(FTy, "__cxa_bad_typeid"); 116559486a2dSAnders Carlsson Builder.CreateCall(F)->setDoesNotReturn(); 116659486a2dSAnders Carlsson Builder.CreateUnreachable(); 116759486a2dSAnders Carlsson EmitBlock(NonZeroBlock); 116859486a2dSAnders Carlsson } 116959486a2dSAnders Carlsson V = Builder.CreateLoad(V, "vtable"); 117059486a2dSAnders Carlsson V = Builder.CreateConstInBoundsGEP1_64(V, -1ULL); 117159486a2dSAnders Carlsson V = Builder.CreateLoad(V); 117259486a2dSAnders Carlsson return V; 117359486a2dSAnders Carlsson } 117459486a2dSAnders Carlsson } 11753f4336cbSAnders Carlsson return Builder.CreateBitCast(CGM.GetAddrOfRTTIDescriptor(Ty), LTy); 117659486a2dSAnders Carlsson } 117759486a2dSAnders Carlsson 117859486a2dSAnders Carlsson llvm::Value *CodeGenFunction::EmitDynamicCast(llvm::Value *V, 117959486a2dSAnders Carlsson const CXXDynamicCastExpr *DCE) { 11803f4336cbSAnders Carlsson QualType SrcTy = DCE->getSubExpr()->getType(); 11813f4336cbSAnders Carlsson QualType DestTy = DCE->getTypeAsWritten(); 11823f4336cbSAnders Carlsson QualType InnerType = DestTy->getPointeeType(); 11833f4336cbSAnders Carlsson 118459486a2dSAnders Carlsson const llvm::Type *LTy = ConvertType(DCE->getType()); 118559486a2dSAnders Carlsson 118659486a2dSAnders Carlsson bool CanBeZero = false; 118759486a2dSAnders Carlsson bool ToVoid = false; 118859486a2dSAnders Carlsson bool ThrowOnBad = false; 11893f4336cbSAnders Carlsson if (DestTy->isPointerType()) { 119059486a2dSAnders Carlsson // FIXME: if PointerType->hasAttr<NonNullAttr>(), we don't set this 119159486a2dSAnders Carlsson CanBeZero = true; 119259486a2dSAnders Carlsson if (InnerType->isVoidType()) 119359486a2dSAnders Carlsson ToVoid = true; 119459486a2dSAnders Carlsson } else { 119559486a2dSAnders Carlsson LTy = LTy->getPointerTo(); 1196fa8b4955SDouglas Gregor 1197fa8b4955SDouglas Gregor // FIXME: What if exceptions are disabled? 119859486a2dSAnders Carlsson ThrowOnBad = true; 119959486a2dSAnders Carlsson } 120059486a2dSAnders Carlsson 12013f4336cbSAnders Carlsson if (SrcTy->isPointerType() || SrcTy->isReferenceType()) 12023f4336cbSAnders Carlsson SrcTy = SrcTy->getPointeeType(); 12033f4336cbSAnders Carlsson SrcTy = SrcTy.getUnqualifiedType(); 12043f4336cbSAnders Carlsson 12050087bc85SAnders Carlsson if (DestTy->isPointerType() || DestTy->isReferenceType()) 12063f4336cbSAnders Carlsson DestTy = DestTy->getPointeeType(); 12073f4336cbSAnders Carlsson DestTy = DestTy.getUnqualifiedType(); 120859486a2dSAnders Carlsson 120959486a2dSAnders Carlsson llvm::BasicBlock *ContBlock = createBasicBlock(); 121059486a2dSAnders Carlsson llvm::BasicBlock *NullBlock = 0; 121159486a2dSAnders Carlsson llvm::BasicBlock *NonZeroBlock = 0; 121259486a2dSAnders Carlsson if (CanBeZero) { 121359486a2dSAnders Carlsson NonZeroBlock = createBasicBlock(); 121459486a2dSAnders Carlsson NullBlock = createBasicBlock(); 12153f4336cbSAnders Carlsson Builder.CreateCondBr(Builder.CreateIsNotNull(V), NonZeroBlock, NullBlock); 121659486a2dSAnders Carlsson EmitBlock(NonZeroBlock); 121759486a2dSAnders Carlsson } 121859486a2dSAnders Carlsson 121959486a2dSAnders Carlsson llvm::BasicBlock *BadCastBlock = 0; 122059486a2dSAnders Carlsson 12213f4336cbSAnders Carlsson const llvm::Type *PtrDiffTy = ConvertType(getContext().getPointerDiffType()); 122259486a2dSAnders Carlsson 122359486a2dSAnders Carlsson // See if this is a dynamic_cast(void*) 122459486a2dSAnders Carlsson if (ToVoid) { 122559486a2dSAnders Carlsson llvm::Value *This = V; 122659486a2dSAnders Carlsson V = Builder.CreateBitCast(This, PtrDiffTy->getPointerTo()->getPointerTo()); 122759486a2dSAnders Carlsson V = Builder.CreateLoad(V, "vtable"); 122859486a2dSAnders Carlsson V = Builder.CreateConstInBoundsGEP1_64(V, -2ULL); 122959486a2dSAnders Carlsson V = Builder.CreateLoad(V, "offset to top"); 123059486a2dSAnders Carlsson This = Builder.CreateBitCast(This, llvm::Type::getInt8PtrTy(VMContext)); 123159486a2dSAnders Carlsson V = Builder.CreateInBoundsGEP(This, V); 123259486a2dSAnders Carlsson V = Builder.CreateBitCast(V, LTy); 123359486a2dSAnders Carlsson } else { 123459486a2dSAnders Carlsson /// Call __dynamic_cast 123559486a2dSAnders Carlsson const llvm::Type *ResultType = llvm::Type::getInt8PtrTy(VMContext); 123659486a2dSAnders Carlsson const llvm::FunctionType *FTy; 123759486a2dSAnders Carlsson std::vector<const llvm::Type*> ArgTys; 123859486a2dSAnders Carlsson const llvm::Type *PtrToInt8Ty 123959486a2dSAnders Carlsson = llvm::Type::getInt8Ty(VMContext)->getPointerTo(); 124059486a2dSAnders Carlsson ArgTys.push_back(PtrToInt8Ty); 124159486a2dSAnders Carlsson ArgTys.push_back(PtrToInt8Ty); 124259486a2dSAnders Carlsson ArgTys.push_back(PtrToInt8Ty); 124359486a2dSAnders Carlsson ArgTys.push_back(PtrDiffTy); 124459486a2dSAnders Carlsson FTy = llvm::FunctionType::get(ResultType, ArgTys, false); 124559486a2dSAnders Carlsson 124659486a2dSAnders Carlsson // FIXME: Calculate better hint. 124759486a2dSAnders Carlsson llvm::Value *hint = llvm::ConstantInt::get(PtrDiffTy, -1ULL); 12483f4336cbSAnders Carlsson 12493f4336cbSAnders Carlsson assert(SrcTy->isRecordType() && "Src type must be record type!"); 12503f4336cbSAnders Carlsson assert(DestTy->isRecordType() && "Dest type must be record type!"); 12513f4336cbSAnders Carlsson 1252247894b3SDouglas Gregor llvm::Value *SrcArg 1253247894b3SDouglas Gregor = CGM.GetAddrOfRTTIDescriptor(SrcTy.getUnqualifiedType()); 1254247894b3SDouglas Gregor llvm::Value *DestArg 1255247894b3SDouglas Gregor = CGM.GetAddrOfRTTIDescriptor(DestTy.getUnqualifiedType()); 12563f4336cbSAnders Carlsson 125759486a2dSAnders Carlsson V = Builder.CreateBitCast(V, PtrToInt8Ty); 125859486a2dSAnders Carlsson V = Builder.CreateCall4(CGM.CreateRuntimeFunction(FTy, "__dynamic_cast"), 12593f4336cbSAnders Carlsson V, SrcArg, DestArg, hint); 126059486a2dSAnders Carlsson V = Builder.CreateBitCast(V, LTy); 126159486a2dSAnders Carlsson 126259486a2dSAnders Carlsson if (ThrowOnBad) { 126359486a2dSAnders Carlsson BadCastBlock = createBasicBlock(); 12643f4336cbSAnders Carlsson Builder.CreateCondBr(Builder.CreateIsNotNull(V), ContBlock, BadCastBlock); 126559486a2dSAnders Carlsson EmitBlock(BadCastBlock); 1266fa8b4955SDouglas Gregor /// Invoke __cxa_bad_cast 126759486a2dSAnders Carlsson ResultType = llvm::Type::getVoidTy(VMContext); 126859486a2dSAnders Carlsson const llvm::FunctionType *FBadTy; 126959486a2dSAnders Carlsson FBadTy = llvm::FunctionType::get(ResultType, false); 127059486a2dSAnders Carlsson llvm::Value *F = CGM.CreateRuntimeFunction(FBadTy, "__cxa_bad_cast"); 1271fa8b4955SDouglas Gregor if (llvm::BasicBlock *InvokeDest = getInvokeDest()) { 1272fa8b4955SDouglas Gregor llvm::BasicBlock *Cont = createBasicBlock("invoke.cont"); 1273fa8b4955SDouglas Gregor Builder.CreateInvoke(F, Cont, InvokeDest)->setDoesNotReturn(); 1274fa8b4955SDouglas Gregor EmitBlock(Cont); 1275fa8b4955SDouglas Gregor } else { 1276fa8b4955SDouglas Gregor // FIXME: Does this ever make sense? 127759486a2dSAnders Carlsson Builder.CreateCall(F)->setDoesNotReturn(); 1278fa8b4955SDouglas Gregor } 127959486a2dSAnders Carlsson Builder.CreateUnreachable(); 128059486a2dSAnders Carlsson } 128159486a2dSAnders Carlsson } 128259486a2dSAnders Carlsson 128359486a2dSAnders Carlsson if (CanBeZero) { 128459486a2dSAnders Carlsson Builder.CreateBr(ContBlock); 128559486a2dSAnders Carlsson EmitBlock(NullBlock); 128659486a2dSAnders Carlsson Builder.CreateBr(ContBlock); 128759486a2dSAnders Carlsson } 128859486a2dSAnders Carlsson EmitBlock(ContBlock); 128959486a2dSAnders Carlsson if (CanBeZero) { 129059486a2dSAnders Carlsson llvm::PHINode *PHI = Builder.CreatePHI(LTy); 129159486a2dSAnders Carlsson PHI->reserveOperandSpace(2); 129259486a2dSAnders Carlsson PHI->addIncoming(V, NonZeroBlock); 129359486a2dSAnders Carlsson PHI->addIncoming(llvm::Constant::getNullValue(LTy), NullBlock); 129459486a2dSAnders Carlsson V = PHI; 129559486a2dSAnders Carlsson } 129659486a2dSAnders Carlsson 129759486a2dSAnders Carlsson return V; 129859486a2dSAnders Carlsson } 1299