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()) { 2221b8b8bf2SFariborz Jahanian llvm::Value *AggLoc = CreateMemTemp(E->getArg(1)->getType()); 223e1b45a5eSFariborz Jahanian EmitAggExpr(E->getArg(1), AggLoc, false /*VolatileDest*/); 22461a31241SFariborz Jahanian if (LV.isPropertyRef()) 225e1b45a5eSFariborz Jahanian EmitObjCPropertySet(LV.getPropertyRefExpr(), 22661a31241SFariborz Jahanian RValue::getAggregate(AggLoc, 22761a31241SFariborz Jahanian false /*VolatileDest*/)); 22861a31241SFariborz Jahanian else 22961a31241SFariborz Jahanian EmitObjCPropertySet(LV.getKVCRefExpr(), 23061a31241SFariborz Jahanian RValue::getAggregate(AggLoc, 23161a31241SFariborz Jahanian false /*VolatileDest*/)); 232e1b45a5eSFariborz Jahanian return RValue::getAggregate(0, false); 23343a40f93SFariborz Jahanian } 23443a40f93SFariborz Jahanian else 23543a40f93SFariborz Jahanian This = LV.getAddress(); 23643a40f93SFariborz Jahanian 23727da15baSAnders Carlsson llvm::Value *Src = EmitLValue(E->getArg(1)).getAddress(); 23827da15baSAnders Carlsson QualType Ty = E->getType(); 23927da15baSAnders Carlsson EmitAggregateCopy(This, Src, Ty); 24027da15baSAnders Carlsson return RValue::get(This); 24127da15baSAnders Carlsson } 24227da15baSAnders Carlsson } 24327da15baSAnders Carlsson 24427da15baSAnders Carlsson const FunctionProtoType *FPT = MD->getType()->getAs<FunctionProtoType>(); 24527da15baSAnders Carlsson const llvm::Type *Ty = 24627da15baSAnders Carlsson CGM.getTypes().GetFunctionType(CGM.getTypes().getFunctionInfo(MD), 24727da15baSAnders Carlsson FPT->isVariadic()); 248fdf474b0SFariborz Jahanian LValue LV = EmitLValue(E->getArg(0)); 249fdf474b0SFariborz Jahanian llvm::Value *This; 25061a31241SFariborz Jahanian if (LV.isPropertyRef() || LV.isKVCRef()) { 25161a31241SFariborz Jahanian QualType QT = E->getArg(0)->getType(); 25261a31241SFariborz Jahanian RValue RV = 25361a31241SFariborz Jahanian LV.isPropertyRef() ? EmitLoadOfPropertyRefLValue(LV, QT) 25461a31241SFariborz Jahanian : EmitLoadOfKVCRefLValue(LV, QT); 2556855ba2cSFariborz Jahanian assert (!RV.isScalar() && "EmitCXXOperatorMemberCallExpr"); 2566855ba2cSFariborz Jahanian This = RV.getAggregateAddr(); 257fdf474b0SFariborz Jahanian } 258fdf474b0SFariborz Jahanian else 259fdf474b0SFariborz Jahanian This = LV.getAddress(); 26027da15baSAnders Carlsson 26127da15baSAnders Carlsson llvm::Value *Callee; 26227da15baSAnders Carlsson if (MD->isVirtual() && !canDevirtualizeMemberFunctionCalls(E->getArg(0))) 26327da15baSAnders Carlsson Callee = BuildVirtualCall(MD, This, Ty); 26427da15baSAnders Carlsson else 26527da15baSAnders Carlsson Callee = CGM.GetAddrOfFunction(MD, Ty); 26627da15baSAnders Carlsson 267e36a6b3eSAnders Carlsson return EmitCXXMemberCall(MD, Callee, ReturnValue, This, /*VTT=*/0, 26827da15baSAnders Carlsson E->arg_begin() + 1, E->arg_end()); 26927da15baSAnders Carlsson } 27027da15baSAnders Carlsson 27127da15baSAnders Carlsson void 27227da15baSAnders Carlsson CodeGenFunction::EmitCXXConstructExpr(llvm::Value *Dest, 27327da15baSAnders Carlsson const CXXConstructExpr *E) { 27427da15baSAnders Carlsson assert(Dest && "Must have a destination!"); 27527da15baSAnders Carlsson const CXXConstructorDecl *CD = E->getConstructor(); 276630c76efSDouglas Gregor 277630c76efSDouglas Gregor // If we require zero initialization before (or instead of) calling the 278630c76efSDouglas Gregor // constructor, as can be the case with a non-user-provided default 279630c76efSDouglas Gregor // constructor, emit the zero initialization now. 280e3b3464dSDouglas Gregor if (E->requiresZeroInitialization()) 281e3b3464dSDouglas Gregor EmitNullInitialization(Dest, E->getType()); 282e3b3464dSDouglas Gregor 283630c76efSDouglas Gregor 284630c76efSDouglas Gregor // If this is a call to a trivial default constructor, do nothing. 285630c76efSDouglas Gregor if (CD->isTrivial() && CD->isDefaultConstructor()) 28627da15baSAnders Carlsson return; 287630c76efSDouglas Gregor 28827da15baSAnders Carlsson // Code gen optimization to eliminate copy constructor and return 289222cf0efSDouglas Gregor // its first argument instead, if in fact that argument is a temporary 290222cf0efSDouglas Gregor // object. 29127da15baSAnders Carlsson if (getContext().getLangOptions().ElideConstructors && E->isElidable()) { 292222cf0efSDouglas Gregor if (const Expr *Arg = E->getArg(0)->getTemporaryObject()) { 29327da15baSAnders Carlsson EmitAggExpr(Arg, Dest, false); 29427da15baSAnders Carlsson return; 29527da15baSAnders Carlsson } 296222cf0efSDouglas Gregor } 297630c76efSDouglas Gregor 298630c76efSDouglas Gregor const ConstantArrayType *Array 299630c76efSDouglas Gregor = getContext().getAsConstantArrayType(E->getType()); 30027da15baSAnders Carlsson if (Array) { 30127da15baSAnders Carlsson QualType BaseElementTy = getContext().getBaseElementType(Array); 30227da15baSAnders Carlsson const llvm::Type *BasePtr = ConvertType(BaseElementTy); 30327da15baSAnders Carlsson BasePtr = llvm::PointerType::getUnqual(BasePtr); 30427da15baSAnders Carlsson llvm::Value *BaseAddrPtr = 30527da15baSAnders Carlsson Builder.CreateBitCast(Dest, BasePtr); 30627da15baSAnders Carlsson 30727da15baSAnders Carlsson EmitCXXAggrConstructorCall(CD, Array, BaseAddrPtr, 30827da15baSAnders Carlsson E->arg_begin(), E->arg_end()); 30927da15baSAnders Carlsson } 310e11f9ce9SAnders Carlsson else { 311e11f9ce9SAnders Carlsson CXXCtorType Type = 312e11f9ce9SAnders Carlsson (E->getConstructionKind() == CXXConstructExpr::CK_Complete) 313e11f9ce9SAnders Carlsson ? Ctor_Complete : Ctor_Base; 314e11f9ce9SAnders Carlsson bool ForVirtualBase = 315e11f9ce9SAnders Carlsson E->getConstructionKind() == CXXConstructExpr::CK_VirtualBase; 316e11f9ce9SAnders Carlsson 31727da15baSAnders Carlsson // Call the constructor. 318e11f9ce9SAnders Carlsson EmitCXXConstructorCall(CD, Type, ForVirtualBase, Dest, 31927da15baSAnders Carlsson E->arg_begin(), E->arg_end()); 32027da15baSAnders Carlsson } 321e11f9ce9SAnders Carlsson } 32227da15baSAnders Carlsson 323aa4149a2SJohn McCall /// Check whether the given operator new[] is the global placement 324aa4149a2SJohn McCall /// operator new[]. 325aa4149a2SJohn McCall static bool IsPlacementOperatorNewArray(ASTContext &Ctx, 326aa4149a2SJohn McCall const FunctionDecl *Fn) { 327aa4149a2SJohn McCall // Must be in global scope. Note that allocation functions can't be 328aa4149a2SJohn McCall // declared in namespaces. 32950c68258SSebastian Redl if (!Fn->getDeclContext()->getRedeclContext()->isFileContext()) 330aa4149a2SJohn McCall return false; 331aa4149a2SJohn McCall 332aa4149a2SJohn McCall // Signature must be void *operator new[](size_t, void*). 333aa4149a2SJohn McCall // The size_t is common to all operator new[]s. 334aa4149a2SJohn McCall if (Fn->getNumParams() != 2) 335aa4149a2SJohn McCall return false; 336aa4149a2SJohn McCall 337aa4149a2SJohn McCall CanQualType ParamType = Ctx.getCanonicalType(Fn->getParamDecl(1)->getType()); 338aa4149a2SJohn McCall return (ParamType == Ctx.VoidPtrTy); 339aa4149a2SJohn McCall } 340aa4149a2SJohn McCall 3418ed55a54SJohn McCall static CharUnits CalculateCookiePadding(CodeGenFunction &CGF, 3428ed55a54SJohn McCall const CXXNewExpr *E) { 34321122cf6SAnders Carlsson if (!E->isArray()) 3443eb55cfeSKen Dyck return CharUnits::Zero(); 34521122cf6SAnders Carlsson 346399f499fSAnders Carlsson // No cookie is required if the new operator being used is 347399f499fSAnders Carlsson // ::operator new[](size_t, void*). 348399f499fSAnders Carlsson const FunctionDecl *OperatorNew = E->getOperatorNew(); 3498ed55a54SJohn McCall if (IsPlacementOperatorNewArray(CGF.getContext(), OperatorNew)) 3503eb55cfeSKen Dyck return CharUnits::Zero(); 351399f499fSAnders Carlsson 3528ed55a54SJohn McCall return CGF.CGM.getCXXABI().GetArrayCookieSize(E->getAllocatedType()); 35359486a2dSAnders Carlsson } 35459486a2dSAnders Carlsson 35547b4629bSFariborz Jahanian static llvm::Value *EmitCXXNewAllocSize(ASTContext &Context, 35647b4629bSFariborz Jahanian CodeGenFunction &CGF, 35759486a2dSAnders Carlsson const CXXNewExpr *E, 35805fc5be3SDouglas Gregor llvm::Value *&NumElements, 35905fc5be3SDouglas Gregor llvm::Value *&SizeWithoutCookie) { 3607648fb46SArgyrios Kyrtzidis QualType ElemType = E->getAllocatedType(); 36159486a2dSAnders Carlsson 3628ed55a54SJohn McCall const llvm::IntegerType *SizeTy = 3638ed55a54SJohn McCall cast<llvm::IntegerType>(CGF.ConvertType(CGF.getContext().getSizeType())); 3648ed55a54SJohn McCall 3657648fb46SArgyrios Kyrtzidis CharUnits TypeSize = CGF.getContext().getTypeSizeInChars(ElemType); 3668ed55a54SJohn McCall 3678ed55a54SJohn McCall if (!E->isArray()) { 36805fc5be3SDouglas Gregor SizeWithoutCookie = llvm::ConstantInt::get(SizeTy, TypeSize.getQuantity()); 36905fc5be3SDouglas Gregor return SizeWithoutCookie; 37005fc5be3SDouglas Gregor } 37159486a2dSAnders Carlsson 3728ed55a54SJohn McCall // Figure out the cookie size. 3738ed55a54SJohn McCall CharUnits CookieSize = CalculateCookiePadding(CGF, E); 3748ed55a54SJohn McCall 37559486a2dSAnders Carlsson // Emit the array size expression. 3767648fb46SArgyrios Kyrtzidis // We multiply the size of all dimensions for NumElements. 3777648fb46SArgyrios Kyrtzidis // e.g for 'int[2][3]', ElemType is 'int' and NumElements is 6. 37859486a2dSAnders Carlsson NumElements = CGF.EmitScalarExpr(E->getArraySize()); 3798ed55a54SJohn McCall assert(NumElements->getType() == SizeTy && "element count not a size_t"); 3808ed55a54SJohn McCall 3818ed55a54SJohn McCall uint64_t ArraySizeMultiplier = 1; 3827648fb46SArgyrios Kyrtzidis while (const ConstantArrayType *CAT 3837648fb46SArgyrios Kyrtzidis = CGF.getContext().getAsConstantArrayType(ElemType)) { 3847648fb46SArgyrios Kyrtzidis ElemType = CAT->getElementType(); 3858ed55a54SJohn McCall ArraySizeMultiplier *= CAT->getSize().getZExtValue(); 3867648fb46SArgyrios Kyrtzidis } 38759486a2dSAnders Carlsson 3888ed55a54SJohn McCall llvm::Value *Size; 38932ac583dSChris Lattner 39032ac583dSChris Lattner // If someone is doing 'new int[42]' there is no need to do a dynamic check. 39132ac583dSChris Lattner // Don't bloat the -O0 code. 39232ac583dSChris Lattner if (llvm::ConstantInt *NumElementsC = 39332ac583dSChris Lattner dyn_cast<llvm::ConstantInt>(NumElements)) { 39432ac583dSChris Lattner llvm::APInt NEC = NumElementsC->getValue(); 3958ed55a54SJohn McCall unsigned SizeWidth = NEC.getBitWidth(); 39632ac583dSChris Lattner 3978ed55a54SJohn McCall // Determine if there is an overflow here by doing an extended multiply. 3988ed55a54SJohn McCall NEC.zext(SizeWidth*2); 3998ed55a54SJohn McCall llvm::APInt SC(SizeWidth*2, TypeSize.getQuantity()); 40032ac583dSChris Lattner SC *= NEC; 40132ac583dSChris Lattner 4028ed55a54SJohn McCall if (!CookieSize.isZero()) { 4038ed55a54SJohn McCall // Save the current size without a cookie. We don't care if an 4048ed55a54SJohn McCall // overflow's already happened because SizeWithoutCookie isn't 4058ed55a54SJohn McCall // used if the allocator returns null or throws, as it should 4068ed55a54SJohn McCall // always do on an overflow. 4078ed55a54SJohn McCall llvm::APInt SWC = SC; 4088ed55a54SJohn McCall SWC.trunc(SizeWidth); 4098ed55a54SJohn McCall SizeWithoutCookie = llvm::ConstantInt::get(SizeTy, SWC); 4108ed55a54SJohn McCall 4118ed55a54SJohn McCall // Add the cookie size. 4128ed55a54SJohn McCall SC += llvm::APInt(SizeWidth*2, CookieSize.getQuantity()); 4138ed55a54SJohn McCall } 4148ed55a54SJohn McCall 4158ed55a54SJohn McCall if (SC.countLeadingZeros() >= SizeWidth) { 4168ed55a54SJohn McCall SC.trunc(SizeWidth); 4178ed55a54SJohn McCall Size = llvm::ConstantInt::get(SizeTy, SC); 41832ac583dSChris Lattner } else { 41932ac583dSChris Lattner // On overflow, produce a -1 so operator new throws. 4208ed55a54SJohn McCall Size = llvm::Constant::getAllOnesValue(SizeTy); 42132ac583dSChris Lattner } 42232ac583dSChris Lattner 4238ed55a54SJohn McCall // Scale NumElements while we're at it. 4248ed55a54SJohn McCall uint64_t N = NEC.getZExtValue() * ArraySizeMultiplier; 4258ed55a54SJohn McCall NumElements = llvm::ConstantInt::get(SizeTy, N); 42647b4629bSFariborz Jahanian 4278ed55a54SJohn McCall // Otherwise, we don't need to do an overflow-checked multiplication if 4288ed55a54SJohn McCall // we're multiplying by one. 4298ed55a54SJohn McCall } else if (TypeSize.isOne()) { 4308ed55a54SJohn McCall assert(ArraySizeMultiplier == 1); 431f2f38701SChris Lattner 4328ed55a54SJohn McCall Size = NumElements; 433f2f38701SChris Lattner 4348ed55a54SJohn McCall // If we need a cookie, add its size in with an overflow check. 4358ed55a54SJohn McCall // This is maybe a little paranoid. 4368ed55a54SJohn McCall if (!CookieSize.isZero()) { 43705fc5be3SDouglas Gregor SizeWithoutCookie = Size; 438f2f38701SChris Lattner 4398ed55a54SJohn McCall llvm::Value *CookieSizeV 4408ed55a54SJohn McCall = llvm::ConstantInt::get(SizeTy, CookieSize.getQuantity()); 4418ed55a54SJohn McCall 4428ed55a54SJohn McCall const llvm::Type *Types[] = { SizeTy }; 4438ed55a54SJohn McCall llvm::Value *UAddF 4448ed55a54SJohn McCall = CGF.CGM.getIntrinsic(llvm::Intrinsic::uadd_with_overflow, Types, 1); 4458ed55a54SJohn McCall llvm::Value *AddRes 4468ed55a54SJohn McCall = CGF.Builder.CreateCall2(UAddF, Size, CookieSizeV); 4478ed55a54SJohn McCall 4488ed55a54SJohn McCall Size = CGF.Builder.CreateExtractValue(AddRes, 0); 4498ed55a54SJohn McCall llvm::Value *DidOverflow = CGF.Builder.CreateExtractValue(AddRes, 1); 4508ed55a54SJohn McCall Size = CGF.Builder.CreateSelect(DidOverflow, 4518ed55a54SJohn McCall llvm::ConstantInt::get(SizeTy, -1), 4528ed55a54SJohn McCall Size); 4538ed55a54SJohn McCall } 4548ed55a54SJohn McCall 4558ed55a54SJohn McCall // Otherwise use the int.umul.with.overflow intrinsic. 4568ed55a54SJohn McCall } else { 4578ed55a54SJohn McCall llvm::Value *OutermostElementSize 4588ed55a54SJohn McCall = llvm::ConstantInt::get(SizeTy, TypeSize.getQuantity()); 4598ed55a54SJohn McCall 4608ed55a54SJohn McCall llvm::Value *NumOutermostElements = NumElements; 4618ed55a54SJohn McCall 4628ed55a54SJohn McCall // Scale NumElements by the array size multiplier. This might 4638ed55a54SJohn McCall // overflow, but only if the multiplication below also overflows, 4648ed55a54SJohn McCall // in which case this multiplication isn't used. 4658ed55a54SJohn McCall if (ArraySizeMultiplier != 1) 4668ed55a54SJohn McCall NumElements = CGF.Builder.CreateMul(NumElements, 4678ed55a54SJohn McCall llvm::ConstantInt::get(SizeTy, ArraySizeMultiplier)); 4688ed55a54SJohn McCall 4698ed55a54SJohn McCall // The requested size of the outermost array is non-constant. 4708ed55a54SJohn McCall // Multiply that by the static size of the elements of that array; 4718ed55a54SJohn McCall // on unsigned overflow, set the size to -1 to trigger an 4728ed55a54SJohn McCall // exception from the allocation routine. This is sufficient to 4738ed55a54SJohn McCall // prevent buffer overruns from the allocator returning a 4748ed55a54SJohn McCall // seemingly valid pointer to insufficient space. This idea comes 4758ed55a54SJohn McCall // originally from MSVC, and GCC has an open bug requesting 4768ed55a54SJohn McCall // similar behavior: 4778ed55a54SJohn McCall // http://gcc.gnu.org/bugzilla/show_bug.cgi?id=19351 4788ed55a54SJohn McCall // 4798ed55a54SJohn McCall // This will not be sufficient for C++0x, which requires a 4808ed55a54SJohn McCall // specific exception class (std::bad_array_new_length). 4818ed55a54SJohn McCall // That will require ABI support that has not yet been specified. 4828ed55a54SJohn McCall const llvm::Type *Types[] = { SizeTy }; 4838ed55a54SJohn McCall llvm::Value *UMulF 4848ed55a54SJohn McCall = CGF.CGM.getIntrinsic(llvm::Intrinsic::umul_with_overflow, Types, 1); 4858ed55a54SJohn McCall llvm::Value *MulRes = CGF.Builder.CreateCall2(UMulF, NumOutermostElements, 4868ed55a54SJohn McCall OutermostElementSize); 4878ed55a54SJohn McCall 4888ed55a54SJohn McCall // The overflow bit. 4898ed55a54SJohn McCall llvm::Value *DidOverflow = CGF.Builder.CreateExtractValue(MulRes, 1); 4908ed55a54SJohn McCall 4918ed55a54SJohn McCall // The result of the multiplication. 4928ed55a54SJohn McCall Size = CGF.Builder.CreateExtractValue(MulRes, 0); 4938ed55a54SJohn McCall 4948ed55a54SJohn McCall // If we have a cookie, we need to add that size in, too. 4958ed55a54SJohn McCall if (!CookieSize.isZero()) { 4968ed55a54SJohn McCall SizeWithoutCookie = Size; 4978ed55a54SJohn McCall 4988ed55a54SJohn McCall llvm::Value *CookieSizeV 4998ed55a54SJohn McCall = llvm::ConstantInt::get(SizeTy, CookieSize.getQuantity()); 5008ed55a54SJohn McCall llvm::Value *UAddF 5018ed55a54SJohn McCall = CGF.CGM.getIntrinsic(llvm::Intrinsic::uadd_with_overflow, Types, 1); 5028ed55a54SJohn McCall llvm::Value *AddRes 5038ed55a54SJohn McCall = CGF.Builder.CreateCall2(UAddF, SizeWithoutCookie, CookieSizeV); 5048ed55a54SJohn McCall 5058ed55a54SJohn McCall Size = CGF.Builder.CreateExtractValue(AddRes, 0); 5068ed55a54SJohn McCall 5078ed55a54SJohn McCall llvm::Value *AddDidOverflow = CGF.Builder.CreateExtractValue(AddRes, 1); 5088ed55a54SJohn McCall DidOverflow = CGF.Builder.CreateAnd(DidOverflow, AddDidOverflow); 5098ed55a54SJohn McCall } 5108ed55a54SJohn McCall 5118ed55a54SJohn McCall Size = CGF.Builder.CreateSelect(DidOverflow, 5128ed55a54SJohn McCall llvm::ConstantInt::get(SizeTy, -1), 5138ed55a54SJohn McCall Size); 5148ed55a54SJohn McCall } 5158ed55a54SJohn McCall 5168ed55a54SJohn McCall if (CookieSize.isZero()) 5178ed55a54SJohn McCall SizeWithoutCookie = Size; 5188ed55a54SJohn McCall else 5198ed55a54SJohn McCall assert(SizeWithoutCookie && "didn't set SizeWithoutCookie?"); 52059486a2dSAnders Carlsson 52132ac583dSChris Lattner return Size; 52259486a2dSAnders Carlsson } 52359486a2dSAnders Carlsson 524d5202e09SFariborz Jahanian static void StoreAnyExprIntoOneUnit(CodeGenFunction &CGF, const CXXNewExpr *E, 525d5202e09SFariborz Jahanian llvm::Value *NewPtr) { 526d5202e09SFariborz Jahanian 527d5202e09SFariborz Jahanian assert(E->getNumConstructorArgs() == 1 && 528d5202e09SFariborz Jahanian "Can only have one argument to initializer of POD type."); 529d5202e09SFariborz Jahanian 530d5202e09SFariborz Jahanian const Expr *Init = E->getConstructorArg(0); 531d5202e09SFariborz Jahanian QualType AllocType = E->getAllocatedType(); 532d5202e09SFariborz Jahanian 5330381634aSDaniel Dunbar unsigned Alignment = 5340381634aSDaniel Dunbar CGF.getContext().getTypeAlignInChars(AllocType).getQuantity(); 535d5202e09SFariborz Jahanian if (!CGF.hasAggregateLLVMType(AllocType)) 536d5202e09SFariborz Jahanian CGF.EmitStoreOfScalar(CGF.EmitScalarExpr(Init), NewPtr, 5370381634aSDaniel Dunbar AllocType.isVolatileQualified(), Alignment, 5380381634aSDaniel Dunbar AllocType); 539d5202e09SFariborz Jahanian else if (AllocType->isAnyComplexType()) 540d5202e09SFariborz Jahanian CGF.EmitComplexExprIntoAddr(Init, NewPtr, 541d5202e09SFariborz Jahanian AllocType.isVolatileQualified()); 542d5202e09SFariborz Jahanian else 543d5202e09SFariborz Jahanian CGF.EmitAggExpr(Init, NewPtr, AllocType.isVolatileQualified()); 544d5202e09SFariborz Jahanian } 545d5202e09SFariborz Jahanian 546d5202e09SFariborz Jahanian void 547d5202e09SFariborz Jahanian CodeGenFunction::EmitNewArrayInitializer(const CXXNewExpr *E, 548d5202e09SFariborz Jahanian llvm::Value *NewPtr, 549d5202e09SFariborz Jahanian llvm::Value *NumElements) { 550b66b08efSFariborz Jahanian // We have a POD type. 551b66b08efSFariborz Jahanian if (E->getNumConstructorArgs() == 0) 552b66b08efSFariborz Jahanian return; 553b66b08efSFariborz Jahanian 554d5202e09SFariborz Jahanian const llvm::Type *SizeTy = ConvertType(getContext().getSizeType()); 555d5202e09SFariborz Jahanian 556d5202e09SFariborz Jahanian // Create a temporary for the loop index and initialize it with 0. 557d5202e09SFariborz Jahanian llvm::Value *IndexPtr = CreateTempAlloca(SizeTy, "loop.index"); 558d5202e09SFariborz Jahanian llvm::Value *Zero = llvm::Constant::getNullValue(SizeTy); 559d5202e09SFariborz Jahanian Builder.CreateStore(Zero, IndexPtr); 560d5202e09SFariborz Jahanian 561d5202e09SFariborz Jahanian // Start the loop with a block that tests the condition. 562d5202e09SFariborz Jahanian llvm::BasicBlock *CondBlock = createBasicBlock("for.cond"); 563d5202e09SFariborz Jahanian llvm::BasicBlock *AfterFor = createBasicBlock("for.end"); 564d5202e09SFariborz Jahanian 565d5202e09SFariborz Jahanian EmitBlock(CondBlock); 566d5202e09SFariborz Jahanian 567d5202e09SFariborz Jahanian llvm::BasicBlock *ForBody = createBasicBlock("for.body"); 568d5202e09SFariborz Jahanian 569d5202e09SFariborz Jahanian // Generate: if (loop-index < number-of-elements fall to the loop body, 570d5202e09SFariborz Jahanian // otherwise, go to the block after the for-loop. 571d5202e09SFariborz Jahanian llvm::Value *Counter = Builder.CreateLoad(IndexPtr); 572d5202e09SFariborz Jahanian llvm::Value *IsLess = Builder.CreateICmpULT(Counter, NumElements, "isless"); 573d5202e09SFariborz Jahanian // If the condition is true, execute the body. 574d5202e09SFariborz Jahanian Builder.CreateCondBr(IsLess, ForBody, AfterFor); 575d5202e09SFariborz Jahanian 576d5202e09SFariborz Jahanian EmitBlock(ForBody); 577d5202e09SFariborz Jahanian 578d5202e09SFariborz Jahanian llvm::BasicBlock *ContinueBlock = createBasicBlock("for.inc"); 579d5202e09SFariborz Jahanian // Inside the loop body, emit the constructor call on the array element. 580d5202e09SFariborz Jahanian Counter = Builder.CreateLoad(IndexPtr); 581d5202e09SFariborz Jahanian llvm::Value *Address = Builder.CreateInBoundsGEP(NewPtr, Counter, 582d5202e09SFariborz Jahanian "arrayidx"); 583d5202e09SFariborz Jahanian StoreAnyExprIntoOneUnit(*this, E, Address); 584d5202e09SFariborz Jahanian 585d5202e09SFariborz Jahanian EmitBlock(ContinueBlock); 586d5202e09SFariborz Jahanian 587d5202e09SFariborz Jahanian // Emit the increment of the loop counter. 588d5202e09SFariborz Jahanian llvm::Value *NextVal = llvm::ConstantInt::get(SizeTy, 1); 589d5202e09SFariborz Jahanian Counter = Builder.CreateLoad(IndexPtr); 590d5202e09SFariborz Jahanian NextVal = Builder.CreateAdd(Counter, NextVal, "inc"); 591d5202e09SFariborz Jahanian Builder.CreateStore(NextVal, IndexPtr); 592d5202e09SFariborz Jahanian 593d5202e09SFariborz Jahanian // Finally, branch back up to the condition for the next iteration. 594d5202e09SFariborz Jahanian EmitBranch(CondBlock); 595d5202e09SFariborz Jahanian 596d5202e09SFariborz Jahanian // Emit the fall-through block. 597d5202e09SFariborz Jahanian EmitBlock(AfterFor, true); 598d5202e09SFariborz Jahanian } 599d5202e09SFariborz Jahanian 60005fc5be3SDouglas Gregor static void EmitZeroMemSet(CodeGenFunction &CGF, QualType T, 60105fc5be3SDouglas Gregor llvm::Value *NewPtr, llvm::Value *Size) { 60205fc5be3SDouglas Gregor llvm::LLVMContext &VMContext = CGF.CGM.getLLVMContext(); 60305fc5be3SDouglas Gregor const llvm::Type *BP = llvm::Type::getInt8PtrTy(VMContext); 60405fc5be3SDouglas Gregor if (NewPtr->getType() != BP) 60505fc5be3SDouglas Gregor NewPtr = CGF.Builder.CreateBitCast(NewPtr, BP, "tmp"); 60605fc5be3SDouglas Gregor 60705fc5be3SDouglas Gregor CGF.Builder.CreateCall5(CGF.CGM.getMemSetFn(BP, CGF.IntPtrTy), NewPtr, 60805fc5be3SDouglas Gregor llvm::Constant::getNullValue(llvm::Type::getInt8Ty(VMContext)), 60905fc5be3SDouglas Gregor Size, 61005fc5be3SDouglas Gregor llvm::ConstantInt::get(CGF.Int32Ty, 61105fc5be3SDouglas Gregor CGF.getContext().getTypeAlign(T)/8), 61205fc5be3SDouglas Gregor llvm::ConstantInt::get(llvm::Type::getInt1Ty(VMContext), 61305fc5be3SDouglas Gregor 0)); 61405fc5be3SDouglas Gregor } 61505fc5be3SDouglas Gregor 61659486a2dSAnders Carlsson static void EmitNewInitializer(CodeGenFunction &CGF, const CXXNewExpr *E, 61759486a2dSAnders Carlsson llvm::Value *NewPtr, 61805fc5be3SDouglas Gregor llvm::Value *NumElements, 61905fc5be3SDouglas Gregor llvm::Value *AllocSizeWithoutCookie) { 6203a202f60SAnders Carlsson if (E->isArray()) { 621d040e6b2SAnders Carlsson if (CXXConstructorDecl *Ctor = E->getConstructor()) { 62205fc5be3SDouglas Gregor bool RequiresZeroInitialization = false; 62305fc5be3SDouglas Gregor if (Ctor->getParent()->hasTrivialConstructor()) { 62405fc5be3SDouglas Gregor // If new expression did not specify value-initialization, then there 62505fc5be3SDouglas Gregor // is no initialization. 62605fc5be3SDouglas Gregor if (!E->hasInitializer() || Ctor->getParent()->isEmpty()) 62705fc5be3SDouglas Gregor return; 62805fc5be3SDouglas Gregor 629614dbdcdSJohn McCall if (CGF.CGM.getTypes().isZeroInitializable(E->getAllocatedType())) { 63005fc5be3SDouglas Gregor // Optimization: since zero initialization will just set the memory 63105fc5be3SDouglas Gregor // to all zeroes, generate a single memset to do it in one shot. 63205fc5be3SDouglas Gregor EmitZeroMemSet(CGF, E->getAllocatedType(), NewPtr, 63305fc5be3SDouglas Gregor AllocSizeWithoutCookie); 6343a202f60SAnders Carlsson return; 6353a202f60SAnders Carlsson } 63605fc5be3SDouglas Gregor 63705fc5be3SDouglas Gregor RequiresZeroInitialization = true; 63805fc5be3SDouglas Gregor } 63905fc5be3SDouglas Gregor 64005fc5be3SDouglas Gregor CGF.EmitCXXAggrConstructorCall(Ctor, NumElements, NewPtr, 64105fc5be3SDouglas Gregor E->constructor_arg_begin(), 64205fc5be3SDouglas Gregor E->constructor_arg_end(), 64305fc5be3SDouglas Gregor RequiresZeroInitialization); 64405fc5be3SDouglas Gregor return; 64505fc5be3SDouglas Gregor } else if (E->getNumConstructorArgs() == 1 && 64605fc5be3SDouglas Gregor isa<ImplicitValueInitExpr>(E->getConstructorArg(0))) { 64705fc5be3SDouglas Gregor // Optimization: since zero initialization will just set the memory 64805fc5be3SDouglas Gregor // to all zeroes, generate a single memset to do it in one shot. 64905fc5be3SDouglas Gregor EmitZeroMemSet(CGF, E->getAllocatedType(), NewPtr, 65005fc5be3SDouglas Gregor AllocSizeWithoutCookie); 65105fc5be3SDouglas Gregor return; 65205fc5be3SDouglas Gregor } else { 653d5202e09SFariborz Jahanian CGF.EmitNewArrayInitializer(E, NewPtr, NumElements); 654d5202e09SFariborz Jahanian return; 655d040e6b2SAnders Carlsson } 656d5202e09SFariborz Jahanian } 65759486a2dSAnders Carlsson 65859486a2dSAnders Carlsson if (CXXConstructorDecl *Ctor = E->getConstructor()) { 659747eb784SDouglas Gregor // Per C++ [expr.new]p15, if we have an initializer, then we're performing 660747eb784SDouglas Gregor // direct initialization. C++ [dcl.init]p5 requires that we 661747eb784SDouglas Gregor // zero-initialize storage if there are no user-declared constructors. 662747eb784SDouglas Gregor if (E->hasInitializer() && 663747eb784SDouglas Gregor !Ctor->getParent()->hasUserDeclaredConstructor() && 664747eb784SDouglas Gregor !Ctor->getParent()->isEmpty()) 665747eb784SDouglas Gregor CGF.EmitNullInitialization(NewPtr, E->getAllocatedType()); 666747eb784SDouglas Gregor 667e11f9ce9SAnders Carlsson CGF.EmitCXXConstructorCall(Ctor, Ctor_Complete, /*ForVirtualBase=*/false, 668e11f9ce9SAnders Carlsson NewPtr, E->constructor_arg_begin(), 66959486a2dSAnders Carlsson E->constructor_arg_end()); 67059486a2dSAnders Carlsson 67159486a2dSAnders Carlsson return; 67259486a2dSAnders Carlsson } 673b66b08efSFariborz Jahanian // We have a POD type. 674b66b08efSFariborz Jahanian if (E->getNumConstructorArgs() == 0) 675b66b08efSFariborz Jahanian return; 67659486a2dSAnders Carlsson 677d5202e09SFariborz Jahanian StoreAnyExprIntoOneUnit(CGF, E, NewPtr); 67859486a2dSAnders Carlsson } 67959486a2dSAnders Carlsson 680824c2f53SJohn McCall namespace { 681824c2f53SJohn McCall /// A cleanup to call the given 'operator delete' function upon 682824c2f53SJohn McCall /// abnormal exit from a new expression. 683824c2f53SJohn McCall class CallDeleteDuringNew : public EHScopeStack::Cleanup { 684824c2f53SJohn McCall size_t NumPlacementArgs; 685824c2f53SJohn McCall const FunctionDecl *OperatorDelete; 686824c2f53SJohn McCall llvm::Value *Ptr; 687824c2f53SJohn McCall llvm::Value *AllocSize; 688824c2f53SJohn McCall 689824c2f53SJohn McCall RValue *getPlacementArgs() { return reinterpret_cast<RValue*>(this+1); } 690824c2f53SJohn McCall 691824c2f53SJohn McCall public: 692824c2f53SJohn McCall static size_t getExtraSize(size_t NumPlacementArgs) { 693824c2f53SJohn McCall return NumPlacementArgs * sizeof(RValue); 694824c2f53SJohn McCall } 695824c2f53SJohn McCall 696824c2f53SJohn McCall CallDeleteDuringNew(size_t NumPlacementArgs, 697824c2f53SJohn McCall const FunctionDecl *OperatorDelete, 698824c2f53SJohn McCall llvm::Value *Ptr, 699824c2f53SJohn McCall llvm::Value *AllocSize) 700824c2f53SJohn McCall : NumPlacementArgs(NumPlacementArgs), OperatorDelete(OperatorDelete), 701824c2f53SJohn McCall Ptr(Ptr), AllocSize(AllocSize) {} 702824c2f53SJohn McCall 703824c2f53SJohn McCall void setPlacementArg(unsigned I, RValue Arg) { 704824c2f53SJohn McCall assert(I < NumPlacementArgs && "index out of range"); 705824c2f53SJohn McCall getPlacementArgs()[I] = Arg; 706824c2f53SJohn McCall } 707824c2f53SJohn McCall 708824c2f53SJohn McCall void Emit(CodeGenFunction &CGF, bool IsForEH) { 709824c2f53SJohn McCall const FunctionProtoType *FPT 710824c2f53SJohn McCall = OperatorDelete->getType()->getAs<FunctionProtoType>(); 711824c2f53SJohn McCall assert(FPT->getNumArgs() == NumPlacementArgs + 1 || 712*d441b1e6SJohn McCall (FPT->getNumArgs() == 2 && NumPlacementArgs == 0)); 713824c2f53SJohn McCall 714824c2f53SJohn McCall CallArgList DeleteArgs; 715824c2f53SJohn McCall 716824c2f53SJohn McCall // The first argument is always a void*. 717824c2f53SJohn McCall FunctionProtoType::arg_type_iterator AI = FPT->arg_type_begin(); 718824c2f53SJohn McCall DeleteArgs.push_back(std::make_pair(RValue::get(Ptr), *AI++)); 719824c2f53SJohn McCall 720824c2f53SJohn McCall // A member 'operator delete' can take an extra 'size_t' argument. 721824c2f53SJohn McCall if (FPT->getNumArgs() == NumPlacementArgs + 2) 722824c2f53SJohn McCall DeleteArgs.push_back(std::make_pair(RValue::get(AllocSize), *AI++)); 723824c2f53SJohn McCall 724824c2f53SJohn McCall // Pass the rest of the arguments, which must match exactly. 725824c2f53SJohn McCall for (unsigned I = 0; I != NumPlacementArgs; ++I) 726824c2f53SJohn McCall DeleteArgs.push_back(std::make_pair(getPlacementArgs()[I], *AI++)); 727824c2f53SJohn McCall 728824c2f53SJohn McCall // Call 'operator delete'. 729824c2f53SJohn McCall CGF.EmitCall(CGF.CGM.getTypes().getFunctionInfo(DeleteArgs, FPT), 730824c2f53SJohn McCall CGF.CGM.GetAddrOfFunction(OperatorDelete), 731824c2f53SJohn McCall ReturnValueSlot(), DeleteArgs, OperatorDelete); 732824c2f53SJohn McCall } 733824c2f53SJohn McCall }; 734824c2f53SJohn McCall } 735824c2f53SJohn McCall 73659486a2dSAnders Carlsson llvm::Value *CodeGenFunction::EmitCXXNewExpr(const CXXNewExpr *E) { 73759486a2dSAnders Carlsson QualType AllocType = E->getAllocatedType(); 7388ed55a54SJohn McCall if (AllocType->isArrayType()) 7398ed55a54SJohn McCall while (const ArrayType *AType = getContext().getAsArrayType(AllocType)) 7408ed55a54SJohn McCall AllocType = AType->getElementType(); 7418ed55a54SJohn McCall 74259486a2dSAnders Carlsson FunctionDecl *NewFD = E->getOperatorNew(); 74359486a2dSAnders Carlsson const FunctionProtoType *NewFTy = NewFD->getType()->getAs<FunctionProtoType>(); 74459486a2dSAnders Carlsson 74559486a2dSAnders Carlsson CallArgList NewArgs; 74659486a2dSAnders Carlsson 74759486a2dSAnders Carlsson // The allocation size is the first argument. 74859486a2dSAnders Carlsson QualType SizeTy = getContext().getSizeType(); 74959486a2dSAnders Carlsson 75059486a2dSAnders Carlsson llvm::Value *NumElements = 0; 75105fc5be3SDouglas Gregor llvm::Value *AllocSizeWithoutCookie = 0; 75247b4629bSFariborz Jahanian llvm::Value *AllocSize = EmitCXXNewAllocSize(getContext(), 75305fc5be3SDouglas Gregor *this, E, NumElements, 75405fc5be3SDouglas Gregor AllocSizeWithoutCookie); 75559486a2dSAnders Carlsson 75659486a2dSAnders Carlsson NewArgs.push_back(std::make_pair(RValue::get(AllocSize), SizeTy)); 75759486a2dSAnders Carlsson 75859486a2dSAnders Carlsson // Emit the rest of the arguments. 75959486a2dSAnders Carlsson // FIXME: Ideally, this should just use EmitCallArgs. 76059486a2dSAnders Carlsson CXXNewExpr::const_arg_iterator NewArg = E->placement_arg_begin(); 76159486a2dSAnders Carlsson 76259486a2dSAnders Carlsson // First, use the types from the function type. 76359486a2dSAnders Carlsson // We start at 1 here because the first argument (the allocation size) 76459486a2dSAnders Carlsson // has already been emitted. 76559486a2dSAnders Carlsson for (unsigned i = 1, e = NewFTy->getNumArgs(); i != e; ++i, ++NewArg) { 76659486a2dSAnders Carlsson QualType ArgType = NewFTy->getArgType(i); 76759486a2dSAnders Carlsson 76859486a2dSAnders Carlsson assert(getContext().getCanonicalType(ArgType.getNonReferenceType()). 76959486a2dSAnders Carlsson getTypePtr() == 77059486a2dSAnders Carlsson getContext().getCanonicalType(NewArg->getType()).getTypePtr() && 77159486a2dSAnders Carlsson "type mismatch in call argument!"); 77259486a2dSAnders Carlsson 77359486a2dSAnders Carlsson NewArgs.push_back(std::make_pair(EmitCallArg(*NewArg, ArgType), 77459486a2dSAnders Carlsson ArgType)); 77559486a2dSAnders Carlsson 77659486a2dSAnders Carlsson } 77759486a2dSAnders Carlsson 77859486a2dSAnders Carlsson // Either we've emitted all the call args, or we have a call to a 77959486a2dSAnders Carlsson // variadic function. 78059486a2dSAnders Carlsson assert((NewArg == E->placement_arg_end() || NewFTy->isVariadic()) && 78159486a2dSAnders Carlsson "Extra arguments in non-variadic function!"); 78259486a2dSAnders Carlsson 78359486a2dSAnders Carlsson // If we still have any arguments, emit them using the type of the argument. 78459486a2dSAnders Carlsson for (CXXNewExpr::const_arg_iterator NewArgEnd = E->placement_arg_end(); 78559486a2dSAnders Carlsson NewArg != NewArgEnd; ++NewArg) { 78659486a2dSAnders Carlsson QualType ArgType = NewArg->getType(); 78759486a2dSAnders Carlsson NewArgs.push_back(std::make_pair(EmitCallArg(*NewArg, ArgType), 78859486a2dSAnders Carlsson ArgType)); 78959486a2dSAnders Carlsson } 79059486a2dSAnders Carlsson 79159486a2dSAnders Carlsson // Emit the call to new. 79259486a2dSAnders Carlsson RValue RV = 793ab26cfa5SJohn McCall EmitCall(CGM.getTypes().getFunctionInfo(NewArgs, NewFTy), 79461a401caSAnders Carlsson CGM.GetAddrOfFunction(NewFD), ReturnValueSlot(), NewArgs, NewFD); 79559486a2dSAnders Carlsson 79659486a2dSAnders Carlsson // If an allocation function is declared with an empty exception specification 79759486a2dSAnders Carlsson // it returns null to indicate failure to allocate storage. [expr.new]p13. 79859486a2dSAnders Carlsson // (We don't need to check for null when there's no new initializer and 79959486a2dSAnders Carlsson // we're allocating a POD type). 80059486a2dSAnders Carlsson bool NullCheckResult = NewFTy->hasEmptyExceptionSpec() && 80159486a2dSAnders Carlsson !(AllocType->isPODType() && !E->hasInitializer()); 80259486a2dSAnders Carlsson 8038ed55a54SJohn McCall llvm::BasicBlock *NullCheckSource = 0; 80459486a2dSAnders Carlsson llvm::BasicBlock *NewNotNull = 0; 80559486a2dSAnders Carlsson llvm::BasicBlock *NewEnd = 0; 80659486a2dSAnders Carlsson 80759486a2dSAnders Carlsson llvm::Value *NewPtr = RV.getScalarVal(); 8088ed55a54SJohn McCall unsigned AS = cast<llvm::PointerType>(NewPtr->getType())->getAddressSpace(); 80959486a2dSAnders Carlsson 81059486a2dSAnders Carlsson if (NullCheckResult) { 8118ed55a54SJohn McCall NullCheckSource = Builder.GetInsertBlock(); 81259486a2dSAnders Carlsson NewNotNull = createBasicBlock("new.notnull"); 81359486a2dSAnders Carlsson NewEnd = createBasicBlock("new.end"); 81459486a2dSAnders Carlsson 8158ed55a54SJohn McCall llvm::Value *IsNull = Builder.CreateIsNull(NewPtr, "new.isnull"); 8168ed55a54SJohn McCall Builder.CreateCondBr(IsNull, NewEnd, NewNotNull); 81759486a2dSAnders Carlsson EmitBlock(NewNotNull); 81859486a2dSAnders Carlsson } 81959486a2dSAnders Carlsson 8208ed55a54SJohn McCall assert((AllocSize == AllocSizeWithoutCookie) == 8218ed55a54SJohn McCall CalculateCookiePadding(*this, E).isZero()); 8228ed55a54SJohn McCall if (AllocSize != AllocSizeWithoutCookie) { 8238ed55a54SJohn McCall assert(E->isArray()); 8248ed55a54SJohn McCall NewPtr = CGM.getCXXABI().InitializeArrayCookie(CGF, NewPtr, NumElements, 8258ed55a54SJohn McCall AllocType); 82659486a2dSAnders Carlsson } 82759486a2dSAnders Carlsson 828824c2f53SJohn McCall // If there's an operator delete, enter a cleanup to call it if an 829824c2f53SJohn McCall // exception is thrown. 830824c2f53SJohn McCall EHScopeStack::stable_iterator CallOperatorDelete; 831824c2f53SJohn McCall if (E->getOperatorDelete()) { 832824c2f53SJohn McCall CallDeleteDuringNew *Cleanup = CGF.EHStack 833824c2f53SJohn McCall .pushCleanupWithExtra<CallDeleteDuringNew>(EHCleanup, 834824c2f53SJohn McCall E->getNumPlacementArgs(), 835824c2f53SJohn McCall E->getOperatorDelete(), 836824c2f53SJohn McCall NewPtr, AllocSize); 837824c2f53SJohn McCall for (unsigned I = 0, N = E->getNumPlacementArgs(); I != N; ++I) 838824c2f53SJohn McCall Cleanup->setPlacementArg(I, NewArgs[I+1].first); 839824c2f53SJohn McCall CallOperatorDelete = EHStack.stable_begin(); 840824c2f53SJohn McCall } 841824c2f53SJohn McCall 842040ad500SDouglas Gregor const llvm::Type *ElementPtrTy 843040ad500SDouglas Gregor = ConvertTypeForMem(AllocType)->getPointerTo(AS); 8448ed55a54SJohn McCall NewPtr = Builder.CreateBitCast(NewPtr, ElementPtrTy); 845824c2f53SJohn McCall 8468ed55a54SJohn McCall if (E->isArray()) { 84705fc5be3SDouglas Gregor EmitNewInitializer(*this, E, NewPtr, NumElements, AllocSizeWithoutCookie); 8488ed55a54SJohn McCall 8498ed55a54SJohn McCall // NewPtr is a pointer to the base element type. If we're 8508ed55a54SJohn McCall // allocating an array of arrays, we'll need to cast back to the 8518ed55a54SJohn McCall // array pointer type. 852040ad500SDouglas Gregor const llvm::Type *ResultTy = ConvertTypeForMem(E->getType()); 8538ed55a54SJohn McCall if (NewPtr->getType() != ResultTy) 8548ed55a54SJohn McCall NewPtr = Builder.CreateBitCast(NewPtr, ResultTy); 8558ed55a54SJohn McCall } else { 85605fc5be3SDouglas Gregor EmitNewInitializer(*this, E, NewPtr, NumElements, AllocSizeWithoutCookie); 85747b4629bSFariborz Jahanian } 85859486a2dSAnders Carlsson 859824c2f53SJohn McCall // Deactivate the 'operator delete' cleanup if we finished 860824c2f53SJohn McCall // initialization. 861824c2f53SJohn McCall if (CallOperatorDelete.isValid()) 862824c2f53SJohn McCall DeactivateCleanupBlock(CallOperatorDelete); 863824c2f53SJohn McCall 86459486a2dSAnders Carlsson if (NullCheckResult) { 86559486a2dSAnders Carlsson Builder.CreateBr(NewEnd); 8668ed55a54SJohn McCall llvm::BasicBlock *NotNullSource = Builder.GetInsertBlock(); 86759486a2dSAnders Carlsson EmitBlock(NewEnd); 86859486a2dSAnders Carlsson 86959486a2dSAnders Carlsson llvm::PHINode *PHI = Builder.CreatePHI(NewPtr->getType()); 87059486a2dSAnders Carlsson PHI->reserveOperandSpace(2); 8718ed55a54SJohn McCall PHI->addIncoming(NewPtr, NotNullSource); 8728ed55a54SJohn McCall PHI->addIncoming(llvm::Constant::getNullValue(NewPtr->getType()), 8738ed55a54SJohn McCall NullCheckSource); 87459486a2dSAnders Carlsson 87559486a2dSAnders Carlsson NewPtr = PHI; 87659486a2dSAnders Carlsson } 87759486a2dSAnders Carlsson 87859486a2dSAnders Carlsson return NewPtr; 87959486a2dSAnders Carlsson } 88059486a2dSAnders Carlsson 88159486a2dSAnders Carlsson void CodeGenFunction::EmitDeleteCall(const FunctionDecl *DeleteFD, 88259486a2dSAnders Carlsson llvm::Value *Ptr, 88359486a2dSAnders Carlsson QualType DeleteTy) { 8848ed55a54SJohn McCall assert(DeleteFD->getOverloadedOperator() == OO_Delete); 8858ed55a54SJohn McCall 88659486a2dSAnders Carlsson const FunctionProtoType *DeleteFTy = 88759486a2dSAnders Carlsson DeleteFD->getType()->getAs<FunctionProtoType>(); 88859486a2dSAnders Carlsson 88959486a2dSAnders Carlsson CallArgList DeleteArgs; 89059486a2dSAnders Carlsson 89121122cf6SAnders Carlsson // Check if we need to pass the size to the delete operator. 89221122cf6SAnders Carlsson llvm::Value *Size = 0; 89321122cf6SAnders Carlsson QualType SizeTy; 89421122cf6SAnders Carlsson if (DeleteFTy->getNumArgs() == 2) { 89521122cf6SAnders Carlsson SizeTy = DeleteFTy->getArgType(1); 8967df3cbebSKen Dyck CharUnits DeleteTypeSize = getContext().getTypeSizeInChars(DeleteTy); 8977df3cbebSKen Dyck Size = llvm::ConstantInt::get(ConvertType(SizeTy), 8987df3cbebSKen Dyck DeleteTypeSize.getQuantity()); 89921122cf6SAnders Carlsson } 90021122cf6SAnders Carlsson 90159486a2dSAnders Carlsson QualType ArgTy = DeleteFTy->getArgType(0); 90259486a2dSAnders Carlsson llvm::Value *DeletePtr = Builder.CreateBitCast(Ptr, ConvertType(ArgTy)); 90359486a2dSAnders Carlsson DeleteArgs.push_back(std::make_pair(RValue::get(DeletePtr), ArgTy)); 90459486a2dSAnders Carlsson 90521122cf6SAnders Carlsson if (Size) 90659486a2dSAnders Carlsson DeleteArgs.push_back(std::make_pair(RValue::get(Size), SizeTy)); 90759486a2dSAnders Carlsson 90859486a2dSAnders Carlsson // Emit the call to delete. 909ab26cfa5SJohn McCall EmitCall(CGM.getTypes().getFunctionInfo(DeleteArgs, DeleteFTy), 91061a401caSAnders Carlsson CGM.GetAddrOfFunction(DeleteFD), ReturnValueSlot(), 91159486a2dSAnders Carlsson DeleteArgs, DeleteFD); 91259486a2dSAnders Carlsson } 91359486a2dSAnders Carlsson 9148ed55a54SJohn McCall namespace { 9158ed55a54SJohn McCall /// Calls the given 'operator delete' on a single object. 9168ed55a54SJohn McCall struct CallObjectDelete : EHScopeStack::Cleanup { 9178ed55a54SJohn McCall llvm::Value *Ptr; 9188ed55a54SJohn McCall const FunctionDecl *OperatorDelete; 9198ed55a54SJohn McCall QualType ElementType; 9208ed55a54SJohn McCall 9218ed55a54SJohn McCall CallObjectDelete(llvm::Value *Ptr, 9228ed55a54SJohn McCall const FunctionDecl *OperatorDelete, 9238ed55a54SJohn McCall QualType ElementType) 9248ed55a54SJohn McCall : Ptr(Ptr), OperatorDelete(OperatorDelete), ElementType(ElementType) {} 9258ed55a54SJohn McCall 9268ed55a54SJohn McCall void Emit(CodeGenFunction &CGF, bool IsForEH) { 9278ed55a54SJohn McCall CGF.EmitDeleteCall(OperatorDelete, Ptr, ElementType); 9288ed55a54SJohn McCall } 9298ed55a54SJohn McCall }; 9308ed55a54SJohn McCall } 9318ed55a54SJohn McCall 9328ed55a54SJohn McCall /// Emit the code for deleting a single object. 9338ed55a54SJohn McCall static void EmitObjectDelete(CodeGenFunction &CGF, 9348ed55a54SJohn McCall const FunctionDecl *OperatorDelete, 9358ed55a54SJohn McCall llvm::Value *Ptr, 9368ed55a54SJohn McCall QualType ElementType) { 9378ed55a54SJohn McCall // Find the destructor for the type, if applicable. If the 9388ed55a54SJohn McCall // destructor is virtual, we'll just emit the vcall and return. 9398ed55a54SJohn McCall const CXXDestructorDecl *Dtor = 0; 9408ed55a54SJohn McCall if (const RecordType *RT = ElementType->getAs<RecordType>()) { 9418ed55a54SJohn McCall CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl()); 9428ed55a54SJohn McCall if (!RD->hasTrivialDestructor()) { 9438ed55a54SJohn McCall Dtor = RD->getDestructor(); 9448ed55a54SJohn McCall 9458ed55a54SJohn McCall if (Dtor->isVirtual()) { 9468ed55a54SJohn McCall const llvm::Type *Ty = 9470d635f53SJohn McCall CGF.getTypes().GetFunctionType(CGF.getTypes().getFunctionInfo(Dtor, 9480d635f53SJohn McCall Dtor_Complete), 9498ed55a54SJohn McCall /*isVariadic=*/false); 9508ed55a54SJohn McCall 9518ed55a54SJohn McCall llvm::Value *Callee 9528ed55a54SJohn McCall = CGF.BuildVirtualCall(Dtor, Dtor_Deleting, Ptr, Ty); 9538ed55a54SJohn McCall CGF.EmitCXXMemberCall(Dtor, Callee, ReturnValueSlot(), Ptr, /*VTT=*/0, 9548ed55a54SJohn McCall 0, 0); 9558ed55a54SJohn McCall 9568ed55a54SJohn McCall // The dtor took care of deleting the object. 9578ed55a54SJohn McCall return; 9588ed55a54SJohn McCall } 9598ed55a54SJohn McCall } 9608ed55a54SJohn McCall } 9618ed55a54SJohn McCall 9628ed55a54SJohn McCall // Make sure that we call delete even if the dtor throws. 9638ed55a54SJohn McCall CGF.EHStack.pushCleanup<CallObjectDelete>(NormalAndEHCleanup, 9648ed55a54SJohn McCall Ptr, OperatorDelete, ElementType); 9658ed55a54SJohn McCall 9668ed55a54SJohn McCall if (Dtor) 9678ed55a54SJohn McCall CGF.EmitCXXDestructorCall(Dtor, Dtor_Complete, 9688ed55a54SJohn McCall /*ForVirtualBase=*/false, Ptr); 9698ed55a54SJohn McCall 9708ed55a54SJohn McCall CGF.PopCleanupBlock(); 9718ed55a54SJohn McCall } 9728ed55a54SJohn McCall 9738ed55a54SJohn McCall namespace { 9748ed55a54SJohn McCall /// Calls the given 'operator delete' on an array of objects. 9758ed55a54SJohn McCall struct CallArrayDelete : EHScopeStack::Cleanup { 9768ed55a54SJohn McCall llvm::Value *Ptr; 9778ed55a54SJohn McCall const FunctionDecl *OperatorDelete; 9788ed55a54SJohn McCall llvm::Value *NumElements; 9798ed55a54SJohn McCall QualType ElementType; 9808ed55a54SJohn McCall CharUnits CookieSize; 9818ed55a54SJohn McCall 9828ed55a54SJohn McCall CallArrayDelete(llvm::Value *Ptr, 9838ed55a54SJohn McCall const FunctionDecl *OperatorDelete, 9848ed55a54SJohn McCall llvm::Value *NumElements, 9858ed55a54SJohn McCall QualType ElementType, 9868ed55a54SJohn McCall CharUnits CookieSize) 9878ed55a54SJohn McCall : Ptr(Ptr), OperatorDelete(OperatorDelete), NumElements(NumElements), 9888ed55a54SJohn McCall ElementType(ElementType), CookieSize(CookieSize) {} 9898ed55a54SJohn McCall 9908ed55a54SJohn McCall void Emit(CodeGenFunction &CGF, bool IsForEH) { 9918ed55a54SJohn McCall const FunctionProtoType *DeleteFTy = 9928ed55a54SJohn McCall OperatorDelete->getType()->getAs<FunctionProtoType>(); 9938ed55a54SJohn McCall assert(DeleteFTy->getNumArgs() == 1 || DeleteFTy->getNumArgs() == 2); 9948ed55a54SJohn McCall 9958ed55a54SJohn McCall CallArgList Args; 9968ed55a54SJohn McCall 9978ed55a54SJohn McCall // Pass the pointer as the first argument. 9988ed55a54SJohn McCall QualType VoidPtrTy = DeleteFTy->getArgType(0); 9998ed55a54SJohn McCall llvm::Value *DeletePtr 10008ed55a54SJohn McCall = CGF.Builder.CreateBitCast(Ptr, CGF.ConvertType(VoidPtrTy)); 10018ed55a54SJohn McCall Args.push_back(std::make_pair(RValue::get(DeletePtr), VoidPtrTy)); 10028ed55a54SJohn McCall 10038ed55a54SJohn McCall // Pass the original requested size as the second argument. 10048ed55a54SJohn McCall if (DeleteFTy->getNumArgs() == 2) { 10058ed55a54SJohn McCall QualType size_t = DeleteFTy->getArgType(1); 10068ed55a54SJohn McCall const llvm::IntegerType *SizeTy 10078ed55a54SJohn McCall = cast<llvm::IntegerType>(CGF.ConvertType(size_t)); 10088ed55a54SJohn McCall 10098ed55a54SJohn McCall CharUnits ElementTypeSize = 10108ed55a54SJohn McCall CGF.CGM.getContext().getTypeSizeInChars(ElementType); 10118ed55a54SJohn McCall 10128ed55a54SJohn McCall // The size of an element, multiplied by the number of elements. 10138ed55a54SJohn McCall llvm::Value *Size 10148ed55a54SJohn McCall = llvm::ConstantInt::get(SizeTy, ElementTypeSize.getQuantity()); 10158ed55a54SJohn McCall Size = CGF.Builder.CreateMul(Size, NumElements); 10168ed55a54SJohn McCall 10178ed55a54SJohn McCall // Plus the size of the cookie if applicable. 10188ed55a54SJohn McCall if (!CookieSize.isZero()) { 10198ed55a54SJohn McCall llvm::Value *CookieSizeV 10208ed55a54SJohn McCall = llvm::ConstantInt::get(SizeTy, CookieSize.getQuantity()); 10218ed55a54SJohn McCall Size = CGF.Builder.CreateAdd(Size, CookieSizeV); 10228ed55a54SJohn McCall } 10238ed55a54SJohn McCall 10248ed55a54SJohn McCall Args.push_back(std::make_pair(RValue::get(Size), size_t)); 10258ed55a54SJohn McCall } 10268ed55a54SJohn McCall 10278ed55a54SJohn McCall // Emit the call to delete. 10288ed55a54SJohn McCall CGF.EmitCall(CGF.getTypes().getFunctionInfo(Args, DeleteFTy), 10298ed55a54SJohn McCall CGF.CGM.GetAddrOfFunction(OperatorDelete), 10308ed55a54SJohn McCall ReturnValueSlot(), Args, OperatorDelete); 10318ed55a54SJohn McCall } 10328ed55a54SJohn McCall }; 10338ed55a54SJohn McCall } 10348ed55a54SJohn McCall 10358ed55a54SJohn McCall /// Emit the code for deleting an array of objects. 10368ed55a54SJohn McCall static void EmitArrayDelete(CodeGenFunction &CGF, 10378ed55a54SJohn McCall const FunctionDecl *OperatorDelete, 10388ed55a54SJohn McCall llvm::Value *Ptr, 10398ed55a54SJohn McCall QualType ElementType) { 10408ed55a54SJohn McCall llvm::Value *NumElements = 0; 10418ed55a54SJohn McCall llvm::Value *AllocatedPtr = 0; 10428ed55a54SJohn McCall CharUnits CookieSize; 10438ed55a54SJohn McCall CGF.CGM.getCXXABI().ReadArrayCookie(CGF, Ptr, ElementType, 10448ed55a54SJohn McCall NumElements, AllocatedPtr, CookieSize); 10458ed55a54SJohn McCall 10468ed55a54SJohn McCall assert(AllocatedPtr && "ReadArrayCookie didn't set AllocatedPtr"); 10478ed55a54SJohn McCall 10488ed55a54SJohn McCall // Make sure that we call delete even if one of the dtors throws. 10498ed55a54SJohn McCall CGF.EHStack.pushCleanup<CallArrayDelete>(NormalAndEHCleanup, 10508ed55a54SJohn McCall AllocatedPtr, OperatorDelete, 10518ed55a54SJohn McCall NumElements, ElementType, 10528ed55a54SJohn McCall CookieSize); 10538ed55a54SJohn McCall 10548ed55a54SJohn McCall if (const CXXRecordDecl *RD = ElementType->getAsCXXRecordDecl()) { 10558ed55a54SJohn McCall if (!RD->hasTrivialDestructor()) { 10568ed55a54SJohn McCall assert(NumElements && "ReadArrayCookie didn't find element count" 10578ed55a54SJohn McCall " for a class with destructor"); 10588ed55a54SJohn McCall CGF.EmitCXXAggrDestructorCall(RD->getDestructor(), NumElements, Ptr); 10598ed55a54SJohn McCall } 10608ed55a54SJohn McCall } 10618ed55a54SJohn McCall 10628ed55a54SJohn McCall CGF.PopCleanupBlock(); 10638ed55a54SJohn McCall } 10648ed55a54SJohn McCall 106559486a2dSAnders Carlsson void CodeGenFunction::EmitCXXDeleteExpr(const CXXDeleteExpr *E) { 106659486a2dSAnders Carlsson 106759486a2dSAnders Carlsson // Get at the argument before we performed the implicit conversion 106859486a2dSAnders Carlsson // to void*. 106959486a2dSAnders Carlsson const Expr *Arg = E->getArgument(); 107059486a2dSAnders Carlsson while (const ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(Arg)) { 1071e302792bSJohn McCall if (ICE->getCastKind() != CK_UserDefinedConversion && 107259486a2dSAnders Carlsson ICE->getType()->isVoidPointerType()) 107359486a2dSAnders Carlsson Arg = ICE->getSubExpr(); 107459486a2dSAnders Carlsson else 107559486a2dSAnders Carlsson break; 107659486a2dSAnders Carlsson } 107759486a2dSAnders Carlsson 107859486a2dSAnders Carlsson llvm::Value *Ptr = EmitScalarExpr(Arg); 107959486a2dSAnders Carlsson 108059486a2dSAnders Carlsson // Null check the pointer. 108159486a2dSAnders Carlsson llvm::BasicBlock *DeleteNotNull = createBasicBlock("delete.notnull"); 108259486a2dSAnders Carlsson llvm::BasicBlock *DeleteEnd = createBasicBlock("delete.end"); 108359486a2dSAnders Carlsson 108459486a2dSAnders Carlsson llvm::Value *IsNull = 108559486a2dSAnders Carlsson Builder.CreateICmpEQ(Ptr, llvm::Constant::getNullValue(Ptr->getType()), 108659486a2dSAnders Carlsson "isnull"); 108759486a2dSAnders Carlsson 108859486a2dSAnders Carlsson Builder.CreateCondBr(IsNull, DeleteEnd, DeleteNotNull); 108959486a2dSAnders Carlsson EmitBlock(DeleteNotNull); 109059486a2dSAnders Carlsson 10918ed55a54SJohn McCall // We might be deleting a pointer to array. If so, GEP down to the 10928ed55a54SJohn McCall // first non-array element. 10938ed55a54SJohn McCall // (this assumes that A(*)[3][7] is converted to [3 x [7 x %A]]*) 10948ed55a54SJohn McCall QualType DeleteTy = Arg->getType()->getAs<PointerType>()->getPointeeType(); 10958ed55a54SJohn McCall if (DeleteTy->isConstantArrayType()) { 10968ed55a54SJohn McCall llvm::Value *Zero = Builder.getInt32(0); 10978ed55a54SJohn McCall llvm::SmallVector<llvm::Value*,8> GEP; 109859486a2dSAnders Carlsson 10998ed55a54SJohn McCall GEP.push_back(Zero); // point at the outermost array 11008ed55a54SJohn McCall 11018ed55a54SJohn McCall // For each layer of array type we're pointing at: 11028ed55a54SJohn McCall while (const ConstantArrayType *Arr 11038ed55a54SJohn McCall = getContext().getAsConstantArrayType(DeleteTy)) { 11048ed55a54SJohn McCall // 1. Unpeel the array type. 11058ed55a54SJohn McCall DeleteTy = Arr->getElementType(); 11068ed55a54SJohn McCall 11078ed55a54SJohn McCall // 2. GEP to the first element of the array. 11088ed55a54SJohn McCall GEP.push_back(Zero); 11098ed55a54SJohn McCall } 11108ed55a54SJohn McCall 11118ed55a54SJohn McCall Ptr = Builder.CreateInBoundsGEP(Ptr, GEP.begin(), GEP.end(), "del.first"); 11128ed55a54SJohn McCall } 11138ed55a54SJohn McCall 111404f36218SDouglas Gregor assert(ConvertTypeForMem(DeleteTy) == 111504f36218SDouglas Gregor cast<llvm::PointerType>(Ptr->getType())->getElementType()); 11168ed55a54SJohn McCall 111759486a2dSAnders Carlsson if (E->isArrayForm()) { 11188ed55a54SJohn McCall EmitArrayDelete(*this, E->getOperatorDelete(), Ptr, DeleteTy); 11198ed55a54SJohn McCall } else { 11208ed55a54SJohn McCall EmitObjectDelete(*this, E->getOperatorDelete(), Ptr, DeleteTy); 112159486a2dSAnders Carlsson } 112259486a2dSAnders Carlsson 112359486a2dSAnders Carlsson EmitBlock(DeleteEnd); 112459486a2dSAnders Carlsson } 112559486a2dSAnders Carlsson 112659486a2dSAnders Carlsson llvm::Value * CodeGenFunction::EmitCXXTypeidExpr(const CXXTypeidExpr *E) { 112759486a2dSAnders Carlsson QualType Ty = E->getType(); 112859486a2dSAnders Carlsson const llvm::Type *LTy = ConvertType(Ty)->getPointerTo(); 1129fd7dfeb7SAnders Carlsson 11303f4336cbSAnders Carlsson if (E->isTypeOperand()) { 11313f4336cbSAnders Carlsson llvm::Constant *TypeInfo = 11323f4336cbSAnders Carlsson CGM.GetAddrOfRTTIDescriptor(E->getTypeOperand()); 11333f4336cbSAnders Carlsson return Builder.CreateBitCast(TypeInfo, LTy); 11343f4336cbSAnders Carlsson } 1135fd7dfeb7SAnders Carlsson 113659486a2dSAnders Carlsson Expr *subE = E->getExprOperand(); 113759486a2dSAnders Carlsson Ty = subE->getType(); 113859486a2dSAnders Carlsson CanQualType CanTy = CGM.getContext().getCanonicalType(Ty); 113959486a2dSAnders Carlsson Ty = CanTy.getUnqualifiedType().getNonReferenceType(); 114059486a2dSAnders Carlsson if (const RecordType *RT = Ty->getAs<RecordType>()) { 114159486a2dSAnders Carlsson const CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl()); 114259486a2dSAnders Carlsson if (RD->isPolymorphic()) { 114359486a2dSAnders Carlsson // FIXME: if subE is an lvalue do 114459486a2dSAnders Carlsson LValue Obj = EmitLValue(subE); 114559486a2dSAnders Carlsson llvm::Value *This = Obj.getAddress(); 114659486a2dSAnders Carlsson LTy = LTy->getPointerTo()->getPointerTo(); 114759486a2dSAnders Carlsson llvm::Value *V = Builder.CreateBitCast(This, LTy); 114859486a2dSAnders Carlsson // We need to do a zero check for *p, unless it has NonNullAttr. 114959486a2dSAnders Carlsson // FIXME: PointerType->hasAttr<NonNullAttr>() 115059486a2dSAnders Carlsson bool CanBeZero = false; 115159486a2dSAnders Carlsson if (UnaryOperator *UO = dyn_cast<UnaryOperator>(subE->IgnoreParens())) 1152e302792bSJohn McCall if (UO->getOpcode() == UO_Deref) 115359486a2dSAnders Carlsson CanBeZero = true; 115459486a2dSAnders Carlsson if (CanBeZero) { 115559486a2dSAnders Carlsson llvm::BasicBlock *NonZeroBlock = createBasicBlock(); 115659486a2dSAnders Carlsson llvm::BasicBlock *ZeroBlock = createBasicBlock(); 115759486a2dSAnders Carlsson 115859486a2dSAnders Carlsson llvm::Value *Zero = llvm::Constant::getNullValue(LTy); 115959486a2dSAnders Carlsson Builder.CreateCondBr(Builder.CreateICmpNE(V, Zero), 116059486a2dSAnders Carlsson NonZeroBlock, ZeroBlock); 116159486a2dSAnders Carlsson EmitBlock(ZeroBlock); 116259486a2dSAnders Carlsson /// Call __cxa_bad_typeid 116359486a2dSAnders Carlsson const llvm::Type *ResultType = llvm::Type::getVoidTy(VMContext); 116459486a2dSAnders Carlsson const llvm::FunctionType *FTy; 116559486a2dSAnders Carlsson FTy = llvm::FunctionType::get(ResultType, false); 116659486a2dSAnders Carlsson llvm::Value *F = CGM.CreateRuntimeFunction(FTy, "__cxa_bad_typeid"); 116759486a2dSAnders Carlsson Builder.CreateCall(F)->setDoesNotReturn(); 116859486a2dSAnders Carlsson Builder.CreateUnreachable(); 116959486a2dSAnders Carlsson EmitBlock(NonZeroBlock); 117059486a2dSAnders Carlsson } 117159486a2dSAnders Carlsson V = Builder.CreateLoad(V, "vtable"); 117259486a2dSAnders Carlsson V = Builder.CreateConstInBoundsGEP1_64(V, -1ULL); 117359486a2dSAnders Carlsson V = Builder.CreateLoad(V); 117459486a2dSAnders Carlsson return V; 117559486a2dSAnders Carlsson } 117659486a2dSAnders Carlsson } 11773f4336cbSAnders Carlsson return Builder.CreateBitCast(CGM.GetAddrOfRTTIDescriptor(Ty), LTy); 117859486a2dSAnders Carlsson } 117959486a2dSAnders Carlsson 118059486a2dSAnders Carlsson llvm::Value *CodeGenFunction::EmitDynamicCast(llvm::Value *V, 118159486a2dSAnders Carlsson const CXXDynamicCastExpr *DCE) { 11823f4336cbSAnders Carlsson QualType SrcTy = DCE->getSubExpr()->getType(); 11833f4336cbSAnders Carlsson QualType DestTy = DCE->getTypeAsWritten(); 11843f4336cbSAnders Carlsson QualType InnerType = DestTy->getPointeeType(); 11853f4336cbSAnders Carlsson 118659486a2dSAnders Carlsson const llvm::Type *LTy = ConvertType(DCE->getType()); 118759486a2dSAnders Carlsson 118859486a2dSAnders Carlsson bool CanBeZero = false; 118959486a2dSAnders Carlsson bool ToVoid = false; 119059486a2dSAnders Carlsson bool ThrowOnBad = false; 11913f4336cbSAnders Carlsson if (DestTy->isPointerType()) { 119259486a2dSAnders Carlsson // FIXME: if PointerType->hasAttr<NonNullAttr>(), we don't set this 119359486a2dSAnders Carlsson CanBeZero = true; 119459486a2dSAnders Carlsson if (InnerType->isVoidType()) 119559486a2dSAnders Carlsson ToVoid = true; 119659486a2dSAnders Carlsson } else { 119759486a2dSAnders Carlsson LTy = LTy->getPointerTo(); 1198fa8b4955SDouglas Gregor 1199fa8b4955SDouglas Gregor // FIXME: What if exceptions are disabled? 120059486a2dSAnders Carlsson ThrowOnBad = true; 120159486a2dSAnders Carlsson } 120259486a2dSAnders Carlsson 12033f4336cbSAnders Carlsson if (SrcTy->isPointerType() || SrcTy->isReferenceType()) 12043f4336cbSAnders Carlsson SrcTy = SrcTy->getPointeeType(); 12053f4336cbSAnders Carlsson SrcTy = SrcTy.getUnqualifiedType(); 12063f4336cbSAnders Carlsson 12070087bc85SAnders Carlsson if (DestTy->isPointerType() || DestTy->isReferenceType()) 12083f4336cbSAnders Carlsson DestTy = DestTy->getPointeeType(); 12093f4336cbSAnders Carlsson DestTy = DestTy.getUnqualifiedType(); 121059486a2dSAnders Carlsson 121159486a2dSAnders Carlsson llvm::BasicBlock *ContBlock = createBasicBlock(); 121259486a2dSAnders Carlsson llvm::BasicBlock *NullBlock = 0; 121359486a2dSAnders Carlsson llvm::BasicBlock *NonZeroBlock = 0; 121459486a2dSAnders Carlsson if (CanBeZero) { 121559486a2dSAnders Carlsson NonZeroBlock = createBasicBlock(); 121659486a2dSAnders Carlsson NullBlock = createBasicBlock(); 12173f4336cbSAnders Carlsson Builder.CreateCondBr(Builder.CreateIsNotNull(V), NonZeroBlock, NullBlock); 121859486a2dSAnders Carlsson EmitBlock(NonZeroBlock); 121959486a2dSAnders Carlsson } 122059486a2dSAnders Carlsson 122159486a2dSAnders Carlsson llvm::BasicBlock *BadCastBlock = 0; 122259486a2dSAnders Carlsson 12233f4336cbSAnders Carlsson const llvm::Type *PtrDiffTy = ConvertType(getContext().getPointerDiffType()); 122459486a2dSAnders Carlsson 122559486a2dSAnders Carlsson // See if this is a dynamic_cast(void*) 122659486a2dSAnders Carlsson if (ToVoid) { 122759486a2dSAnders Carlsson llvm::Value *This = V; 122859486a2dSAnders Carlsson V = Builder.CreateBitCast(This, PtrDiffTy->getPointerTo()->getPointerTo()); 122959486a2dSAnders Carlsson V = Builder.CreateLoad(V, "vtable"); 123059486a2dSAnders Carlsson V = Builder.CreateConstInBoundsGEP1_64(V, -2ULL); 123159486a2dSAnders Carlsson V = Builder.CreateLoad(V, "offset to top"); 123259486a2dSAnders Carlsson This = Builder.CreateBitCast(This, llvm::Type::getInt8PtrTy(VMContext)); 123359486a2dSAnders Carlsson V = Builder.CreateInBoundsGEP(This, V); 123459486a2dSAnders Carlsson V = Builder.CreateBitCast(V, LTy); 123559486a2dSAnders Carlsson } else { 123659486a2dSAnders Carlsson /// Call __dynamic_cast 123759486a2dSAnders Carlsson const llvm::Type *ResultType = llvm::Type::getInt8PtrTy(VMContext); 123859486a2dSAnders Carlsson const llvm::FunctionType *FTy; 123959486a2dSAnders Carlsson std::vector<const llvm::Type*> ArgTys; 124059486a2dSAnders Carlsson const llvm::Type *PtrToInt8Ty 124159486a2dSAnders Carlsson = llvm::Type::getInt8Ty(VMContext)->getPointerTo(); 124259486a2dSAnders Carlsson ArgTys.push_back(PtrToInt8Ty); 124359486a2dSAnders Carlsson ArgTys.push_back(PtrToInt8Ty); 124459486a2dSAnders Carlsson ArgTys.push_back(PtrToInt8Ty); 124559486a2dSAnders Carlsson ArgTys.push_back(PtrDiffTy); 124659486a2dSAnders Carlsson FTy = llvm::FunctionType::get(ResultType, ArgTys, false); 124759486a2dSAnders Carlsson 124859486a2dSAnders Carlsson // FIXME: Calculate better hint. 124959486a2dSAnders Carlsson llvm::Value *hint = llvm::ConstantInt::get(PtrDiffTy, -1ULL); 12503f4336cbSAnders Carlsson 12513f4336cbSAnders Carlsson assert(SrcTy->isRecordType() && "Src type must be record type!"); 12523f4336cbSAnders Carlsson assert(DestTy->isRecordType() && "Dest type must be record type!"); 12533f4336cbSAnders Carlsson 1254247894b3SDouglas Gregor llvm::Value *SrcArg 1255247894b3SDouglas Gregor = CGM.GetAddrOfRTTIDescriptor(SrcTy.getUnqualifiedType()); 1256247894b3SDouglas Gregor llvm::Value *DestArg 1257247894b3SDouglas Gregor = CGM.GetAddrOfRTTIDescriptor(DestTy.getUnqualifiedType()); 12583f4336cbSAnders Carlsson 125959486a2dSAnders Carlsson V = Builder.CreateBitCast(V, PtrToInt8Ty); 126059486a2dSAnders Carlsson V = Builder.CreateCall4(CGM.CreateRuntimeFunction(FTy, "__dynamic_cast"), 12613f4336cbSAnders Carlsson V, SrcArg, DestArg, hint); 126259486a2dSAnders Carlsson V = Builder.CreateBitCast(V, LTy); 126359486a2dSAnders Carlsson 126459486a2dSAnders Carlsson if (ThrowOnBad) { 126559486a2dSAnders Carlsson BadCastBlock = createBasicBlock(); 12663f4336cbSAnders Carlsson Builder.CreateCondBr(Builder.CreateIsNotNull(V), ContBlock, BadCastBlock); 126759486a2dSAnders Carlsson EmitBlock(BadCastBlock); 1268fa8b4955SDouglas Gregor /// Invoke __cxa_bad_cast 126959486a2dSAnders Carlsson ResultType = llvm::Type::getVoidTy(VMContext); 127059486a2dSAnders Carlsson const llvm::FunctionType *FBadTy; 127159486a2dSAnders Carlsson FBadTy = llvm::FunctionType::get(ResultType, false); 127259486a2dSAnders Carlsson llvm::Value *F = CGM.CreateRuntimeFunction(FBadTy, "__cxa_bad_cast"); 1273fa8b4955SDouglas Gregor if (llvm::BasicBlock *InvokeDest = getInvokeDest()) { 1274fa8b4955SDouglas Gregor llvm::BasicBlock *Cont = createBasicBlock("invoke.cont"); 1275fa8b4955SDouglas Gregor Builder.CreateInvoke(F, Cont, InvokeDest)->setDoesNotReturn(); 1276fa8b4955SDouglas Gregor EmitBlock(Cont); 1277fa8b4955SDouglas Gregor } else { 1278fa8b4955SDouglas Gregor // FIXME: Does this ever make sense? 127959486a2dSAnders Carlsson Builder.CreateCall(F)->setDoesNotReturn(); 1280fa8b4955SDouglas Gregor } 128159486a2dSAnders Carlsson Builder.CreateUnreachable(); 128259486a2dSAnders Carlsson } 128359486a2dSAnders Carlsson } 128459486a2dSAnders Carlsson 128559486a2dSAnders Carlsson if (CanBeZero) { 128659486a2dSAnders Carlsson Builder.CreateBr(ContBlock); 128759486a2dSAnders Carlsson EmitBlock(NullBlock); 128859486a2dSAnders Carlsson Builder.CreateBr(ContBlock); 128959486a2dSAnders Carlsson } 129059486a2dSAnders Carlsson EmitBlock(ContBlock); 129159486a2dSAnders Carlsson if (CanBeZero) { 129259486a2dSAnders Carlsson llvm::PHINode *PHI = Builder.CreatePHI(LTy); 129359486a2dSAnders Carlsson PHI->reserveOperandSpace(2); 129459486a2dSAnders Carlsson PHI->addIncoming(V, NonZeroBlock); 129559486a2dSAnders Carlsson PHI->addIncoming(llvm::Constant::getNullValue(LTy), NullBlock); 129659486a2dSAnders Carlsson V = PHI; 129759486a2dSAnders Carlsson } 129859486a2dSAnders Carlsson 129959486a2dSAnders Carlsson return V; 130059486a2dSAnders Carlsson } 1301