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" 1560d215b6SFariborz Jahanian #include "CGObjCRuntime.h" 1626008e07SChris Lattner #include "llvm/Intrinsics.h" 1759486a2dSAnders Carlsson using namespace clang; 1859486a2dSAnders Carlsson using namespace CodeGen; 1959486a2dSAnders Carlsson 2027da15baSAnders Carlsson RValue CodeGenFunction::EmitCXXMemberCall(const CXXMethodDecl *MD, 2127da15baSAnders Carlsson llvm::Value *Callee, 2227da15baSAnders Carlsson ReturnValueSlot ReturnValue, 2327da15baSAnders Carlsson llvm::Value *This, 24e36a6b3eSAnders Carlsson llvm::Value *VTT, 2527da15baSAnders Carlsson CallExpr::const_arg_iterator ArgBeg, 2627da15baSAnders Carlsson CallExpr::const_arg_iterator ArgEnd) { 2727da15baSAnders Carlsson assert(MD->isInstance() && 2827da15baSAnders Carlsson "Trying to emit a member call expr on a static method!"); 2927da15baSAnders Carlsson 3027da15baSAnders Carlsson const FunctionProtoType *FPT = MD->getType()->getAs<FunctionProtoType>(); 3127da15baSAnders Carlsson 3227da15baSAnders Carlsson CallArgList Args; 3327da15baSAnders Carlsson 3427da15baSAnders Carlsson // Push the this ptr. 3527da15baSAnders Carlsson Args.push_back(std::make_pair(RValue::get(This), 3627da15baSAnders Carlsson MD->getThisType(getContext()))); 3727da15baSAnders Carlsson 38e36a6b3eSAnders Carlsson // If there is a VTT parameter, emit it. 39e36a6b3eSAnders Carlsson if (VTT) { 40e36a6b3eSAnders Carlsson QualType T = getContext().getPointerType(getContext().VoidPtrTy); 41e36a6b3eSAnders Carlsson Args.push_back(std::make_pair(RValue::get(VTT), T)); 42e36a6b3eSAnders Carlsson } 43e36a6b3eSAnders Carlsson 4427da15baSAnders Carlsson // And the rest of the call args 4527da15baSAnders Carlsson EmitCallArgs(Args, FPT, ArgBeg, ArgEnd); 4627da15baSAnders Carlsson 47ab26cfa5SJohn McCall QualType ResultType = FPT->getResultType(); 48ab26cfa5SJohn McCall return EmitCall(CGM.getTypes().getFunctionInfo(ResultType, Args, 49c50c27ccSRafael Espindola FPT->getExtInfo()), 50c50c27ccSRafael Espindola Callee, ReturnValue, Args, MD); 5127da15baSAnders Carlsson } 5227da15baSAnders Carlsson 5327da15baSAnders Carlsson /// canDevirtualizeMemberFunctionCalls - Checks whether virtual calls on given 5427da15baSAnders Carlsson /// expr can be devirtualized. 5527da15baSAnders Carlsson static bool canDevirtualizeMemberFunctionCalls(const Expr *Base) { 5627da15baSAnders Carlsson if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(Base)) { 5727da15baSAnders Carlsson if (const VarDecl *VD = dyn_cast<VarDecl>(DRE->getDecl())) { 5827da15baSAnders Carlsson // This is a record decl. We know the type and can devirtualize it. 5927da15baSAnders Carlsson return VD->getType()->isRecordType(); 6027da15baSAnders Carlsson } 6127da15baSAnders Carlsson 6227da15baSAnders Carlsson return false; 6327da15baSAnders Carlsson } 6427da15baSAnders Carlsson 6527da15baSAnders Carlsson // We can always devirtualize calls on temporary object expressions. 66a682427eSEli Friedman if (isa<CXXConstructExpr>(Base)) 6727da15baSAnders Carlsson return true; 6827da15baSAnders Carlsson 6927da15baSAnders Carlsson // And calls on bound temporaries. 7027da15baSAnders Carlsson if (isa<CXXBindTemporaryExpr>(Base)) 7127da15baSAnders Carlsson return true; 7227da15baSAnders Carlsson 7327da15baSAnders Carlsson // Check if this is a call expr that returns a record type. 7427da15baSAnders Carlsson if (const CallExpr *CE = dyn_cast<CallExpr>(Base)) 7527da15baSAnders Carlsson return CE->getCallReturnType()->isRecordType(); 7627da15baSAnders Carlsson 7727da15baSAnders Carlsson // We can't devirtualize the call. 7827da15baSAnders Carlsson return false; 7927da15baSAnders Carlsson } 8027da15baSAnders Carlsson 8127da15baSAnders Carlsson RValue CodeGenFunction::EmitCXXMemberCallExpr(const CXXMemberCallExpr *CE, 8227da15baSAnders Carlsson ReturnValueSlot ReturnValue) { 8327da15baSAnders Carlsson if (isa<BinaryOperator>(CE->getCallee()->IgnoreParens())) 8427da15baSAnders Carlsson return EmitCXXMemberPointerCallExpr(CE, ReturnValue); 8527da15baSAnders Carlsson 8627da15baSAnders Carlsson const MemberExpr *ME = cast<MemberExpr>(CE->getCallee()->IgnoreParens()); 8727da15baSAnders Carlsson const CXXMethodDecl *MD = cast<CXXMethodDecl>(ME->getMemberDecl()); 8827da15baSAnders Carlsson 8927da15baSAnders Carlsson if (MD->isStatic()) { 9027da15baSAnders Carlsson // The method is static, emit it as we would a regular call. 9127da15baSAnders Carlsson llvm::Value *Callee = CGM.GetAddrOfFunction(MD); 9227da15baSAnders Carlsson return EmitCall(getContext().getPointerType(MD->getType()), Callee, 9327da15baSAnders Carlsson ReturnValue, CE->arg_begin(), CE->arg_end()); 9427da15baSAnders Carlsson } 9527da15baSAnders Carlsson 9627da15baSAnders Carlsson const FunctionProtoType *FPT = MD->getType()->getAs<FunctionProtoType>(); 9727da15baSAnders Carlsson 9827da15baSAnders Carlsson const llvm::Type *Ty = 9927da15baSAnders Carlsson CGM.getTypes().GetFunctionType(CGM.getTypes().getFunctionInfo(MD), 10027da15baSAnders Carlsson FPT->isVariadic()); 10127da15baSAnders Carlsson llvm::Value *This; 10227da15baSAnders Carlsson 10327da15baSAnders Carlsson if (ME->isArrow()) 10427da15baSAnders Carlsson This = EmitScalarExpr(ME->getBase()); 10527da15baSAnders Carlsson else { 10627da15baSAnders Carlsson LValue BaseLV = EmitLValue(ME->getBase()); 10727da15baSAnders Carlsson This = BaseLV.getAddress(); 10827da15baSAnders Carlsson } 10927da15baSAnders Carlsson 11027da15baSAnders Carlsson if (MD->isCopyAssignment() && MD->isTrivial()) { 11127da15baSAnders Carlsson // We don't like to generate the trivial copy assignment operator when 11227da15baSAnders Carlsson // it isn't necessary; just produce the proper effect here. 11327da15baSAnders Carlsson llvm::Value *RHS = EmitLValue(*CE->arg_begin()).getAddress(); 11427da15baSAnders Carlsson EmitAggregateCopy(This, RHS, CE->getType()); 11527da15baSAnders Carlsson return RValue::get(This); 11627da15baSAnders Carlsson } 11727da15baSAnders Carlsson 11827da15baSAnders Carlsson // C++ [class.virtual]p12: 11927da15baSAnders Carlsson // Explicit qualification with the scope operator (5.1) suppresses the 12027da15baSAnders Carlsson // virtual call mechanism. 12127da15baSAnders Carlsson // 12227da15baSAnders Carlsson // We also don't emit a virtual call if the base expression has a record type 12327da15baSAnders Carlsson // because then we know what the type is. 12427da15baSAnders Carlsson llvm::Value *Callee; 12527da15baSAnders Carlsson if (const CXXDestructorDecl *Destructor 12627da15baSAnders Carlsson = dyn_cast<CXXDestructorDecl>(MD)) { 12727da15baSAnders Carlsson if (Destructor->isTrivial()) 12827da15baSAnders Carlsson return RValue::get(0); 12927da15baSAnders Carlsson if (MD->isVirtual() && !ME->hasQualifier() && 13027da15baSAnders Carlsson !canDevirtualizeMemberFunctionCalls(ME->getBase())) { 13127da15baSAnders Carlsson Callee = BuildVirtualCall(Destructor, Dtor_Complete, This, Ty); 13227da15baSAnders Carlsson } else { 13327da15baSAnders Carlsson Callee = CGM.GetAddrOfFunction(GlobalDecl(Destructor, Dtor_Complete), Ty); 13427da15baSAnders Carlsson } 13527da15baSAnders Carlsson } else if (MD->isVirtual() && !ME->hasQualifier() && 13627da15baSAnders Carlsson !canDevirtualizeMemberFunctionCalls(ME->getBase())) { 13727da15baSAnders Carlsson Callee = BuildVirtualCall(MD, This, Ty); 13827da15baSAnders Carlsson } else { 13927da15baSAnders Carlsson Callee = CGM.GetAddrOfFunction(MD, Ty); 14027da15baSAnders Carlsson } 14127da15baSAnders Carlsson 142e36a6b3eSAnders Carlsson return EmitCXXMemberCall(MD, Callee, ReturnValue, This, /*VTT=*/0, 14327da15baSAnders Carlsson CE->arg_begin(), CE->arg_end()); 14427da15baSAnders Carlsson } 14527da15baSAnders Carlsson 14627da15baSAnders Carlsson RValue 14727da15baSAnders Carlsson CodeGenFunction::EmitCXXMemberPointerCallExpr(const CXXMemberCallExpr *E, 14827da15baSAnders Carlsson ReturnValueSlot ReturnValue) { 14927da15baSAnders Carlsson const BinaryOperator *BO = 15027da15baSAnders Carlsson cast<BinaryOperator>(E->getCallee()->IgnoreParens()); 15127da15baSAnders Carlsson const Expr *BaseExpr = BO->getLHS(); 15227da15baSAnders Carlsson const Expr *MemFnExpr = BO->getRHS(); 15327da15baSAnders Carlsson 15427da15baSAnders Carlsson const MemberPointerType *MPT = 15527da15baSAnders Carlsson MemFnExpr->getType()->getAs<MemberPointerType>(); 156475999dcSJohn McCall 15727da15baSAnders Carlsson const FunctionProtoType *FPT = 15827da15baSAnders Carlsson MPT->getPointeeType()->getAs<FunctionProtoType>(); 15927da15baSAnders Carlsson const CXXRecordDecl *RD = 16027da15baSAnders Carlsson cast<CXXRecordDecl>(MPT->getClass()->getAs<RecordType>()->getDecl()); 16127da15baSAnders Carlsson 16227da15baSAnders Carlsson // Get the member function pointer. 163a7566f16SDaniel Dunbar llvm::Value *MemFnPtr = CreateMemTemp(MemFnExpr->getType(), "mem.fn"); 16427da15baSAnders Carlsson EmitAggExpr(MemFnExpr, MemFnPtr, /*VolatileDest=*/false); 16527da15baSAnders Carlsson 16627da15baSAnders Carlsson // Emit the 'this' pointer. 16727da15baSAnders Carlsson llvm::Value *This; 16827da15baSAnders Carlsson 16927da15baSAnders Carlsson if (BO->getOpcode() == BinaryOperator::PtrMemI) 17027da15baSAnders Carlsson This = EmitScalarExpr(BaseExpr); 17127da15baSAnders Carlsson else 17227da15baSAnders Carlsson This = EmitLValue(BaseExpr).getAddress(); 17327da15baSAnders Carlsson 174475999dcSJohn McCall // Ask the ABI to load the callee. Note that This is modified. 175475999dcSJohn McCall llvm::Value *Callee = 176475999dcSJohn McCall CGM.getCXXABI().EmitLoadOfMemberFunctionPointer(CGF, This, MemFnPtr, MPT); 17727da15baSAnders Carlsson 17827da15baSAnders Carlsson CallArgList Args; 17927da15baSAnders Carlsson 18027da15baSAnders Carlsson QualType ThisType = 18127da15baSAnders Carlsson getContext().getPointerType(getContext().getTagDeclType(RD)); 18227da15baSAnders Carlsson 18327da15baSAnders Carlsson // Push the this ptr. 18427da15baSAnders Carlsson Args.push_back(std::make_pair(RValue::get(This), ThisType)); 18527da15baSAnders Carlsson 18627da15baSAnders Carlsson // And the rest of the call args 18727da15baSAnders Carlsson EmitCallArgs(Args, FPT, E->arg_begin(), E->arg_end()); 188ab26cfa5SJohn McCall const FunctionType *BO_FPT = BO->getType()->getAs<FunctionProtoType>(); 189ab26cfa5SJohn McCall return EmitCall(CGM.getTypes().getFunctionInfo(Args, BO_FPT), Callee, 19027da15baSAnders Carlsson ReturnValue, Args); 19127da15baSAnders Carlsson } 19227da15baSAnders Carlsson 19327da15baSAnders Carlsson RValue 19427da15baSAnders Carlsson CodeGenFunction::EmitCXXOperatorMemberCallExpr(const CXXOperatorCallExpr *E, 19527da15baSAnders Carlsson const CXXMethodDecl *MD, 19627da15baSAnders Carlsson ReturnValueSlot ReturnValue) { 19727da15baSAnders Carlsson assert(MD->isInstance() && 19827da15baSAnders Carlsson "Trying to emit a member call expr on a static method!"); 19927da15baSAnders Carlsson if (MD->isCopyAssignment()) { 20027da15baSAnders Carlsson const CXXRecordDecl *ClassDecl = cast<CXXRecordDecl>(MD->getDeclContext()); 20127da15baSAnders Carlsson if (ClassDecl->hasTrivialCopyAssignment()) { 20227da15baSAnders Carlsson assert(!ClassDecl->hasUserDeclaredCopyAssignment() && 20327da15baSAnders Carlsson "EmitCXXOperatorMemberCallExpr - user declared copy assignment"); 20443a40f93SFariborz Jahanian LValue LV = EmitLValue(E->getArg(0)); 20543a40f93SFariborz Jahanian llvm::Value *This; 20643a40f93SFariborz Jahanian if (LV.isPropertyRef()) { 2071b8b8bf2SFariborz Jahanian llvm::Value *AggLoc = CreateMemTemp(E->getArg(1)->getType()); 208e1b45a5eSFariborz Jahanian EmitAggExpr(E->getArg(1), AggLoc, false /*VolatileDest*/); 209e1b45a5eSFariborz Jahanian EmitObjCPropertySet(LV.getPropertyRefExpr(), 210e1b45a5eSFariborz Jahanian RValue::getAggregate(AggLoc, false /*VolatileDest*/)); 211e1b45a5eSFariborz Jahanian return RValue::getAggregate(0, false); 21243a40f93SFariborz Jahanian } 21343a40f93SFariborz Jahanian else 21443a40f93SFariborz Jahanian This = LV.getAddress(); 21543a40f93SFariborz Jahanian 21627da15baSAnders Carlsson llvm::Value *Src = EmitLValue(E->getArg(1)).getAddress(); 21727da15baSAnders Carlsson QualType Ty = E->getType(); 21827da15baSAnders Carlsson EmitAggregateCopy(This, Src, Ty); 21927da15baSAnders Carlsson return RValue::get(This); 22027da15baSAnders Carlsson } 22127da15baSAnders Carlsson } 22227da15baSAnders Carlsson 22327da15baSAnders Carlsson const FunctionProtoType *FPT = MD->getType()->getAs<FunctionProtoType>(); 22427da15baSAnders Carlsson const llvm::Type *Ty = 22527da15baSAnders Carlsson CGM.getTypes().GetFunctionType(CGM.getTypes().getFunctionInfo(MD), 22627da15baSAnders Carlsson FPT->isVariadic()); 227fdf474b0SFariborz Jahanian LValue LV = EmitLValue(E->getArg(0)); 228fdf474b0SFariborz Jahanian llvm::Value *This; 229fdf474b0SFariborz Jahanian if (LV.isPropertyRef()) { 2306855ba2cSFariborz Jahanian RValue RV = EmitLoadOfPropertyRefLValue(LV, E->getArg(0)->getType()); 2316855ba2cSFariborz Jahanian assert (!RV.isScalar() && "EmitCXXOperatorMemberCallExpr"); 2326855ba2cSFariborz Jahanian This = RV.getAggregateAddr(); 233fdf474b0SFariborz Jahanian } 234fdf474b0SFariborz Jahanian else 235fdf474b0SFariborz Jahanian This = LV.getAddress(); 23627da15baSAnders Carlsson 23727da15baSAnders Carlsson llvm::Value *Callee; 23827da15baSAnders Carlsson if (MD->isVirtual() && !canDevirtualizeMemberFunctionCalls(E->getArg(0))) 23927da15baSAnders Carlsson Callee = BuildVirtualCall(MD, This, Ty); 24027da15baSAnders Carlsson else 24127da15baSAnders Carlsson Callee = CGM.GetAddrOfFunction(MD, Ty); 24227da15baSAnders Carlsson 243e36a6b3eSAnders Carlsson return EmitCXXMemberCall(MD, Callee, ReturnValue, This, /*VTT=*/0, 24427da15baSAnders Carlsson E->arg_begin() + 1, E->arg_end()); 24527da15baSAnders Carlsson } 24627da15baSAnders Carlsson 24727da15baSAnders Carlsson void 24827da15baSAnders Carlsson CodeGenFunction::EmitCXXConstructExpr(llvm::Value *Dest, 24927da15baSAnders Carlsson const CXXConstructExpr *E) { 25027da15baSAnders Carlsson assert(Dest && "Must have a destination!"); 25127da15baSAnders Carlsson const CXXConstructorDecl *CD = E->getConstructor(); 25227da15baSAnders Carlsson const ConstantArrayType *Array = 25327da15baSAnders Carlsson getContext().getAsConstantArrayType(E->getType()); 25427da15baSAnders Carlsson // For a copy constructor, even if it is trivial, must fall thru so 25527da15baSAnders Carlsson // its argument is code-gen'ed. 25627da15baSAnders Carlsson if (!CD->isCopyConstructor()) { 25727da15baSAnders Carlsson QualType InitType = E->getType(); 25827da15baSAnders Carlsson if (Array) 25927da15baSAnders Carlsson InitType = getContext().getBaseElementType(Array); 26027da15baSAnders Carlsson const CXXRecordDecl *RD = 26127da15baSAnders Carlsson cast<CXXRecordDecl>(InitType->getAs<RecordType>()->getDecl()); 262e3b3464dSDouglas Gregor if (RD->hasTrivialConstructor()) { 263e3b3464dSDouglas Gregor // The constructor is trivial, but we may still need to zero-initialize 264e3b3464dSDouglas Gregor // the class. 265e3b3464dSDouglas Gregor if (E->requiresZeroInitialization()) 266e3b3464dSDouglas Gregor EmitNullInitialization(Dest, E->getType()); 267e3b3464dSDouglas Gregor 26827da15baSAnders Carlsson return; 26927da15baSAnders Carlsson } 270e3b3464dSDouglas Gregor } 27127da15baSAnders Carlsson // Code gen optimization to eliminate copy constructor and return 272222cf0efSDouglas Gregor // its first argument instead, if in fact that argument is a temporary 273222cf0efSDouglas Gregor // object. 27427da15baSAnders Carlsson if (getContext().getLangOptions().ElideConstructors && E->isElidable()) { 275222cf0efSDouglas Gregor if (const Expr *Arg = E->getArg(0)->getTemporaryObject()) { 27627da15baSAnders Carlsson EmitAggExpr(Arg, Dest, false); 27727da15baSAnders Carlsson return; 27827da15baSAnders Carlsson } 279222cf0efSDouglas Gregor } 28027da15baSAnders Carlsson if (Array) { 28127da15baSAnders Carlsson QualType BaseElementTy = getContext().getBaseElementType(Array); 28227da15baSAnders Carlsson const llvm::Type *BasePtr = ConvertType(BaseElementTy); 28327da15baSAnders Carlsson BasePtr = llvm::PointerType::getUnqual(BasePtr); 28427da15baSAnders Carlsson llvm::Value *BaseAddrPtr = 28527da15baSAnders Carlsson Builder.CreateBitCast(Dest, BasePtr); 28627da15baSAnders Carlsson 28727da15baSAnders Carlsson EmitCXXAggrConstructorCall(CD, Array, BaseAddrPtr, 28827da15baSAnders Carlsson E->arg_begin(), E->arg_end()); 28927da15baSAnders Carlsson } 290e11f9ce9SAnders Carlsson else { 291e11f9ce9SAnders Carlsson CXXCtorType Type = 292e11f9ce9SAnders Carlsson (E->getConstructionKind() == CXXConstructExpr::CK_Complete) 293e11f9ce9SAnders Carlsson ? Ctor_Complete : Ctor_Base; 294e11f9ce9SAnders Carlsson bool ForVirtualBase = 295e11f9ce9SAnders Carlsson E->getConstructionKind() == CXXConstructExpr::CK_VirtualBase; 296e11f9ce9SAnders Carlsson 29727da15baSAnders Carlsson // Call the constructor. 298e11f9ce9SAnders Carlsson EmitCXXConstructorCall(CD, Type, ForVirtualBase, Dest, 29927da15baSAnders Carlsson E->arg_begin(), E->arg_end()); 30027da15baSAnders Carlsson } 301e11f9ce9SAnders Carlsson } 30227da15baSAnders Carlsson 3033eb55cfeSKen Dyck static CharUnits CalculateCookiePadding(ASTContext &Ctx, QualType ElementType) { 30421122cf6SAnders Carlsson const RecordType *RT = ElementType->getAs<RecordType>(); 30559486a2dSAnders Carlsson if (!RT) 3063eb55cfeSKen Dyck return CharUnits::Zero(); 30759486a2dSAnders Carlsson 30859486a2dSAnders Carlsson const CXXRecordDecl *RD = dyn_cast<CXXRecordDecl>(RT->getDecl()); 30959486a2dSAnders Carlsson if (!RD) 3103eb55cfeSKen Dyck return CharUnits::Zero(); 31159486a2dSAnders Carlsson 31259486a2dSAnders Carlsson // Check if the class has a trivial destructor. 31359486a2dSAnders Carlsson if (RD->hasTrivialDestructor()) { 31421122cf6SAnders Carlsson // Check if the usual deallocation function takes two arguments. 315adbe4249SAnders Carlsson const CXXMethodDecl *UsualDeallocationFunction = 0; 316adbe4249SAnders Carlsson 31721122cf6SAnders Carlsson DeclarationName OpName = 31821122cf6SAnders Carlsson Ctx.DeclarationNames.getCXXOperatorName(OO_Array_Delete); 31921122cf6SAnders Carlsson DeclContext::lookup_const_iterator Op, OpEnd; 32021122cf6SAnders Carlsson for (llvm::tie(Op, OpEnd) = RD->lookup(OpName); 32121122cf6SAnders Carlsson Op != OpEnd; ++Op) { 322adbe4249SAnders Carlsson const CXXMethodDecl *Delete = cast<CXXMethodDecl>(*Op); 32321122cf6SAnders Carlsson 32421122cf6SAnders Carlsson if (Delete->isUsualDeallocationFunction()) { 325adbe4249SAnders Carlsson UsualDeallocationFunction = Delete; 32621122cf6SAnders Carlsson break; 32721122cf6SAnders Carlsson } 32821122cf6SAnders Carlsson } 329adbe4249SAnders Carlsson 330adbe4249SAnders Carlsson // No usual deallocation function, we don't need a cookie. 331adbe4249SAnders Carlsson if (!UsualDeallocationFunction) 3323eb55cfeSKen Dyck return CharUnits::Zero(); 333adbe4249SAnders Carlsson 334adbe4249SAnders Carlsson // The usual deallocation function doesn't take a size_t argument, so we 335adbe4249SAnders Carlsson // don't need a cookie. 336adbe4249SAnders Carlsson if (UsualDeallocationFunction->getNumParams() == 1) 3373eb55cfeSKen Dyck return CharUnits::Zero(); 338adbe4249SAnders Carlsson 339adbe4249SAnders Carlsson assert(UsualDeallocationFunction->getNumParams() == 2 && 340adbe4249SAnders Carlsson "Unexpected deallocation function type!"); 34159486a2dSAnders Carlsson } 34259486a2dSAnders Carlsson 34321122cf6SAnders Carlsson // Padding is the maximum of sizeof(size_t) and alignof(ElementType) 3443eb55cfeSKen Dyck return std::max(Ctx.getTypeSizeInChars(Ctx.getSizeType()), 3453eb55cfeSKen Dyck Ctx.getTypeAlignInChars(ElementType)); 34621122cf6SAnders Carlsson } 34721122cf6SAnders Carlsson 3483eb55cfeSKen Dyck static CharUnits CalculateCookiePadding(ASTContext &Ctx, const CXXNewExpr *E) { 34921122cf6SAnders Carlsson if (!E->isArray()) 3503eb55cfeSKen Dyck return CharUnits::Zero(); 35121122cf6SAnders Carlsson 352399f499fSAnders Carlsson // No cookie is required if the new operator being used is 353399f499fSAnders Carlsson // ::operator new[](size_t, void*). 354399f499fSAnders Carlsson const FunctionDecl *OperatorNew = E->getOperatorNew(); 355399f499fSAnders Carlsson if (OperatorNew->getDeclContext()->getLookupContext()->isFileContext()) { 356399f499fSAnders Carlsson if (OperatorNew->getNumParams() == 2) { 357399f499fSAnders Carlsson CanQualType ParamType = 358399f499fSAnders Carlsson Ctx.getCanonicalType(OperatorNew->getParamDecl(1)->getType()); 359399f499fSAnders Carlsson 360399f499fSAnders Carlsson if (ParamType == Ctx.VoidPtrTy) 3613eb55cfeSKen Dyck return CharUnits::Zero(); 362399f499fSAnders Carlsson } 363399f499fSAnders Carlsson } 364399f499fSAnders Carlsson 36521122cf6SAnders Carlsson return CalculateCookiePadding(Ctx, E->getAllocatedType()); 36659486a2dSAnders Carlsson } 36759486a2dSAnders Carlsson 36847b4629bSFariborz Jahanian static llvm::Value *EmitCXXNewAllocSize(ASTContext &Context, 36947b4629bSFariborz Jahanian CodeGenFunction &CGF, 37059486a2dSAnders Carlsson const CXXNewExpr *E, 37105fc5be3SDouglas Gregor llvm::Value *&NumElements, 37205fc5be3SDouglas Gregor llvm::Value *&SizeWithoutCookie) { 37359486a2dSAnders Carlsson QualType Type = E->getAllocatedType(); 3743eb55cfeSKen Dyck CharUnits TypeSize = CGF.getContext().getTypeSizeInChars(Type); 37559486a2dSAnders Carlsson const llvm::Type *SizeTy = CGF.ConvertType(CGF.getContext().getSizeType()); 37659486a2dSAnders Carlsson 37705fc5be3SDouglas Gregor if (!E->isArray()) { 37805fc5be3SDouglas Gregor SizeWithoutCookie = llvm::ConstantInt::get(SizeTy, TypeSize.getQuantity()); 37905fc5be3SDouglas Gregor return SizeWithoutCookie; 38005fc5be3SDouglas Gregor } 38159486a2dSAnders Carlsson 38259486a2dSAnders Carlsson // Emit the array size expression. 38359486a2dSAnders Carlsson NumElements = CGF.EmitScalarExpr(E->getArraySize()); 38459486a2dSAnders Carlsson 38532ac583dSChris Lattner llvm::Value *Size = llvm::ConstantInt::get(SizeTy, TypeSize.getQuantity()); 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 // Determine if there is an overflow here by doing an extended multiply. 39232ac583dSChris Lattner llvm::APInt NEC = NumElementsC->getValue(); 39332ac583dSChris Lattner NEC.zext(NEC.getBitWidth()*2); 39432ac583dSChris Lattner 39532ac583dSChris Lattner llvm::APInt SC = cast<llvm::ConstantInt>(Size)->getValue(); 39632ac583dSChris Lattner SC.zext(SC.getBitWidth()*2); 39732ac583dSChris Lattner SC *= NEC; 39832ac583dSChris Lattner 39932ac583dSChris Lattner if (SC.countLeadingZeros() >= NumElementsC->getValue().getBitWidth()) { 40032ac583dSChris Lattner SC.trunc(NumElementsC->getValue().getBitWidth()); 40132ac583dSChris Lattner Size = llvm::ConstantInt::get(Size->getContext(), SC); 40232ac583dSChris Lattner } else { 40332ac583dSChris Lattner // On overflow, produce a -1 so operator new throws. 40432ac583dSChris Lattner Size = llvm::Constant::getAllOnesValue(Size->getType()); 40532ac583dSChris Lattner } 40632ac583dSChris Lattner 40732ac583dSChris Lattner } else { 40826008e07SChris Lattner // Multiply with the type size. This multiply can overflow, e.g. in: 40926008e07SChris Lattner // new double[n] 41026008e07SChris Lattner // where n is 2^30 on a 32-bit machine or 2^62 on a 64-bit machine. Because 41126008e07SChris Lattner // of this, we need to detect the overflow and ensure that an exception is 412f2f38701SChris Lattner // called by forcing the size to -1 on overflow. 41332ac583dSChris Lattner llvm::Value *UMulF = 41432ac583dSChris Lattner CGF.CGM.getIntrinsic(llvm::Intrinsic::umul_with_overflow, &SizeTy, 1); 41532ac583dSChris Lattner llvm::Value *MulRes = CGF.Builder.CreateCall2(UMulF, NumElements, Size); 41632ac583dSChris Lattner // Branch on the overflow bit to the overflow block, which is lazily 41732ac583dSChris Lattner // created. 41826008e07SChris Lattner llvm::Value *DidOverflow = CGF.Builder.CreateExtractValue(MulRes, 1); 41926008e07SChris Lattner // Get the normal result of the multiplication. 42026008e07SChris Lattner llvm::Value *V = CGF.Builder.CreateExtractValue(MulRes, 0); 42147b4629bSFariborz Jahanian 422f2f38701SChris Lattner llvm::BasicBlock *NormalBB = CGF.createBasicBlock("no_overflow"); 423f2f38701SChris Lattner llvm::BasicBlock *OverflowBB = CGF.createBasicBlock("overflow"); 424f2f38701SChris Lattner 425f2f38701SChris Lattner CGF.Builder.CreateCondBr(DidOverflow, OverflowBB, NormalBB); 426f2f38701SChris Lattner 427f2f38701SChris Lattner llvm::BasicBlock *PrevBB = CGF.Builder.GetInsertBlock(); 428f2f38701SChris Lattner 429f2f38701SChris Lattner // We just need the overflow block to build a PHI node. 430f2f38701SChris Lattner CGF.EmitBlock(OverflowBB); 431f2f38701SChris Lattner CGF.EmitBlock(NormalBB); 432f2f38701SChris Lattner 433f2f38701SChris Lattner llvm::PHINode *PN = CGF.Builder.CreatePHI(V->getType()); 434f2f38701SChris Lattner 435f2f38701SChris Lattner PN->addIncoming(V, PrevBB); 436f2f38701SChris Lattner PN->addIncoming(llvm::Constant::getAllOnesValue(V->getType()), OverflowBB); 43732ac583dSChris Lattner Size = PN; 43832ac583dSChris Lattner } 43905fc5be3SDouglas Gregor SizeWithoutCookie = Size; 440f2f38701SChris Lattner 44132ac583dSChris Lattner // Add the cookie padding if necessary. 44232ac583dSChris Lattner CharUnits CookiePadding = CalculateCookiePadding(CGF.getContext(), E); 4433eb55cfeSKen Dyck if (!CookiePadding.isZero()) 44432ac583dSChris Lattner Size = CGF.Builder.CreateAdd(Size, 4453eb55cfeSKen Dyck llvm::ConstantInt::get(SizeTy, CookiePadding.getQuantity())); 44659486a2dSAnders Carlsson 44732ac583dSChris Lattner return Size; 44859486a2dSAnders Carlsson } 44959486a2dSAnders Carlsson 450d5202e09SFariborz Jahanian static void StoreAnyExprIntoOneUnit(CodeGenFunction &CGF, const CXXNewExpr *E, 451d5202e09SFariborz Jahanian llvm::Value *NewPtr) { 452d5202e09SFariborz Jahanian 453d5202e09SFariborz Jahanian assert(E->getNumConstructorArgs() == 1 && 454d5202e09SFariborz Jahanian "Can only have one argument to initializer of POD type."); 455d5202e09SFariborz Jahanian 456d5202e09SFariborz Jahanian const Expr *Init = E->getConstructorArg(0); 457d5202e09SFariborz Jahanian QualType AllocType = E->getAllocatedType(); 458d5202e09SFariborz Jahanian 4590381634aSDaniel Dunbar unsigned Alignment = 4600381634aSDaniel Dunbar CGF.getContext().getTypeAlignInChars(AllocType).getQuantity(); 461d5202e09SFariborz Jahanian if (!CGF.hasAggregateLLVMType(AllocType)) 462d5202e09SFariborz Jahanian CGF.EmitStoreOfScalar(CGF.EmitScalarExpr(Init), NewPtr, 4630381634aSDaniel Dunbar AllocType.isVolatileQualified(), Alignment, 4640381634aSDaniel Dunbar AllocType); 465d5202e09SFariborz Jahanian else if (AllocType->isAnyComplexType()) 466d5202e09SFariborz Jahanian CGF.EmitComplexExprIntoAddr(Init, NewPtr, 467d5202e09SFariborz Jahanian AllocType.isVolatileQualified()); 468d5202e09SFariborz Jahanian else 469d5202e09SFariborz Jahanian CGF.EmitAggExpr(Init, NewPtr, AllocType.isVolatileQualified()); 470d5202e09SFariborz Jahanian } 471d5202e09SFariborz Jahanian 472d5202e09SFariborz Jahanian void 473d5202e09SFariborz Jahanian CodeGenFunction::EmitNewArrayInitializer(const CXXNewExpr *E, 474d5202e09SFariborz Jahanian llvm::Value *NewPtr, 475d5202e09SFariborz Jahanian llvm::Value *NumElements) { 476b66b08efSFariborz Jahanian // We have a POD type. 477b66b08efSFariborz Jahanian if (E->getNumConstructorArgs() == 0) 478b66b08efSFariborz Jahanian return; 479b66b08efSFariborz Jahanian 480d5202e09SFariborz Jahanian const llvm::Type *SizeTy = ConvertType(getContext().getSizeType()); 481d5202e09SFariborz Jahanian 482d5202e09SFariborz Jahanian // Create a temporary for the loop index and initialize it with 0. 483d5202e09SFariborz Jahanian llvm::Value *IndexPtr = CreateTempAlloca(SizeTy, "loop.index"); 484d5202e09SFariborz Jahanian llvm::Value *Zero = llvm::Constant::getNullValue(SizeTy); 485d5202e09SFariborz Jahanian Builder.CreateStore(Zero, IndexPtr); 486d5202e09SFariborz Jahanian 487d5202e09SFariborz Jahanian // Start the loop with a block that tests the condition. 488d5202e09SFariborz Jahanian llvm::BasicBlock *CondBlock = createBasicBlock("for.cond"); 489d5202e09SFariborz Jahanian llvm::BasicBlock *AfterFor = createBasicBlock("for.end"); 490d5202e09SFariborz Jahanian 491d5202e09SFariborz Jahanian EmitBlock(CondBlock); 492d5202e09SFariborz Jahanian 493d5202e09SFariborz Jahanian llvm::BasicBlock *ForBody = createBasicBlock("for.body"); 494d5202e09SFariborz Jahanian 495d5202e09SFariborz Jahanian // Generate: if (loop-index < number-of-elements fall to the loop body, 496d5202e09SFariborz Jahanian // otherwise, go to the block after the for-loop. 497d5202e09SFariborz Jahanian llvm::Value *Counter = Builder.CreateLoad(IndexPtr); 498d5202e09SFariborz Jahanian llvm::Value *IsLess = Builder.CreateICmpULT(Counter, NumElements, "isless"); 499d5202e09SFariborz Jahanian // If the condition is true, execute the body. 500d5202e09SFariborz Jahanian Builder.CreateCondBr(IsLess, ForBody, AfterFor); 501d5202e09SFariborz Jahanian 502d5202e09SFariborz Jahanian EmitBlock(ForBody); 503d5202e09SFariborz Jahanian 504d5202e09SFariborz Jahanian llvm::BasicBlock *ContinueBlock = createBasicBlock("for.inc"); 505d5202e09SFariborz Jahanian // Inside the loop body, emit the constructor call on the array element. 506d5202e09SFariborz Jahanian Counter = Builder.CreateLoad(IndexPtr); 507d5202e09SFariborz Jahanian llvm::Value *Address = Builder.CreateInBoundsGEP(NewPtr, Counter, 508d5202e09SFariborz Jahanian "arrayidx"); 509d5202e09SFariborz Jahanian StoreAnyExprIntoOneUnit(*this, E, Address); 510d5202e09SFariborz Jahanian 511d5202e09SFariborz Jahanian EmitBlock(ContinueBlock); 512d5202e09SFariborz Jahanian 513d5202e09SFariborz Jahanian // Emit the increment of the loop counter. 514d5202e09SFariborz Jahanian llvm::Value *NextVal = llvm::ConstantInt::get(SizeTy, 1); 515d5202e09SFariborz Jahanian Counter = Builder.CreateLoad(IndexPtr); 516d5202e09SFariborz Jahanian NextVal = Builder.CreateAdd(Counter, NextVal, "inc"); 517d5202e09SFariborz Jahanian Builder.CreateStore(NextVal, IndexPtr); 518d5202e09SFariborz Jahanian 519d5202e09SFariborz Jahanian // Finally, branch back up to the condition for the next iteration. 520d5202e09SFariborz Jahanian EmitBranch(CondBlock); 521d5202e09SFariborz Jahanian 522d5202e09SFariborz Jahanian // Emit the fall-through block. 523d5202e09SFariborz Jahanian EmitBlock(AfterFor, true); 524d5202e09SFariborz Jahanian } 525d5202e09SFariborz Jahanian 52605fc5be3SDouglas Gregor static void EmitZeroMemSet(CodeGenFunction &CGF, QualType T, 52705fc5be3SDouglas Gregor llvm::Value *NewPtr, llvm::Value *Size) { 52805fc5be3SDouglas Gregor llvm::LLVMContext &VMContext = CGF.CGM.getLLVMContext(); 52905fc5be3SDouglas Gregor const llvm::Type *BP = llvm::Type::getInt8PtrTy(VMContext); 53005fc5be3SDouglas Gregor if (NewPtr->getType() != BP) 53105fc5be3SDouglas Gregor NewPtr = CGF.Builder.CreateBitCast(NewPtr, BP, "tmp"); 53205fc5be3SDouglas Gregor 53305fc5be3SDouglas Gregor CGF.Builder.CreateCall5(CGF.CGM.getMemSetFn(BP, CGF.IntPtrTy), NewPtr, 53405fc5be3SDouglas Gregor llvm::Constant::getNullValue(llvm::Type::getInt8Ty(VMContext)), 53505fc5be3SDouglas Gregor Size, 53605fc5be3SDouglas Gregor llvm::ConstantInt::get(CGF.Int32Ty, 53705fc5be3SDouglas Gregor CGF.getContext().getTypeAlign(T)/8), 53805fc5be3SDouglas Gregor llvm::ConstantInt::get(llvm::Type::getInt1Ty(VMContext), 53905fc5be3SDouglas Gregor 0)); 54005fc5be3SDouglas Gregor } 54105fc5be3SDouglas Gregor 54259486a2dSAnders Carlsson static void EmitNewInitializer(CodeGenFunction &CGF, const CXXNewExpr *E, 54359486a2dSAnders Carlsson llvm::Value *NewPtr, 54405fc5be3SDouglas Gregor llvm::Value *NumElements, 54505fc5be3SDouglas Gregor llvm::Value *AllocSizeWithoutCookie) { 5463a202f60SAnders Carlsson if (E->isArray()) { 547d040e6b2SAnders Carlsson if (CXXConstructorDecl *Ctor = E->getConstructor()) { 54805fc5be3SDouglas Gregor bool RequiresZeroInitialization = false; 54905fc5be3SDouglas Gregor if (Ctor->getParent()->hasTrivialConstructor()) { 55005fc5be3SDouglas Gregor // If new expression did not specify value-initialization, then there 55105fc5be3SDouglas Gregor // is no initialization. 55205fc5be3SDouglas Gregor if (!E->hasInitializer() || Ctor->getParent()->isEmpty()) 55305fc5be3SDouglas Gregor return; 55405fc5be3SDouglas Gregor 555*84fa510aSJohn McCall if (!CGF.CGM.getCXXABI().RequiresNonZeroInitializer( 55605fc5be3SDouglas Gregor E->getAllocatedType())) { 55705fc5be3SDouglas Gregor // Optimization: since zero initialization will just set the memory 55805fc5be3SDouglas Gregor // to all zeroes, generate a single memset to do it in one shot. 55905fc5be3SDouglas Gregor EmitZeroMemSet(CGF, E->getAllocatedType(), NewPtr, 56005fc5be3SDouglas Gregor AllocSizeWithoutCookie); 5613a202f60SAnders Carlsson return; 5623a202f60SAnders Carlsson } 56305fc5be3SDouglas Gregor 56405fc5be3SDouglas Gregor RequiresZeroInitialization = true; 56505fc5be3SDouglas Gregor } 56605fc5be3SDouglas Gregor 56705fc5be3SDouglas Gregor CGF.EmitCXXAggrConstructorCall(Ctor, NumElements, NewPtr, 56805fc5be3SDouglas Gregor E->constructor_arg_begin(), 56905fc5be3SDouglas Gregor E->constructor_arg_end(), 57005fc5be3SDouglas Gregor RequiresZeroInitialization); 57105fc5be3SDouglas Gregor return; 57205fc5be3SDouglas Gregor } else if (E->getNumConstructorArgs() == 1 && 57305fc5be3SDouglas Gregor isa<ImplicitValueInitExpr>(E->getConstructorArg(0))) { 57405fc5be3SDouglas Gregor // Optimization: since zero initialization will just set the memory 57505fc5be3SDouglas Gregor // to all zeroes, generate a single memset to do it in one shot. 57605fc5be3SDouglas Gregor EmitZeroMemSet(CGF, E->getAllocatedType(), NewPtr, 57705fc5be3SDouglas Gregor AllocSizeWithoutCookie); 57805fc5be3SDouglas Gregor return; 57905fc5be3SDouglas Gregor } else { 580d5202e09SFariborz Jahanian CGF.EmitNewArrayInitializer(E, NewPtr, NumElements); 581d5202e09SFariborz Jahanian return; 582d040e6b2SAnders Carlsson } 583d5202e09SFariborz Jahanian } 58459486a2dSAnders Carlsson 58559486a2dSAnders Carlsson if (CXXConstructorDecl *Ctor = E->getConstructor()) { 586747eb784SDouglas Gregor // Per C++ [expr.new]p15, if we have an initializer, then we're performing 587747eb784SDouglas Gregor // direct initialization. C++ [dcl.init]p5 requires that we 588747eb784SDouglas Gregor // zero-initialize storage if there are no user-declared constructors. 589747eb784SDouglas Gregor if (E->hasInitializer() && 590747eb784SDouglas Gregor !Ctor->getParent()->hasUserDeclaredConstructor() && 591747eb784SDouglas Gregor !Ctor->getParent()->isEmpty()) 592747eb784SDouglas Gregor CGF.EmitNullInitialization(NewPtr, E->getAllocatedType()); 593747eb784SDouglas Gregor 594e11f9ce9SAnders Carlsson CGF.EmitCXXConstructorCall(Ctor, Ctor_Complete, /*ForVirtualBase=*/false, 595e11f9ce9SAnders Carlsson NewPtr, E->constructor_arg_begin(), 59659486a2dSAnders Carlsson E->constructor_arg_end()); 59759486a2dSAnders Carlsson 59859486a2dSAnders Carlsson return; 59959486a2dSAnders Carlsson } 600b66b08efSFariborz Jahanian // We have a POD type. 601b66b08efSFariborz Jahanian if (E->getNumConstructorArgs() == 0) 602b66b08efSFariborz Jahanian return; 60359486a2dSAnders Carlsson 604d5202e09SFariborz Jahanian StoreAnyExprIntoOneUnit(CGF, E, NewPtr); 60559486a2dSAnders Carlsson } 60659486a2dSAnders Carlsson 60759486a2dSAnders Carlsson llvm::Value *CodeGenFunction::EmitCXXNewExpr(const CXXNewExpr *E) { 60859486a2dSAnders Carlsson QualType AllocType = E->getAllocatedType(); 60959486a2dSAnders Carlsson FunctionDecl *NewFD = E->getOperatorNew(); 61059486a2dSAnders Carlsson const FunctionProtoType *NewFTy = NewFD->getType()->getAs<FunctionProtoType>(); 61159486a2dSAnders Carlsson 61259486a2dSAnders Carlsson CallArgList NewArgs; 61359486a2dSAnders Carlsson 61459486a2dSAnders Carlsson // The allocation size is the first argument. 61559486a2dSAnders Carlsson QualType SizeTy = getContext().getSizeType(); 61659486a2dSAnders Carlsson 61759486a2dSAnders Carlsson llvm::Value *NumElements = 0; 61805fc5be3SDouglas Gregor llvm::Value *AllocSizeWithoutCookie = 0; 61947b4629bSFariborz Jahanian llvm::Value *AllocSize = EmitCXXNewAllocSize(getContext(), 62005fc5be3SDouglas Gregor *this, E, NumElements, 62105fc5be3SDouglas Gregor AllocSizeWithoutCookie); 62259486a2dSAnders Carlsson 62359486a2dSAnders Carlsson NewArgs.push_back(std::make_pair(RValue::get(AllocSize), SizeTy)); 62459486a2dSAnders Carlsson 62559486a2dSAnders Carlsson // Emit the rest of the arguments. 62659486a2dSAnders Carlsson // FIXME: Ideally, this should just use EmitCallArgs. 62759486a2dSAnders Carlsson CXXNewExpr::const_arg_iterator NewArg = E->placement_arg_begin(); 62859486a2dSAnders Carlsson 62959486a2dSAnders Carlsson // First, use the types from the function type. 63059486a2dSAnders Carlsson // We start at 1 here because the first argument (the allocation size) 63159486a2dSAnders Carlsson // has already been emitted. 63259486a2dSAnders Carlsson for (unsigned i = 1, e = NewFTy->getNumArgs(); i != e; ++i, ++NewArg) { 63359486a2dSAnders Carlsson QualType ArgType = NewFTy->getArgType(i); 63459486a2dSAnders Carlsson 63559486a2dSAnders Carlsson assert(getContext().getCanonicalType(ArgType.getNonReferenceType()). 63659486a2dSAnders Carlsson getTypePtr() == 63759486a2dSAnders Carlsson getContext().getCanonicalType(NewArg->getType()).getTypePtr() && 63859486a2dSAnders Carlsson "type mismatch in call argument!"); 63959486a2dSAnders Carlsson 64059486a2dSAnders Carlsson NewArgs.push_back(std::make_pair(EmitCallArg(*NewArg, ArgType), 64159486a2dSAnders Carlsson ArgType)); 64259486a2dSAnders Carlsson 64359486a2dSAnders Carlsson } 64459486a2dSAnders Carlsson 64559486a2dSAnders Carlsson // Either we've emitted all the call args, or we have a call to a 64659486a2dSAnders Carlsson // variadic function. 64759486a2dSAnders Carlsson assert((NewArg == E->placement_arg_end() || NewFTy->isVariadic()) && 64859486a2dSAnders Carlsson "Extra arguments in non-variadic function!"); 64959486a2dSAnders Carlsson 65059486a2dSAnders Carlsson // If we still have any arguments, emit them using the type of the argument. 65159486a2dSAnders Carlsson for (CXXNewExpr::const_arg_iterator NewArgEnd = E->placement_arg_end(); 65259486a2dSAnders Carlsson NewArg != NewArgEnd; ++NewArg) { 65359486a2dSAnders Carlsson QualType ArgType = NewArg->getType(); 65459486a2dSAnders Carlsson NewArgs.push_back(std::make_pair(EmitCallArg(*NewArg, ArgType), 65559486a2dSAnders Carlsson ArgType)); 65659486a2dSAnders Carlsson } 65759486a2dSAnders Carlsson 65859486a2dSAnders Carlsson // Emit the call to new. 65959486a2dSAnders Carlsson RValue RV = 660ab26cfa5SJohn McCall EmitCall(CGM.getTypes().getFunctionInfo(NewArgs, NewFTy), 66161a401caSAnders Carlsson CGM.GetAddrOfFunction(NewFD), ReturnValueSlot(), NewArgs, NewFD); 66259486a2dSAnders Carlsson 66359486a2dSAnders Carlsson // If an allocation function is declared with an empty exception specification 66459486a2dSAnders Carlsson // it returns null to indicate failure to allocate storage. [expr.new]p13. 66559486a2dSAnders Carlsson // (We don't need to check for null when there's no new initializer and 66659486a2dSAnders Carlsson // we're allocating a POD type). 66759486a2dSAnders Carlsson bool NullCheckResult = NewFTy->hasEmptyExceptionSpec() && 66859486a2dSAnders Carlsson !(AllocType->isPODType() && !E->hasInitializer()); 66959486a2dSAnders Carlsson 67059486a2dSAnders Carlsson llvm::BasicBlock *NewNull = 0; 67159486a2dSAnders Carlsson llvm::BasicBlock *NewNotNull = 0; 67259486a2dSAnders Carlsson llvm::BasicBlock *NewEnd = 0; 67359486a2dSAnders Carlsson 67459486a2dSAnders Carlsson llvm::Value *NewPtr = RV.getScalarVal(); 67559486a2dSAnders Carlsson 67659486a2dSAnders Carlsson if (NullCheckResult) { 67759486a2dSAnders Carlsson NewNull = createBasicBlock("new.null"); 67859486a2dSAnders Carlsson NewNotNull = createBasicBlock("new.notnull"); 67959486a2dSAnders Carlsson NewEnd = createBasicBlock("new.end"); 68059486a2dSAnders Carlsson 68159486a2dSAnders Carlsson llvm::Value *IsNull = 68259486a2dSAnders Carlsson Builder.CreateICmpEQ(NewPtr, 68359486a2dSAnders Carlsson llvm::Constant::getNullValue(NewPtr->getType()), 68459486a2dSAnders Carlsson "isnull"); 68559486a2dSAnders Carlsson 68659486a2dSAnders Carlsson Builder.CreateCondBr(IsNull, NewNull, NewNotNull); 68759486a2dSAnders Carlsson EmitBlock(NewNotNull); 68859486a2dSAnders Carlsson } 68959486a2dSAnders Carlsson 6903eb55cfeSKen Dyck CharUnits CookiePadding = CalculateCookiePadding(getContext(), E); 6913eb55cfeSKen Dyck if (!CookiePadding.isZero()) { 6923eb55cfeSKen Dyck CharUnits CookieOffset = 6933eb55cfeSKen Dyck CookiePadding - getContext().getTypeSizeInChars(SizeTy); 69459486a2dSAnders Carlsson 69559486a2dSAnders Carlsson llvm::Value *NumElementsPtr = 6963eb55cfeSKen Dyck Builder.CreateConstInBoundsGEP1_64(NewPtr, CookieOffset.getQuantity()); 69759486a2dSAnders Carlsson 69859486a2dSAnders Carlsson NumElementsPtr = Builder.CreateBitCast(NumElementsPtr, 69959486a2dSAnders Carlsson ConvertType(SizeTy)->getPointerTo()); 70059486a2dSAnders Carlsson Builder.CreateStore(NumElements, NumElementsPtr); 70159486a2dSAnders Carlsson 70259486a2dSAnders Carlsson // Now add the padding to the new ptr. 7033eb55cfeSKen Dyck NewPtr = Builder.CreateConstInBoundsGEP1_64(NewPtr, 7043eb55cfeSKen Dyck CookiePadding.getQuantity()); 70559486a2dSAnders Carlsson } 70659486a2dSAnders Carlsson 70747b4629bSFariborz Jahanian if (AllocType->isArrayType()) { 70847b4629bSFariborz Jahanian while (const ArrayType *AType = getContext().getAsArrayType(AllocType)) 70947b4629bSFariborz Jahanian AllocType = AType->getElementType(); 71047b4629bSFariborz Jahanian NewPtr = 71147b4629bSFariborz Jahanian Builder.CreateBitCast(NewPtr, 71247b4629bSFariborz Jahanian ConvertType(getContext().getPointerType(AllocType))); 71305fc5be3SDouglas Gregor EmitNewInitializer(*this, E, NewPtr, NumElements, AllocSizeWithoutCookie); 71447b4629bSFariborz Jahanian NewPtr = Builder.CreateBitCast(NewPtr, ConvertType(E->getType())); 71547b4629bSFariborz Jahanian } 71647b4629bSFariborz Jahanian else { 71747b4629bSFariborz Jahanian NewPtr = Builder.CreateBitCast(NewPtr, ConvertType(E->getType())); 71805fc5be3SDouglas Gregor EmitNewInitializer(*this, E, NewPtr, NumElements, AllocSizeWithoutCookie); 71947b4629bSFariborz Jahanian } 72059486a2dSAnders Carlsson 72159486a2dSAnders Carlsson if (NullCheckResult) { 72259486a2dSAnders Carlsson Builder.CreateBr(NewEnd); 72359486a2dSAnders Carlsson NewNotNull = Builder.GetInsertBlock(); 72459486a2dSAnders Carlsson EmitBlock(NewNull); 72559486a2dSAnders Carlsson Builder.CreateBr(NewEnd); 72659486a2dSAnders Carlsson EmitBlock(NewEnd); 72759486a2dSAnders Carlsson 72859486a2dSAnders Carlsson llvm::PHINode *PHI = Builder.CreatePHI(NewPtr->getType()); 72959486a2dSAnders Carlsson PHI->reserveOperandSpace(2); 73059486a2dSAnders Carlsson PHI->addIncoming(NewPtr, NewNotNull); 73159486a2dSAnders Carlsson PHI->addIncoming(llvm::Constant::getNullValue(NewPtr->getType()), NewNull); 73259486a2dSAnders Carlsson 73359486a2dSAnders Carlsson NewPtr = PHI; 73459486a2dSAnders Carlsson } 73559486a2dSAnders Carlsson 73659486a2dSAnders Carlsson return NewPtr; 73759486a2dSAnders Carlsson } 73859486a2dSAnders Carlsson 73921122cf6SAnders Carlsson static std::pair<llvm::Value *, llvm::Value *> 74021122cf6SAnders Carlsson GetAllocatedObjectPtrAndNumElements(CodeGenFunction &CGF, 74121122cf6SAnders Carlsson llvm::Value *Ptr, QualType DeleteTy) { 74221122cf6SAnders Carlsson QualType SizeTy = CGF.getContext().getSizeType(); 74321122cf6SAnders Carlsson const llvm::Type *SizeLTy = CGF.ConvertType(SizeTy); 74421122cf6SAnders Carlsson 7453eb55cfeSKen Dyck CharUnits DeleteTypeAlign = CGF.getContext().getTypeAlignInChars(DeleteTy); 7463eb55cfeSKen Dyck CharUnits CookiePadding = 7473eb55cfeSKen Dyck std::max(CGF.getContext().getTypeSizeInChars(SizeTy), 7483eb55cfeSKen Dyck DeleteTypeAlign); 7493eb55cfeSKen Dyck assert(!CookiePadding.isZero() && "CookiePadding should not be 0."); 75021122cf6SAnders Carlsson 75121122cf6SAnders Carlsson const llvm::Type *Int8PtrTy = llvm::Type::getInt8PtrTy(CGF.getLLVMContext()); 7523eb55cfeSKen Dyck CharUnits CookieOffset = 7533eb55cfeSKen Dyck CookiePadding - CGF.getContext().getTypeSizeInChars(SizeTy); 75421122cf6SAnders Carlsson 75521122cf6SAnders Carlsson llvm::Value *AllocatedObjectPtr = CGF.Builder.CreateBitCast(Ptr, Int8PtrTy); 75621122cf6SAnders Carlsson AllocatedObjectPtr = 75721122cf6SAnders Carlsson CGF.Builder.CreateConstInBoundsGEP1_64(AllocatedObjectPtr, 7583eb55cfeSKen Dyck -CookiePadding.getQuantity()); 75921122cf6SAnders Carlsson 76021122cf6SAnders Carlsson llvm::Value *NumElementsPtr = 76121122cf6SAnders Carlsson CGF.Builder.CreateConstInBoundsGEP1_64(AllocatedObjectPtr, 7623eb55cfeSKen Dyck CookieOffset.getQuantity()); 76321122cf6SAnders Carlsson NumElementsPtr = 76421122cf6SAnders Carlsson CGF.Builder.CreateBitCast(NumElementsPtr, SizeLTy->getPointerTo()); 76521122cf6SAnders Carlsson 76621122cf6SAnders Carlsson llvm::Value *NumElements = CGF.Builder.CreateLoad(NumElementsPtr); 76721122cf6SAnders Carlsson NumElements = 76821122cf6SAnders Carlsson CGF.Builder.CreateIntCast(NumElements, SizeLTy, /*isSigned=*/false); 76921122cf6SAnders Carlsson 77021122cf6SAnders Carlsson return std::make_pair(AllocatedObjectPtr, NumElements); 77121122cf6SAnders Carlsson } 77221122cf6SAnders Carlsson 77359486a2dSAnders Carlsson void CodeGenFunction::EmitDeleteCall(const FunctionDecl *DeleteFD, 77459486a2dSAnders Carlsson llvm::Value *Ptr, 77559486a2dSAnders Carlsson QualType DeleteTy) { 77659486a2dSAnders Carlsson const FunctionProtoType *DeleteFTy = 77759486a2dSAnders Carlsson DeleteFD->getType()->getAs<FunctionProtoType>(); 77859486a2dSAnders Carlsson 77959486a2dSAnders Carlsson CallArgList DeleteArgs; 78059486a2dSAnders Carlsson 78121122cf6SAnders Carlsson // Check if we need to pass the size to the delete operator. 78221122cf6SAnders Carlsson llvm::Value *Size = 0; 78321122cf6SAnders Carlsson QualType SizeTy; 78421122cf6SAnders Carlsson if (DeleteFTy->getNumArgs() == 2) { 78521122cf6SAnders Carlsson SizeTy = DeleteFTy->getArgType(1); 7867df3cbebSKen Dyck CharUnits DeleteTypeSize = getContext().getTypeSizeInChars(DeleteTy); 7877df3cbebSKen Dyck Size = llvm::ConstantInt::get(ConvertType(SizeTy), 7887df3cbebSKen Dyck DeleteTypeSize.getQuantity()); 78921122cf6SAnders Carlsson } 79021122cf6SAnders Carlsson 79121122cf6SAnders Carlsson if (DeleteFD->getOverloadedOperator() == OO_Array_Delete && 7923eb55cfeSKen Dyck !CalculateCookiePadding(getContext(), DeleteTy).isZero()) { 79321122cf6SAnders Carlsson // We need to get the number of elements in the array from the cookie. 79421122cf6SAnders Carlsson llvm::Value *AllocatedObjectPtr; 79521122cf6SAnders Carlsson llvm::Value *NumElements; 79621122cf6SAnders Carlsson llvm::tie(AllocatedObjectPtr, NumElements) = 79721122cf6SAnders Carlsson GetAllocatedObjectPtrAndNumElements(*this, Ptr, DeleteTy); 79821122cf6SAnders Carlsson 79921122cf6SAnders Carlsson // Multiply the size with the number of elements. 80021122cf6SAnders Carlsson if (Size) 80121122cf6SAnders Carlsson Size = Builder.CreateMul(NumElements, Size); 80221122cf6SAnders Carlsson 80321122cf6SAnders Carlsson Ptr = AllocatedObjectPtr; 80421122cf6SAnders Carlsson } 80521122cf6SAnders Carlsson 80659486a2dSAnders Carlsson QualType ArgTy = DeleteFTy->getArgType(0); 80759486a2dSAnders Carlsson llvm::Value *DeletePtr = Builder.CreateBitCast(Ptr, ConvertType(ArgTy)); 80859486a2dSAnders Carlsson DeleteArgs.push_back(std::make_pair(RValue::get(DeletePtr), ArgTy)); 80959486a2dSAnders Carlsson 81021122cf6SAnders Carlsson if (Size) 81159486a2dSAnders Carlsson DeleteArgs.push_back(std::make_pair(RValue::get(Size), SizeTy)); 81259486a2dSAnders Carlsson 81359486a2dSAnders Carlsson // Emit the call to delete. 814ab26cfa5SJohn McCall EmitCall(CGM.getTypes().getFunctionInfo(DeleteArgs, DeleteFTy), 81561a401caSAnders Carlsson CGM.GetAddrOfFunction(DeleteFD), ReturnValueSlot(), 81659486a2dSAnders Carlsson DeleteArgs, DeleteFD); 81759486a2dSAnders Carlsson } 81859486a2dSAnders Carlsson 81959486a2dSAnders Carlsson void CodeGenFunction::EmitCXXDeleteExpr(const CXXDeleteExpr *E) { 82059486a2dSAnders Carlsson 82159486a2dSAnders Carlsson // Get at the argument before we performed the implicit conversion 82259486a2dSAnders Carlsson // to void*. 82359486a2dSAnders Carlsson const Expr *Arg = E->getArgument(); 82459486a2dSAnders Carlsson while (const ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(Arg)) { 82559486a2dSAnders Carlsson if (ICE->getCastKind() != CastExpr::CK_UserDefinedConversion && 82659486a2dSAnders Carlsson ICE->getType()->isVoidPointerType()) 82759486a2dSAnders Carlsson Arg = ICE->getSubExpr(); 82859486a2dSAnders Carlsson else 82959486a2dSAnders Carlsson break; 83059486a2dSAnders Carlsson } 83159486a2dSAnders Carlsson 83259486a2dSAnders Carlsson QualType DeleteTy = Arg->getType()->getAs<PointerType>()->getPointeeType(); 83359486a2dSAnders Carlsson 83459486a2dSAnders Carlsson llvm::Value *Ptr = EmitScalarExpr(Arg); 83559486a2dSAnders Carlsson 83659486a2dSAnders Carlsson // Null check the pointer. 83759486a2dSAnders Carlsson llvm::BasicBlock *DeleteNotNull = createBasicBlock("delete.notnull"); 83859486a2dSAnders Carlsson llvm::BasicBlock *DeleteEnd = createBasicBlock("delete.end"); 83959486a2dSAnders Carlsson 84059486a2dSAnders Carlsson llvm::Value *IsNull = 84159486a2dSAnders Carlsson Builder.CreateICmpEQ(Ptr, llvm::Constant::getNullValue(Ptr->getType()), 84259486a2dSAnders Carlsson "isnull"); 84359486a2dSAnders Carlsson 84459486a2dSAnders Carlsson Builder.CreateCondBr(IsNull, DeleteEnd, DeleteNotNull); 84559486a2dSAnders Carlsson EmitBlock(DeleteNotNull); 84659486a2dSAnders Carlsson 84759486a2dSAnders Carlsson bool ShouldCallDelete = true; 84859486a2dSAnders Carlsson 84959486a2dSAnders Carlsson // Call the destructor if necessary. 85059486a2dSAnders Carlsson if (const RecordType *RT = DeleteTy->getAs<RecordType>()) { 85159486a2dSAnders Carlsson if (CXXRecordDecl *RD = dyn_cast<CXXRecordDecl>(RT->getDecl())) { 85259486a2dSAnders Carlsson if (!RD->hasTrivialDestructor()) { 853bac7490fSDouglas Gregor const CXXDestructorDecl *Dtor = RD->getDestructor(); 85459486a2dSAnders Carlsson if (E->isArrayForm()) { 85521122cf6SAnders Carlsson llvm::Value *AllocatedObjectPtr; 85621122cf6SAnders Carlsson llvm::Value *NumElements; 85721122cf6SAnders Carlsson llvm::tie(AllocatedObjectPtr, NumElements) = 85821122cf6SAnders Carlsson GetAllocatedObjectPtrAndNumElements(*this, Ptr, DeleteTy); 859f5af8deaSAnders Carlsson 86059486a2dSAnders Carlsson EmitCXXAggrDestructorCall(Dtor, NumElements, Ptr); 861f5af8deaSAnders Carlsson } else if (Dtor->isVirtual()) { 86259486a2dSAnders Carlsson const llvm::Type *Ty = 86359486a2dSAnders Carlsson CGM.getTypes().GetFunctionType(CGM.getTypes().getFunctionInfo(Dtor), 86459486a2dSAnders Carlsson /*isVariadic=*/false); 86559486a2dSAnders Carlsson 86659486a2dSAnders Carlsson llvm::Value *Callee = BuildVirtualCall(Dtor, Dtor_Deleting, Ptr, Ty); 867e36a6b3eSAnders Carlsson EmitCXXMemberCall(Dtor, Callee, ReturnValueSlot(), Ptr, /*VTT=*/0, 868e36a6b3eSAnders Carlsson 0, 0); 86959486a2dSAnders Carlsson 87059486a2dSAnders Carlsson // The dtor took care of deleting the object. 87159486a2dSAnders Carlsson ShouldCallDelete = false; 87259486a2dSAnders Carlsson } else 873f8a71f08SAnders Carlsson EmitCXXDestructorCall(Dtor, Dtor_Complete, /*ForVirtualBase=*/false, 874f8a71f08SAnders Carlsson Ptr); 87559486a2dSAnders Carlsson } 87659486a2dSAnders Carlsson } 87759486a2dSAnders Carlsson } 87859486a2dSAnders Carlsson 87959486a2dSAnders Carlsson if (ShouldCallDelete) 88059486a2dSAnders Carlsson EmitDeleteCall(E->getOperatorDelete(), Ptr, DeleteTy); 88159486a2dSAnders Carlsson 88259486a2dSAnders Carlsson EmitBlock(DeleteEnd); 88359486a2dSAnders Carlsson } 88459486a2dSAnders Carlsson 88559486a2dSAnders Carlsson llvm::Value * CodeGenFunction::EmitCXXTypeidExpr(const CXXTypeidExpr *E) { 88659486a2dSAnders Carlsson QualType Ty = E->getType(); 88759486a2dSAnders Carlsson const llvm::Type *LTy = ConvertType(Ty)->getPointerTo(); 888fd7dfeb7SAnders Carlsson 8893f4336cbSAnders Carlsson if (E->isTypeOperand()) { 8903f4336cbSAnders Carlsson llvm::Constant *TypeInfo = 8913f4336cbSAnders Carlsson CGM.GetAddrOfRTTIDescriptor(E->getTypeOperand()); 8923f4336cbSAnders Carlsson return Builder.CreateBitCast(TypeInfo, LTy); 8933f4336cbSAnders Carlsson } 894fd7dfeb7SAnders Carlsson 89559486a2dSAnders Carlsson Expr *subE = E->getExprOperand(); 89659486a2dSAnders Carlsson Ty = subE->getType(); 89759486a2dSAnders Carlsson CanQualType CanTy = CGM.getContext().getCanonicalType(Ty); 89859486a2dSAnders Carlsson Ty = CanTy.getUnqualifiedType().getNonReferenceType(); 89959486a2dSAnders Carlsson if (const RecordType *RT = Ty->getAs<RecordType>()) { 90059486a2dSAnders Carlsson const CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl()); 90159486a2dSAnders Carlsson if (RD->isPolymorphic()) { 90259486a2dSAnders Carlsson // FIXME: if subE is an lvalue do 90359486a2dSAnders Carlsson LValue Obj = EmitLValue(subE); 90459486a2dSAnders Carlsson llvm::Value *This = Obj.getAddress(); 90559486a2dSAnders Carlsson LTy = LTy->getPointerTo()->getPointerTo(); 90659486a2dSAnders Carlsson llvm::Value *V = Builder.CreateBitCast(This, LTy); 90759486a2dSAnders Carlsson // We need to do a zero check for *p, unless it has NonNullAttr. 90859486a2dSAnders Carlsson // FIXME: PointerType->hasAttr<NonNullAttr>() 90959486a2dSAnders Carlsson bool CanBeZero = false; 91059486a2dSAnders Carlsson if (UnaryOperator *UO = dyn_cast<UnaryOperator>(subE->IgnoreParens())) 91159486a2dSAnders Carlsson if (UO->getOpcode() == UnaryOperator::Deref) 91259486a2dSAnders Carlsson CanBeZero = true; 91359486a2dSAnders Carlsson if (CanBeZero) { 91459486a2dSAnders Carlsson llvm::BasicBlock *NonZeroBlock = createBasicBlock(); 91559486a2dSAnders Carlsson llvm::BasicBlock *ZeroBlock = createBasicBlock(); 91659486a2dSAnders Carlsson 91759486a2dSAnders Carlsson llvm::Value *Zero = llvm::Constant::getNullValue(LTy); 91859486a2dSAnders Carlsson Builder.CreateCondBr(Builder.CreateICmpNE(V, Zero), 91959486a2dSAnders Carlsson NonZeroBlock, ZeroBlock); 92059486a2dSAnders Carlsson EmitBlock(ZeroBlock); 92159486a2dSAnders Carlsson /// Call __cxa_bad_typeid 92259486a2dSAnders Carlsson const llvm::Type *ResultType = llvm::Type::getVoidTy(VMContext); 92359486a2dSAnders Carlsson const llvm::FunctionType *FTy; 92459486a2dSAnders Carlsson FTy = llvm::FunctionType::get(ResultType, false); 92559486a2dSAnders Carlsson llvm::Value *F = CGM.CreateRuntimeFunction(FTy, "__cxa_bad_typeid"); 92659486a2dSAnders Carlsson Builder.CreateCall(F)->setDoesNotReturn(); 92759486a2dSAnders Carlsson Builder.CreateUnreachable(); 92859486a2dSAnders Carlsson EmitBlock(NonZeroBlock); 92959486a2dSAnders Carlsson } 93059486a2dSAnders Carlsson V = Builder.CreateLoad(V, "vtable"); 93159486a2dSAnders Carlsson V = Builder.CreateConstInBoundsGEP1_64(V, -1ULL); 93259486a2dSAnders Carlsson V = Builder.CreateLoad(V); 93359486a2dSAnders Carlsson return V; 93459486a2dSAnders Carlsson } 93559486a2dSAnders Carlsson } 9363f4336cbSAnders Carlsson return Builder.CreateBitCast(CGM.GetAddrOfRTTIDescriptor(Ty), LTy); 93759486a2dSAnders Carlsson } 93859486a2dSAnders Carlsson 93959486a2dSAnders Carlsson llvm::Value *CodeGenFunction::EmitDynamicCast(llvm::Value *V, 94059486a2dSAnders Carlsson const CXXDynamicCastExpr *DCE) { 9413f4336cbSAnders Carlsson QualType SrcTy = DCE->getSubExpr()->getType(); 9423f4336cbSAnders Carlsson QualType DestTy = DCE->getTypeAsWritten(); 9433f4336cbSAnders Carlsson QualType InnerType = DestTy->getPointeeType(); 9443f4336cbSAnders Carlsson 94559486a2dSAnders Carlsson const llvm::Type *LTy = ConvertType(DCE->getType()); 94659486a2dSAnders Carlsson 94759486a2dSAnders Carlsson bool CanBeZero = false; 94859486a2dSAnders Carlsson bool ToVoid = false; 94959486a2dSAnders Carlsson bool ThrowOnBad = false; 9503f4336cbSAnders Carlsson if (DestTy->isPointerType()) { 95159486a2dSAnders Carlsson // FIXME: if PointerType->hasAttr<NonNullAttr>(), we don't set this 95259486a2dSAnders Carlsson CanBeZero = true; 95359486a2dSAnders Carlsson if (InnerType->isVoidType()) 95459486a2dSAnders Carlsson ToVoid = true; 95559486a2dSAnders Carlsson } else { 95659486a2dSAnders Carlsson LTy = LTy->getPointerTo(); 957fa8b4955SDouglas Gregor 958fa8b4955SDouglas Gregor // FIXME: What if exceptions are disabled? 95959486a2dSAnders Carlsson ThrowOnBad = true; 96059486a2dSAnders Carlsson } 96159486a2dSAnders Carlsson 9623f4336cbSAnders Carlsson if (SrcTy->isPointerType() || SrcTy->isReferenceType()) 9633f4336cbSAnders Carlsson SrcTy = SrcTy->getPointeeType(); 9643f4336cbSAnders Carlsson SrcTy = SrcTy.getUnqualifiedType(); 9653f4336cbSAnders Carlsson 9660087bc85SAnders Carlsson if (DestTy->isPointerType() || DestTy->isReferenceType()) 9673f4336cbSAnders Carlsson DestTy = DestTy->getPointeeType(); 9683f4336cbSAnders Carlsson DestTy = DestTy.getUnqualifiedType(); 96959486a2dSAnders Carlsson 97059486a2dSAnders Carlsson llvm::BasicBlock *ContBlock = createBasicBlock(); 97159486a2dSAnders Carlsson llvm::BasicBlock *NullBlock = 0; 97259486a2dSAnders Carlsson llvm::BasicBlock *NonZeroBlock = 0; 97359486a2dSAnders Carlsson if (CanBeZero) { 97459486a2dSAnders Carlsson NonZeroBlock = createBasicBlock(); 97559486a2dSAnders Carlsson NullBlock = createBasicBlock(); 9763f4336cbSAnders Carlsson Builder.CreateCondBr(Builder.CreateIsNotNull(V), NonZeroBlock, NullBlock); 97759486a2dSAnders Carlsson EmitBlock(NonZeroBlock); 97859486a2dSAnders Carlsson } 97959486a2dSAnders Carlsson 98059486a2dSAnders Carlsson llvm::BasicBlock *BadCastBlock = 0; 98159486a2dSAnders Carlsson 9823f4336cbSAnders Carlsson const llvm::Type *PtrDiffTy = ConvertType(getContext().getPointerDiffType()); 98359486a2dSAnders Carlsson 98459486a2dSAnders Carlsson // See if this is a dynamic_cast(void*) 98559486a2dSAnders Carlsson if (ToVoid) { 98659486a2dSAnders Carlsson llvm::Value *This = V; 98759486a2dSAnders Carlsson V = Builder.CreateBitCast(This, PtrDiffTy->getPointerTo()->getPointerTo()); 98859486a2dSAnders Carlsson V = Builder.CreateLoad(V, "vtable"); 98959486a2dSAnders Carlsson V = Builder.CreateConstInBoundsGEP1_64(V, -2ULL); 99059486a2dSAnders Carlsson V = Builder.CreateLoad(V, "offset to top"); 99159486a2dSAnders Carlsson This = Builder.CreateBitCast(This, llvm::Type::getInt8PtrTy(VMContext)); 99259486a2dSAnders Carlsson V = Builder.CreateInBoundsGEP(This, V); 99359486a2dSAnders Carlsson V = Builder.CreateBitCast(V, LTy); 99459486a2dSAnders Carlsson } else { 99559486a2dSAnders Carlsson /// Call __dynamic_cast 99659486a2dSAnders Carlsson const llvm::Type *ResultType = llvm::Type::getInt8PtrTy(VMContext); 99759486a2dSAnders Carlsson const llvm::FunctionType *FTy; 99859486a2dSAnders Carlsson std::vector<const llvm::Type*> ArgTys; 99959486a2dSAnders Carlsson const llvm::Type *PtrToInt8Ty 100059486a2dSAnders Carlsson = llvm::Type::getInt8Ty(VMContext)->getPointerTo(); 100159486a2dSAnders Carlsson ArgTys.push_back(PtrToInt8Ty); 100259486a2dSAnders Carlsson ArgTys.push_back(PtrToInt8Ty); 100359486a2dSAnders Carlsson ArgTys.push_back(PtrToInt8Ty); 100459486a2dSAnders Carlsson ArgTys.push_back(PtrDiffTy); 100559486a2dSAnders Carlsson FTy = llvm::FunctionType::get(ResultType, ArgTys, false); 100659486a2dSAnders Carlsson 100759486a2dSAnders Carlsson // FIXME: Calculate better hint. 100859486a2dSAnders Carlsson llvm::Value *hint = llvm::ConstantInt::get(PtrDiffTy, -1ULL); 10093f4336cbSAnders Carlsson 10103f4336cbSAnders Carlsson assert(SrcTy->isRecordType() && "Src type must be record type!"); 10113f4336cbSAnders Carlsson assert(DestTy->isRecordType() && "Dest type must be record type!"); 10123f4336cbSAnders Carlsson 1013247894b3SDouglas Gregor llvm::Value *SrcArg 1014247894b3SDouglas Gregor = CGM.GetAddrOfRTTIDescriptor(SrcTy.getUnqualifiedType()); 1015247894b3SDouglas Gregor llvm::Value *DestArg 1016247894b3SDouglas Gregor = CGM.GetAddrOfRTTIDescriptor(DestTy.getUnqualifiedType()); 10173f4336cbSAnders Carlsson 101859486a2dSAnders Carlsson V = Builder.CreateBitCast(V, PtrToInt8Ty); 101959486a2dSAnders Carlsson V = Builder.CreateCall4(CGM.CreateRuntimeFunction(FTy, "__dynamic_cast"), 10203f4336cbSAnders Carlsson V, SrcArg, DestArg, hint); 102159486a2dSAnders Carlsson V = Builder.CreateBitCast(V, LTy); 102259486a2dSAnders Carlsson 102359486a2dSAnders Carlsson if (ThrowOnBad) { 102459486a2dSAnders Carlsson BadCastBlock = createBasicBlock(); 10253f4336cbSAnders Carlsson Builder.CreateCondBr(Builder.CreateIsNotNull(V), ContBlock, BadCastBlock); 102659486a2dSAnders Carlsson EmitBlock(BadCastBlock); 1027fa8b4955SDouglas Gregor /// Invoke __cxa_bad_cast 102859486a2dSAnders Carlsson ResultType = llvm::Type::getVoidTy(VMContext); 102959486a2dSAnders Carlsson const llvm::FunctionType *FBadTy; 103059486a2dSAnders Carlsson FBadTy = llvm::FunctionType::get(ResultType, false); 103159486a2dSAnders Carlsson llvm::Value *F = CGM.CreateRuntimeFunction(FBadTy, "__cxa_bad_cast"); 1032fa8b4955SDouglas Gregor if (llvm::BasicBlock *InvokeDest = getInvokeDest()) { 1033fa8b4955SDouglas Gregor llvm::BasicBlock *Cont = createBasicBlock("invoke.cont"); 1034fa8b4955SDouglas Gregor Builder.CreateInvoke(F, Cont, InvokeDest)->setDoesNotReturn(); 1035fa8b4955SDouglas Gregor EmitBlock(Cont); 1036fa8b4955SDouglas Gregor } else { 1037fa8b4955SDouglas Gregor // FIXME: Does this ever make sense? 103859486a2dSAnders Carlsson Builder.CreateCall(F)->setDoesNotReturn(); 1039fa8b4955SDouglas Gregor } 104059486a2dSAnders Carlsson Builder.CreateUnreachable(); 104159486a2dSAnders Carlsson } 104259486a2dSAnders Carlsson } 104359486a2dSAnders Carlsson 104459486a2dSAnders Carlsson if (CanBeZero) { 104559486a2dSAnders Carlsson Builder.CreateBr(ContBlock); 104659486a2dSAnders Carlsson EmitBlock(NullBlock); 104759486a2dSAnders Carlsson Builder.CreateBr(ContBlock); 104859486a2dSAnders Carlsson } 104959486a2dSAnders Carlsson EmitBlock(ContBlock); 105059486a2dSAnders Carlsson if (CanBeZero) { 105159486a2dSAnders Carlsson llvm::PHINode *PHI = Builder.CreatePHI(LTy); 105259486a2dSAnders Carlsson PHI->reserveOperandSpace(2); 105359486a2dSAnders Carlsson PHI->addIncoming(V, NonZeroBlock); 105459486a2dSAnders Carlsson PHI->addIncoming(llvm::Constant::getNullValue(LTy), NullBlock); 105559486a2dSAnders Carlsson V = PHI; 105659486a2dSAnders Carlsson } 105759486a2dSAnders Carlsson 105859486a2dSAnders Carlsson return V; 105959486a2dSAnders Carlsson } 1060