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" 1559486a2dSAnders Carlsson using namespace clang; 1659486a2dSAnders Carlsson using namespace CodeGen; 1759486a2dSAnders Carlsson 1827da15baSAnders Carlsson RValue CodeGenFunction::EmitCXXMemberCall(const CXXMethodDecl *MD, 1927da15baSAnders Carlsson llvm::Value *Callee, 2027da15baSAnders Carlsson ReturnValueSlot ReturnValue, 2127da15baSAnders Carlsson llvm::Value *This, 22e36a6b3eSAnders Carlsson llvm::Value *VTT, 2327da15baSAnders Carlsson CallExpr::const_arg_iterator ArgBeg, 2427da15baSAnders Carlsson CallExpr::const_arg_iterator ArgEnd) { 2527da15baSAnders Carlsson assert(MD->isInstance() && 2627da15baSAnders Carlsson "Trying to emit a member call expr on a static method!"); 2727da15baSAnders Carlsson 2827da15baSAnders Carlsson const FunctionProtoType *FPT = MD->getType()->getAs<FunctionProtoType>(); 2927da15baSAnders Carlsson 3027da15baSAnders Carlsson CallArgList Args; 3127da15baSAnders Carlsson 3227da15baSAnders Carlsson // Push the this ptr. 3327da15baSAnders Carlsson Args.push_back(std::make_pair(RValue::get(This), 3427da15baSAnders Carlsson MD->getThisType(getContext()))); 3527da15baSAnders Carlsson 36e36a6b3eSAnders Carlsson // If there is a VTT parameter, emit it. 37e36a6b3eSAnders Carlsson if (VTT) { 38e36a6b3eSAnders Carlsson QualType T = getContext().getPointerType(getContext().VoidPtrTy); 39e36a6b3eSAnders Carlsson Args.push_back(std::make_pair(RValue::get(VTT), T)); 40e36a6b3eSAnders Carlsson } 41e36a6b3eSAnders Carlsson 4227da15baSAnders Carlsson // And the rest of the call args 4327da15baSAnders Carlsson EmitCallArgs(Args, FPT, ArgBeg, ArgEnd); 4427da15baSAnders Carlsson 45ab26cfa5SJohn McCall QualType ResultType = FPT->getResultType(); 46ab26cfa5SJohn McCall return EmitCall(CGM.getTypes().getFunctionInfo(ResultType, Args, 47c50c27ccSRafael Espindola FPT->getExtInfo()), 48c50c27ccSRafael Espindola Callee, ReturnValue, Args, MD); 4927da15baSAnders Carlsson } 5027da15baSAnders Carlsson 5127da15baSAnders Carlsson /// canDevirtualizeMemberFunctionCalls - Checks whether virtual calls on given 5227da15baSAnders Carlsson /// expr can be devirtualized. 5327da15baSAnders Carlsson static bool canDevirtualizeMemberFunctionCalls(const Expr *Base) { 5427da15baSAnders Carlsson if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(Base)) { 5527da15baSAnders Carlsson if (const VarDecl *VD = dyn_cast<VarDecl>(DRE->getDecl())) { 5627da15baSAnders Carlsson // This is a record decl. We know the type and can devirtualize it. 5727da15baSAnders Carlsson return VD->getType()->isRecordType(); 5827da15baSAnders Carlsson } 5927da15baSAnders Carlsson 6027da15baSAnders Carlsson return false; 6127da15baSAnders Carlsson } 6227da15baSAnders Carlsson 6327da15baSAnders Carlsson // We can always devirtualize calls on temporary object expressions. 64a682427eSEli Friedman if (isa<CXXConstructExpr>(Base)) 6527da15baSAnders Carlsson return true; 6627da15baSAnders Carlsson 6727da15baSAnders Carlsson // And calls on bound temporaries. 6827da15baSAnders Carlsson if (isa<CXXBindTemporaryExpr>(Base)) 6927da15baSAnders Carlsson return true; 7027da15baSAnders Carlsson 7127da15baSAnders Carlsson // Check if this is a call expr that returns a record type. 7227da15baSAnders Carlsson if (const CallExpr *CE = dyn_cast<CallExpr>(Base)) 7327da15baSAnders Carlsson return CE->getCallReturnType()->isRecordType(); 7427da15baSAnders Carlsson 7527da15baSAnders Carlsson // We can't devirtualize the call. 7627da15baSAnders Carlsson return false; 7727da15baSAnders Carlsson } 7827da15baSAnders Carlsson 7927da15baSAnders Carlsson RValue CodeGenFunction::EmitCXXMemberCallExpr(const CXXMemberCallExpr *CE, 8027da15baSAnders Carlsson ReturnValueSlot ReturnValue) { 8127da15baSAnders Carlsson if (isa<BinaryOperator>(CE->getCallee()->IgnoreParens())) 8227da15baSAnders Carlsson return EmitCXXMemberPointerCallExpr(CE, ReturnValue); 8327da15baSAnders Carlsson 8427da15baSAnders Carlsson const MemberExpr *ME = cast<MemberExpr>(CE->getCallee()->IgnoreParens()); 8527da15baSAnders Carlsson const CXXMethodDecl *MD = cast<CXXMethodDecl>(ME->getMemberDecl()); 8627da15baSAnders Carlsson 8727da15baSAnders Carlsson if (MD->isStatic()) { 8827da15baSAnders Carlsson // The method is static, emit it as we would a regular call. 8927da15baSAnders Carlsson llvm::Value *Callee = CGM.GetAddrOfFunction(MD); 9027da15baSAnders Carlsson return EmitCall(getContext().getPointerType(MD->getType()), Callee, 9127da15baSAnders Carlsson ReturnValue, CE->arg_begin(), CE->arg_end()); 9227da15baSAnders Carlsson } 9327da15baSAnders Carlsson 9427da15baSAnders Carlsson const FunctionProtoType *FPT = MD->getType()->getAs<FunctionProtoType>(); 9527da15baSAnders Carlsson 9627da15baSAnders Carlsson const llvm::Type *Ty = 9727da15baSAnders Carlsson CGM.getTypes().GetFunctionType(CGM.getTypes().getFunctionInfo(MD), 9827da15baSAnders Carlsson FPT->isVariadic()); 9927da15baSAnders Carlsson llvm::Value *This; 10027da15baSAnders Carlsson 10127da15baSAnders Carlsson if (ME->isArrow()) 10227da15baSAnders Carlsson This = EmitScalarExpr(ME->getBase()); 10327da15baSAnders Carlsson else { 10427da15baSAnders Carlsson LValue BaseLV = EmitLValue(ME->getBase()); 10527da15baSAnders Carlsson This = BaseLV.getAddress(); 10627da15baSAnders Carlsson } 10727da15baSAnders Carlsson 10827da15baSAnders Carlsson if (MD->isCopyAssignment() && MD->isTrivial()) { 10927da15baSAnders Carlsson // We don't like to generate the trivial copy assignment operator when 11027da15baSAnders Carlsson // it isn't necessary; just produce the proper effect here. 11127da15baSAnders Carlsson llvm::Value *RHS = EmitLValue(*CE->arg_begin()).getAddress(); 11227da15baSAnders Carlsson EmitAggregateCopy(This, RHS, CE->getType()); 11327da15baSAnders Carlsson return RValue::get(This); 11427da15baSAnders Carlsson } 11527da15baSAnders Carlsson 11627da15baSAnders Carlsson // C++ [class.virtual]p12: 11727da15baSAnders Carlsson // Explicit qualification with the scope operator (5.1) suppresses the 11827da15baSAnders Carlsson // virtual call mechanism. 11927da15baSAnders Carlsson // 12027da15baSAnders Carlsson // We also don't emit a virtual call if the base expression has a record type 12127da15baSAnders Carlsson // because then we know what the type is. 12227da15baSAnders Carlsson llvm::Value *Callee; 12327da15baSAnders Carlsson if (const CXXDestructorDecl *Destructor 12427da15baSAnders Carlsson = dyn_cast<CXXDestructorDecl>(MD)) { 12527da15baSAnders Carlsson if (Destructor->isTrivial()) 12627da15baSAnders Carlsson return RValue::get(0); 12727da15baSAnders Carlsson if (MD->isVirtual() && !ME->hasQualifier() && 12827da15baSAnders Carlsson !canDevirtualizeMemberFunctionCalls(ME->getBase())) { 12927da15baSAnders Carlsson Callee = BuildVirtualCall(Destructor, Dtor_Complete, This, Ty); 13027da15baSAnders Carlsson } else { 13127da15baSAnders Carlsson Callee = CGM.GetAddrOfFunction(GlobalDecl(Destructor, Dtor_Complete), Ty); 13227da15baSAnders Carlsson } 13327da15baSAnders Carlsson } else if (MD->isVirtual() && !ME->hasQualifier() && 13427da15baSAnders Carlsson !canDevirtualizeMemberFunctionCalls(ME->getBase())) { 13527da15baSAnders Carlsson Callee = BuildVirtualCall(MD, This, Ty); 13627da15baSAnders Carlsson } else { 13727da15baSAnders Carlsson Callee = CGM.GetAddrOfFunction(MD, Ty); 13827da15baSAnders Carlsson } 13927da15baSAnders Carlsson 140e36a6b3eSAnders Carlsson return EmitCXXMemberCall(MD, Callee, ReturnValue, This, /*VTT=*/0, 14127da15baSAnders Carlsson CE->arg_begin(), CE->arg_end()); 14227da15baSAnders Carlsson } 14327da15baSAnders Carlsson 14427da15baSAnders Carlsson RValue 14527da15baSAnders Carlsson CodeGenFunction::EmitCXXMemberPointerCallExpr(const CXXMemberCallExpr *E, 14627da15baSAnders Carlsson ReturnValueSlot ReturnValue) { 14727da15baSAnders Carlsson const BinaryOperator *BO = 14827da15baSAnders Carlsson cast<BinaryOperator>(E->getCallee()->IgnoreParens()); 14927da15baSAnders Carlsson const Expr *BaseExpr = BO->getLHS(); 15027da15baSAnders Carlsson const Expr *MemFnExpr = BO->getRHS(); 15127da15baSAnders Carlsson 15227da15baSAnders Carlsson const MemberPointerType *MPT = 15327da15baSAnders Carlsson MemFnExpr->getType()->getAs<MemberPointerType>(); 15427da15baSAnders Carlsson const FunctionProtoType *FPT = 15527da15baSAnders Carlsson MPT->getPointeeType()->getAs<FunctionProtoType>(); 15627da15baSAnders Carlsson const CXXRecordDecl *RD = 15727da15baSAnders Carlsson cast<CXXRecordDecl>(MPT->getClass()->getAs<RecordType>()->getDecl()); 15827da15baSAnders Carlsson 15927da15baSAnders Carlsson const llvm::FunctionType *FTy = 16027da15baSAnders Carlsson CGM.getTypes().GetFunctionType(CGM.getTypes().getFunctionInfo(RD, FPT), 16127da15baSAnders Carlsson FPT->isVariadic()); 16227da15baSAnders Carlsson 163c4325486SAnders Carlsson const llvm::Type *Int8PtrTy = llvm::Type::getInt8PtrTy(VMContext); 16427da15baSAnders Carlsson 16527da15baSAnders Carlsson // Get the member function pointer. 166a7566f16SDaniel Dunbar llvm::Value *MemFnPtr = CreateMemTemp(MemFnExpr->getType(), "mem.fn"); 16727da15baSAnders Carlsson EmitAggExpr(MemFnExpr, MemFnPtr, /*VolatileDest=*/false); 16827da15baSAnders Carlsson 16927da15baSAnders Carlsson // Emit the 'this' pointer. 17027da15baSAnders Carlsson llvm::Value *This; 17127da15baSAnders Carlsson 17227da15baSAnders Carlsson if (BO->getOpcode() == BinaryOperator::PtrMemI) 17327da15baSAnders Carlsson This = EmitScalarExpr(BaseExpr); 17427da15baSAnders Carlsson else 17527da15baSAnders Carlsson This = EmitLValue(BaseExpr).getAddress(); 17627da15baSAnders Carlsson 17727da15baSAnders Carlsson // Adjust it. 17827da15baSAnders Carlsson llvm::Value *Adj = Builder.CreateStructGEP(MemFnPtr, 1); 17927da15baSAnders Carlsson Adj = Builder.CreateLoad(Adj, "mem.fn.adj"); 18027da15baSAnders Carlsson 18127da15baSAnders Carlsson llvm::Value *Ptr = Builder.CreateBitCast(This, Int8PtrTy, "ptr"); 18227da15baSAnders Carlsson Ptr = Builder.CreateGEP(Ptr, Adj, "adj"); 18327da15baSAnders Carlsson 18427da15baSAnders Carlsson This = Builder.CreateBitCast(Ptr, This->getType(), "this"); 18527da15baSAnders Carlsson 18627da15baSAnders Carlsson llvm::Value *FnPtr = Builder.CreateStructGEP(MemFnPtr, 0, "mem.fn.ptr"); 18727da15baSAnders Carlsson 18827da15baSAnders Carlsson const llvm::Type *PtrDiffTy = ConvertType(getContext().getPointerDiffType()); 18927da15baSAnders Carlsson 19027da15baSAnders Carlsson llvm::Value *FnAsInt = Builder.CreateLoad(FnPtr, "fn"); 19127da15baSAnders Carlsson 19227da15baSAnders Carlsson // If the LSB in the function pointer is 1, the function pointer points to 19327da15baSAnders Carlsson // a virtual function. 19427da15baSAnders Carlsson llvm::Value *IsVirtual 19527da15baSAnders Carlsson = Builder.CreateAnd(FnAsInt, llvm::ConstantInt::get(PtrDiffTy, 1), 19627da15baSAnders Carlsson "and"); 19727da15baSAnders Carlsson 19827da15baSAnders Carlsson IsVirtual = Builder.CreateTrunc(IsVirtual, 19927da15baSAnders Carlsson llvm::Type::getInt1Ty(VMContext)); 20027da15baSAnders Carlsson 20127da15baSAnders Carlsson llvm::BasicBlock *FnVirtual = createBasicBlock("fn.virtual"); 20227da15baSAnders Carlsson llvm::BasicBlock *FnNonVirtual = createBasicBlock("fn.nonvirtual"); 20327da15baSAnders Carlsson llvm::BasicBlock *FnEnd = createBasicBlock("fn.end"); 20427da15baSAnders Carlsson 20527da15baSAnders Carlsson Builder.CreateCondBr(IsVirtual, FnVirtual, FnNonVirtual); 20627da15baSAnders Carlsson EmitBlock(FnVirtual); 20727da15baSAnders Carlsson 208c4325486SAnders Carlsson const llvm::Type *VtableTy = 209c4325486SAnders Carlsson FTy->getPointerTo()->getPointerTo(); 21027da15baSAnders Carlsson 211c4325486SAnders Carlsson llvm::Value *Vtable = Builder.CreateBitCast(This, VtableTy->getPointerTo()); 212c4325486SAnders Carlsson Vtable = Builder.CreateLoad(Vtable); 21327da15baSAnders Carlsson 214c4325486SAnders Carlsson Vtable = Builder.CreateBitCast(Vtable, Int8PtrTy); 215c4325486SAnders Carlsson llvm::Value *VtableOffset = 216c4325486SAnders Carlsson Builder.CreateSub(FnAsInt, llvm::ConstantInt::get(PtrDiffTy, 1)); 21727da15baSAnders Carlsson 218c4325486SAnders Carlsson Vtable = Builder.CreateGEP(Vtable, VtableOffset, "fn"); 219c4325486SAnders Carlsson Vtable = Builder.CreateBitCast(Vtable, VtableTy); 22027da15baSAnders Carlsson 221c4325486SAnders Carlsson llvm::Value *VirtualFn = Builder.CreateLoad(Vtable, "virtualfn"); 22227da15baSAnders Carlsson 22327da15baSAnders Carlsson EmitBranch(FnEnd); 22427da15baSAnders Carlsson EmitBlock(FnNonVirtual); 22527da15baSAnders Carlsson 22627da15baSAnders Carlsson // If the function is not virtual, just load the pointer. 22727da15baSAnders Carlsson llvm::Value *NonVirtualFn = Builder.CreateLoad(FnPtr, "fn"); 22827da15baSAnders Carlsson NonVirtualFn = Builder.CreateIntToPtr(NonVirtualFn, FTy->getPointerTo()); 22927da15baSAnders Carlsson 23027da15baSAnders Carlsson EmitBlock(FnEnd); 23127da15baSAnders Carlsson 23227da15baSAnders Carlsson llvm::PHINode *Callee = Builder.CreatePHI(FTy->getPointerTo()); 23327da15baSAnders Carlsson Callee->reserveOperandSpace(2); 23427da15baSAnders Carlsson Callee->addIncoming(VirtualFn, FnVirtual); 23527da15baSAnders Carlsson Callee->addIncoming(NonVirtualFn, FnNonVirtual); 23627da15baSAnders Carlsson 23727da15baSAnders Carlsson CallArgList Args; 23827da15baSAnders Carlsson 23927da15baSAnders Carlsson QualType ThisType = 24027da15baSAnders Carlsson getContext().getPointerType(getContext().getTagDeclType(RD)); 24127da15baSAnders Carlsson 24227da15baSAnders Carlsson // Push the this ptr. 24327da15baSAnders Carlsson Args.push_back(std::make_pair(RValue::get(This), ThisType)); 24427da15baSAnders Carlsson 24527da15baSAnders Carlsson // And the rest of the call args 24627da15baSAnders Carlsson EmitCallArgs(Args, FPT, E->arg_begin(), E->arg_end()); 247ab26cfa5SJohn McCall const FunctionType *BO_FPT = BO->getType()->getAs<FunctionProtoType>(); 248ab26cfa5SJohn McCall return EmitCall(CGM.getTypes().getFunctionInfo(Args, BO_FPT), Callee, 24927da15baSAnders Carlsson ReturnValue, Args); 25027da15baSAnders Carlsson } 25127da15baSAnders Carlsson 25227da15baSAnders Carlsson RValue 25327da15baSAnders Carlsson CodeGenFunction::EmitCXXOperatorMemberCallExpr(const CXXOperatorCallExpr *E, 25427da15baSAnders Carlsson const CXXMethodDecl *MD, 25527da15baSAnders Carlsson ReturnValueSlot ReturnValue) { 25627da15baSAnders Carlsson assert(MD->isInstance() && 25727da15baSAnders Carlsson "Trying to emit a member call expr on a static method!"); 25827da15baSAnders Carlsson 25927da15baSAnders Carlsson if (MD->isCopyAssignment()) { 26027da15baSAnders Carlsson const CXXRecordDecl *ClassDecl = cast<CXXRecordDecl>(MD->getDeclContext()); 26127da15baSAnders Carlsson if (ClassDecl->hasTrivialCopyAssignment()) { 26227da15baSAnders Carlsson assert(!ClassDecl->hasUserDeclaredCopyAssignment() && 26327da15baSAnders Carlsson "EmitCXXOperatorMemberCallExpr - user declared copy assignment"); 26427da15baSAnders Carlsson llvm::Value *This = EmitLValue(E->getArg(0)).getAddress(); 26527da15baSAnders Carlsson llvm::Value *Src = EmitLValue(E->getArg(1)).getAddress(); 26627da15baSAnders Carlsson QualType Ty = E->getType(); 26727da15baSAnders Carlsson EmitAggregateCopy(This, Src, Ty); 26827da15baSAnders Carlsson return RValue::get(This); 26927da15baSAnders Carlsson } 27027da15baSAnders Carlsson } 27127da15baSAnders Carlsson 27227da15baSAnders Carlsson const FunctionProtoType *FPT = MD->getType()->getAs<FunctionProtoType>(); 27327da15baSAnders Carlsson const llvm::Type *Ty = 27427da15baSAnders Carlsson CGM.getTypes().GetFunctionType(CGM.getTypes().getFunctionInfo(MD), 27527da15baSAnders Carlsson FPT->isVariadic()); 27627da15baSAnders Carlsson 27727da15baSAnders Carlsson llvm::Value *This = EmitLValue(E->getArg(0)).getAddress(); 27827da15baSAnders Carlsson 27927da15baSAnders Carlsson llvm::Value *Callee; 28027da15baSAnders Carlsson if (MD->isVirtual() && !canDevirtualizeMemberFunctionCalls(E->getArg(0))) 28127da15baSAnders Carlsson Callee = BuildVirtualCall(MD, This, Ty); 28227da15baSAnders Carlsson else 28327da15baSAnders Carlsson Callee = CGM.GetAddrOfFunction(MD, Ty); 28427da15baSAnders Carlsson 285e36a6b3eSAnders Carlsson return EmitCXXMemberCall(MD, Callee, ReturnValue, This, /*VTT=*/0, 28627da15baSAnders Carlsson E->arg_begin() + 1, E->arg_end()); 28727da15baSAnders Carlsson } 28827da15baSAnders Carlsson 28927da15baSAnders Carlsson void 29027da15baSAnders Carlsson CodeGenFunction::EmitCXXConstructExpr(llvm::Value *Dest, 29127da15baSAnders Carlsson const CXXConstructExpr *E) { 29227da15baSAnders Carlsson assert(Dest && "Must have a destination!"); 29327da15baSAnders Carlsson const CXXConstructorDecl *CD = E->getConstructor(); 29427da15baSAnders Carlsson const ConstantArrayType *Array = 29527da15baSAnders Carlsson getContext().getAsConstantArrayType(E->getType()); 29627da15baSAnders Carlsson // For a copy constructor, even if it is trivial, must fall thru so 29727da15baSAnders Carlsson // its argument is code-gen'ed. 29827da15baSAnders Carlsson if (!CD->isCopyConstructor()) { 29927da15baSAnders Carlsson QualType InitType = E->getType(); 30027da15baSAnders Carlsson if (Array) 30127da15baSAnders Carlsson InitType = getContext().getBaseElementType(Array); 30227da15baSAnders Carlsson const CXXRecordDecl *RD = 30327da15baSAnders Carlsson cast<CXXRecordDecl>(InitType->getAs<RecordType>()->getDecl()); 30427da15baSAnders Carlsson if (RD->hasTrivialConstructor()) 30527da15baSAnders Carlsson return; 30627da15baSAnders Carlsson } 30727da15baSAnders Carlsson // Code gen optimization to eliminate copy constructor and return 30827da15baSAnders Carlsson // its first argument instead. 30927da15baSAnders Carlsson if (getContext().getLangOptions().ElideConstructors && E->isElidable()) { 310*45cf7e3dSDouglas Gregor const Expr *Arg = E->getArg(0)->getTemporaryObject(); 31127da15baSAnders Carlsson EmitAggExpr(Arg, Dest, false); 31227da15baSAnders Carlsson return; 31327da15baSAnders Carlsson } 31427da15baSAnders Carlsson if (Array) { 31527da15baSAnders Carlsson QualType BaseElementTy = getContext().getBaseElementType(Array); 31627da15baSAnders Carlsson const llvm::Type *BasePtr = ConvertType(BaseElementTy); 31727da15baSAnders Carlsson BasePtr = llvm::PointerType::getUnqual(BasePtr); 31827da15baSAnders Carlsson llvm::Value *BaseAddrPtr = 31927da15baSAnders Carlsson Builder.CreateBitCast(Dest, BasePtr); 32027da15baSAnders Carlsson 32127da15baSAnders Carlsson EmitCXXAggrConstructorCall(CD, Array, BaseAddrPtr, 32227da15baSAnders Carlsson E->arg_begin(), E->arg_end()); 32327da15baSAnders Carlsson } 32427da15baSAnders Carlsson else 32527da15baSAnders Carlsson // Call the constructor. 3267ae2d775SDouglas Gregor EmitCXXConstructorCall(CD, 3277ae2d775SDouglas Gregor E->isBaseInitialization()? Ctor_Base : Ctor_Complete, 3287ae2d775SDouglas Gregor Dest, 32927da15baSAnders Carlsson E->arg_begin(), E->arg_end()); 33027da15baSAnders Carlsson } 33127da15baSAnders Carlsson 3323eb55cfeSKen Dyck static CharUnits CalculateCookiePadding(ASTContext &Ctx, QualType ElementType) { 33321122cf6SAnders Carlsson const RecordType *RT = ElementType->getAs<RecordType>(); 33459486a2dSAnders Carlsson if (!RT) 3353eb55cfeSKen Dyck return CharUnits::Zero(); 33659486a2dSAnders Carlsson 33759486a2dSAnders Carlsson const CXXRecordDecl *RD = dyn_cast<CXXRecordDecl>(RT->getDecl()); 33859486a2dSAnders Carlsson if (!RD) 3393eb55cfeSKen Dyck return CharUnits::Zero(); 34059486a2dSAnders Carlsson 34159486a2dSAnders Carlsson // Check if the class has a trivial destructor. 34259486a2dSAnders Carlsson if (RD->hasTrivialDestructor()) { 34321122cf6SAnders Carlsson // Check if the usual deallocation function takes two arguments. 344adbe4249SAnders Carlsson const CXXMethodDecl *UsualDeallocationFunction = 0; 345adbe4249SAnders Carlsson 34621122cf6SAnders Carlsson DeclarationName OpName = 34721122cf6SAnders Carlsson Ctx.DeclarationNames.getCXXOperatorName(OO_Array_Delete); 34821122cf6SAnders Carlsson DeclContext::lookup_const_iterator Op, OpEnd; 34921122cf6SAnders Carlsson for (llvm::tie(Op, OpEnd) = RD->lookup(OpName); 35021122cf6SAnders Carlsson Op != OpEnd; ++Op) { 351adbe4249SAnders Carlsson const CXXMethodDecl *Delete = cast<CXXMethodDecl>(*Op); 35221122cf6SAnders Carlsson 35321122cf6SAnders Carlsson if (Delete->isUsualDeallocationFunction()) { 354adbe4249SAnders Carlsson UsualDeallocationFunction = Delete; 35521122cf6SAnders Carlsson break; 35621122cf6SAnders Carlsson } 35721122cf6SAnders Carlsson } 358adbe4249SAnders Carlsson 359adbe4249SAnders Carlsson // No usual deallocation function, we don't need a cookie. 360adbe4249SAnders Carlsson if (!UsualDeallocationFunction) 3613eb55cfeSKen Dyck return CharUnits::Zero(); 362adbe4249SAnders Carlsson 363adbe4249SAnders Carlsson // The usual deallocation function doesn't take a size_t argument, so we 364adbe4249SAnders Carlsson // don't need a cookie. 365adbe4249SAnders Carlsson if (UsualDeallocationFunction->getNumParams() == 1) 3663eb55cfeSKen Dyck return CharUnits::Zero(); 367adbe4249SAnders Carlsson 368adbe4249SAnders Carlsson assert(UsualDeallocationFunction->getNumParams() == 2 && 369adbe4249SAnders Carlsson "Unexpected deallocation function type!"); 37059486a2dSAnders Carlsson } 37159486a2dSAnders Carlsson 37221122cf6SAnders Carlsson // Padding is the maximum of sizeof(size_t) and alignof(ElementType) 3733eb55cfeSKen Dyck return std::max(Ctx.getTypeSizeInChars(Ctx.getSizeType()), 3743eb55cfeSKen Dyck Ctx.getTypeAlignInChars(ElementType)); 37521122cf6SAnders Carlsson } 37621122cf6SAnders Carlsson 3773eb55cfeSKen Dyck static CharUnits CalculateCookiePadding(ASTContext &Ctx, const CXXNewExpr *E) { 37821122cf6SAnders Carlsson if (!E->isArray()) 3793eb55cfeSKen Dyck return CharUnits::Zero(); 38021122cf6SAnders Carlsson 381399f499fSAnders Carlsson // No cookie is required if the new operator being used is 382399f499fSAnders Carlsson // ::operator new[](size_t, void*). 383399f499fSAnders Carlsson const FunctionDecl *OperatorNew = E->getOperatorNew(); 384399f499fSAnders Carlsson if (OperatorNew->getDeclContext()->getLookupContext()->isFileContext()) { 385399f499fSAnders Carlsson if (OperatorNew->getNumParams() == 2) { 386399f499fSAnders Carlsson CanQualType ParamType = 387399f499fSAnders Carlsson Ctx.getCanonicalType(OperatorNew->getParamDecl(1)->getType()); 388399f499fSAnders Carlsson 389399f499fSAnders Carlsson if (ParamType == Ctx.VoidPtrTy) 3903eb55cfeSKen Dyck return CharUnits::Zero(); 391399f499fSAnders Carlsson } 392399f499fSAnders Carlsson } 393399f499fSAnders Carlsson 39421122cf6SAnders Carlsson return CalculateCookiePadding(Ctx, E->getAllocatedType()); 39559486a2dSAnders Carlsson } 39659486a2dSAnders Carlsson 39747b4629bSFariborz Jahanian static llvm::Value *EmitCXXNewAllocSize(ASTContext &Context, 39847b4629bSFariborz Jahanian CodeGenFunction &CGF, 39959486a2dSAnders Carlsson const CXXNewExpr *E, 40059486a2dSAnders Carlsson llvm::Value *& NumElements) { 40159486a2dSAnders Carlsson QualType Type = E->getAllocatedType(); 4023eb55cfeSKen Dyck CharUnits TypeSize = CGF.getContext().getTypeSizeInChars(Type); 40359486a2dSAnders Carlsson const llvm::Type *SizeTy = CGF.ConvertType(CGF.getContext().getSizeType()); 40459486a2dSAnders Carlsson 40559486a2dSAnders Carlsson if (!E->isArray()) 4063eb55cfeSKen Dyck return llvm::ConstantInt::get(SizeTy, TypeSize.getQuantity()); 40759486a2dSAnders Carlsson 4083eb55cfeSKen Dyck CharUnits CookiePadding = CalculateCookiePadding(CGF.getContext(), E); 40959486a2dSAnders Carlsson 41059486a2dSAnders Carlsson Expr::EvalResult Result; 41159486a2dSAnders Carlsson if (E->getArraySize()->Evaluate(Result, CGF.getContext()) && 41259486a2dSAnders Carlsson !Result.HasSideEffects && Result.Val.isInt()) { 41359486a2dSAnders Carlsson 4143eb55cfeSKen Dyck CharUnits AllocSize = 4153eb55cfeSKen Dyck Result.Val.getInt().getZExtValue() * TypeSize + CookiePadding; 41659486a2dSAnders Carlsson 41759486a2dSAnders Carlsson NumElements = 41859486a2dSAnders Carlsson llvm::ConstantInt::get(SizeTy, Result.Val.getInt().getZExtValue()); 41947b4629bSFariborz Jahanian while (const ArrayType *AType = Context.getAsArrayType(Type)) { 42047b4629bSFariborz Jahanian const llvm::ArrayType *llvmAType = 42147b4629bSFariborz Jahanian cast<llvm::ArrayType>(CGF.ConvertType(Type)); 42247b4629bSFariborz Jahanian NumElements = 42347b4629bSFariborz Jahanian CGF.Builder.CreateMul(NumElements, 42447b4629bSFariborz Jahanian llvm::ConstantInt::get( 42547b4629bSFariborz Jahanian SizeTy, llvmAType->getNumElements())); 42647b4629bSFariborz Jahanian Type = AType->getElementType(); 42747b4629bSFariborz Jahanian } 42859486a2dSAnders Carlsson 4293eb55cfeSKen Dyck return llvm::ConstantInt::get(SizeTy, AllocSize.getQuantity()); 43059486a2dSAnders Carlsson } 43159486a2dSAnders Carlsson 43259486a2dSAnders Carlsson // Emit the array size expression. 43359486a2dSAnders Carlsson NumElements = CGF.EmitScalarExpr(E->getArraySize()); 43459486a2dSAnders Carlsson 43559486a2dSAnders Carlsson // Multiply with the type size. 43659486a2dSAnders Carlsson llvm::Value *V = 43759486a2dSAnders Carlsson CGF.Builder.CreateMul(NumElements, 4383eb55cfeSKen Dyck llvm::ConstantInt::get(SizeTy, 4393eb55cfeSKen Dyck TypeSize.getQuantity())); 44059486a2dSAnders Carlsson 44147b4629bSFariborz Jahanian while (const ArrayType *AType = Context.getAsArrayType(Type)) { 44247b4629bSFariborz Jahanian const llvm::ArrayType *llvmAType = 44347b4629bSFariborz Jahanian cast<llvm::ArrayType>(CGF.ConvertType(Type)); 44447b4629bSFariborz Jahanian NumElements = 44547b4629bSFariborz Jahanian CGF.Builder.CreateMul(NumElements, 44647b4629bSFariborz Jahanian llvm::ConstantInt::get( 44747b4629bSFariborz Jahanian SizeTy, llvmAType->getNumElements())); 44847b4629bSFariborz Jahanian Type = AType->getElementType(); 44947b4629bSFariborz Jahanian } 45047b4629bSFariborz Jahanian 45159486a2dSAnders Carlsson // And add the cookie padding if necessary. 4523eb55cfeSKen Dyck if (!CookiePadding.isZero()) 4533eb55cfeSKen Dyck V = CGF.Builder.CreateAdd(V, 4543eb55cfeSKen Dyck llvm::ConstantInt::get(SizeTy, CookiePadding.getQuantity())); 45559486a2dSAnders Carlsson 45659486a2dSAnders Carlsson return V; 45759486a2dSAnders Carlsson } 45859486a2dSAnders Carlsson 45959486a2dSAnders Carlsson static void EmitNewInitializer(CodeGenFunction &CGF, const CXXNewExpr *E, 46059486a2dSAnders Carlsson llvm::Value *NewPtr, 46159486a2dSAnders Carlsson llvm::Value *NumElements) { 4623a202f60SAnders Carlsson if (E->isArray()) { 4633a202f60SAnders Carlsson if (CXXConstructorDecl *Ctor = E->getConstructor()) 4643a202f60SAnders Carlsson CGF.EmitCXXAggrConstructorCall(Ctor, NumElements, NewPtr, 4653a202f60SAnders Carlsson E->constructor_arg_begin(), 4663a202f60SAnders Carlsson E->constructor_arg_end()); 4673a202f60SAnders Carlsson return; 4683a202f60SAnders Carlsson } 4693a202f60SAnders Carlsson 47059486a2dSAnders Carlsson QualType AllocType = E->getAllocatedType(); 47159486a2dSAnders Carlsson 47259486a2dSAnders Carlsson if (CXXConstructorDecl *Ctor = E->getConstructor()) { 47359486a2dSAnders Carlsson CGF.EmitCXXConstructorCall(Ctor, Ctor_Complete, NewPtr, 47459486a2dSAnders Carlsson E->constructor_arg_begin(), 47559486a2dSAnders Carlsson E->constructor_arg_end()); 47659486a2dSAnders Carlsson 47759486a2dSAnders Carlsson return; 47859486a2dSAnders Carlsson } 47959486a2dSAnders Carlsson 48059486a2dSAnders Carlsson // We have a POD type. 48159486a2dSAnders Carlsson if (E->getNumConstructorArgs() == 0) 48259486a2dSAnders Carlsson return; 48359486a2dSAnders Carlsson 48459486a2dSAnders Carlsson assert(E->getNumConstructorArgs() == 1 && 48559486a2dSAnders Carlsson "Can only have one argument to initializer of POD type."); 48659486a2dSAnders Carlsson 48759486a2dSAnders Carlsson const Expr *Init = E->getConstructorArg(0); 48859486a2dSAnders Carlsson 48959486a2dSAnders Carlsson if (!CGF.hasAggregateLLVMType(AllocType)) 49059486a2dSAnders Carlsson CGF.EmitStoreOfScalar(CGF.EmitScalarExpr(Init), NewPtr, 49159486a2dSAnders Carlsson AllocType.isVolatileQualified(), AllocType); 49259486a2dSAnders Carlsson else if (AllocType->isAnyComplexType()) 49359486a2dSAnders Carlsson CGF.EmitComplexExprIntoAddr(Init, NewPtr, 49459486a2dSAnders Carlsson AllocType.isVolatileQualified()); 49559486a2dSAnders Carlsson else 49659486a2dSAnders Carlsson CGF.EmitAggExpr(Init, NewPtr, AllocType.isVolatileQualified()); 49759486a2dSAnders Carlsson } 49859486a2dSAnders Carlsson 49959486a2dSAnders Carlsson llvm::Value *CodeGenFunction::EmitCXXNewExpr(const CXXNewExpr *E) { 50059486a2dSAnders Carlsson QualType AllocType = E->getAllocatedType(); 50159486a2dSAnders Carlsson FunctionDecl *NewFD = E->getOperatorNew(); 50259486a2dSAnders Carlsson const FunctionProtoType *NewFTy = NewFD->getType()->getAs<FunctionProtoType>(); 50359486a2dSAnders Carlsson 50459486a2dSAnders Carlsson CallArgList NewArgs; 50559486a2dSAnders Carlsson 50659486a2dSAnders Carlsson // The allocation size is the first argument. 50759486a2dSAnders Carlsson QualType SizeTy = getContext().getSizeType(); 50859486a2dSAnders Carlsson 50959486a2dSAnders Carlsson llvm::Value *NumElements = 0; 51047b4629bSFariborz Jahanian llvm::Value *AllocSize = EmitCXXNewAllocSize(getContext(), 51147b4629bSFariborz Jahanian *this, E, NumElements); 51259486a2dSAnders Carlsson 51359486a2dSAnders Carlsson NewArgs.push_back(std::make_pair(RValue::get(AllocSize), SizeTy)); 51459486a2dSAnders Carlsson 51559486a2dSAnders Carlsson // Emit the rest of the arguments. 51659486a2dSAnders Carlsson // FIXME: Ideally, this should just use EmitCallArgs. 51759486a2dSAnders Carlsson CXXNewExpr::const_arg_iterator NewArg = E->placement_arg_begin(); 51859486a2dSAnders Carlsson 51959486a2dSAnders Carlsson // First, use the types from the function type. 52059486a2dSAnders Carlsson // We start at 1 here because the first argument (the allocation size) 52159486a2dSAnders Carlsson // has already been emitted. 52259486a2dSAnders Carlsson for (unsigned i = 1, e = NewFTy->getNumArgs(); i != e; ++i, ++NewArg) { 52359486a2dSAnders Carlsson QualType ArgType = NewFTy->getArgType(i); 52459486a2dSAnders Carlsson 52559486a2dSAnders Carlsson assert(getContext().getCanonicalType(ArgType.getNonReferenceType()). 52659486a2dSAnders Carlsson getTypePtr() == 52759486a2dSAnders Carlsson getContext().getCanonicalType(NewArg->getType()).getTypePtr() && 52859486a2dSAnders Carlsson "type mismatch in call argument!"); 52959486a2dSAnders Carlsson 53059486a2dSAnders Carlsson NewArgs.push_back(std::make_pair(EmitCallArg(*NewArg, ArgType), 53159486a2dSAnders Carlsson ArgType)); 53259486a2dSAnders Carlsson 53359486a2dSAnders Carlsson } 53459486a2dSAnders Carlsson 53559486a2dSAnders Carlsson // Either we've emitted all the call args, or we have a call to a 53659486a2dSAnders Carlsson // variadic function. 53759486a2dSAnders Carlsson assert((NewArg == E->placement_arg_end() || NewFTy->isVariadic()) && 53859486a2dSAnders Carlsson "Extra arguments in non-variadic function!"); 53959486a2dSAnders Carlsson 54059486a2dSAnders Carlsson // If we still have any arguments, emit them using the type of the argument. 54159486a2dSAnders Carlsson for (CXXNewExpr::const_arg_iterator NewArgEnd = E->placement_arg_end(); 54259486a2dSAnders Carlsson NewArg != NewArgEnd; ++NewArg) { 54359486a2dSAnders Carlsson QualType ArgType = NewArg->getType(); 54459486a2dSAnders Carlsson NewArgs.push_back(std::make_pair(EmitCallArg(*NewArg, ArgType), 54559486a2dSAnders Carlsson ArgType)); 54659486a2dSAnders Carlsson } 54759486a2dSAnders Carlsson 54859486a2dSAnders Carlsson // Emit the call to new. 54959486a2dSAnders Carlsson RValue RV = 550ab26cfa5SJohn McCall EmitCall(CGM.getTypes().getFunctionInfo(NewArgs, NewFTy), 55161a401caSAnders Carlsson CGM.GetAddrOfFunction(NewFD), ReturnValueSlot(), NewArgs, NewFD); 55259486a2dSAnders Carlsson 55359486a2dSAnders Carlsson // If an allocation function is declared with an empty exception specification 55459486a2dSAnders Carlsson // it returns null to indicate failure to allocate storage. [expr.new]p13. 55559486a2dSAnders Carlsson // (We don't need to check for null when there's no new initializer and 55659486a2dSAnders Carlsson // we're allocating a POD type). 55759486a2dSAnders Carlsson bool NullCheckResult = NewFTy->hasEmptyExceptionSpec() && 55859486a2dSAnders Carlsson !(AllocType->isPODType() && !E->hasInitializer()); 55959486a2dSAnders Carlsson 56059486a2dSAnders Carlsson llvm::BasicBlock *NewNull = 0; 56159486a2dSAnders Carlsson llvm::BasicBlock *NewNotNull = 0; 56259486a2dSAnders Carlsson llvm::BasicBlock *NewEnd = 0; 56359486a2dSAnders Carlsson 56459486a2dSAnders Carlsson llvm::Value *NewPtr = RV.getScalarVal(); 56559486a2dSAnders Carlsson 56659486a2dSAnders Carlsson if (NullCheckResult) { 56759486a2dSAnders Carlsson NewNull = createBasicBlock("new.null"); 56859486a2dSAnders Carlsson NewNotNull = createBasicBlock("new.notnull"); 56959486a2dSAnders Carlsson NewEnd = createBasicBlock("new.end"); 57059486a2dSAnders Carlsson 57159486a2dSAnders Carlsson llvm::Value *IsNull = 57259486a2dSAnders Carlsson Builder.CreateICmpEQ(NewPtr, 57359486a2dSAnders Carlsson llvm::Constant::getNullValue(NewPtr->getType()), 57459486a2dSAnders Carlsson "isnull"); 57559486a2dSAnders Carlsson 57659486a2dSAnders Carlsson Builder.CreateCondBr(IsNull, NewNull, NewNotNull); 57759486a2dSAnders Carlsson EmitBlock(NewNotNull); 57859486a2dSAnders Carlsson } 57959486a2dSAnders Carlsson 5803eb55cfeSKen Dyck CharUnits CookiePadding = CalculateCookiePadding(getContext(), E); 5813eb55cfeSKen Dyck if (!CookiePadding.isZero()) { 5823eb55cfeSKen Dyck CharUnits CookieOffset = 5833eb55cfeSKen Dyck CookiePadding - getContext().getTypeSizeInChars(SizeTy); 58459486a2dSAnders Carlsson 58559486a2dSAnders Carlsson llvm::Value *NumElementsPtr = 5863eb55cfeSKen Dyck Builder.CreateConstInBoundsGEP1_64(NewPtr, CookieOffset.getQuantity()); 58759486a2dSAnders Carlsson 58859486a2dSAnders Carlsson NumElementsPtr = Builder.CreateBitCast(NumElementsPtr, 58959486a2dSAnders Carlsson ConvertType(SizeTy)->getPointerTo()); 59059486a2dSAnders Carlsson Builder.CreateStore(NumElements, NumElementsPtr); 59159486a2dSAnders Carlsson 59259486a2dSAnders Carlsson // Now add the padding to the new ptr. 5933eb55cfeSKen Dyck NewPtr = Builder.CreateConstInBoundsGEP1_64(NewPtr, 5943eb55cfeSKen Dyck CookiePadding.getQuantity()); 59559486a2dSAnders Carlsson } 59659486a2dSAnders Carlsson 59747b4629bSFariborz Jahanian if (AllocType->isArrayType()) { 59847b4629bSFariborz Jahanian while (const ArrayType *AType = getContext().getAsArrayType(AllocType)) 59947b4629bSFariborz Jahanian AllocType = AType->getElementType(); 60047b4629bSFariborz Jahanian NewPtr = 60147b4629bSFariborz Jahanian Builder.CreateBitCast(NewPtr, 60247b4629bSFariborz Jahanian ConvertType(getContext().getPointerType(AllocType))); 60359486a2dSAnders Carlsson EmitNewInitializer(*this, E, NewPtr, NumElements); 60447b4629bSFariborz Jahanian NewPtr = Builder.CreateBitCast(NewPtr, ConvertType(E->getType())); 60547b4629bSFariborz Jahanian } 60647b4629bSFariborz Jahanian else { 60747b4629bSFariborz Jahanian NewPtr = Builder.CreateBitCast(NewPtr, ConvertType(E->getType())); 60847b4629bSFariborz Jahanian EmitNewInitializer(*this, E, NewPtr, NumElements); 60947b4629bSFariborz Jahanian } 61059486a2dSAnders Carlsson 61159486a2dSAnders Carlsson if (NullCheckResult) { 61259486a2dSAnders Carlsson Builder.CreateBr(NewEnd); 61359486a2dSAnders Carlsson NewNotNull = Builder.GetInsertBlock(); 61459486a2dSAnders Carlsson EmitBlock(NewNull); 61559486a2dSAnders Carlsson Builder.CreateBr(NewEnd); 61659486a2dSAnders Carlsson EmitBlock(NewEnd); 61759486a2dSAnders Carlsson 61859486a2dSAnders Carlsson llvm::PHINode *PHI = Builder.CreatePHI(NewPtr->getType()); 61959486a2dSAnders Carlsson PHI->reserveOperandSpace(2); 62059486a2dSAnders Carlsson PHI->addIncoming(NewPtr, NewNotNull); 62159486a2dSAnders Carlsson PHI->addIncoming(llvm::Constant::getNullValue(NewPtr->getType()), NewNull); 62259486a2dSAnders Carlsson 62359486a2dSAnders Carlsson NewPtr = PHI; 62459486a2dSAnders Carlsson } 62559486a2dSAnders Carlsson 62659486a2dSAnders Carlsson return NewPtr; 62759486a2dSAnders Carlsson } 62859486a2dSAnders Carlsson 62921122cf6SAnders Carlsson static std::pair<llvm::Value *, llvm::Value *> 63021122cf6SAnders Carlsson GetAllocatedObjectPtrAndNumElements(CodeGenFunction &CGF, 63121122cf6SAnders Carlsson llvm::Value *Ptr, QualType DeleteTy) { 63221122cf6SAnders Carlsson QualType SizeTy = CGF.getContext().getSizeType(); 63321122cf6SAnders Carlsson const llvm::Type *SizeLTy = CGF.ConvertType(SizeTy); 63421122cf6SAnders Carlsson 6353eb55cfeSKen Dyck CharUnits DeleteTypeAlign = CGF.getContext().getTypeAlignInChars(DeleteTy); 6363eb55cfeSKen Dyck CharUnits CookiePadding = 6373eb55cfeSKen Dyck std::max(CGF.getContext().getTypeSizeInChars(SizeTy), 6383eb55cfeSKen Dyck DeleteTypeAlign); 6393eb55cfeSKen Dyck assert(!CookiePadding.isZero() && "CookiePadding should not be 0."); 64021122cf6SAnders Carlsson 64121122cf6SAnders Carlsson const llvm::Type *Int8PtrTy = llvm::Type::getInt8PtrTy(CGF.getLLVMContext()); 6423eb55cfeSKen Dyck CharUnits CookieOffset = 6433eb55cfeSKen Dyck CookiePadding - CGF.getContext().getTypeSizeInChars(SizeTy); 64421122cf6SAnders Carlsson 64521122cf6SAnders Carlsson llvm::Value *AllocatedObjectPtr = CGF.Builder.CreateBitCast(Ptr, Int8PtrTy); 64621122cf6SAnders Carlsson AllocatedObjectPtr = 64721122cf6SAnders Carlsson CGF.Builder.CreateConstInBoundsGEP1_64(AllocatedObjectPtr, 6483eb55cfeSKen Dyck -CookiePadding.getQuantity()); 64921122cf6SAnders Carlsson 65021122cf6SAnders Carlsson llvm::Value *NumElementsPtr = 65121122cf6SAnders Carlsson CGF.Builder.CreateConstInBoundsGEP1_64(AllocatedObjectPtr, 6523eb55cfeSKen Dyck CookieOffset.getQuantity()); 65321122cf6SAnders Carlsson NumElementsPtr = 65421122cf6SAnders Carlsson CGF.Builder.CreateBitCast(NumElementsPtr, SizeLTy->getPointerTo()); 65521122cf6SAnders Carlsson 65621122cf6SAnders Carlsson llvm::Value *NumElements = CGF.Builder.CreateLoad(NumElementsPtr); 65721122cf6SAnders Carlsson NumElements = 65821122cf6SAnders Carlsson CGF.Builder.CreateIntCast(NumElements, SizeLTy, /*isSigned=*/false); 65921122cf6SAnders Carlsson 66021122cf6SAnders Carlsson return std::make_pair(AllocatedObjectPtr, NumElements); 66121122cf6SAnders Carlsson } 66221122cf6SAnders Carlsson 66359486a2dSAnders Carlsson void CodeGenFunction::EmitDeleteCall(const FunctionDecl *DeleteFD, 66459486a2dSAnders Carlsson llvm::Value *Ptr, 66559486a2dSAnders Carlsson QualType DeleteTy) { 66659486a2dSAnders Carlsson const FunctionProtoType *DeleteFTy = 66759486a2dSAnders Carlsson DeleteFD->getType()->getAs<FunctionProtoType>(); 66859486a2dSAnders Carlsson 66959486a2dSAnders Carlsson CallArgList DeleteArgs; 67059486a2dSAnders Carlsson 67121122cf6SAnders Carlsson // Check if we need to pass the size to the delete operator. 67221122cf6SAnders Carlsson llvm::Value *Size = 0; 67321122cf6SAnders Carlsson QualType SizeTy; 67421122cf6SAnders Carlsson if (DeleteFTy->getNumArgs() == 2) { 67521122cf6SAnders Carlsson SizeTy = DeleteFTy->getArgType(1); 6767df3cbebSKen Dyck CharUnits DeleteTypeSize = getContext().getTypeSizeInChars(DeleteTy); 6777df3cbebSKen Dyck Size = llvm::ConstantInt::get(ConvertType(SizeTy), 6787df3cbebSKen Dyck DeleteTypeSize.getQuantity()); 67921122cf6SAnders Carlsson } 68021122cf6SAnders Carlsson 68121122cf6SAnders Carlsson if (DeleteFD->getOverloadedOperator() == OO_Array_Delete && 6823eb55cfeSKen Dyck !CalculateCookiePadding(getContext(), DeleteTy).isZero()) { 68321122cf6SAnders Carlsson // We need to get the number of elements in the array from the cookie. 68421122cf6SAnders Carlsson llvm::Value *AllocatedObjectPtr; 68521122cf6SAnders Carlsson llvm::Value *NumElements; 68621122cf6SAnders Carlsson llvm::tie(AllocatedObjectPtr, NumElements) = 68721122cf6SAnders Carlsson GetAllocatedObjectPtrAndNumElements(*this, Ptr, DeleteTy); 68821122cf6SAnders Carlsson 68921122cf6SAnders Carlsson // Multiply the size with the number of elements. 69021122cf6SAnders Carlsson if (Size) 69121122cf6SAnders Carlsson Size = Builder.CreateMul(NumElements, Size); 69221122cf6SAnders Carlsson 69321122cf6SAnders Carlsson Ptr = AllocatedObjectPtr; 69421122cf6SAnders Carlsson } 69521122cf6SAnders Carlsson 69659486a2dSAnders Carlsson QualType ArgTy = DeleteFTy->getArgType(0); 69759486a2dSAnders Carlsson llvm::Value *DeletePtr = Builder.CreateBitCast(Ptr, ConvertType(ArgTy)); 69859486a2dSAnders Carlsson DeleteArgs.push_back(std::make_pair(RValue::get(DeletePtr), ArgTy)); 69959486a2dSAnders Carlsson 70021122cf6SAnders Carlsson if (Size) 70159486a2dSAnders Carlsson DeleteArgs.push_back(std::make_pair(RValue::get(Size), SizeTy)); 70259486a2dSAnders Carlsson 70359486a2dSAnders Carlsson // Emit the call to delete. 704ab26cfa5SJohn McCall EmitCall(CGM.getTypes().getFunctionInfo(DeleteArgs, DeleteFTy), 70561a401caSAnders Carlsson CGM.GetAddrOfFunction(DeleteFD), ReturnValueSlot(), 70659486a2dSAnders Carlsson DeleteArgs, DeleteFD); 70759486a2dSAnders Carlsson } 70859486a2dSAnders Carlsson 70959486a2dSAnders Carlsson void CodeGenFunction::EmitCXXDeleteExpr(const CXXDeleteExpr *E) { 71059486a2dSAnders Carlsson 71159486a2dSAnders Carlsson // Get at the argument before we performed the implicit conversion 71259486a2dSAnders Carlsson // to void*. 71359486a2dSAnders Carlsson const Expr *Arg = E->getArgument(); 71459486a2dSAnders Carlsson while (const ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(Arg)) { 71559486a2dSAnders Carlsson if (ICE->getCastKind() != CastExpr::CK_UserDefinedConversion && 71659486a2dSAnders Carlsson ICE->getType()->isVoidPointerType()) 71759486a2dSAnders Carlsson Arg = ICE->getSubExpr(); 71859486a2dSAnders Carlsson else 71959486a2dSAnders Carlsson break; 72059486a2dSAnders Carlsson } 72159486a2dSAnders Carlsson 72259486a2dSAnders Carlsson QualType DeleteTy = Arg->getType()->getAs<PointerType>()->getPointeeType(); 72359486a2dSAnders Carlsson 72459486a2dSAnders Carlsson llvm::Value *Ptr = EmitScalarExpr(Arg); 72559486a2dSAnders Carlsson 72659486a2dSAnders Carlsson // Null check the pointer. 72759486a2dSAnders Carlsson llvm::BasicBlock *DeleteNotNull = createBasicBlock("delete.notnull"); 72859486a2dSAnders Carlsson llvm::BasicBlock *DeleteEnd = createBasicBlock("delete.end"); 72959486a2dSAnders Carlsson 73059486a2dSAnders Carlsson llvm::Value *IsNull = 73159486a2dSAnders Carlsson Builder.CreateICmpEQ(Ptr, llvm::Constant::getNullValue(Ptr->getType()), 73259486a2dSAnders Carlsson "isnull"); 73359486a2dSAnders Carlsson 73459486a2dSAnders Carlsson Builder.CreateCondBr(IsNull, DeleteEnd, DeleteNotNull); 73559486a2dSAnders Carlsson EmitBlock(DeleteNotNull); 73659486a2dSAnders Carlsson 73759486a2dSAnders Carlsson bool ShouldCallDelete = true; 73859486a2dSAnders Carlsson 73959486a2dSAnders Carlsson // Call the destructor if necessary. 74059486a2dSAnders Carlsson if (const RecordType *RT = DeleteTy->getAs<RecordType>()) { 74159486a2dSAnders Carlsson if (CXXRecordDecl *RD = dyn_cast<CXXRecordDecl>(RT->getDecl())) { 74259486a2dSAnders Carlsson if (!RD->hasTrivialDestructor()) { 74359486a2dSAnders Carlsson const CXXDestructorDecl *Dtor = RD->getDestructor(getContext()); 74459486a2dSAnders Carlsson if (E->isArrayForm()) { 74521122cf6SAnders Carlsson llvm::Value *AllocatedObjectPtr; 74621122cf6SAnders Carlsson llvm::Value *NumElements; 74721122cf6SAnders Carlsson llvm::tie(AllocatedObjectPtr, NumElements) = 74821122cf6SAnders Carlsson GetAllocatedObjectPtrAndNumElements(*this, Ptr, DeleteTy); 749f5af8deaSAnders Carlsson 75059486a2dSAnders Carlsson EmitCXXAggrDestructorCall(Dtor, NumElements, Ptr); 751f5af8deaSAnders Carlsson } else if (Dtor->isVirtual()) { 75259486a2dSAnders Carlsson const llvm::Type *Ty = 75359486a2dSAnders Carlsson CGM.getTypes().GetFunctionType(CGM.getTypes().getFunctionInfo(Dtor), 75459486a2dSAnders Carlsson /*isVariadic=*/false); 75559486a2dSAnders Carlsson 75659486a2dSAnders Carlsson llvm::Value *Callee = BuildVirtualCall(Dtor, Dtor_Deleting, Ptr, Ty); 757e36a6b3eSAnders Carlsson EmitCXXMemberCall(Dtor, Callee, ReturnValueSlot(), Ptr, /*VTT=*/0, 758e36a6b3eSAnders Carlsson 0, 0); 75959486a2dSAnders Carlsson 76059486a2dSAnders Carlsson // The dtor took care of deleting the object. 76159486a2dSAnders Carlsson ShouldCallDelete = false; 76259486a2dSAnders Carlsson } else 76359486a2dSAnders Carlsson EmitCXXDestructorCall(Dtor, Dtor_Complete, Ptr); 76459486a2dSAnders Carlsson } 76559486a2dSAnders Carlsson } 76659486a2dSAnders Carlsson } 76759486a2dSAnders Carlsson 76859486a2dSAnders Carlsson if (ShouldCallDelete) 76959486a2dSAnders Carlsson EmitDeleteCall(E->getOperatorDelete(), Ptr, DeleteTy); 77059486a2dSAnders Carlsson 77159486a2dSAnders Carlsson EmitBlock(DeleteEnd); 77259486a2dSAnders Carlsson } 77359486a2dSAnders Carlsson 77459486a2dSAnders Carlsson llvm::Value * CodeGenFunction::EmitCXXTypeidExpr(const CXXTypeidExpr *E) { 77559486a2dSAnders Carlsson QualType Ty = E->getType(); 77659486a2dSAnders Carlsson const llvm::Type *LTy = ConvertType(Ty)->getPointerTo(); 777fd7dfeb7SAnders Carlsson 7783f4336cbSAnders Carlsson if (E->isTypeOperand()) { 7793f4336cbSAnders Carlsson llvm::Constant *TypeInfo = 7803f4336cbSAnders Carlsson CGM.GetAddrOfRTTIDescriptor(E->getTypeOperand()); 7813f4336cbSAnders Carlsson return Builder.CreateBitCast(TypeInfo, LTy); 7823f4336cbSAnders Carlsson } 783fd7dfeb7SAnders Carlsson 78459486a2dSAnders Carlsson Expr *subE = E->getExprOperand(); 78559486a2dSAnders Carlsson Ty = subE->getType(); 78659486a2dSAnders Carlsson CanQualType CanTy = CGM.getContext().getCanonicalType(Ty); 78759486a2dSAnders Carlsson Ty = CanTy.getUnqualifiedType().getNonReferenceType(); 78859486a2dSAnders Carlsson if (const RecordType *RT = Ty->getAs<RecordType>()) { 78959486a2dSAnders Carlsson const CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl()); 79059486a2dSAnders Carlsson if (RD->isPolymorphic()) { 79159486a2dSAnders Carlsson // FIXME: if subE is an lvalue do 79259486a2dSAnders Carlsson LValue Obj = EmitLValue(subE); 79359486a2dSAnders Carlsson llvm::Value *This = Obj.getAddress(); 79459486a2dSAnders Carlsson LTy = LTy->getPointerTo()->getPointerTo(); 79559486a2dSAnders Carlsson llvm::Value *V = Builder.CreateBitCast(This, LTy); 79659486a2dSAnders Carlsson // We need to do a zero check for *p, unless it has NonNullAttr. 79759486a2dSAnders Carlsson // FIXME: PointerType->hasAttr<NonNullAttr>() 79859486a2dSAnders Carlsson bool CanBeZero = false; 79959486a2dSAnders Carlsson if (UnaryOperator *UO = dyn_cast<UnaryOperator>(subE->IgnoreParens())) 80059486a2dSAnders Carlsson if (UO->getOpcode() == UnaryOperator::Deref) 80159486a2dSAnders Carlsson CanBeZero = true; 80259486a2dSAnders Carlsson if (CanBeZero) { 80359486a2dSAnders Carlsson llvm::BasicBlock *NonZeroBlock = createBasicBlock(); 80459486a2dSAnders Carlsson llvm::BasicBlock *ZeroBlock = createBasicBlock(); 80559486a2dSAnders Carlsson 80659486a2dSAnders Carlsson llvm::Value *Zero = llvm::Constant::getNullValue(LTy); 80759486a2dSAnders Carlsson Builder.CreateCondBr(Builder.CreateICmpNE(V, Zero), 80859486a2dSAnders Carlsson NonZeroBlock, ZeroBlock); 80959486a2dSAnders Carlsson EmitBlock(ZeroBlock); 81059486a2dSAnders Carlsson /// Call __cxa_bad_typeid 81159486a2dSAnders Carlsson const llvm::Type *ResultType = llvm::Type::getVoidTy(VMContext); 81259486a2dSAnders Carlsson const llvm::FunctionType *FTy; 81359486a2dSAnders Carlsson FTy = llvm::FunctionType::get(ResultType, false); 81459486a2dSAnders Carlsson llvm::Value *F = CGM.CreateRuntimeFunction(FTy, "__cxa_bad_typeid"); 81559486a2dSAnders Carlsson Builder.CreateCall(F)->setDoesNotReturn(); 81659486a2dSAnders Carlsson Builder.CreateUnreachable(); 81759486a2dSAnders Carlsson EmitBlock(NonZeroBlock); 81859486a2dSAnders Carlsson } 81959486a2dSAnders Carlsson V = Builder.CreateLoad(V, "vtable"); 82059486a2dSAnders Carlsson V = Builder.CreateConstInBoundsGEP1_64(V, -1ULL); 82159486a2dSAnders Carlsson V = Builder.CreateLoad(V); 82259486a2dSAnders Carlsson return V; 82359486a2dSAnders Carlsson } 82459486a2dSAnders Carlsson } 8253f4336cbSAnders Carlsson return Builder.CreateBitCast(CGM.GetAddrOfRTTIDescriptor(Ty), LTy); 82659486a2dSAnders Carlsson } 82759486a2dSAnders Carlsson 82859486a2dSAnders Carlsson llvm::Value *CodeGenFunction::EmitDynamicCast(llvm::Value *V, 82959486a2dSAnders Carlsson const CXXDynamicCastExpr *DCE) { 8303f4336cbSAnders Carlsson QualType SrcTy = DCE->getSubExpr()->getType(); 8313f4336cbSAnders Carlsson QualType DestTy = DCE->getTypeAsWritten(); 8323f4336cbSAnders Carlsson QualType InnerType = DestTy->getPointeeType(); 8333f4336cbSAnders Carlsson 83459486a2dSAnders Carlsson const llvm::Type *LTy = ConvertType(DCE->getType()); 83559486a2dSAnders Carlsson 83659486a2dSAnders Carlsson bool CanBeZero = false; 83759486a2dSAnders Carlsson bool ToVoid = false; 83859486a2dSAnders Carlsson bool ThrowOnBad = false; 8393f4336cbSAnders Carlsson if (DestTy->isPointerType()) { 84059486a2dSAnders Carlsson // FIXME: if PointerType->hasAttr<NonNullAttr>(), we don't set this 84159486a2dSAnders Carlsson CanBeZero = true; 84259486a2dSAnders Carlsson if (InnerType->isVoidType()) 84359486a2dSAnders Carlsson ToVoid = true; 84459486a2dSAnders Carlsson } else { 84559486a2dSAnders Carlsson LTy = LTy->getPointerTo(); 84659486a2dSAnders Carlsson ThrowOnBad = true; 84759486a2dSAnders Carlsson } 84859486a2dSAnders Carlsson 8493f4336cbSAnders Carlsson if (SrcTy->isPointerType() || SrcTy->isReferenceType()) 8503f4336cbSAnders Carlsson SrcTy = SrcTy->getPointeeType(); 8513f4336cbSAnders Carlsson SrcTy = SrcTy.getUnqualifiedType(); 8523f4336cbSAnders Carlsson 8530087bc85SAnders Carlsson if (DestTy->isPointerType() || DestTy->isReferenceType()) 8543f4336cbSAnders Carlsson DestTy = DestTy->getPointeeType(); 8553f4336cbSAnders Carlsson DestTy = DestTy.getUnqualifiedType(); 85659486a2dSAnders Carlsson 85759486a2dSAnders Carlsson llvm::BasicBlock *ContBlock = createBasicBlock(); 85859486a2dSAnders Carlsson llvm::BasicBlock *NullBlock = 0; 85959486a2dSAnders Carlsson llvm::BasicBlock *NonZeroBlock = 0; 86059486a2dSAnders Carlsson if (CanBeZero) { 86159486a2dSAnders Carlsson NonZeroBlock = createBasicBlock(); 86259486a2dSAnders Carlsson NullBlock = createBasicBlock(); 8633f4336cbSAnders Carlsson Builder.CreateCondBr(Builder.CreateIsNotNull(V), NonZeroBlock, NullBlock); 86459486a2dSAnders Carlsson EmitBlock(NonZeroBlock); 86559486a2dSAnders Carlsson } 86659486a2dSAnders Carlsson 86759486a2dSAnders Carlsson llvm::BasicBlock *BadCastBlock = 0; 86859486a2dSAnders Carlsson 8693f4336cbSAnders Carlsson const llvm::Type *PtrDiffTy = ConvertType(getContext().getPointerDiffType()); 87059486a2dSAnders Carlsson 87159486a2dSAnders Carlsson // See if this is a dynamic_cast(void*) 87259486a2dSAnders Carlsson if (ToVoid) { 87359486a2dSAnders Carlsson llvm::Value *This = V; 87459486a2dSAnders Carlsson V = Builder.CreateBitCast(This, PtrDiffTy->getPointerTo()->getPointerTo()); 87559486a2dSAnders Carlsson V = Builder.CreateLoad(V, "vtable"); 87659486a2dSAnders Carlsson V = Builder.CreateConstInBoundsGEP1_64(V, -2ULL); 87759486a2dSAnders Carlsson V = Builder.CreateLoad(V, "offset to top"); 87859486a2dSAnders Carlsson This = Builder.CreateBitCast(This, llvm::Type::getInt8PtrTy(VMContext)); 87959486a2dSAnders Carlsson V = Builder.CreateInBoundsGEP(This, V); 88059486a2dSAnders Carlsson V = Builder.CreateBitCast(V, LTy); 88159486a2dSAnders Carlsson } else { 88259486a2dSAnders Carlsson /// Call __dynamic_cast 88359486a2dSAnders Carlsson const llvm::Type *ResultType = llvm::Type::getInt8PtrTy(VMContext); 88459486a2dSAnders Carlsson const llvm::FunctionType *FTy; 88559486a2dSAnders Carlsson std::vector<const llvm::Type*> ArgTys; 88659486a2dSAnders Carlsson const llvm::Type *PtrToInt8Ty 88759486a2dSAnders Carlsson = llvm::Type::getInt8Ty(VMContext)->getPointerTo(); 88859486a2dSAnders Carlsson ArgTys.push_back(PtrToInt8Ty); 88959486a2dSAnders Carlsson ArgTys.push_back(PtrToInt8Ty); 89059486a2dSAnders Carlsson ArgTys.push_back(PtrToInt8Ty); 89159486a2dSAnders Carlsson ArgTys.push_back(PtrDiffTy); 89259486a2dSAnders Carlsson FTy = llvm::FunctionType::get(ResultType, ArgTys, false); 89359486a2dSAnders Carlsson 89459486a2dSAnders Carlsson // FIXME: Calculate better hint. 89559486a2dSAnders Carlsson llvm::Value *hint = llvm::ConstantInt::get(PtrDiffTy, -1ULL); 8963f4336cbSAnders Carlsson 8973f4336cbSAnders Carlsson assert(SrcTy->isRecordType() && "Src type must be record type!"); 8983f4336cbSAnders Carlsson assert(DestTy->isRecordType() && "Dest type must be record type!"); 8993f4336cbSAnders Carlsson 900247894b3SDouglas Gregor llvm::Value *SrcArg 901247894b3SDouglas Gregor = CGM.GetAddrOfRTTIDescriptor(SrcTy.getUnqualifiedType()); 902247894b3SDouglas Gregor llvm::Value *DestArg 903247894b3SDouglas Gregor = CGM.GetAddrOfRTTIDescriptor(DestTy.getUnqualifiedType()); 9043f4336cbSAnders Carlsson 90559486a2dSAnders Carlsson V = Builder.CreateBitCast(V, PtrToInt8Ty); 90659486a2dSAnders Carlsson V = Builder.CreateCall4(CGM.CreateRuntimeFunction(FTy, "__dynamic_cast"), 9073f4336cbSAnders Carlsson V, SrcArg, DestArg, hint); 90859486a2dSAnders Carlsson V = Builder.CreateBitCast(V, LTy); 90959486a2dSAnders Carlsson 91059486a2dSAnders Carlsson if (ThrowOnBad) { 91159486a2dSAnders Carlsson BadCastBlock = createBasicBlock(); 91259486a2dSAnders Carlsson 9133f4336cbSAnders Carlsson Builder.CreateCondBr(Builder.CreateIsNotNull(V), ContBlock, BadCastBlock); 91459486a2dSAnders Carlsson EmitBlock(BadCastBlock); 91559486a2dSAnders Carlsson /// Call __cxa_bad_cast 91659486a2dSAnders Carlsson ResultType = llvm::Type::getVoidTy(VMContext); 91759486a2dSAnders Carlsson const llvm::FunctionType *FBadTy; 91859486a2dSAnders Carlsson FBadTy = llvm::FunctionType::get(ResultType, false); 91959486a2dSAnders Carlsson llvm::Value *F = CGM.CreateRuntimeFunction(FBadTy, "__cxa_bad_cast"); 92059486a2dSAnders Carlsson Builder.CreateCall(F)->setDoesNotReturn(); 92159486a2dSAnders Carlsson Builder.CreateUnreachable(); 92259486a2dSAnders Carlsson } 92359486a2dSAnders Carlsson } 92459486a2dSAnders Carlsson 92559486a2dSAnders Carlsson if (CanBeZero) { 92659486a2dSAnders Carlsson Builder.CreateBr(ContBlock); 92759486a2dSAnders Carlsson EmitBlock(NullBlock); 92859486a2dSAnders Carlsson Builder.CreateBr(ContBlock); 92959486a2dSAnders Carlsson } 93059486a2dSAnders Carlsson EmitBlock(ContBlock); 93159486a2dSAnders Carlsson if (CanBeZero) { 93259486a2dSAnders Carlsson llvm::PHINode *PHI = Builder.CreatePHI(LTy); 93359486a2dSAnders Carlsson PHI->reserveOperandSpace(2); 93459486a2dSAnders Carlsson PHI->addIncoming(V, NonZeroBlock); 93559486a2dSAnders Carlsson PHI->addIncoming(llvm::Constant::getNullValue(LTy), NullBlock); 93659486a2dSAnders Carlsson V = PHI; 93759486a2dSAnders Carlsson } 93859486a2dSAnders Carlsson 93959486a2dSAnders Carlsson return V; 94059486a2dSAnders Carlsson } 941