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()) { 2227a626f63SJohn McCall AggValueSlot Slot = CreateAggTemp(E->getArg(1)->getType()); 2237a626f63SJohn McCall EmitAggExpr(E->getArg(1), Slot); 22461a31241SFariborz Jahanian if (LV.isPropertyRef()) 2257a626f63SJohn McCall EmitObjCPropertySet(LV.getPropertyRefExpr(), Slot.asRValue()); 22661a31241SFariborz Jahanian else 2277a626f63SJohn 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 2687a626f63SJohn McCall CodeGenFunction::EmitCXXConstructExpr(const CXXConstructExpr *E, 2697a626f63SJohn McCall AggValueSlot Dest) { 2707a626f63SJohn 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()) 2777a626f63SJohn 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 283*8ea46b66SJohn McCall // Elide the constructor if we're constructing from a temporary. 284*8ea46b66SJohn McCall // The temporary check is required because Sema sets this on NRVO 285*8ea46b66SJohn McCall // returns. 28627da15baSAnders Carlsson if (getContext().getLangOptions().ElideConstructors && E->isElidable()) { 287*8ea46b66SJohn McCall assert(getContext().hasSameUnqualifiedType(E->getType(), 288*8ea46b66SJohn McCall E->getArg(0)->getType())); 2897a626f63SJohn McCall if (E->getArg(0)->isTemporaryObject(getContext(), CD->getParent())) { 2907a626f63SJohn McCall EmitAggExpr(E->getArg(0), Dest); 29127da15baSAnders Carlsson return; 29227da15baSAnders Carlsson } 293222cf0efSDouglas Gregor } 294630c76efSDouglas Gregor 295630c76efSDouglas Gregor const ConstantArrayType *Array 296630c76efSDouglas Gregor = getContext().getAsConstantArrayType(E->getType()); 29727da15baSAnders Carlsson if (Array) { 29827da15baSAnders Carlsson QualType BaseElementTy = getContext().getBaseElementType(Array); 29927da15baSAnders Carlsson const llvm::Type *BasePtr = ConvertType(BaseElementTy); 30027da15baSAnders Carlsson BasePtr = llvm::PointerType::getUnqual(BasePtr); 30127da15baSAnders Carlsson llvm::Value *BaseAddrPtr = 3027a626f63SJohn McCall Builder.CreateBitCast(Dest.getAddr(), BasePtr); 30327da15baSAnders Carlsson 30427da15baSAnders Carlsson EmitCXXAggrConstructorCall(CD, Array, BaseAddrPtr, 30527da15baSAnders Carlsson E->arg_begin(), E->arg_end()); 30627da15baSAnders Carlsson } 307e11f9ce9SAnders Carlsson else { 308e11f9ce9SAnders Carlsson CXXCtorType Type = 309e11f9ce9SAnders Carlsson (E->getConstructionKind() == CXXConstructExpr::CK_Complete) 310e11f9ce9SAnders Carlsson ? Ctor_Complete : Ctor_Base; 311e11f9ce9SAnders Carlsson bool ForVirtualBase = 312e11f9ce9SAnders Carlsson E->getConstructionKind() == CXXConstructExpr::CK_VirtualBase; 313e11f9ce9SAnders Carlsson 31427da15baSAnders Carlsson // Call the constructor. 3157a626f63SJohn McCall EmitCXXConstructorCall(CD, Type, ForVirtualBase, Dest.getAddr(), 31627da15baSAnders Carlsson E->arg_begin(), E->arg_end()); 31727da15baSAnders Carlsson } 318e11f9ce9SAnders Carlsson } 31927da15baSAnders Carlsson 320aa4149a2SJohn McCall /// Check whether the given operator new[] is the global placement 321aa4149a2SJohn McCall /// operator new[]. 322aa4149a2SJohn McCall static bool IsPlacementOperatorNewArray(ASTContext &Ctx, 323aa4149a2SJohn McCall const FunctionDecl *Fn) { 324aa4149a2SJohn McCall // Must be in global scope. Note that allocation functions can't be 325aa4149a2SJohn McCall // declared in namespaces. 32650c68258SSebastian Redl if (!Fn->getDeclContext()->getRedeclContext()->isFileContext()) 327aa4149a2SJohn McCall return false; 328aa4149a2SJohn McCall 329aa4149a2SJohn McCall // Signature must be void *operator new[](size_t, void*). 330aa4149a2SJohn McCall // The size_t is common to all operator new[]s. 331aa4149a2SJohn McCall if (Fn->getNumParams() != 2) 332aa4149a2SJohn McCall return false; 333aa4149a2SJohn McCall 334aa4149a2SJohn McCall CanQualType ParamType = Ctx.getCanonicalType(Fn->getParamDecl(1)->getType()); 335aa4149a2SJohn McCall return (ParamType == Ctx.VoidPtrTy); 336aa4149a2SJohn McCall } 337aa4149a2SJohn McCall 3388ed55a54SJohn McCall static CharUnits CalculateCookiePadding(CodeGenFunction &CGF, 3398ed55a54SJohn McCall const CXXNewExpr *E) { 34021122cf6SAnders Carlsson if (!E->isArray()) 3413eb55cfeSKen Dyck return CharUnits::Zero(); 34221122cf6SAnders Carlsson 343399f499fSAnders Carlsson // No cookie is required if the new operator being used is 344399f499fSAnders Carlsson // ::operator new[](size_t, void*). 345399f499fSAnders Carlsson const FunctionDecl *OperatorNew = E->getOperatorNew(); 3468ed55a54SJohn McCall if (IsPlacementOperatorNewArray(CGF.getContext(), OperatorNew)) 3473eb55cfeSKen Dyck return CharUnits::Zero(); 348399f499fSAnders Carlsson 3498ed55a54SJohn McCall return CGF.CGM.getCXXABI().GetArrayCookieSize(E->getAllocatedType()); 35059486a2dSAnders Carlsson } 35159486a2dSAnders Carlsson 35247b4629bSFariborz Jahanian static llvm::Value *EmitCXXNewAllocSize(ASTContext &Context, 35347b4629bSFariborz Jahanian CodeGenFunction &CGF, 35459486a2dSAnders Carlsson const CXXNewExpr *E, 35505fc5be3SDouglas Gregor llvm::Value *&NumElements, 35605fc5be3SDouglas Gregor llvm::Value *&SizeWithoutCookie) { 3577648fb46SArgyrios Kyrtzidis QualType ElemType = E->getAllocatedType(); 35859486a2dSAnders Carlsson 3598ed55a54SJohn McCall const llvm::IntegerType *SizeTy = 3608ed55a54SJohn McCall cast<llvm::IntegerType>(CGF.ConvertType(CGF.getContext().getSizeType())); 3618ed55a54SJohn McCall 3627648fb46SArgyrios Kyrtzidis CharUnits TypeSize = CGF.getContext().getTypeSizeInChars(ElemType); 3638ed55a54SJohn McCall 3648ed55a54SJohn McCall if (!E->isArray()) { 36505fc5be3SDouglas Gregor SizeWithoutCookie = llvm::ConstantInt::get(SizeTy, TypeSize.getQuantity()); 36605fc5be3SDouglas Gregor return SizeWithoutCookie; 36705fc5be3SDouglas Gregor } 36859486a2dSAnders Carlsson 3698ed55a54SJohn McCall // Figure out the cookie size. 3708ed55a54SJohn McCall CharUnits CookieSize = CalculateCookiePadding(CGF, E); 3718ed55a54SJohn McCall 37259486a2dSAnders Carlsson // Emit the array size expression. 3737648fb46SArgyrios Kyrtzidis // We multiply the size of all dimensions for NumElements. 3747648fb46SArgyrios Kyrtzidis // e.g for 'int[2][3]', ElemType is 'int' and NumElements is 6. 37559486a2dSAnders Carlsson NumElements = CGF.EmitScalarExpr(E->getArraySize()); 3768ed55a54SJohn McCall assert(NumElements->getType() == SizeTy && "element count not a size_t"); 3778ed55a54SJohn McCall 3788ed55a54SJohn McCall uint64_t ArraySizeMultiplier = 1; 3797648fb46SArgyrios Kyrtzidis while (const ConstantArrayType *CAT 3807648fb46SArgyrios Kyrtzidis = CGF.getContext().getAsConstantArrayType(ElemType)) { 3817648fb46SArgyrios Kyrtzidis ElemType = CAT->getElementType(); 3828ed55a54SJohn McCall ArraySizeMultiplier *= CAT->getSize().getZExtValue(); 3837648fb46SArgyrios Kyrtzidis } 38459486a2dSAnders Carlsson 3858ed55a54SJohn McCall llvm::Value *Size; 38632ac583dSChris Lattner 38732ac583dSChris Lattner // If someone is doing 'new int[42]' there is no need to do a dynamic check. 38832ac583dSChris Lattner // Don't bloat the -O0 code. 38932ac583dSChris Lattner if (llvm::ConstantInt *NumElementsC = 39032ac583dSChris Lattner dyn_cast<llvm::ConstantInt>(NumElements)) { 39132ac583dSChris Lattner llvm::APInt NEC = NumElementsC->getValue(); 3928ed55a54SJohn McCall unsigned SizeWidth = NEC.getBitWidth(); 39332ac583dSChris Lattner 3948ed55a54SJohn McCall // Determine if there is an overflow here by doing an extended multiply. 3958ed55a54SJohn McCall NEC.zext(SizeWidth*2); 3968ed55a54SJohn McCall llvm::APInt SC(SizeWidth*2, TypeSize.getQuantity()); 39732ac583dSChris Lattner SC *= NEC; 39832ac583dSChris Lattner 3998ed55a54SJohn McCall if (!CookieSize.isZero()) { 4008ed55a54SJohn McCall // Save the current size without a cookie. We don't care if an 4018ed55a54SJohn McCall // overflow's already happened because SizeWithoutCookie isn't 4028ed55a54SJohn McCall // used if the allocator returns null or throws, as it should 4038ed55a54SJohn McCall // always do on an overflow. 4048ed55a54SJohn McCall llvm::APInt SWC = SC; 4058ed55a54SJohn McCall SWC.trunc(SizeWidth); 4068ed55a54SJohn McCall SizeWithoutCookie = llvm::ConstantInt::get(SizeTy, SWC); 4078ed55a54SJohn McCall 4088ed55a54SJohn McCall // Add the cookie size. 4098ed55a54SJohn McCall SC += llvm::APInt(SizeWidth*2, CookieSize.getQuantity()); 4108ed55a54SJohn McCall } 4118ed55a54SJohn McCall 4128ed55a54SJohn McCall if (SC.countLeadingZeros() >= SizeWidth) { 4138ed55a54SJohn McCall SC.trunc(SizeWidth); 4148ed55a54SJohn McCall Size = llvm::ConstantInt::get(SizeTy, SC); 41532ac583dSChris Lattner } else { 41632ac583dSChris Lattner // On overflow, produce a -1 so operator new throws. 4178ed55a54SJohn McCall Size = llvm::Constant::getAllOnesValue(SizeTy); 41832ac583dSChris Lattner } 41932ac583dSChris Lattner 4208ed55a54SJohn McCall // Scale NumElements while we're at it. 4218ed55a54SJohn McCall uint64_t N = NEC.getZExtValue() * ArraySizeMultiplier; 4228ed55a54SJohn McCall NumElements = llvm::ConstantInt::get(SizeTy, N); 42347b4629bSFariborz Jahanian 4248ed55a54SJohn McCall // Otherwise, we don't need to do an overflow-checked multiplication if 4258ed55a54SJohn McCall // we're multiplying by one. 4268ed55a54SJohn McCall } else if (TypeSize.isOne()) { 4278ed55a54SJohn McCall assert(ArraySizeMultiplier == 1); 428f2f38701SChris Lattner 4298ed55a54SJohn McCall Size = NumElements; 430f2f38701SChris Lattner 4318ed55a54SJohn McCall // If we need a cookie, add its size in with an overflow check. 4328ed55a54SJohn McCall // This is maybe a little paranoid. 4338ed55a54SJohn McCall if (!CookieSize.isZero()) { 43405fc5be3SDouglas Gregor SizeWithoutCookie = Size; 435f2f38701SChris Lattner 4368ed55a54SJohn McCall llvm::Value *CookieSizeV 4378ed55a54SJohn McCall = llvm::ConstantInt::get(SizeTy, CookieSize.getQuantity()); 4388ed55a54SJohn McCall 4398ed55a54SJohn McCall const llvm::Type *Types[] = { SizeTy }; 4408ed55a54SJohn McCall llvm::Value *UAddF 4418ed55a54SJohn McCall = CGF.CGM.getIntrinsic(llvm::Intrinsic::uadd_with_overflow, Types, 1); 4428ed55a54SJohn McCall llvm::Value *AddRes 4438ed55a54SJohn McCall = CGF.Builder.CreateCall2(UAddF, Size, CookieSizeV); 4448ed55a54SJohn McCall 4458ed55a54SJohn McCall Size = CGF.Builder.CreateExtractValue(AddRes, 0); 4468ed55a54SJohn McCall llvm::Value *DidOverflow = CGF.Builder.CreateExtractValue(AddRes, 1); 4478ed55a54SJohn McCall Size = CGF.Builder.CreateSelect(DidOverflow, 4488ed55a54SJohn McCall llvm::ConstantInt::get(SizeTy, -1), 4498ed55a54SJohn McCall Size); 4508ed55a54SJohn McCall } 4518ed55a54SJohn McCall 4528ed55a54SJohn McCall // Otherwise use the int.umul.with.overflow intrinsic. 4538ed55a54SJohn McCall } else { 4548ed55a54SJohn McCall llvm::Value *OutermostElementSize 4558ed55a54SJohn McCall = llvm::ConstantInt::get(SizeTy, TypeSize.getQuantity()); 4568ed55a54SJohn McCall 4578ed55a54SJohn McCall llvm::Value *NumOutermostElements = NumElements; 4588ed55a54SJohn McCall 4598ed55a54SJohn McCall // Scale NumElements by the array size multiplier. This might 4608ed55a54SJohn McCall // overflow, but only if the multiplication below also overflows, 4618ed55a54SJohn McCall // in which case this multiplication isn't used. 4628ed55a54SJohn McCall if (ArraySizeMultiplier != 1) 4638ed55a54SJohn McCall NumElements = CGF.Builder.CreateMul(NumElements, 4648ed55a54SJohn McCall llvm::ConstantInt::get(SizeTy, ArraySizeMultiplier)); 4658ed55a54SJohn McCall 4668ed55a54SJohn McCall // The requested size of the outermost array is non-constant. 4678ed55a54SJohn McCall // Multiply that by the static size of the elements of that array; 4688ed55a54SJohn McCall // on unsigned overflow, set the size to -1 to trigger an 4698ed55a54SJohn McCall // exception from the allocation routine. This is sufficient to 4708ed55a54SJohn McCall // prevent buffer overruns from the allocator returning a 4718ed55a54SJohn McCall // seemingly valid pointer to insufficient space. This idea comes 4728ed55a54SJohn McCall // originally from MSVC, and GCC has an open bug requesting 4738ed55a54SJohn McCall // similar behavior: 4748ed55a54SJohn McCall // http://gcc.gnu.org/bugzilla/show_bug.cgi?id=19351 4758ed55a54SJohn McCall // 4768ed55a54SJohn McCall // This will not be sufficient for C++0x, which requires a 4778ed55a54SJohn McCall // specific exception class (std::bad_array_new_length). 4788ed55a54SJohn McCall // That will require ABI support that has not yet been specified. 4798ed55a54SJohn McCall const llvm::Type *Types[] = { SizeTy }; 4808ed55a54SJohn McCall llvm::Value *UMulF 4818ed55a54SJohn McCall = CGF.CGM.getIntrinsic(llvm::Intrinsic::umul_with_overflow, Types, 1); 4828ed55a54SJohn McCall llvm::Value *MulRes = CGF.Builder.CreateCall2(UMulF, NumOutermostElements, 4838ed55a54SJohn McCall OutermostElementSize); 4848ed55a54SJohn McCall 4858ed55a54SJohn McCall // The overflow bit. 4868ed55a54SJohn McCall llvm::Value *DidOverflow = CGF.Builder.CreateExtractValue(MulRes, 1); 4878ed55a54SJohn McCall 4888ed55a54SJohn McCall // The result of the multiplication. 4898ed55a54SJohn McCall Size = CGF.Builder.CreateExtractValue(MulRes, 0); 4908ed55a54SJohn McCall 4918ed55a54SJohn McCall // If we have a cookie, we need to add that size in, too. 4928ed55a54SJohn McCall if (!CookieSize.isZero()) { 4938ed55a54SJohn McCall SizeWithoutCookie = Size; 4948ed55a54SJohn McCall 4958ed55a54SJohn McCall llvm::Value *CookieSizeV 4968ed55a54SJohn McCall = llvm::ConstantInt::get(SizeTy, CookieSize.getQuantity()); 4978ed55a54SJohn McCall llvm::Value *UAddF 4988ed55a54SJohn McCall = CGF.CGM.getIntrinsic(llvm::Intrinsic::uadd_with_overflow, Types, 1); 4998ed55a54SJohn McCall llvm::Value *AddRes 5008ed55a54SJohn McCall = CGF.Builder.CreateCall2(UAddF, SizeWithoutCookie, CookieSizeV); 5018ed55a54SJohn McCall 5028ed55a54SJohn McCall Size = CGF.Builder.CreateExtractValue(AddRes, 0); 5038ed55a54SJohn McCall 5048ed55a54SJohn McCall llvm::Value *AddDidOverflow = CGF.Builder.CreateExtractValue(AddRes, 1); 5058ed55a54SJohn McCall DidOverflow = CGF.Builder.CreateAnd(DidOverflow, AddDidOverflow); 5068ed55a54SJohn McCall } 5078ed55a54SJohn McCall 5088ed55a54SJohn McCall Size = CGF.Builder.CreateSelect(DidOverflow, 5098ed55a54SJohn McCall llvm::ConstantInt::get(SizeTy, -1), 5108ed55a54SJohn McCall Size); 5118ed55a54SJohn McCall } 5128ed55a54SJohn McCall 5138ed55a54SJohn McCall if (CookieSize.isZero()) 5148ed55a54SJohn McCall SizeWithoutCookie = Size; 5158ed55a54SJohn McCall else 5168ed55a54SJohn McCall assert(SizeWithoutCookie && "didn't set SizeWithoutCookie?"); 51759486a2dSAnders Carlsson 51832ac583dSChris Lattner return Size; 51959486a2dSAnders Carlsson } 52059486a2dSAnders Carlsson 521d5202e09SFariborz Jahanian static void StoreAnyExprIntoOneUnit(CodeGenFunction &CGF, const CXXNewExpr *E, 522d5202e09SFariborz Jahanian llvm::Value *NewPtr) { 523d5202e09SFariborz Jahanian 524d5202e09SFariborz Jahanian assert(E->getNumConstructorArgs() == 1 && 525d5202e09SFariborz Jahanian "Can only have one argument to initializer of POD type."); 526d5202e09SFariborz Jahanian 527d5202e09SFariborz Jahanian const Expr *Init = E->getConstructorArg(0); 528d5202e09SFariborz Jahanian QualType AllocType = E->getAllocatedType(); 529d5202e09SFariborz Jahanian 5300381634aSDaniel Dunbar unsigned Alignment = 5310381634aSDaniel Dunbar CGF.getContext().getTypeAlignInChars(AllocType).getQuantity(); 532d5202e09SFariborz Jahanian if (!CGF.hasAggregateLLVMType(AllocType)) 533d5202e09SFariborz Jahanian CGF.EmitStoreOfScalar(CGF.EmitScalarExpr(Init), NewPtr, 5340381634aSDaniel Dunbar AllocType.isVolatileQualified(), Alignment, 5350381634aSDaniel Dunbar AllocType); 536d5202e09SFariborz Jahanian else if (AllocType->isAnyComplexType()) 537d5202e09SFariborz Jahanian CGF.EmitComplexExprIntoAddr(Init, NewPtr, 538d5202e09SFariborz Jahanian AllocType.isVolatileQualified()); 5397a626f63SJohn McCall else { 5407a626f63SJohn McCall AggValueSlot Slot 5417a626f63SJohn McCall = AggValueSlot::forAddr(NewPtr, AllocType.isVolatileQualified(), true); 5427a626f63SJohn McCall CGF.EmitAggExpr(Init, Slot); 5437a626f63SJohn McCall } 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 6807f9c92a9SJohn McCall /// A utility class for saving an rvalue. 6817f9c92a9SJohn McCall class SavedRValue { 6827f9c92a9SJohn McCall public: 6837f9c92a9SJohn McCall enum Kind { ScalarLiteral, ScalarAddress, 6847f9c92a9SJohn McCall AggregateLiteral, AggregateAddress, 6857f9c92a9SJohn McCall Complex }; 6867f9c92a9SJohn McCall 6877f9c92a9SJohn McCall private: 6887f9c92a9SJohn McCall llvm::Value *Value; 6897f9c92a9SJohn McCall Kind K; 6907f9c92a9SJohn McCall 6917f9c92a9SJohn McCall SavedRValue(llvm::Value *V, Kind K) : Value(V), K(K) {} 6927f9c92a9SJohn McCall 6937f9c92a9SJohn McCall public: 6947f9c92a9SJohn McCall SavedRValue() {} 6957f9c92a9SJohn McCall 6967f9c92a9SJohn McCall static SavedRValue forScalarLiteral(llvm::Value *V) { 6977f9c92a9SJohn McCall return SavedRValue(V, ScalarLiteral); 6987f9c92a9SJohn McCall } 6997f9c92a9SJohn McCall 7007f9c92a9SJohn McCall static SavedRValue forScalarAddress(llvm::Value *Addr) { 7017f9c92a9SJohn McCall return SavedRValue(Addr, ScalarAddress); 7027f9c92a9SJohn McCall } 7037f9c92a9SJohn McCall 7047f9c92a9SJohn McCall static SavedRValue forAggregateLiteral(llvm::Value *V) { 7057f9c92a9SJohn McCall return SavedRValue(V, AggregateLiteral); 7067f9c92a9SJohn McCall } 7077f9c92a9SJohn McCall 7087f9c92a9SJohn McCall static SavedRValue forAggregateAddress(llvm::Value *Addr) { 7097f9c92a9SJohn McCall return SavedRValue(Addr, AggregateAddress); 7107f9c92a9SJohn McCall } 7117f9c92a9SJohn McCall 7127f9c92a9SJohn McCall static SavedRValue forComplexAddress(llvm::Value *Addr) { 7137f9c92a9SJohn McCall return SavedRValue(Addr, Complex); 7147f9c92a9SJohn McCall } 7157f9c92a9SJohn McCall 7167f9c92a9SJohn McCall Kind getKind() const { return K; } 7177f9c92a9SJohn McCall llvm::Value *getValue() const { return Value; } 7187f9c92a9SJohn McCall }; 7197f9c92a9SJohn McCall 7207f9c92a9SJohn McCall /// Given an r-value, perform the code necessary to make sure that a 7217f9c92a9SJohn McCall /// future RestoreRValue will be able to load the value without 7227f9c92a9SJohn McCall /// domination concerns. 7237f9c92a9SJohn McCall static SavedRValue SaveRValue(CodeGenFunction &CGF, RValue RV) { 7247f9c92a9SJohn McCall if (RV.isScalar()) { 7257f9c92a9SJohn McCall llvm::Value *V = RV.getScalarVal(); 7267f9c92a9SJohn McCall 7277f9c92a9SJohn McCall // These automatically dominate and don't need to be saved. 7287f9c92a9SJohn McCall if (isa<llvm::Constant>(V) || isa<llvm::AllocaInst>(V)) 7297f9c92a9SJohn McCall return SavedRValue::forScalarLiteral(V); 7307f9c92a9SJohn McCall 7317f9c92a9SJohn McCall // Everything else needs an alloca. 7327f9c92a9SJohn McCall llvm::Value *Addr = CGF.CreateTempAlloca(V->getType(), "saved-rvalue"); 7337f9c92a9SJohn McCall CGF.Builder.CreateStore(V, Addr); 7347f9c92a9SJohn McCall return SavedRValue::forScalarAddress(Addr); 7357f9c92a9SJohn McCall } 7367f9c92a9SJohn McCall 7377f9c92a9SJohn McCall if (RV.isComplex()) { 7387f9c92a9SJohn McCall CodeGenFunction::ComplexPairTy V = RV.getComplexVal(); 7397f9c92a9SJohn McCall const llvm::Type *ComplexTy = 7407f9c92a9SJohn McCall llvm::StructType::get(CGF.getLLVMContext(), 7417f9c92a9SJohn McCall V.first->getType(), V.second->getType(), 7427f9c92a9SJohn McCall (void*) 0); 7437f9c92a9SJohn McCall llvm::Value *Addr = CGF.CreateTempAlloca(ComplexTy, "saved-complex"); 7447f9c92a9SJohn McCall CGF.StoreComplexToAddr(V, Addr, /*volatile*/ false); 7457f9c92a9SJohn McCall return SavedRValue::forComplexAddress(Addr); 7467f9c92a9SJohn McCall } 7477f9c92a9SJohn McCall 7487f9c92a9SJohn McCall assert(RV.isAggregate()); 7497f9c92a9SJohn McCall llvm::Value *V = RV.getAggregateAddr(); // TODO: volatile? 7507f9c92a9SJohn McCall if (isa<llvm::Constant>(V) || isa<llvm::AllocaInst>(V)) 7517f9c92a9SJohn McCall return SavedRValue::forAggregateLiteral(V); 7527f9c92a9SJohn McCall 7537f9c92a9SJohn McCall llvm::Value *Addr = CGF.CreateTempAlloca(V->getType(), "saved-rvalue"); 7547f9c92a9SJohn McCall CGF.Builder.CreateStore(V, Addr); 7557f9c92a9SJohn McCall return SavedRValue::forAggregateAddress(Addr); 7567f9c92a9SJohn McCall } 7577f9c92a9SJohn McCall 7587f9c92a9SJohn McCall /// Given a saved r-value produced by SaveRValue, perform the code 7597f9c92a9SJohn McCall /// necessary to restore it to usability at the current insertion 7607f9c92a9SJohn McCall /// point. 7617f9c92a9SJohn McCall static RValue RestoreRValue(CodeGenFunction &CGF, SavedRValue RV) { 7627f9c92a9SJohn McCall switch (RV.getKind()) { 7637f9c92a9SJohn McCall case SavedRValue::ScalarLiteral: 7647f9c92a9SJohn McCall return RValue::get(RV.getValue()); 7657f9c92a9SJohn McCall case SavedRValue::ScalarAddress: 7667f9c92a9SJohn McCall return RValue::get(CGF.Builder.CreateLoad(RV.getValue())); 7677f9c92a9SJohn McCall case SavedRValue::AggregateLiteral: 7687f9c92a9SJohn McCall return RValue::getAggregate(RV.getValue()); 7697f9c92a9SJohn McCall case SavedRValue::AggregateAddress: 7707f9c92a9SJohn McCall return RValue::getAggregate(CGF.Builder.CreateLoad(RV.getValue())); 7717f9c92a9SJohn McCall case SavedRValue::Complex: 7727f9c92a9SJohn McCall return RValue::getComplex(CGF.LoadComplexFromAddr(RV.getValue(), false)); 7737f9c92a9SJohn McCall } 7747f9c92a9SJohn McCall 7757f9c92a9SJohn McCall llvm_unreachable("bad saved r-value kind"); 7767f9c92a9SJohn McCall return RValue(); 7777f9c92a9SJohn McCall } 7787f9c92a9SJohn McCall 779824c2f53SJohn McCall namespace { 780824c2f53SJohn McCall /// A cleanup to call the given 'operator delete' function upon 781824c2f53SJohn McCall /// abnormal exit from a new expression. 782824c2f53SJohn McCall class CallDeleteDuringNew : public EHScopeStack::Cleanup { 783824c2f53SJohn McCall size_t NumPlacementArgs; 784824c2f53SJohn McCall const FunctionDecl *OperatorDelete; 785824c2f53SJohn McCall llvm::Value *Ptr; 786824c2f53SJohn McCall llvm::Value *AllocSize; 787824c2f53SJohn McCall 788824c2f53SJohn McCall RValue *getPlacementArgs() { return reinterpret_cast<RValue*>(this+1); } 789824c2f53SJohn McCall 790824c2f53SJohn McCall public: 791824c2f53SJohn McCall static size_t getExtraSize(size_t NumPlacementArgs) { 792824c2f53SJohn McCall return NumPlacementArgs * sizeof(RValue); 793824c2f53SJohn McCall } 794824c2f53SJohn McCall 795824c2f53SJohn McCall CallDeleteDuringNew(size_t NumPlacementArgs, 796824c2f53SJohn McCall const FunctionDecl *OperatorDelete, 797824c2f53SJohn McCall llvm::Value *Ptr, 798824c2f53SJohn McCall llvm::Value *AllocSize) 799824c2f53SJohn McCall : NumPlacementArgs(NumPlacementArgs), OperatorDelete(OperatorDelete), 800824c2f53SJohn McCall Ptr(Ptr), AllocSize(AllocSize) {} 801824c2f53SJohn McCall 802824c2f53SJohn McCall void setPlacementArg(unsigned I, RValue Arg) { 803824c2f53SJohn McCall assert(I < NumPlacementArgs && "index out of range"); 804824c2f53SJohn McCall getPlacementArgs()[I] = Arg; 805824c2f53SJohn McCall } 806824c2f53SJohn McCall 807824c2f53SJohn McCall void Emit(CodeGenFunction &CGF, bool IsForEH) { 808824c2f53SJohn McCall const FunctionProtoType *FPT 809824c2f53SJohn McCall = OperatorDelete->getType()->getAs<FunctionProtoType>(); 810824c2f53SJohn McCall assert(FPT->getNumArgs() == NumPlacementArgs + 1 || 811d441b1e6SJohn McCall (FPT->getNumArgs() == 2 && NumPlacementArgs == 0)); 812824c2f53SJohn McCall 813824c2f53SJohn McCall CallArgList DeleteArgs; 814824c2f53SJohn McCall 815824c2f53SJohn McCall // The first argument is always a void*. 816824c2f53SJohn McCall FunctionProtoType::arg_type_iterator AI = FPT->arg_type_begin(); 817824c2f53SJohn McCall DeleteArgs.push_back(std::make_pair(RValue::get(Ptr), *AI++)); 818824c2f53SJohn McCall 819824c2f53SJohn McCall // A member 'operator delete' can take an extra 'size_t' argument. 820824c2f53SJohn McCall if (FPT->getNumArgs() == NumPlacementArgs + 2) 821824c2f53SJohn McCall DeleteArgs.push_back(std::make_pair(RValue::get(AllocSize), *AI++)); 822824c2f53SJohn McCall 823824c2f53SJohn McCall // Pass the rest of the arguments, which must match exactly. 824824c2f53SJohn McCall for (unsigned I = 0; I != NumPlacementArgs; ++I) 825824c2f53SJohn McCall DeleteArgs.push_back(std::make_pair(getPlacementArgs()[I], *AI++)); 826824c2f53SJohn McCall 827824c2f53SJohn McCall // Call 'operator delete'. 828824c2f53SJohn McCall CGF.EmitCall(CGF.CGM.getTypes().getFunctionInfo(DeleteArgs, FPT), 829824c2f53SJohn McCall CGF.CGM.GetAddrOfFunction(OperatorDelete), 830824c2f53SJohn McCall ReturnValueSlot(), DeleteArgs, OperatorDelete); 831824c2f53SJohn McCall } 832824c2f53SJohn McCall }; 8337f9c92a9SJohn McCall 8347f9c92a9SJohn McCall /// A cleanup to call the given 'operator delete' function upon 8357f9c92a9SJohn McCall /// abnormal exit from a new expression when the new expression is 8367f9c92a9SJohn McCall /// conditional. 8377f9c92a9SJohn McCall class CallDeleteDuringConditionalNew : public EHScopeStack::Cleanup { 8387f9c92a9SJohn McCall size_t NumPlacementArgs; 8397f9c92a9SJohn McCall const FunctionDecl *OperatorDelete; 8407f9c92a9SJohn McCall SavedRValue Ptr; 8417f9c92a9SJohn McCall SavedRValue AllocSize; 8427f9c92a9SJohn McCall 8437f9c92a9SJohn McCall SavedRValue *getPlacementArgs() { 8447f9c92a9SJohn McCall return reinterpret_cast<SavedRValue*>(this+1); 8457f9c92a9SJohn McCall } 8467f9c92a9SJohn McCall 8477f9c92a9SJohn McCall public: 8487f9c92a9SJohn McCall static size_t getExtraSize(size_t NumPlacementArgs) { 8497f9c92a9SJohn McCall return NumPlacementArgs * sizeof(SavedRValue); 8507f9c92a9SJohn McCall } 8517f9c92a9SJohn McCall 8527f9c92a9SJohn McCall CallDeleteDuringConditionalNew(size_t NumPlacementArgs, 8537f9c92a9SJohn McCall const FunctionDecl *OperatorDelete, 8547f9c92a9SJohn McCall SavedRValue Ptr, 8557f9c92a9SJohn McCall SavedRValue AllocSize) 8567f9c92a9SJohn McCall : NumPlacementArgs(NumPlacementArgs), OperatorDelete(OperatorDelete), 8577f9c92a9SJohn McCall Ptr(Ptr), AllocSize(AllocSize) {} 8587f9c92a9SJohn McCall 8597f9c92a9SJohn McCall void setPlacementArg(unsigned I, SavedRValue Arg) { 8607f9c92a9SJohn McCall assert(I < NumPlacementArgs && "index out of range"); 8617f9c92a9SJohn McCall getPlacementArgs()[I] = Arg; 8627f9c92a9SJohn McCall } 8637f9c92a9SJohn McCall 8647f9c92a9SJohn McCall void Emit(CodeGenFunction &CGF, bool IsForEH) { 8657f9c92a9SJohn McCall const FunctionProtoType *FPT 8667f9c92a9SJohn McCall = OperatorDelete->getType()->getAs<FunctionProtoType>(); 8677f9c92a9SJohn McCall assert(FPT->getNumArgs() == NumPlacementArgs + 1 || 8687f9c92a9SJohn McCall (FPT->getNumArgs() == 2 && NumPlacementArgs == 0)); 8697f9c92a9SJohn McCall 8707f9c92a9SJohn McCall CallArgList DeleteArgs; 8717f9c92a9SJohn McCall 8727f9c92a9SJohn McCall // The first argument is always a void*. 8737f9c92a9SJohn McCall FunctionProtoType::arg_type_iterator AI = FPT->arg_type_begin(); 8747f9c92a9SJohn McCall DeleteArgs.push_back(std::make_pair(RestoreRValue(CGF, Ptr), *AI++)); 8757f9c92a9SJohn McCall 8767f9c92a9SJohn McCall // A member 'operator delete' can take an extra 'size_t' argument. 8777f9c92a9SJohn McCall if (FPT->getNumArgs() == NumPlacementArgs + 2) { 8787f9c92a9SJohn McCall RValue RV = RestoreRValue(CGF, AllocSize); 8797f9c92a9SJohn McCall DeleteArgs.push_back(std::make_pair(RV, *AI++)); 8807f9c92a9SJohn McCall } 8817f9c92a9SJohn McCall 8827f9c92a9SJohn McCall // Pass the rest of the arguments, which must match exactly. 8837f9c92a9SJohn McCall for (unsigned I = 0; I != NumPlacementArgs; ++I) { 8847f9c92a9SJohn McCall RValue RV = RestoreRValue(CGF, getPlacementArgs()[I]); 8857f9c92a9SJohn McCall DeleteArgs.push_back(std::make_pair(RV, *AI++)); 8867f9c92a9SJohn McCall } 8877f9c92a9SJohn McCall 8887f9c92a9SJohn McCall // Call 'operator delete'. 8897f9c92a9SJohn McCall CGF.EmitCall(CGF.CGM.getTypes().getFunctionInfo(DeleteArgs, FPT), 8907f9c92a9SJohn McCall CGF.CGM.GetAddrOfFunction(OperatorDelete), 8917f9c92a9SJohn McCall ReturnValueSlot(), DeleteArgs, OperatorDelete); 8927f9c92a9SJohn McCall } 8937f9c92a9SJohn McCall }; 8947f9c92a9SJohn McCall } 8957f9c92a9SJohn McCall 8967f9c92a9SJohn McCall /// Enter a cleanup to call 'operator delete' if the initializer in a 8977f9c92a9SJohn McCall /// new-expression throws. 8987f9c92a9SJohn McCall static void EnterNewDeleteCleanup(CodeGenFunction &CGF, 8997f9c92a9SJohn McCall const CXXNewExpr *E, 9007f9c92a9SJohn McCall llvm::Value *NewPtr, 9017f9c92a9SJohn McCall llvm::Value *AllocSize, 9027f9c92a9SJohn McCall const CallArgList &NewArgs) { 9037f9c92a9SJohn McCall // If we're not inside a conditional branch, then the cleanup will 9047f9c92a9SJohn McCall // dominate and we can do the easier (and more efficient) thing. 9057f9c92a9SJohn McCall if (!CGF.isInConditionalBranch()) { 9067f9c92a9SJohn McCall CallDeleteDuringNew *Cleanup = CGF.EHStack 9077f9c92a9SJohn McCall .pushCleanupWithExtra<CallDeleteDuringNew>(EHCleanup, 9087f9c92a9SJohn McCall E->getNumPlacementArgs(), 9097f9c92a9SJohn McCall E->getOperatorDelete(), 9107f9c92a9SJohn McCall NewPtr, AllocSize); 9117f9c92a9SJohn McCall for (unsigned I = 0, N = E->getNumPlacementArgs(); I != N; ++I) 9127f9c92a9SJohn McCall Cleanup->setPlacementArg(I, NewArgs[I+1].first); 9137f9c92a9SJohn McCall 9147f9c92a9SJohn McCall return; 9157f9c92a9SJohn McCall } 9167f9c92a9SJohn McCall 9177f9c92a9SJohn McCall // Otherwise, we need to save all this stuff. 9187f9c92a9SJohn McCall SavedRValue SavedNewPtr = SaveRValue(CGF, RValue::get(NewPtr)); 9197f9c92a9SJohn McCall SavedRValue SavedAllocSize = SaveRValue(CGF, RValue::get(AllocSize)); 9207f9c92a9SJohn McCall 9217f9c92a9SJohn McCall CallDeleteDuringConditionalNew *Cleanup = CGF.EHStack 9227f9c92a9SJohn McCall .pushCleanupWithExtra<CallDeleteDuringConditionalNew>(InactiveEHCleanup, 9237f9c92a9SJohn McCall E->getNumPlacementArgs(), 9247f9c92a9SJohn McCall E->getOperatorDelete(), 9257f9c92a9SJohn McCall SavedNewPtr, 9267f9c92a9SJohn McCall SavedAllocSize); 9277f9c92a9SJohn McCall for (unsigned I = 0, N = E->getNumPlacementArgs(); I != N; ++I) 9287f9c92a9SJohn McCall Cleanup->setPlacementArg(I, SaveRValue(CGF, NewArgs[I+1].first)); 9297f9c92a9SJohn McCall 9307f9c92a9SJohn McCall CGF.ActivateCleanupBlock(CGF.EHStack.stable_begin()); 931824c2f53SJohn McCall } 932824c2f53SJohn McCall 93359486a2dSAnders Carlsson llvm::Value *CodeGenFunction::EmitCXXNewExpr(const CXXNewExpr *E) { 93459486a2dSAnders Carlsson QualType AllocType = E->getAllocatedType(); 9358ed55a54SJohn McCall if (AllocType->isArrayType()) 9368ed55a54SJohn McCall while (const ArrayType *AType = getContext().getAsArrayType(AllocType)) 9378ed55a54SJohn McCall AllocType = AType->getElementType(); 9388ed55a54SJohn McCall 93959486a2dSAnders Carlsson FunctionDecl *NewFD = E->getOperatorNew(); 94059486a2dSAnders Carlsson const FunctionProtoType *NewFTy = NewFD->getType()->getAs<FunctionProtoType>(); 94159486a2dSAnders Carlsson 94259486a2dSAnders Carlsson CallArgList NewArgs; 94359486a2dSAnders Carlsson 94459486a2dSAnders Carlsson // The allocation size is the first argument. 94559486a2dSAnders Carlsson QualType SizeTy = getContext().getSizeType(); 94659486a2dSAnders Carlsson 94759486a2dSAnders Carlsson llvm::Value *NumElements = 0; 94805fc5be3SDouglas Gregor llvm::Value *AllocSizeWithoutCookie = 0; 94947b4629bSFariborz Jahanian llvm::Value *AllocSize = EmitCXXNewAllocSize(getContext(), 95005fc5be3SDouglas Gregor *this, E, NumElements, 95105fc5be3SDouglas Gregor AllocSizeWithoutCookie); 95259486a2dSAnders Carlsson 95359486a2dSAnders Carlsson NewArgs.push_back(std::make_pair(RValue::get(AllocSize), SizeTy)); 95459486a2dSAnders Carlsson 95559486a2dSAnders Carlsson // Emit the rest of the arguments. 95659486a2dSAnders Carlsson // FIXME: Ideally, this should just use EmitCallArgs. 95759486a2dSAnders Carlsson CXXNewExpr::const_arg_iterator NewArg = E->placement_arg_begin(); 95859486a2dSAnders Carlsson 95959486a2dSAnders Carlsson // First, use the types from the function type. 96059486a2dSAnders Carlsson // We start at 1 here because the first argument (the allocation size) 96159486a2dSAnders Carlsson // has already been emitted. 96259486a2dSAnders Carlsson for (unsigned i = 1, e = NewFTy->getNumArgs(); i != e; ++i, ++NewArg) { 96359486a2dSAnders Carlsson QualType ArgType = NewFTy->getArgType(i); 96459486a2dSAnders Carlsson 96559486a2dSAnders Carlsson assert(getContext().getCanonicalType(ArgType.getNonReferenceType()). 96659486a2dSAnders Carlsson getTypePtr() == 96759486a2dSAnders Carlsson getContext().getCanonicalType(NewArg->getType()).getTypePtr() && 96859486a2dSAnders Carlsson "type mismatch in call argument!"); 96959486a2dSAnders Carlsson 97059486a2dSAnders Carlsson NewArgs.push_back(std::make_pair(EmitCallArg(*NewArg, ArgType), 97159486a2dSAnders Carlsson ArgType)); 97259486a2dSAnders Carlsson 97359486a2dSAnders Carlsson } 97459486a2dSAnders Carlsson 97559486a2dSAnders Carlsson // Either we've emitted all the call args, or we have a call to a 97659486a2dSAnders Carlsson // variadic function. 97759486a2dSAnders Carlsson assert((NewArg == E->placement_arg_end() || NewFTy->isVariadic()) && 97859486a2dSAnders Carlsson "Extra arguments in non-variadic function!"); 97959486a2dSAnders Carlsson 98059486a2dSAnders Carlsson // If we still have any arguments, emit them using the type of the argument. 98159486a2dSAnders Carlsson for (CXXNewExpr::const_arg_iterator NewArgEnd = E->placement_arg_end(); 98259486a2dSAnders Carlsson NewArg != NewArgEnd; ++NewArg) { 98359486a2dSAnders Carlsson QualType ArgType = NewArg->getType(); 98459486a2dSAnders Carlsson NewArgs.push_back(std::make_pair(EmitCallArg(*NewArg, ArgType), 98559486a2dSAnders Carlsson ArgType)); 98659486a2dSAnders Carlsson } 98759486a2dSAnders Carlsson 98859486a2dSAnders Carlsson // Emit the call to new. 98959486a2dSAnders Carlsson RValue RV = 990ab26cfa5SJohn McCall EmitCall(CGM.getTypes().getFunctionInfo(NewArgs, NewFTy), 99161a401caSAnders Carlsson CGM.GetAddrOfFunction(NewFD), ReturnValueSlot(), NewArgs, NewFD); 99259486a2dSAnders Carlsson 99359486a2dSAnders Carlsson // If an allocation function is declared with an empty exception specification 99459486a2dSAnders Carlsson // it returns null to indicate failure to allocate storage. [expr.new]p13. 99559486a2dSAnders Carlsson // (We don't need to check for null when there's no new initializer and 99659486a2dSAnders Carlsson // we're allocating a POD type). 99759486a2dSAnders Carlsson bool NullCheckResult = NewFTy->hasEmptyExceptionSpec() && 99859486a2dSAnders Carlsson !(AllocType->isPODType() && !E->hasInitializer()); 99959486a2dSAnders Carlsson 10008ed55a54SJohn McCall llvm::BasicBlock *NullCheckSource = 0; 100159486a2dSAnders Carlsson llvm::BasicBlock *NewNotNull = 0; 100259486a2dSAnders Carlsson llvm::BasicBlock *NewEnd = 0; 100359486a2dSAnders Carlsson 100459486a2dSAnders Carlsson llvm::Value *NewPtr = RV.getScalarVal(); 10058ed55a54SJohn McCall unsigned AS = cast<llvm::PointerType>(NewPtr->getType())->getAddressSpace(); 100659486a2dSAnders Carlsson 100759486a2dSAnders Carlsson if (NullCheckResult) { 10088ed55a54SJohn McCall NullCheckSource = Builder.GetInsertBlock(); 100959486a2dSAnders Carlsson NewNotNull = createBasicBlock("new.notnull"); 101059486a2dSAnders Carlsson NewEnd = createBasicBlock("new.end"); 101159486a2dSAnders Carlsson 10128ed55a54SJohn McCall llvm::Value *IsNull = Builder.CreateIsNull(NewPtr, "new.isnull"); 10138ed55a54SJohn McCall Builder.CreateCondBr(IsNull, NewEnd, NewNotNull); 101459486a2dSAnders Carlsson EmitBlock(NewNotNull); 101559486a2dSAnders Carlsson } 101659486a2dSAnders Carlsson 10178ed55a54SJohn McCall assert((AllocSize == AllocSizeWithoutCookie) == 10188ed55a54SJohn McCall CalculateCookiePadding(*this, E).isZero()); 10198ed55a54SJohn McCall if (AllocSize != AllocSizeWithoutCookie) { 10208ed55a54SJohn McCall assert(E->isArray()); 10218ed55a54SJohn McCall NewPtr = CGM.getCXXABI().InitializeArrayCookie(CGF, NewPtr, NumElements, 10228ed55a54SJohn McCall AllocType); 102359486a2dSAnders Carlsson } 102459486a2dSAnders Carlsson 1025824c2f53SJohn McCall // If there's an operator delete, enter a cleanup to call it if an 1026824c2f53SJohn McCall // exception is thrown. 1027824c2f53SJohn McCall EHScopeStack::stable_iterator CallOperatorDelete; 1028824c2f53SJohn McCall if (E->getOperatorDelete()) { 10297f9c92a9SJohn McCall EnterNewDeleteCleanup(*this, E, NewPtr, AllocSize, NewArgs); 1030824c2f53SJohn McCall CallOperatorDelete = EHStack.stable_begin(); 1031824c2f53SJohn McCall } 1032824c2f53SJohn McCall 1033040ad500SDouglas Gregor const llvm::Type *ElementPtrTy 1034040ad500SDouglas Gregor = ConvertTypeForMem(AllocType)->getPointerTo(AS); 10358ed55a54SJohn McCall NewPtr = Builder.CreateBitCast(NewPtr, ElementPtrTy); 1036824c2f53SJohn McCall 10378ed55a54SJohn McCall if (E->isArray()) { 103805fc5be3SDouglas Gregor EmitNewInitializer(*this, E, NewPtr, NumElements, AllocSizeWithoutCookie); 10398ed55a54SJohn McCall 10408ed55a54SJohn McCall // NewPtr is a pointer to the base element type. If we're 10418ed55a54SJohn McCall // allocating an array of arrays, we'll need to cast back to the 10428ed55a54SJohn McCall // array pointer type. 1043040ad500SDouglas Gregor const llvm::Type *ResultTy = ConvertTypeForMem(E->getType()); 10448ed55a54SJohn McCall if (NewPtr->getType() != ResultTy) 10458ed55a54SJohn McCall NewPtr = Builder.CreateBitCast(NewPtr, ResultTy); 10468ed55a54SJohn McCall } else { 104705fc5be3SDouglas Gregor EmitNewInitializer(*this, E, NewPtr, NumElements, AllocSizeWithoutCookie); 104847b4629bSFariborz Jahanian } 104959486a2dSAnders Carlsson 1050824c2f53SJohn McCall // Deactivate the 'operator delete' cleanup if we finished 1051824c2f53SJohn McCall // initialization. 1052824c2f53SJohn McCall if (CallOperatorDelete.isValid()) 1053824c2f53SJohn McCall DeactivateCleanupBlock(CallOperatorDelete); 1054824c2f53SJohn McCall 105559486a2dSAnders Carlsson if (NullCheckResult) { 105659486a2dSAnders Carlsson Builder.CreateBr(NewEnd); 10578ed55a54SJohn McCall llvm::BasicBlock *NotNullSource = Builder.GetInsertBlock(); 105859486a2dSAnders Carlsson EmitBlock(NewEnd); 105959486a2dSAnders Carlsson 106059486a2dSAnders Carlsson llvm::PHINode *PHI = Builder.CreatePHI(NewPtr->getType()); 106159486a2dSAnders Carlsson PHI->reserveOperandSpace(2); 10628ed55a54SJohn McCall PHI->addIncoming(NewPtr, NotNullSource); 10638ed55a54SJohn McCall PHI->addIncoming(llvm::Constant::getNullValue(NewPtr->getType()), 10648ed55a54SJohn McCall NullCheckSource); 106559486a2dSAnders Carlsson 106659486a2dSAnders Carlsson NewPtr = PHI; 106759486a2dSAnders Carlsson } 106859486a2dSAnders Carlsson 106959486a2dSAnders Carlsson return NewPtr; 107059486a2dSAnders Carlsson } 107159486a2dSAnders Carlsson 107259486a2dSAnders Carlsson void CodeGenFunction::EmitDeleteCall(const FunctionDecl *DeleteFD, 107359486a2dSAnders Carlsson llvm::Value *Ptr, 107459486a2dSAnders Carlsson QualType DeleteTy) { 10758ed55a54SJohn McCall assert(DeleteFD->getOverloadedOperator() == OO_Delete); 10768ed55a54SJohn McCall 107759486a2dSAnders Carlsson const FunctionProtoType *DeleteFTy = 107859486a2dSAnders Carlsson DeleteFD->getType()->getAs<FunctionProtoType>(); 107959486a2dSAnders Carlsson 108059486a2dSAnders Carlsson CallArgList DeleteArgs; 108159486a2dSAnders Carlsson 108221122cf6SAnders Carlsson // Check if we need to pass the size to the delete operator. 108321122cf6SAnders Carlsson llvm::Value *Size = 0; 108421122cf6SAnders Carlsson QualType SizeTy; 108521122cf6SAnders Carlsson if (DeleteFTy->getNumArgs() == 2) { 108621122cf6SAnders Carlsson SizeTy = DeleteFTy->getArgType(1); 10877df3cbebSKen Dyck CharUnits DeleteTypeSize = getContext().getTypeSizeInChars(DeleteTy); 10887df3cbebSKen Dyck Size = llvm::ConstantInt::get(ConvertType(SizeTy), 10897df3cbebSKen Dyck DeleteTypeSize.getQuantity()); 109021122cf6SAnders Carlsson } 109121122cf6SAnders Carlsson 109259486a2dSAnders Carlsson QualType ArgTy = DeleteFTy->getArgType(0); 109359486a2dSAnders Carlsson llvm::Value *DeletePtr = Builder.CreateBitCast(Ptr, ConvertType(ArgTy)); 109459486a2dSAnders Carlsson DeleteArgs.push_back(std::make_pair(RValue::get(DeletePtr), ArgTy)); 109559486a2dSAnders Carlsson 109621122cf6SAnders Carlsson if (Size) 109759486a2dSAnders Carlsson DeleteArgs.push_back(std::make_pair(RValue::get(Size), SizeTy)); 109859486a2dSAnders Carlsson 109959486a2dSAnders Carlsson // Emit the call to delete. 1100ab26cfa5SJohn McCall EmitCall(CGM.getTypes().getFunctionInfo(DeleteArgs, DeleteFTy), 110161a401caSAnders Carlsson CGM.GetAddrOfFunction(DeleteFD), ReturnValueSlot(), 110259486a2dSAnders Carlsson DeleteArgs, DeleteFD); 110359486a2dSAnders Carlsson } 110459486a2dSAnders Carlsson 11058ed55a54SJohn McCall namespace { 11068ed55a54SJohn McCall /// Calls the given 'operator delete' on a single object. 11078ed55a54SJohn McCall struct CallObjectDelete : EHScopeStack::Cleanup { 11088ed55a54SJohn McCall llvm::Value *Ptr; 11098ed55a54SJohn McCall const FunctionDecl *OperatorDelete; 11108ed55a54SJohn McCall QualType ElementType; 11118ed55a54SJohn McCall 11128ed55a54SJohn McCall CallObjectDelete(llvm::Value *Ptr, 11138ed55a54SJohn McCall const FunctionDecl *OperatorDelete, 11148ed55a54SJohn McCall QualType ElementType) 11158ed55a54SJohn McCall : Ptr(Ptr), OperatorDelete(OperatorDelete), ElementType(ElementType) {} 11168ed55a54SJohn McCall 11178ed55a54SJohn McCall void Emit(CodeGenFunction &CGF, bool IsForEH) { 11188ed55a54SJohn McCall CGF.EmitDeleteCall(OperatorDelete, Ptr, ElementType); 11198ed55a54SJohn McCall } 11208ed55a54SJohn McCall }; 11218ed55a54SJohn McCall } 11228ed55a54SJohn McCall 11238ed55a54SJohn McCall /// Emit the code for deleting a single object. 11248ed55a54SJohn McCall static void EmitObjectDelete(CodeGenFunction &CGF, 11258ed55a54SJohn McCall const FunctionDecl *OperatorDelete, 11268ed55a54SJohn McCall llvm::Value *Ptr, 11278ed55a54SJohn McCall QualType ElementType) { 11288ed55a54SJohn McCall // Find the destructor for the type, if applicable. If the 11298ed55a54SJohn McCall // destructor is virtual, we'll just emit the vcall and return. 11308ed55a54SJohn McCall const CXXDestructorDecl *Dtor = 0; 11318ed55a54SJohn McCall if (const RecordType *RT = ElementType->getAs<RecordType>()) { 11328ed55a54SJohn McCall CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl()); 11338ed55a54SJohn McCall if (!RD->hasTrivialDestructor()) { 11348ed55a54SJohn McCall Dtor = RD->getDestructor(); 11358ed55a54SJohn McCall 11368ed55a54SJohn McCall if (Dtor->isVirtual()) { 11378ed55a54SJohn McCall const llvm::Type *Ty = 11380d635f53SJohn McCall CGF.getTypes().GetFunctionType(CGF.getTypes().getFunctionInfo(Dtor, 11390d635f53SJohn McCall Dtor_Complete), 11408ed55a54SJohn McCall /*isVariadic=*/false); 11418ed55a54SJohn McCall 11428ed55a54SJohn McCall llvm::Value *Callee 11438ed55a54SJohn McCall = CGF.BuildVirtualCall(Dtor, Dtor_Deleting, Ptr, Ty); 11448ed55a54SJohn McCall CGF.EmitCXXMemberCall(Dtor, Callee, ReturnValueSlot(), Ptr, /*VTT=*/0, 11458ed55a54SJohn McCall 0, 0); 11468ed55a54SJohn McCall 11478ed55a54SJohn McCall // The dtor took care of deleting the object. 11488ed55a54SJohn McCall return; 11498ed55a54SJohn McCall } 11508ed55a54SJohn McCall } 11518ed55a54SJohn McCall } 11528ed55a54SJohn McCall 11538ed55a54SJohn McCall // Make sure that we call delete even if the dtor throws. 11548ed55a54SJohn McCall CGF.EHStack.pushCleanup<CallObjectDelete>(NormalAndEHCleanup, 11558ed55a54SJohn McCall Ptr, OperatorDelete, ElementType); 11568ed55a54SJohn McCall 11578ed55a54SJohn McCall if (Dtor) 11588ed55a54SJohn McCall CGF.EmitCXXDestructorCall(Dtor, Dtor_Complete, 11598ed55a54SJohn McCall /*ForVirtualBase=*/false, Ptr); 11608ed55a54SJohn McCall 11618ed55a54SJohn McCall CGF.PopCleanupBlock(); 11628ed55a54SJohn McCall } 11638ed55a54SJohn McCall 11648ed55a54SJohn McCall namespace { 11658ed55a54SJohn McCall /// Calls the given 'operator delete' on an array of objects. 11668ed55a54SJohn McCall struct CallArrayDelete : EHScopeStack::Cleanup { 11678ed55a54SJohn McCall llvm::Value *Ptr; 11688ed55a54SJohn McCall const FunctionDecl *OperatorDelete; 11698ed55a54SJohn McCall llvm::Value *NumElements; 11708ed55a54SJohn McCall QualType ElementType; 11718ed55a54SJohn McCall CharUnits CookieSize; 11728ed55a54SJohn McCall 11738ed55a54SJohn McCall CallArrayDelete(llvm::Value *Ptr, 11748ed55a54SJohn McCall const FunctionDecl *OperatorDelete, 11758ed55a54SJohn McCall llvm::Value *NumElements, 11768ed55a54SJohn McCall QualType ElementType, 11778ed55a54SJohn McCall CharUnits CookieSize) 11788ed55a54SJohn McCall : Ptr(Ptr), OperatorDelete(OperatorDelete), NumElements(NumElements), 11798ed55a54SJohn McCall ElementType(ElementType), CookieSize(CookieSize) {} 11808ed55a54SJohn McCall 11818ed55a54SJohn McCall void Emit(CodeGenFunction &CGF, bool IsForEH) { 11828ed55a54SJohn McCall const FunctionProtoType *DeleteFTy = 11838ed55a54SJohn McCall OperatorDelete->getType()->getAs<FunctionProtoType>(); 11848ed55a54SJohn McCall assert(DeleteFTy->getNumArgs() == 1 || DeleteFTy->getNumArgs() == 2); 11858ed55a54SJohn McCall 11868ed55a54SJohn McCall CallArgList Args; 11878ed55a54SJohn McCall 11888ed55a54SJohn McCall // Pass the pointer as the first argument. 11898ed55a54SJohn McCall QualType VoidPtrTy = DeleteFTy->getArgType(0); 11908ed55a54SJohn McCall llvm::Value *DeletePtr 11918ed55a54SJohn McCall = CGF.Builder.CreateBitCast(Ptr, CGF.ConvertType(VoidPtrTy)); 11928ed55a54SJohn McCall Args.push_back(std::make_pair(RValue::get(DeletePtr), VoidPtrTy)); 11938ed55a54SJohn McCall 11948ed55a54SJohn McCall // Pass the original requested size as the second argument. 11958ed55a54SJohn McCall if (DeleteFTy->getNumArgs() == 2) { 11968ed55a54SJohn McCall QualType size_t = DeleteFTy->getArgType(1); 11978ed55a54SJohn McCall const llvm::IntegerType *SizeTy 11988ed55a54SJohn McCall = cast<llvm::IntegerType>(CGF.ConvertType(size_t)); 11998ed55a54SJohn McCall 12008ed55a54SJohn McCall CharUnits ElementTypeSize = 12018ed55a54SJohn McCall CGF.CGM.getContext().getTypeSizeInChars(ElementType); 12028ed55a54SJohn McCall 12038ed55a54SJohn McCall // The size of an element, multiplied by the number of elements. 12048ed55a54SJohn McCall llvm::Value *Size 12058ed55a54SJohn McCall = llvm::ConstantInt::get(SizeTy, ElementTypeSize.getQuantity()); 12068ed55a54SJohn McCall Size = CGF.Builder.CreateMul(Size, NumElements); 12078ed55a54SJohn McCall 12088ed55a54SJohn McCall // Plus the size of the cookie if applicable. 12098ed55a54SJohn McCall if (!CookieSize.isZero()) { 12108ed55a54SJohn McCall llvm::Value *CookieSizeV 12118ed55a54SJohn McCall = llvm::ConstantInt::get(SizeTy, CookieSize.getQuantity()); 12128ed55a54SJohn McCall Size = CGF.Builder.CreateAdd(Size, CookieSizeV); 12138ed55a54SJohn McCall } 12148ed55a54SJohn McCall 12158ed55a54SJohn McCall Args.push_back(std::make_pair(RValue::get(Size), size_t)); 12168ed55a54SJohn McCall } 12178ed55a54SJohn McCall 12188ed55a54SJohn McCall // Emit the call to delete. 12198ed55a54SJohn McCall CGF.EmitCall(CGF.getTypes().getFunctionInfo(Args, DeleteFTy), 12208ed55a54SJohn McCall CGF.CGM.GetAddrOfFunction(OperatorDelete), 12218ed55a54SJohn McCall ReturnValueSlot(), Args, OperatorDelete); 12228ed55a54SJohn McCall } 12238ed55a54SJohn McCall }; 12248ed55a54SJohn McCall } 12258ed55a54SJohn McCall 12268ed55a54SJohn McCall /// Emit the code for deleting an array of objects. 12278ed55a54SJohn McCall static void EmitArrayDelete(CodeGenFunction &CGF, 12288ed55a54SJohn McCall const FunctionDecl *OperatorDelete, 12298ed55a54SJohn McCall llvm::Value *Ptr, 12308ed55a54SJohn McCall QualType ElementType) { 12318ed55a54SJohn McCall llvm::Value *NumElements = 0; 12328ed55a54SJohn McCall llvm::Value *AllocatedPtr = 0; 12338ed55a54SJohn McCall CharUnits CookieSize; 12348ed55a54SJohn McCall CGF.CGM.getCXXABI().ReadArrayCookie(CGF, Ptr, ElementType, 12358ed55a54SJohn McCall NumElements, AllocatedPtr, CookieSize); 12368ed55a54SJohn McCall 12378ed55a54SJohn McCall assert(AllocatedPtr && "ReadArrayCookie didn't set AllocatedPtr"); 12388ed55a54SJohn McCall 12398ed55a54SJohn McCall // Make sure that we call delete even if one of the dtors throws. 12408ed55a54SJohn McCall CGF.EHStack.pushCleanup<CallArrayDelete>(NormalAndEHCleanup, 12418ed55a54SJohn McCall AllocatedPtr, OperatorDelete, 12428ed55a54SJohn McCall NumElements, ElementType, 12438ed55a54SJohn McCall CookieSize); 12448ed55a54SJohn McCall 12458ed55a54SJohn McCall if (const CXXRecordDecl *RD = ElementType->getAsCXXRecordDecl()) { 12468ed55a54SJohn McCall if (!RD->hasTrivialDestructor()) { 12478ed55a54SJohn McCall assert(NumElements && "ReadArrayCookie didn't find element count" 12488ed55a54SJohn McCall " for a class with destructor"); 12498ed55a54SJohn McCall CGF.EmitCXXAggrDestructorCall(RD->getDestructor(), NumElements, Ptr); 12508ed55a54SJohn McCall } 12518ed55a54SJohn McCall } 12528ed55a54SJohn McCall 12538ed55a54SJohn McCall CGF.PopCleanupBlock(); 12548ed55a54SJohn McCall } 12558ed55a54SJohn McCall 125659486a2dSAnders Carlsson void CodeGenFunction::EmitCXXDeleteExpr(const CXXDeleteExpr *E) { 125759486a2dSAnders Carlsson 125859486a2dSAnders Carlsson // Get at the argument before we performed the implicit conversion 125959486a2dSAnders Carlsson // to void*. 126059486a2dSAnders Carlsson const Expr *Arg = E->getArgument(); 126159486a2dSAnders Carlsson while (const ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(Arg)) { 1262e302792bSJohn McCall if (ICE->getCastKind() != CK_UserDefinedConversion && 126359486a2dSAnders Carlsson ICE->getType()->isVoidPointerType()) 126459486a2dSAnders Carlsson Arg = ICE->getSubExpr(); 126559486a2dSAnders Carlsson else 126659486a2dSAnders Carlsson break; 126759486a2dSAnders Carlsson } 126859486a2dSAnders Carlsson 126959486a2dSAnders Carlsson llvm::Value *Ptr = EmitScalarExpr(Arg); 127059486a2dSAnders Carlsson 127159486a2dSAnders Carlsson // Null check the pointer. 127259486a2dSAnders Carlsson llvm::BasicBlock *DeleteNotNull = createBasicBlock("delete.notnull"); 127359486a2dSAnders Carlsson llvm::BasicBlock *DeleteEnd = createBasicBlock("delete.end"); 127459486a2dSAnders Carlsson 127559486a2dSAnders Carlsson llvm::Value *IsNull = 127659486a2dSAnders Carlsson Builder.CreateICmpEQ(Ptr, llvm::Constant::getNullValue(Ptr->getType()), 127759486a2dSAnders Carlsson "isnull"); 127859486a2dSAnders Carlsson 127959486a2dSAnders Carlsson Builder.CreateCondBr(IsNull, DeleteEnd, DeleteNotNull); 128059486a2dSAnders Carlsson EmitBlock(DeleteNotNull); 128159486a2dSAnders Carlsson 12828ed55a54SJohn McCall // We might be deleting a pointer to array. If so, GEP down to the 12838ed55a54SJohn McCall // first non-array element. 12848ed55a54SJohn McCall // (this assumes that A(*)[3][7] is converted to [3 x [7 x %A]]*) 12858ed55a54SJohn McCall QualType DeleteTy = Arg->getType()->getAs<PointerType>()->getPointeeType(); 12868ed55a54SJohn McCall if (DeleteTy->isConstantArrayType()) { 12878ed55a54SJohn McCall llvm::Value *Zero = Builder.getInt32(0); 12888ed55a54SJohn McCall llvm::SmallVector<llvm::Value*,8> GEP; 128959486a2dSAnders Carlsson 12908ed55a54SJohn McCall GEP.push_back(Zero); // point at the outermost array 12918ed55a54SJohn McCall 12928ed55a54SJohn McCall // For each layer of array type we're pointing at: 12938ed55a54SJohn McCall while (const ConstantArrayType *Arr 12948ed55a54SJohn McCall = getContext().getAsConstantArrayType(DeleteTy)) { 12958ed55a54SJohn McCall // 1. Unpeel the array type. 12968ed55a54SJohn McCall DeleteTy = Arr->getElementType(); 12978ed55a54SJohn McCall 12988ed55a54SJohn McCall // 2. GEP to the first element of the array. 12998ed55a54SJohn McCall GEP.push_back(Zero); 13008ed55a54SJohn McCall } 13018ed55a54SJohn McCall 13028ed55a54SJohn McCall Ptr = Builder.CreateInBoundsGEP(Ptr, GEP.begin(), GEP.end(), "del.first"); 13038ed55a54SJohn McCall } 13048ed55a54SJohn McCall 130504f36218SDouglas Gregor assert(ConvertTypeForMem(DeleteTy) == 130604f36218SDouglas Gregor cast<llvm::PointerType>(Ptr->getType())->getElementType()); 13078ed55a54SJohn McCall 130859486a2dSAnders Carlsson if (E->isArrayForm()) { 13098ed55a54SJohn McCall EmitArrayDelete(*this, E->getOperatorDelete(), Ptr, DeleteTy); 13108ed55a54SJohn McCall } else { 13118ed55a54SJohn McCall EmitObjectDelete(*this, E->getOperatorDelete(), Ptr, DeleteTy); 131259486a2dSAnders Carlsson } 131359486a2dSAnders Carlsson 131459486a2dSAnders Carlsson EmitBlock(DeleteEnd); 131559486a2dSAnders Carlsson } 131659486a2dSAnders Carlsson 131759486a2dSAnders Carlsson llvm::Value * CodeGenFunction::EmitCXXTypeidExpr(const CXXTypeidExpr *E) { 131859486a2dSAnders Carlsson QualType Ty = E->getType(); 131959486a2dSAnders Carlsson const llvm::Type *LTy = ConvertType(Ty)->getPointerTo(); 1320fd7dfeb7SAnders Carlsson 13213f4336cbSAnders Carlsson if (E->isTypeOperand()) { 13223f4336cbSAnders Carlsson llvm::Constant *TypeInfo = 13233f4336cbSAnders Carlsson CGM.GetAddrOfRTTIDescriptor(E->getTypeOperand()); 13243f4336cbSAnders Carlsson return Builder.CreateBitCast(TypeInfo, LTy); 13253f4336cbSAnders Carlsson } 1326fd7dfeb7SAnders Carlsson 132759486a2dSAnders Carlsson Expr *subE = E->getExprOperand(); 132859486a2dSAnders Carlsson Ty = subE->getType(); 132959486a2dSAnders Carlsson CanQualType CanTy = CGM.getContext().getCanonicalType(Ty); 133059486a2dSAnders Carlsson Ty = CanTy.getUnqualifiedType().getNonReferenceType(); 133159486a2dSAnders Carlsson if (const RecordType *RT = Ty->getAs<RecordType>()) { 133259486a2dSAnders Carlsson const CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl()); 133359486a2dSAnders Carlsson if (RD->isPolymorphic()) { 133459486a2dSAnders Carlsson // FIXME: if subE is an lvalue do 133559486a2dSAnders Carlsson LValue Obj = EmitLValue(subE); 133659486a2dSAnders Carlsson llvm::Value *This = Obj.getAddress(); 133759486a2dSAnders Carlsson LTy = LTy->getPointerTo()->getPointerTo(); 133859486a2dSAnders Carlsson llvm::Value *V = Builder.CreateBitCast(This, LTy); 133959486a2dSAnders Carlsson // We need to do a zero check for *p, unless it has NonNullAttr. 134059486a2dSAnders Carlsson // FIXME: PointerType->hasAttr<NonNullAttr>() 134159486a2dSAnders Carlsson bool CanBeZero = false; 134259486a2dSAnders Carlsson if (UnaryOperator *UO = dyn_cast<UnaryOperator>(subE->IgnoreParens())) 1343e302792bSJohn McCall if (UO->getOpcode() == UO_Deref) 134459486a2dSAnders Carlsson CanBeZero = true; 134559486a2dSAnders Carlsson if (CanBeZero) { 134659486a2dSAnders Carlsson llvm::BasicBlock *NonZeroBlock = createBasicBlock(); 134759486a2dSAnders Carlsson llvm::BasicBlock *ZeroBlock = createBasicBlock(); 134859486a2dSAnders Carlsson 134959486a2dSAnders Carlsson llvm::Value *Zero = llvm::Constant::getNullValue(LTy); 135059486a2dSAnders Carlsson Builder.CreateCondBr(Builder.CreateICmpNE(V, Zero), 135159486a2dSAnders Carlsson NonZeroBlock, ZeroBlock); 135259486a2dSAnders Carlsson EmitBlock(ZeroBlock); 135359486a2dSAnders Carlsson /// Call __cxa_bad_typeid 135459486a2dSAnders Carlsson const llvm::Type *ResultType = llvm::Type::getVoidTy(VMContext); 135559486a2dSAnders Carlsson const llvm::FunctionType *FTy; 135659486a2dSAnders Carlsson FTy = llvm::FunctionType::get(ResultType, false); 135759486a2dSAnders Carlsson llvm::Value *F = CGM.CreateRuntimeFunction(FTy, "__cxa_bad_typeid"); 135859486a2dSAnders Carlsson Builder.CreateCall(F)->setDoesNotReturn(); 135959486a2dSAnders Carlsson Builder.CreateUnreachable(); 136059486a2dSAnders Carlsson EmitBlock(NonZeroBlock); 136159486a2dSAnders Carlsson } 136259486a2dSAnders Carlsson V = Builder.CreateLoad(V, "vtable"); 136359486a2dSAnders Carlsson V = Builder.CreateConstInBoundsGEP1_64(V, -1ULL); 136459486a2dSAnders Carlsson V = Builder.CreateLoad(V); 136559486a2dSAnders Carlsson return V; 136659486a2dSAnders Carlsson } 136759486a2dSAnders Carlsson } 13683f4336cbSAnders Carlsson return Builder.CreateBitCast(CGM.GetAddrOfRTTIDescriptor(Ty), LTy); 136959486a2dSAnders Carlsson } 137059486a2dSAnders Carlsson 137159486a2dSAnders Carlsson llvm::Value *CodeGenFunction::EmitDynamicCast(llvm::Value *V, 137259486a2dSAnders Carlsson const CXXDynamicCastExpr *DCE) { 13733f4336cbSAnders Carlsson QualType SrcTy = DCE->getSubExpr()->getType(); 13743f4336cbSAnders Carlsson QualType DestTy = DCE->getTypeAsWritten(); 13753f4336cbSAnders Carlsson QualType InnerType = DestTy->getPointeeType(); 13763f4336cbSAnders Carlsson 137759486a2dSAnders Carlsson const llvm::Type *LTy = ConvertType(DCE->getType()); 137859486a2dSAnders Carlsson 137959486a2dSAnders Carlsson bool CanBeZero = false; 138059486a2dSAnders Carlsson bool ToVoid = false; 138159486a2dSAnders Carlsson bool ThrowOnBad = false; 13823f4336cbSAnders Carlsson if (DestTy->isPointerType()) { 138359486a2dSAnders Carlsson // FIXME: if PointerType->hasAttr<NonNullAttr>(), we don't set this 138459486a2dSAnders Carlsson CanBeZero = true; 138559486a2dSAnders Carlsson if (InnerType->isVoidType()) 138659486a2dSAnders Carlsson ToVoid = true; 138759486a2dSAnders Carlsson } else { 138859486a2dSAnders Carlsson LTy = LTy->getPointerTo(); 1389fa8b4955SDouglas Gregor 1390fa8b4955SDouglas Gregor // FIXME: What if exceptions are disabled? 139159486a2dSAnders Carlsson ThrowOnBad = true; 139259486a2dSAnders Carlsson } 139359486a2dSAnders Carlsson 13943f4336cbSAnders Carlsson if (SrcTy->isPointerType() || SrcTy->isReferenceType()) 13953f4336cbSAnders Carlsson SrcTy = SrcTy->getPointeeType(); 13963f4336cbSAnders Carlsson SrcTy = SrcTy.getUnqualifiedType(); 13973f4336cbSAnders Carlsson 13980087bc85SAnders Carlsson if (DestTy->isPointerType() || DestTy->isReferenceType()) 13993f4336cbSAnders Carlsson DestTy = DestTy->getPointeeType(); 14003f4336cbSAnders Carlsson DestTy = DestTy.getUnqualifiedType(); 140159486a2dSAnders Carlsson 140259486a2dSAnders Carlsson llvm::BasicBlock *ContBlock = createBasicBlock(); 140359486a2dSAnders Carlsson llvm::BasicBlock *NullBlock = 0; 140459486a2dSAnders Carlsson llvm::BasicBlock *NonZeroBlock = 0; 140559486a2dSAnders Carlsson if (CanBeZero) { 140659486a2dSAnders Carlsson NonZeroBlock = createBasicBlock(); 140759486a2dSAnders Carlsson NullBlock = createBasicBlock(); 14083f4336cbSAnders Carlsson Builder.CreateCondBr(Builder.CreateIsNotNull(V), NonZeroBlock, NullBlock); 140959486a2dSAnders Carlsson EmitBlock(NonZeroBlock); 141059486a2dSAnders Carlsson } 141159486a2dSAnders Carlsson 141259486a2dSAnders Carlsson llvm::BasicBlock *BadCastBlock = 0; 141359486a2dSAnders Carlsson 14143f4336cbSAnders Carlsson const llvm::Type *PtrDiffTy = ConvertType(getContext().getPointerDiffType()); 141559486a2dSAnders Carlsson 141659486a2dSAnders Carlsson // See if this is a dynamic_cast(void*) 141759486a2dSAnders Carlsson if (ToVoid) { 141859486a2dSAnders Carlsson llvm::Value *This = V; 141959486a2dSAnders Carlsson V = Builder.CreateBitCast(This, PtrDiffTy->getPointerTo()->getPointerTo()); 142059486a2dSAnders Carlsson V = Builder.CreateLoad(V, "vtable"); 142159486a2dSAnders Carlsson V = Builder.CreateConstInBoundsGEP1_64(V, -2ULL); 142259486a2dSAnders Carlsson V = Builder.CreateLoad(V, "offset to top"); 142359486a2dSAnders Carlsson This = Builder.CreateBitCast(This, llvm::Type::getInt8PtrTy(VMContext)); 142459486a2dSAnders Carlsson V = Builder.CreateInBoundsGEP(This, V); 142559486a2dSAnders Carlsson V = Builder.CreateBitCast(V, LTy); 142659486a2dSAnders Carlsson } else { 142759486a2dSAnders Carlsson /// Call __dynamic_cast 142859486a2dSAnders Carlsson const llvm::Type *ResultType = llvm::Type::getInt8PtrTy(VMContext); 142959486a2dSAnders Carlsson const llvm::FunctionType *FTy; 143059486a2dSAnders Carlsson std::vector<const llvm::Type*> ArgTys; 143159486a2dSAnders Carlsson const llvm::Type *PtrToInt8Ty 143259486a2dSAnders Carlsson = llvm::Type::getInt8Ty(VMContext)->getPointerTo(); 143359486a2dSAnders Carlsson ArgTys.push_back(PtrToInt8Ty); 143459486a2dSAnders Carlsson ArgTys.push_back(PtrToInt8Ty); 143559486a2dSAnders Carlsson ArgTys.push_back(PtrToInt8Ty); 143659486a2dSAnders Carlsson ArgTys.push_back(PtrDiffTy); 143759486a2dSAnders Carlsson FTy = llvm::FunctionType::get(ResultType, ArgTys, false); 143859486a2dSAnders Carlsson 143959486a2dSAnders Carlsson // FIXME: Calculate better hint. 144059486a2dSAnders Carlsson llvm::Value *hint = llvm::ConstantInt::get(PtrDiffTy, -1ULL); 14413f4336cbSAnders Carlsson 14423f4336cbSAnders Carlsson assert(SrcTy->isRecordType() && "Src type must be record type!"); 14433f4336cbSAnders Carlsson assert(DestTy->isRecordType() && "Dest type must be record type!"); 14443f4336cbSAnders Carlsson 1445247894b3SDouglas Gregor llvm::Value *SrcArg 1446247894b3SDouglas Gregor = CGM.GetAddrOfRTTIDescriptor(SrcTy.getUnqualifiedType()); 1447247894b3SDouglas Gregor llvm::Value *DestArg 1448247894b3SDouglas Gregor = CGM.GetAddrOfRTTIDescriptor(DestTy.getUnqualifiedType()); 14493f4336cbSAnders Carlsson 145059486a2dSAnders Carlsson V = Builder.CreateBitCast(V, PtrToInt8Ty); 145159486a2dSAnders Carlsson V = Builder.CreateCall4(CGM.CreateRuntimeFunction(FTy, "__dynamic_cast"), 14523f4336cbSAnders Carlsson V, SrcArg, DestArg, hint); 145359486a2dSAnders Carlsson V = Builder.CreateBitCast(V, LTy); 145459486a2dSAnders Carlsson 145559486a2dSAnders Carlsson if (ThrowOnBad) { 145659486a2dSAnders Carlsson BadCastBlock = createBasicBlock(); 14573f4336cbSAnders Carlsson Builder.CreateCondBr(Builder.CreateIsNotNull(V), ContBlock, BadCastBlock); 145859486a2dSAnders Carlsson EmitBlock(BadCastBlock); 1459fa8b4955SDouglas Gregor /// Invoke __cxa_bad_cast 146059486a2dSAnders Carlsson ResultType = llvm::Type::getVoidTy(VMContext); 146159486a2dSAnders Carlsson const llvm::FunctionType *FBadTy; 146259486a2dSAnders Carlsson FBadTy = llvm::FunctionType::get(ResultType, false); 146359486a2dSAnders Carlsson llvm::Value *F = CGM.CreateRuntimeFunction(FBadTy, "__cxa_bad_cast"); 1464fa8b4955SDouglas Gregor if (llvm::BasicBlock *InvokeDest = getInvokeDest()) { 1465fa8b4955SDouglas Gregor llvm::BasicBlock *Cont = createBasicBlock("invoke.cont"); 1466fa8b4955SDouglas Gregor Builder.CreateInvoke(F, Cont, InvokeDest)->setDoesNotReturn(); 1467fa8b4955SDouglas Gregor EmitBlock(Cont); 1468fa8b4955SDouglas Gregor } else { 1469fa8b4955SDouglas Gregor // FIXME: Does this ever make sense? 147059486a2dSAnders Carlsson Builder.CreateCall(F)->setDoesNotReturn(); 1471fa8b4955SDouglas Gregor } 147259486a2dSAnders Carlsson Builder.CreateUnreachable(); 147359486a2dSAnders Carlsson } 147459486a2dSAnders Carlsson } 147559486a2dSAnders Carlsson 147659486a2dSAnders Carlsson if (CanBeZero) { 147759486a2dSAnders Carlsson Builder.CreateBr(ContBlock); 147859486a2dSAnders Carlsson EmitBlock(NullBlock); 147959486a2dSAnders Carlsson Builder.CreateBr(ContBlock); 148059486a2dSAnders Carlsson } 148159486a2dSAnders Carlsson EmitBlock(ContBlock); 148259486a2dSAnders Carlsson if (CanBeZero) { 148359486a2dSAnders Carlsson llvm::PHINode *PHI = Builder.CreatePHI(LTy); 148459486a2dSAnders Carlsson PHI->reserveOperandSpace(2); 148559486a2dSAnders Carlsson PHI->addIncoming(V, NonZeroBlock); 148659486a2dSAnders Carlsson PHI->addIncoming(llvm::Constant::getNullValue(LTy), NullBlock); 148759486a2dSAnders Carlsson V = PHI; 148859486a2dSAnders Carlsson } 148959486a2dSAnders Carlsson 149059486a2dSAnders Carlsson return V; 149159486a2dSAnders Carlsson } 1492