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. 163a1dee530SJohn McCall llvm::Value *MemFnPtr = EmitScalarExpr(MemFnExpr); 16427da15baSAnders Carlsson 16527da15baSAnders Carlsson // Emit the 'this' pointer. 16627da15baSAnders Carlsson llvm::Value *This; 16727da15baSAnders Carlsson 16827da15baSAnders Carlsson if (BO->getOpcode() == BinaryOperator::PtrMemI) 16927da15baSAnders Carlsson This = EmitScalarExpr(BaseExpr); 17027da15baSAnders Carlsson else 17127da15baSAnders Carlsson This = EmitLValue(BaseExpr).getAddress(); 17227da15baSAnders Carlsson 173475999dcSJohn McCall // Ask the ABI to load the callee. Note that This is modified. 174475999dcSJohn McCall llvm::Value *Callee = 175475999dcSJohn McCall CGM.getCXXABI().EmitLoadOfMemberFunctionPointer(CGF, This, MemFnPtr, MPT); 17627da15baSAnders Carlsson 17727da15baSAnders Carlsson CallArgList Args; 17827da15baSAnders Carlsson 17927da15baSAnders Carlsson QualType ThisType = 18027da15baSAnders Carlsson getContext().getPointerType(getContext().getTagDeclType(RD)); 18127da15baSAnders Carlsson 18227da15baSAnders Carlsson // Push the this ptr. 18327da15baSAnders Carlsson Args.push_back(std::make_pair(RValue::get(This), ThisType)); 18427da15baSAnders Carlsson 18527da15baSAnders Carlsson // And the rest of the call args 18627da15baSAnders Carlsson EmitCallArgs(Args, FPT, E->arg_begin(), E->arg_end()); 187ab26cfa5SJohn McCall const FunctionType *BO_FPT = BO->getType()->getAs<FunctionProtoType>(); 188ab26cfa5SJohn McCall return EmitCall(CGM.getTypes().getFunctionInfo(Args, BO_FPT), Callee, 18927da15baSAnders Carlsson ReturnValue, Args); 19027da15baSAnders Carlsson } 19127da15baSAnders Carlsson 19227da15baSAnders Carlsson RValue 19327da15baSAnders Carlsson CodeGenFunction::EmitCXXOperatorMemberCallExpr(const CXXOperatorCallExpr *E, 19427da15baSAnders Carlsson const CXXMethodDecl *MD, 19527da15baSAnders Carlsson ReturnValueSlot ReturnValue) { 19627da15baSAnders Carlsson assert(MD->isInstance() && 19727da15baSAnders Carlsson "Trying to emit a member call expr on a static method!"); 19827da15baSAnders Carlsson if (MD->isCopyAssignment()) { 19927da15baSAnders Carlsson const CXXRecordDecl *ClassDecl = cast<CXXRecordDecl>(MD->getDeclContext()); 20027da15baSAnders Carlsson if (ClassDecl->hasTrivialCopyAssignment()) { 20127da15baSAnders Carlsson assert(!ClassDecl->hasUserDeclaredCopyAssignment() && 20227da15baSAnders Carlsson "EmitCXXOperatorMemberCallExpr - user declared copy assignment"); 20343a40f93SFariborz Jahanian LValue LV = EmitLValue(E->getArg(0)); 20443a40f93SFariborz Jahanian llvm::Value *This; 20543a40f93SFariborz Jahanian if (LV.isPropertyRef()) { 2061b8b8bf2SFariborz Jahanian llvm::Value *AggLoc = CreateMemTemp(E->getArg(1)->getType()); 207e1b45a5eSFariborz Jahanian EmitAggExpr(E->getArg(1), AggLoc, false /*VolatileDest*/); 208e1b45a5eSFariborz Jahanian EmitObjCPropertySet(LV.getPropertyRefExpr(), 209e1b45a5eSFariborz Jahanian RValue::getAggregate(AggLoc, false /*VolatileDest*/)); 210e1b45a5eSFariborz Jahanian return RValue::getAggregate(0, false); 21143a40f93SFariborz Jahanian } 21243a40f93SFariborz Jahanian else 21343a40f93SFariborz Jahanian This = LV.getAddress(); 21443a40f93SFariborz Jahanian 21527da15baSAnders Carlsson llvm::Value *Src = EmitLValue(E->getArg(1)).getAddress(); 21627da15baSAnders Carlsson QualType Ty = E->getType(); 21727da15baSAnders Carlsson EmitAggregateCopy(This, Src, Ty); 21827da15baSAnders Carlsson return RValue::get(This); 21927da15baSAnders Carlsson } 22027da15baSAnders Carlsson } 22127da15baSAnders Carlsson 22227da15baSAnders Carlsson const FunctionProtoType *FPT = MD->getType()->getAs<FunctionProtoType>(); 22327da15baSAnders Carlsson const llvm::Type *Ty = 22427da15baSAnders Carlsson CGM.getTypes().GetFunctionType(CGM.getTypes().getFunctionInfo(MD), 22527da15baSAnders Carlsson FPT->isVariadic()); 226fdf474b0SFariborz Jahanian LValue LV = EmitLValue(E->getArg(0)); 227fdf474b0SFariborz Jahanian llvm::Value *This; 228fdf474b0SFariborz Jahanian if (LV.isPropertyRef()) { 2296855ba2cSFariborz Jahanian RValue RV = EmitLoadOfPropertyRefLValue(LV, E->getArg(0)->getType()); 2306855ba2cSFariborz Jahanian assert (!RV.isScalar() && "EmitCXXOperatorMemberCallExpr"); 2316855ba2cSFariborz Jahanian This = RV.getAggregateAddr(); 232fdf474b0SFariborz Jahanian } 233fdf474b0SFariborz Jahanian else 234fdf474b0SFariborz Jahanian This = LV.getAddress(); 23527da15baSAnders Carlsson 23627da15baSAnders Carlsson llvm::Value *Callee; 23727da15baSAnders Carlsson if (MD->isVirtual() && !canDevirtualizeMemberFunctionCalls(E->getArg(0))) 23827da15baSAnders Carlsson Callee = BuildVirtualCall(MD, This, Ty); 23927da15baSAnders Carlsson else 24027da15baSAnders Carlsson Callee = CGM.GetAddrOfFunction(MD, Ty); 24127da15baSAnders Carlsson 242e36a6b3eSAnders Carlsson return EmitCXXMemberCall(MD, Callee, ReturnValue, This, /*VTT=*/0, 24327da15baSAnders Carlsson E->arg_begin() + 1, E->arg_end()); 24427da15baSAnders Carlsson } 24527da15baSAnders Carlsson 24627da15baSAnders Carlsson void 24727da15baSAnders Carlsson CodeGenFunction::EmitCXXConstructExpr(llvm::Value *Dest, 24827da15baSAnders Carlsson const CXXConstructExpr *E) { 24927da15baSAnders Carlsson assert(Dest && "Must have a destination!"); 25027da15baSAnders Carlsson const CXXConstructorDecl *CD = E->getConstructor(); 251630c76efSDouglas Gregor 252630c76efSDouglas Gregor // If we require zero initialization before (or instead of) calling the 253630c76efSDouglas Gregor // constructor, as can be the case with a non-user-provided default 254630c76efSDouglas Gregor // constructor, emit the zero initialization now. 255e3b3464dSDouglas Gregor if (E->requiresZeroInitialization()) 256e3b3464dSDouglas Gregor EmitNullInitialization(Dest, E->getType()); 257e3b3464dSDouglas Gregor 258630c76efSDouglas Gregor 259630c76efSDouglas Gregor // If this is a call to a trivial default constructor, do nothing. 260630c76efSDouglas Gregor if (CD->isTrivial() && CD->isDefaultConstructor()) 26127da15baSAnders Carlsson return; 262630c76efSDouglas Gregor 26327da15baSAnders Carlsson // Code gen optimization to eliminate copy constructor and return 264222cf0efSDouglas Gregor // its first argument instead, if in fact that argument is a temporary 265222cf0efSDouglas Gregor // object. 26627da15baSAnders Carlsson if (getContext().getLangOptions().ElideConstructors && E->isElidable()) { 267222cf0efSDouglas Gregor if (const Expr *Arg = E->getArg(0)->getTemporaryObject()) { 26827da15baSAnders Carlsson EmitAggExpr(Arg, Dest, false); 26927da15baSAnders Carlsson return; 27027da15baSAnders Carlsson } 271222cf0efSDouglas Gregor } 272630c76efSDouglas Gregor 273630c76efSDouglas Gregor const ConstantArrayType *Array 274630c76efSDouglas Gregor = getContext().getAsConstantArrayType(E->getType()); 27527da15baSAnders Carlsson if (Array) { 27627da15baSAnders Carlsson QualType BaseElementTy = getContext().getBaseElementType(Array); 27727da15baSAnders Carlsson const llvm::Type *BasePtr = ConvertType(BaseElementTy); 27827da15baSAnders Carlsson BasePtr = llvm::PointerType::getUnqual(BasePtr); 27927da15baSAnders Carlsson llvm::Value *BaseAddrPtr = 28027da15baSAnders Carlsson Builder.CreateBitCast(Dest, BasePtr); 28127da15baSAnders Carlsson 28227da15baSAnders Carlsson EmitCXXAggrConstructorCall(CD, Array, BaseAddrPtr, 28327da15baSAnders Carlsson E->arg_begin(), E->arg_end()); 28427da15baSAnders Carlsson } 285e11f9ce9SAnders Carlsson else { 286e11f9ce9SAnders Carlsson CXXCtorType Type = 287e11f9ce9SAnders Carlsson (E->getConstructionKind() == CXXConstructExpr::CK_Complete) 288e11f9ce9SAnders Carlsson ? Ctor_Complete : Ctor_Base; 289e11f9ce9SAnders Carlsson bool ForVirtualBase = 290e11f9ce9SAnders Carlsson E->getConstructionKind() == CXXConstructExpr::CK_VirtualBase; 291e11f9ce9SAnders Carlsson 29227da15baSAnders Carlsson // Call the constructor. 293e11f9ce9SAnders Carlsson EmitCXXConstructorCall(CD, Type, ForVirtualBase, Dest, 29427da15baSAnders Carlsson E->arg_begin(), E->arg_end()); 29527da15baSAnders Carlsson } 296e11f9ce9SAnders Carlsson } 29727da15baSAnders Carlsson 2983eb55cfeSKen Dyck static CharUnits CalculateCookiePadding(ASTContext &Ctx, QualType ElementType) { 29921122cf6SAnders Carlsson const RecordType *RT = ElementType->getAs<RecordType>(); 30059486a2dSAnders Carlsson if (!RT) 3013eb55cfeSKen Dyck return CharUnits::Zero(); 30259486a2dSAnders Carlsson 30359486a2dSAnders Carlsson const CXXRecordDecl *RD = dyn_cast<CXXRecordDecl>(RT->getDecl()); 30459486a2dSAnders Carlsson if (!RD) 3053eb55cfeSKen Dyck return CharUnits::Zero(); 30659486a2dSAnders Carlsson 30759486a2dSAnders Carlsson // Check if the class has a trivial destructor. 30859486a2dSAnders Carlsson if (RD->hasTrivialDestructor()) { 30921122cf6SAnders Carlsson // Check if the usual deallocation function takes two arguments. 310adbe4249SAnders Carlsson const CXXMethodDecl *UsualDeallocationFunction = 0; 311adbe4249SAnders Carlsson 31221122cf6SAnders Carlsson DeclarationName OpName = 31321122cf6SAnders Carlsson Ctx.DeclarationNames.getCXXOperatorName(OO_Array_Delete); 31421122cf6SAnders Carlsson DeclContext::lookup_const_iterator Op, OpEnd; 31521122cf6SAnders Carlsson for (llvm::tie(Op, OpEnd) = RD->lookup(OpName); 31621122cf6SAnders Carlsson Op != OpEnd; ++Op) { 317adbe4249SAnders Carlsson const CXXMethodDecl *Delete = cast<CXXMethodDecl>(*Op); 31821122cf6SAnders Carlsson 31921122cf6SAnders Carlsson if (Delete->isUsualDeallocationFunction()) { 320adbe4249SAnders Carlsson UsualDeallocationFunction = Delete; 32121122cf6SAnders Carlsson break; 32221122cf6SAnders Carlsson } 32321122cf6SAnders Carlsson } 324adbe4249SAnders Carlsson 325adbe4249SAnders Carlsson // No usual deallocation function, we don't need a cookie. 326adbe4249SAnders Carlsson if (!UsualDeallocationFunction) 3273eb55cfeSKen Dyck return CharUnits::Zero(); 328adbe4249SAnders Carlsson 329adbe4249SAnders Carlsson // The usual deallocation function doesn't take a size_t argument, so we 330adbe4249SAnders Carlsson // don't need a cookie. 331adbe4249SAnders Carlsson if (UsualDeallocationFunction->getNumParams() == 1) 3323eb55cfeSKen Dyck return CharUnits::Zero(); 333adbe4249SAnders Carlsson 334adbe4249SAnders Carlsson assert(UsualDeallocationFunction->getNumParams() == 2 && 335adbe4249SAnders Carlsson "Unexpected deallocation function type!"); 33659486a2dSAnders Carlsson } 33759486a2dSAnders Carlsson 33821122cf6SAnders Carlsson // Padding is the maximum of sizeof(size_t) and alignof(ElementType) 3393eb55cfeSKen Dyck return std::max(Ctx.getTypeSizeInChars(Ctx.getSizeType()), 3403eb55cfeSKen Dyck Ctx.getTypeAlignInChars(ElementType)); 34121122cf6SAnders Carlsson } 34221122cf6SAnders Carlsson 3433eb55cfeSKen Dyck static CharUnits CalculateCookiePadding(ASTContext &Ctx, const CXXNewExpr *E) { 34421122cf6SAnders Carlsson if (!E->isArray()) 3453eb55cfeSKen Dyck return CharUnits::Zero(); 34621122cf6SAnders Carlsson 347399f499fSAnders Carlsson // No cookie is required if the new operator being used is 348399f499fSAnders Carlsson // ::operator new[](size_t, void*). 349399f499fSAnders Carlsson const FunctionDecl *OperatorNew = E->getOperatorNew(); 350399f499fSAnders Carlsson if (OperatorNew->getDeclContext()->getLookupContext()->isFileContext()) { 351399f499fSAnders Carlsson if (OperatorNew->getNumParams() == 2) { 352399f499fSAnders Carlsson CanQualType ParamType = 353399f499fSAnders Carlsson Ctx.getCanonicalType(OperatorNew->getParamDecl(1)->getType()); 354399f499fSAnders Carlsson 355399f499fSAnders Carlsson if (ParamType == Ctx.VoidPtrTy) 3563eb55cfeSKen Dyck return CharUnits::Zero(); 357399f499fSAnders Carlsson } 358399f499fSAnders Carlsson } 359399f499fSAnders Carlsson 36021122cf6SAnders Carlsson return CalculateCookiePadding(Ctx, E->getAllocatedType()); 36159486a2dSAnders Carlsson } 36259486a2dSAnders Carlsson 36347b4629bSFariborz Jahanian static llvm::Value *EmitCXXNewAllocSize(ASTContext &Context, 36447b4629bSFariborz Jahanian CodeGenFunction &CGF, 36559486a2dSAnders Carlsson const CXXNewExpr *E, 36605fc5be3SDouglas Gregor llvm::Value *&NumElements, 36705fc5be3SDouglas Gregor llvm::Value *&SizeWithoutCookie) { 36859486a2dSAnders Carlsson QualType Type = E->getAllocatedType(); 3693eb55cfeSKen Dyck CharUnits TypeSize = CGF.getContext().getTypeSizeInChars(Type); 37059486a2dSAnders Carlsson const llvm::Type *SizeTy = CGF.ConvertType(CGF.getContext().getSizeType()); 37159486a2dSAnders Carlsson 37205fc5be3SDouglas Gregor if (!E->isArray()) { 37305fc5be3SDouglas Gregor SizeWithoutCookie = llvm::ConstantInt::get(SizeTy, TypeSize.getQuantity()); 37405fc5be3SDouglas Gregor return SizeWithoutCookie; 37505fc5be3SDouglas Gregor } 37659486a2dSAnders Carlsson 37759486a2dSAnders Carlsson // Emit the array size expression. 37859486a2dSAnders Carlsson NumElements = CGF.EmitScalarExpr(E->getArraySize()); 37959486a2dSAnders Carlsson 38032ac583dSChris Lattner llvm::Value *Size = llvm::ConstantInt::get(SizeTy, TypeSize.getQuantity()); 38132ac583dSChris Lattner 38232ac583dSChris Lattner // If someone is doing 'new int[42]' there is no need to do a dynamic check. 38332ac583dSChris Lattner // Don't bloat the -O0 code. 38432ac583dSChris Lattner if (llvm::ConstantInt *NumElementsC = 38532ac583dSChris Lattner dyn_cast<llvm::ConstantInt>(NumElements)) { 38632ac583dSChris Lattner // Determine if there is an overflow here by doing an extended multiply. 38732ac583dSChris Lattner llvm::APInt NEC = NumElementsC->getValue(); 38832ac583dSChris Lattner NEC.zext(NEC.getBitWidth()*2); 38932ac583dSChris Lattner 39032ac583dSChris Lattner llvm::APInt SC = cast<llvm::ConstantInt>(Size)->getValue(); 39132ac583dSChris Lattner SC.zext(SC.getBitWidth()*2); 39232ac583dSChris Lattner SC *= NEC; 39332ac583dSChris Lattner 39432ac583dSChris Lattner if (SC.countLeadingZeros() >= NumElementsC->getValue().getBitWidth()) { 39532ac583dSChris Lattner SC.trunc(NumElementsC->getValue().getBitWidth()); 39632ac583dSChris Lattner Size = llvm::ConstantInt::get(Size->getContext(), SC); 39732ac583dSChris Lattner } else { 39832ac583dSChris Lattner // On overflow, produce a -1 so operator new throws. 39932ac583dSChris Lattner Size = llvm::Constant::getAllOnesValue(Size->getType()); 40032ac583dSChris Lattner } 40132ac583dSChris Lattner 40232ac583dSChris Lattner } else { 40326008e07SChris Lattner // Multiply with the type size. This multiply can overflow, e.g. in: 40426008e07SChris Lattner // new double[n] 40526008e07SChris Lattner // where n is 2^30 on a 32-bit machine or 2^62 on a 64-bit machine. Because 40626008e07SChris Lattner // of this, we need to detect the overflow and ensure that an exception is 407f2f38701SChris Lattner // called by forcing the size to -1 on overflow. 40832ac583dSChris Lattner llvm::Value *UMulF = 40932ac583dSChris Lattner CGF.CGM.getIntrinsic(llvm::Intrinsic::umul_with_overflow, &SizeTy, 1); 41032ac583dSChris Lattner llvm::Value *MulRes = CGF.Builder.CreateCall2(UMulF, NumElements, Size); 41132ac583dSChris Lattner // Branch on the overflow bit to the overflow block, which is lazily 41232ac583dSChris Lattner // created. 41326008e07SChris Lattner llvm::Value *DidOverflow = CGF.Builder.CreateExtractValue(MulRes, 1); 41426008e07SChris Lattner // Get the normal result of the multiplication. 41526008e07SChris Lattner llvm::Value *V = CGF.Builder.CreateExtractValue(MulRes, 0); 41647b4629bSFariborz Jahanian 417f2f38701SChris Lattner llvm::BasicBlock *NormalBB = CGF.createBasicBlock("no_overflow"); 418f2f38701SChris Lattner llvm::BasicBlock *OverflowBB = CGF.createBasicBlock("overflow"); 419f2f38701SChris Lattner 420f2f38701SChris Lattner CGF.Builder.CreateCondBr(DidOverflow, OverflowBB, NormalBB); 421f2f38701SChris Lattner 422f2f38701SChris Lattner llvm::BasicBlock *PrevBB = CGF.Builder.GetInsertBlock(); 423f2f38701SChris Lattner 424f2f38701SChris Lattner // We just need the overflow block to build a PHI node. 425f2f38701SChris Lattner CGF.EmitBlock(OverflowBB); 426f2f38701SChris Lattner CGF.EmitBlock(NormalBB); 427f2f38701SChris Lattner 428f2f38701SChris Lattner llvm::PHINode *PN = CGF.Builder.CreatePHI(V->getType()); 429f2f38701SChris Lattner 430f2f38701SChris Lattner PN->addIncoming(V, PrevBB); 431f2f38701SChris Lattner PN->addIncoming(llvm::Constant::getAllOnesValue(V->getType()), OverflowBB); 43232ac583dSChris Lattner Size = PN; 43332ac583dSChris Lattner } 43405fc5be3SDouglas Gregor SizeWithoutCookie = Size; 435f2f38701SChris Lattner 43632ac583dSChris Lattner // Add the cookie padding if necessary. 43732ac583dSChris Lattner CharUnits CookiePadding = CalculateCookiePadding(CGF.getContext(), E); 4383eb55cfeSKen Dyck if (!CookiePadding.isZero()) 43932ac583dSChris Lattner Size = CGF.Builder.CreateAdd(Size, 4403eb55cfeSKen Dyck llvm::ConstantInt::get(SizeTy, CookiePadding.getQuantity())); 44159486a2dSAnders Carlsson 44232ac583dSChris Lattner return Size; 44359486a2dSAnders Carlsson } 44459486a2dSAnders Carlsson 445d5202e09SFariborz Jahanian static void StoreAnyExprIntoOneUnit(CodeGenFunction &CGF, const CXXNewExpr *E, 446d5202e09SFariborz Jahanian llvm::Value *NewPtr) { 447d5202e09SFariborz Jahanian 448d5202e09SFariborz Jahanian assert(E->getNumConstructorArgs() == 1 && 449d5202e09SFariborz Jahanian "Can only have one argument to initializer of POD type."); 450d5202e09SFariborz Jahanian 451d5202e09SFariborz Jahanian const Expr *Init = E->getConstructorArg(0); 452d5202e09SFariborz Jahanian QualType AllocType = E->getAllocatedType(); 453d5202e09SFariborz Jahanian 4540381634aSDaniel Dunbar unsigned Alignment = 4550381634aSDaniel Dunbar CGF.getContext().getTypeAlignInChars(AllocType).getQuantity(); 456d5202e09SFariborz Jahanian if (!CGF.hasAggregateLLVMType(AllocType)) 457d5202e09SFariborz Jahanian CGF.EmitStoreOfScalar(CGF.EmitScalarExpr(Init), NewPtr, 4580381634aSDaniel Dunbar AllocType.isVolatileQualified(), Alignment, 4590381634aSDaniel Dunbar AllocType); 460d5202e09SFariborz Jahanian else if (AllocType->isAnyComplexType()) 461d5202e09SFariborz Jahanian CGF.EmitComplexExprIntoAddr(Init, NewPtr, 462d5202e09SFariborz Jahanian AllocType.isVolatileQualified()); 463d5202e09SFariborz Jahanian else 464d5202e09SFariborz Jahanian CGF.EmitAggExpr(Init, NewPtr, AllocType.isVolatileQualified()); 465d5202e09SFariborz Jahanian } 466d5202e09SFariborz Jahanian 467d5202e09SFariborz Jahanian void 468d5202e09SFariborz Jahanian CodeGenFunction::EmitNewArrayInitializer(const CXXNewExpr *E, 469d5202e09SFariborz Jahanian llvm::Value *NewPtr, 470d5202e09SFariborz Jahanian llvm::Value *NumElements) { 471b66b08efSFariborz Jahanian // We have a POD type. 472b66b08efSFariborz Jahanian if (E->getNumConstructorArgs() == 0) 473b66b08efSFariborz Jahanian return; 474b66b08efSFariborz Jahanian 475d5202e09SFariborz Jahanian const llvm::Type *SizeTy = ConvertType(getContext().getSizeType()); 476d5202e09SFariborz Jahanian 477d5202e09SFariborz Jahanian // Create a temporary for the loop index and initialize it with 0. 478d5202e09SFariborz Jahanian llvm::Value *IndexPtr = CreateTempAlloca(SizeTy, "loop.index"); 479d5202e09SFariborz Jahanian llvm::Value *Zero = llvm::Constant::getNullValue(SizeTy); 480d5202e09SFariborz Jahanian Builder.CreateStore(Zero, IndexPtr); 481d5202e09SFariborz Jahanian 482d5202e09SFariborz Jahanian // Start the loop with a block that tests the condition. 483d5202e09SFariborz Jahanian llvm::BasicBlock *CondBlock = createBasicBlock("for.cond"); 484d5202e09SFariborz Jahanian llvm::BasicBlock *AfterFor = createBasicBlock("for.end"); 485d5202e09SFariborz Jahanian 486d5202e09SFariborz Jahanian EmitBlock(CondBlock); 487d5202e09SFariborz Jahanian 488d5202e09SFariborz Jahanian llvm::BasicBlock *ForBody = createBasicBlock("for.body"); 489d5202e09SFariborz Jahanian 490d5202e09SFariborz Jahanian // Generate: if (loop-index < number-of-elements fall to the loop body, 491d5202e09SFariborz Jahanian // otherwise, go to the block after the for-loop. 492d5202e09SFariborz Jahanian llvm::Value *Counter = Builder.CreateLoad(IndexPtr); 493d5202e09SFariborz Jahanian llvm::Value *IsLess = Builder.CreateICmpULT(Counter, NumElements, "isless"); 494d5202e09SFariborz Jahanian // If the condition is true, execute the body. 495d5202e09SFariborz Jahanian Builder.CreateCondBr(IsLess, ForBody, AfterFor); 496d5202e09SFariborz Jahanian 497d5202e09SFariborz Jahanian EmitBlock(ForBody); 498d5202e09SFariborz Jahanian 499d5202e09SFariborz Jahanian llvm::BasicBlock *ContinueBlock = createBasicBlock("for.inc"); 500d5202e09SFariborz Jahanian // Inside the loop body, emit the constructor call on the array element. 501d5202e09SFariborz Jahanian Counter = Builder.CreateLoad(IndexPtr); 502d5202e09SFariborz Jahanian llvm::Value *Address = Builder.CreateInBoundsGEP(NewPtr, Counter, 503d5202e09SFariborz Jahanian "arrayidx"); 504d5202e09SFariborz Jahanian StoreAnyExprIntoOneUnit(*this, E, Address); 505d5202e09SFariborz Jahanian 506d5202e09SFariborz Jahanian EmitBlock(ContinueBlock); 507d5202e09SFariborz Jahanian 508d5202e09SFariborz Jahanian // Emit the increment of the loop counter. 509d5202e09SFariborz Jahanian llvm::Value *NextVal = llvm::ConstantInt::get(SizeTy, 1); 510d5202e09SFariborz Jahanian Counter = Builder.CreateLoad(IndexPtr); 511d5202e09SFariborz Jahanian NextVal = Builder.CreateAdd(Counter, NextVal, "inc"); 512d5202e09SFariborz Jahanian Builder.CreateStore(NextVal, IndexPtr); 513d5202e09SFariborz Jahanian 514d5202e09SFariborz Jahanian // Finally, branch back up to the condition for the next iteration. 515d5202e09SFariborz Jahanian EmitBranch(CondBlock); 516d5202e09SFariborz Jahanian 517d5202e09SFariborz Jahanian // Emit the fall-through block. 518d5202e09SFariborz Jahanian EmitBlock(AfterFor, true); 519d5202e09SFariborz Jahanian } 520d5202e09SFariborz Jahanian 52105fc5be3SDouglas Gregor static void EmitZeroMemSet(CodeGenFunction &CGF, QualType T, 52205fc5be3SDouglas Gregor llvm::Value *NewPtr, llvm::Value *Size) { 52305fc5be3SDouglas Gregor llvm::LLVMContext &VMContext = CGF.CGM.getLLVMContext(); 52405fc5be3SDouglas Gregor const llvm::Type *BP = llvm::Type::getInt8PtrTy(VMContext); 52505fc5be3SDouglas Gregor if (NewPtr->getType() != BP) 52605fc5be3SDouglas Gregor NewPtr = CGF.Builder.CreateBitCast(NewPtr, BP, "tmp"); 52705fc5be3SDouglas Gregor 52805fc5be3SDouglas Gregor CGF.Builder.CreateCall5(CGF.CGM.getMemSetFn(BP, CGF.IntPtrTy), NewPtr, 52905fc5be3SDouglas Gregor llvm::Constant::getNullValue(llvm::Type::getInt8Ty(VMContext)), 53005fc5be3SDouglas Gregor Size, 53105fc5be3SDouglas Gregor llvm::ConstantInt::get(CGF.Int32Ty, 53205fc5be3SDouglas Gregor CGF.getContext().getTypeAlign(T)/8), 53305fc5be3SDouglas Gregor llvm::ConstantInt::get(llvm::Type::getInt1Ty(VMContext), 53405fc5be3SDouglas Gregor 0)); 53505fc5be3SDouglas Gregor } 53605fc5be3SDouglas Gregor 53759486a2dSAnders Carlsson static void EmitNewInitializer(CodeGenFunction &CGF, const CXXNewExpr *E, 53859486a2dSAnders Carlsson llvm::Value *NewPtr, 53905fc5be3SDouglas Gregor llvm::Value *NumElements, 54005fc5be3SDouglas Gregor llvm::Value *AllocSizeWithoutCookie) { 5413a202f60SAnders Carlsson if (E->isArray()) { 542d040e6b2SAnders Carlsson if (CXXConstructorDecl *Ctor = E->getConstructor()) { 54305fc5be3SDouglas Gregor bool RequiresZeroInitialization = false; 54405fc5be3SDouglas Gregor if (Ctor->getParent()->hasTrivialConstructor()) { 54505fc5be3SDouglas Gregor // If new expression did not specify value-initialization, then there 54605fc5be3SDouglas Gregor // is no initialization. 54705fc5be3SDouglas Gregor if (!E->hasInitializer() || Ctor->getParent()->isEmpty()) 54805fc5be3SDouglas Gregor return; 54905fc5be3SDouglas Gregor 550*614dbdcdSJohn McCall if (CGF.CGM.getTypes().isZeroInitializable(E->getAllocatedType())) { 55105fc5be3SDouglas Gregor // Optimization: since zero initialization will just set the memory 55205fc5be3SDouglas Gregor // to all zeroes, generate a single memset to do it in one shot. 55305fc5be3SDouglas Gregor EmitZeroMemSet(CGF, E->getAllocatedType(), NewPtr, 55405fc5be3SDouglas Gregor AllocSizeWithoutCookie); 5553a202f60SAnders Carlsson return; 5563a202f60SAnders Carlsson } 55705fc5be3SDouglas Gregor 55805fc5be3SDouglas Gregor RequiresZeroInitialization = true; 55905fc5be3SDouglas Gregor } 56005fc5be3SDouglas Gregor 56105fc5be3SDouglas Gregor CGF.EmitCXXAggrConstructorCall(Ctor, NumElements, NewPtr, 56205fc5be3SDouglas Gregor E->constructor_arg_begin(), 56305fc5be3SDouglas Gregor E->constructor_arg_end(), 56405fc5be3SDouglas Gregor RequiresZeroInitialization); 56505fc5be3SDouglas Gregor return; 56605fc5be3SDouglas Gregor } else if (E->getNumConstructorArgs() == 1 && 56705fc5be3SDouglas Gregor isa<ImplicitValueInitExpr>(E->getConstructorArg(0))) { 56805fc5be3SDouglas Gregor // Optimization: since zero initialization will just set the memory 56905fc5be3SDouglas Gregor // to all zeroes, generate a single memset to do it in one shot. 57005fc5be3SDouglas Gregor EmitZeroMemSet(CGF, E->getAllocatedType(), NewPtr, 57105fc5be3SDouglas Gregor AllocSizeWithoutCookie); 57205fc5be3SDouglas Gregor return; 57305fc5be3SDouglas Gregor } else { 574d5202e09SFariborz Jahanian CGF.EmitNewArrayInitializer(E, NewPtr, NumElements); 575d5202e09SFariborz Jahanian return; 576d040e6b2SAnders Carlsson } 577d5202e09SFariborz Jahanian } 57859486a2dSAnders Carlsson 57959486a2dSAnders Carlsson if (CXXConstructorDecl *Ctor = E->getConstructor()) { 580747eb784SDouglas Gregor // Per C++ [expr.new]p15, if we have an initializer, then we're performing 581747eb784SDouglas Gregor // direct initialization. C++ [dcl.init]p5 requires that we 582747eb784SDouglas Gregor // zero-initialize storage if there are no user-declared constructors. 583747eb784SDouglas Gregor if (E->hasInitializer() && 584747eb784SDouglas Gregor !Ctor->getParent()->hasUserDeclaredConstructor() && 585747eb784SDouglas Gregor !Ctor->getParent()->isEmpty()) 586747eb784SDouglas Gregor CGF.EmitNullInitialization(NewPtr, E->getAllocatedType()); 587747eb784SDouglas Gregor 588e11f9ce9SAnders Carlsson CGF.EmitCXXConstructorCall(Ctor, Ctor_Complete, /*ForVirtualBase=*/false, 589e11f9ce9SAnders Carlsson NewPtr, E->constructor_arg_begin(), 59059486a2dSAnders Carlsson E->constructor_arg_end()); 59159486a2dSAnders Carlsson 59259486a2dSAnders Carlsson return; 59359486a2dSAnders Carlsson } 594b66b08efSFariborz Jahanian // We have a POD type. 595b66b08efSFariborz Jahanian if (E->getNumConstructorArgs() == 0) 596b66b08efSFariborz Jahanian return; 59759486a2dSAnders Carlsson 598d5202e09SFariborz Jahanian StoreAnyExprIntoOneUnit(CGF, E, NewPtr); 59959486a2dSAnders Carlsson } 60059486a2dSAnders Carlsson 60159486a2dSAnders Carlsson llvm::Value *CodeGenFunction::EmitCXXNewExpr(const CXXNewExpr *E) { 60259486a2dSAnders Carlsson QualType AllocType = E->getAllocatedType(); 60359486a2dSAnders Carlsson FunctionDecl *NewFD = E->getOperatorNew(); 60459486a2dSAnders Carlsson const FunctionProtoType *NewFTy = NewFD->getType()->getAs<FunctionProtoType>(); 60559486a2dSAnders Carlsson 60659486a2dSAnders Carlsson CallArgList NewArgs; 60759486a2dSAnders Carlsson 60859486a2dSAnders Carlsson // The allocation size is the first argument. 60959486a2dSAnders Carlsson QualType SizeTy = getContext().getSizeType(); 61059486a2dSAnders Carlsson 61159486a2dSAnders Carlsson llvm::Value *NumElements = 0; 61205fc5be3SDouglas Gregor llvm::Value *AllocSizeWithoutCookie = 0; 61347b4629bSFariborz Jahanian llvm::Value *AllocSize = EmitCXXNewAllocSize(getContext(), 61405fc5be3SDouglas Gregor *this, E, NumElements, 61505fc5be3SDouglas Gregor AllocSizeWithoutCookie); 61659486a2dSAnders Carlsson 61759486a2dSAnders Carlsson NewArgs.push_back(std::make_pair(RValue::get(AllocSize), SizeTy)); 61859486a2dSAnders Carlsson 61959486a2dSAnders Carlsson // Emit the rest of the arguments. 62059486a2dSAnders Carlsson // FIXME: Ideally, this should just use EmitCallArgs. 62159486a2dSAnders Carlsson CXXNewExpr::const_arg_iterator NewArg = E->placement_arg_begin(); 62259486a2dSAnders Carlsson 62359486a2dSAnders Carlsson // First, use the types from the function type. 62459486a2dSAnders Carlsson // We start at 1 here because the first argument (the allocation size) 62559486a2dSAnders Carlsson // has already been emitted. 62659486a2dSAnders Carlsson for (unsigned i = 1, e = NewFTy->getNumArgs(); i != e; ++i, ++NewArg) { 62759486a2dSAnders Carlsson QualType ArgType = NewFTy->getArgType(i); 62859486a2dSAnders Carlsson 62959486a2dSAnders Carlsson assert(getContext().getCanonicalType(ArgType.getNonReferenceType()). 63059486a2dSAnders Carlsson getTypePtr() == 63159486a2dSAnders Carlsson getContext().getCanonicalType(NewArg->getType()).getTypePtr() && 63259486a2dSAnders Carlsson "type mismatch in call argument!"); 63359486a2dSAnders Carlsson 63459486a2dSAnders Carlsson NewArgs.push_back(std::make_pair(EmitCallArg(*NewArg, ArgType), 63559486a2dSAnders Carlsson ArgType)); 63659486a2dSAnders Carlsson 63759486a2dSAnders Carlsson } 63859486a2dSAnders Carlsson 63959486a2dSAnders Carlsson // Either we've emitted all the call args, or we have a call to a 64059486a2dSAnders Carlsson // variadic function. 64159486a2dSAnders Carlsson assert((NewArg == E->placement_arg_end() || NewFTy->isVariadic()) && 64259486a2dSAnders Carlsson "Extra arguments in non-variadic function!"); 64359486a2dSAnders Carlsson 64459486a2dSAnders Carlsson // If we still have any arguments, emit them using the type of the argument. 64559486a2dSAnders Carlsson for (CXXNewExpr::const_arg_iterator NewArgEnd = E->placement_arg_end(); 64659486a2dSAnders Carlsson NewArg != NewArgEnd; ++NewArg) { 64759486a2dSAnders Carlsson QualType ArgType = NewArg->getType(); 64859486a2dSAnders Carlsson NewArgs.push_back(std::make_pair(EmitCallArg(*NewArg, ArgType), 64959486a2dSAnders Carlsson ArgType)); 65059486a2dSAnders Carlsson } 65159486a2dSAnders Carlsson 65259486a2dSAnders Carlsson // Emit the call to new. 65359486a2dSAnders Carlsson RValue RV = 654ab26cfa5SJohn McCall EmitCall(CGM.getTypes().getFunctionInfo(NewArgs, NewFTy), 65561a401caSAnders Carlsson CGM.GetAddrOfFunction(NewFD), ReturnValueSlot(), NewArgs, NewFD); 65659486a2dSAnders Carlsson 65759486a2dSAnders Carlsson // If an allocation function is declared with an empty exception specification 65859486a2dSAnders Carlsson // it returns null to indicate failure to allocate storage. [expr.new]p13. 65959486a2dSAnders Carlsson // (We don't need to check for null when there's no new initializer and 66059486a2dSAnders Carlsson // we're allocating a POD type). 66159486a2dSAnders Carlsson bool NullCheckResult = NewFTy->hasEmptyExceptionSpec() && 66259486a2dSAnders Carlsson !(AllocType->isPODType() && !E->hasInitializer()); 66359486a2dSAnders Carlsson 66459486a2dSAnders Carlsson llvm::BasicBlock *NewNull = 0; 66559486a2dSAnders Carlsson llvm::BasicBlock *NewNotNull = 0; 66659486a2dSAnders Carlsson llvm::BasicBlock *NewEnd = 0; 66759486a2dSAnders Carlsson 66859486a2dSAnders Carlsson llvm::Value *NewPtr = RV.getScalarVal(); 66959486a2dSAnders Carlsson 67059486a2dSAnders Carlsson if (NullCheckResult) { 67159486a2dSAnders Carlsson NewNull = createBasicBlock("new.null"); 67259486a2dSAnders Carlsson NewNotNull = createBasicBlock("new.notnull"); 67359486a2dSAnders Carlsson NewEnd = createBasicBlock("new.end"); 67459486a2dSAnders Carlsson 67559486a2dSAnders Carlsson llvm::Value *IsNull = 67659486a2dSAnders Carlsson Builder.CreateICmpEQ(NewPtr, 67759486a2dSAnders Carlsson llvm::Constant::getNullValue(NewPtr->getType()), 67859486a2dSAnders Carlsson "isnull"); 67959486a2dSAnders Carlsson 68059486a2dSAnders Carlsson Builder.CreateCondBr(IsNull, NewNull, NewNotNull); 68159486a2dSAnders Carlsson EmitBlock(NewNotNull); 68259486a2dSAnders Carlsson } 68359486a2dSAnders Carlsson 6843eb55cfeSKen Dyck CharUnits CookiePadding = CalculateCookiePadding(getContext(), E); 6853eb55cfeSKen Dyck if (!CookiePadding.isZero()) { 6863eb55cfeSKen Dyck CharUnits CookieOffset = 6873eb55cfeSKen Dyck CookiePadding - getContext().getTypeSizeInChars(SizeTy); 68859486a2dSAnders Carlsson 68959486a2dSAnders Carlsson llvm::Value *NumElementsPtr = 6903eb55cfeSKen Dyck Builder.CreateConstInBoundsGEP1_64(NewPtr, CookieOffset.getQuantity()); 69159486a2dSAnders Carlsson 69259486a2dSAnders Carlsson NumElementsPtr = Builder.CreateBitCast(NumElementsPtr, 69359486a2dSAnders Carlsson ConvertType(SizeTy)->getPointerTo()); 69459486a2dSAnders Carlsson Builder.CreateStore(NumElements, NumElementsPtr); 69559486a2dSAnders Carlsson 69659486a2dSAnders Carlsson // Now add the padding to the new ptr. 6973eb55cfeSKen Dyck NewPtr = Builder.CreateConstInBoundsGEP1_64(NewPtr, 6983eb55cfeSKen Dyck CookiePadding.getQuantity()); 69959486a2dSAnders Carlsson } 70059486a2dSAnders Carlsson 70147b4629bSFariborz Jahanian if (AllocType->isArrayType()) { 70247b4629bSFariborz Jahanian while (const ArrayType *AType = getContext().getAsArrayType(AllocType)) 70347b4629bSFariborz Jahanian AllocType = AType->getElementType(); 70447b4629bSFariborz Jahanian NewPtr = 70547b4629bSFariborz Jahanian Builder.CreateBitCast(NewPtr, 70647b4629bSFariborz Jahanian ConvertType(getContext().getPointerType(AllocType))); 70705fc5be3SDouglas Gregor EmitNewInitializer(*this, E, NewPtr, NumElements, AllocSizeWithoutCookie); 70847b4629bSFariborz Jahanian NewPtr = Builder.CreateBitCast(NewPtr, ConvertType(E->getType())); 70947b4629bSFariborz Jahanian } 71047b4629bSFariborz Jahanian else { 71147b4629bSFariborz Jahanian NewPtr = Builder.CreateBitCast(NewPtr, ConvertType(E->getType())); 71205fc5be3SDouglas Gregor EmitNewInitializer(*this, E, NewPtr, NumElements, AllocSizeWithoutCookie); 71347b4629bSFariborz Jahanian } 71459486a2dSAnders Carlsson 71559486a2dSAnders Carlsson if (NullCheckResult) { 71659486a2dSAnders Carlsson Builder.CreateBr(NewEnd); 71759486a2dSAnders Carlsson NewNotNull = Builder.GetInsertBlock(); 71859486a2dSAnders Carlsson EmitBlock(NewNull); 71959486a2dSAnders Carlsson Builder.CreateBr(NewEnd); 72059486a2dSAnders Carlsson EmitBlock(NewEnd); 72159486a2dSAnders Carlsson 72259486a2dSAnders Carlsson llvm::PHINode *PHI = Builder.CreatePHI(NewPtr->getType()); 72359486a2dSAnders Carlsson PHI->reserveOperandSpace(2); 72459486a2dSAnders Carlsson PHI->addIncoming(NewPtr, NewNotNull); 72559486a2dSAnders Carlsson PHI->addIncoming(llvm::Constant::getNullValue(NewPtr->getType()), NewNull); 72659486a2dSAnders Carlsson 72759486a2dSAnders Carlsson NewPtr = PHI; 72859486a2dSAnders Carlsson } 72959486a2dSAnders Carlsson 73059486a2dSAnders Carlsson return NewPtr; 73159486a2dSAnders Carlsson } 73259486a2dSAnders Carlsson 73321122cf6SAnders Carlsson static std::pair<llvm::Value *, llvm::Value *> 73421122cf6SAnders Carlsson GetAllocatedObjectPtrAndNumElements(CodeGenFunction &CGF, 73521122cf6SAnders Carlsson llvm::Value *Ptr, QualType DeleteTy) { 73621122cf6SAnders Carlsson QualType SizeTy = CGF.getContext().getSizeType(); 73721122cf6SAnders Carlsson const llvm::Type *SizeLTy = CGF.ConvertType(SizeTy); 73821122cf6SAnders Carlsson 7393eb55cfeSKen Dyck CharUnits DeleteTypeAlign = CGF.getContext().getTypeAlignInChars(DeleteTy); 7403eb55cfeSKen Dyck CharUnits CookiePadding = 7413eb55cfeSKen Dyck std::max(CGF.getContext().getTypeSizeInChars(SizeTy), 7423eb55cfeSKen Dyck DeleteTypeAlign); 7433eb55cfeSKen Dyck assert(!CookiePadding.isZero() && "CookiePadding should not be 0."); 74421122cf6SAnders Carlsson 74521122cf6SAnders Carlsson const llvm::Type *Int8PtrTy = llvm::Type::getInt8PtrTy(CGF.getLLVMContext()); 7463eb55cfeSKen Dyck CharUnits CookieOffset = 7473eb55cfeSKen Dyck CookiePadding - CGF.getContext().getTypeSizeInChars(SizeTy); 74821122cf6SAnders Carlsson 74921122cf6SAnders Carlsson llvm::Value *AllocatedObjectPtr = CGF.Builder.CreateBitCast(Ptr, Int8PtrTy); 75021122cf6SAnders Carlsson AllocatedObjectPtr = 75121122cf6SAnders Carlsson CGF.Builder.CreateConstInBoundsGEP1_64(AllocatedObjectPtr, 7523eb55cfeSKen Dyck -CookiePadding.getQuantity()); 75321122cf6SAnders Carlsson 75421122cf6SAnders Carlsson llvm::Value *NumElementsPtr = 75521122cf6SAnders Carlsson CGF.Builder.CreateConstInBoundsGEP1_64(AllocatedObjectPtr, 7563eb55cfeSKen Dyck CookieOffset.getQuantity()); 75721122cf6SAnders Carlsson NumElementsPtr = 75821122cf6SAnders Carlsson CGF.Builder.CreateBitCast(NumElementsPtr, SizeLTy->getPointerTo()); 75921122cf6SAnders Carlsson 76021122cf6SAnders Carlsson llvm::Value *NumElements = CGF.Builder.CreateLoad(NumElementsPtr); 76121122cf6SAnders Carlsson NumElements = 76221122cf6SAnders Carlsson CGF.Builder.CreateIntCast(NumElements, SizeLTy, /*isSigned=*/false); 76321122cf6SAnders Carlsson 76421122cf6SAnders Carlsson return std::make_pair(AllocatedObjectPtr, NumElements); 76521122cf6SAnders Carlsson } 76621122cf6SAnders Carlsson 76759486a2dSAnders Carlsson void CodeGenFunction::EmitDeleteCall(const FunctionDecl *DeleteFD, 76859486a2dSAnders Carlsson llvm::Value *Ptr, 76959486a2dSAnders Carlsson QualType DeleteTy) { 77059486a2dSAnders Carlsson const FunctionProtoType *DeleteFTy = 77159486a2dSAnders Carlsson DeleteFD->getType()->getAs<FunctionProtoType>(); 77259486a2dSAnders Carlsson 77359486a2dSAnders Carlsson CallArgList DeleteArgs; 77459486a2dSAnders Carlsson 77521122cf6SAnders Carlsson // Check if we need to pass the size to the delete operator. 77621122cf6SAnders Carlsson llvm::Value *Size = 0; 77721122cf6SAnders Carlsson QualType SizeTy; 77821122cf6SAnders Carlsson if (DeleteFTy->getNumArgs() == 2) { 77921122cf6SAnders Carlsson SizeTy = DeleteFTy->getArgType(1); 7807df3cbebSKen Dyck CharUnits DeleteTypeSize = getContext().getTypeSizeInChars(DeleteTy); 7817df3cbebSKen Dyck Size = llvm::ConstantInt::get(ConvertType(SizeTy), 7827df3cbebSKen Dyck DeleteTypeSize.getQuantity()); 78321122cf6SAnders Carlsson } 78421122cf6SAnders Carlsson 78521122cf6SAnders Carlsson if (DeleteFD->getOverloadedOperator() == OO_Array_Delete && 7863eb55cfeSKen Dyck !CalculateCookiePadding(getContext(), DeleteTy).isZero()) { 78721122cf6SAnders Carlsson // We need to get the number of elements in the array from the cookie. 78821122cf6SAnders Carlsson llvm::Value *AllocatedObjectPtr; 78921122cf6SAnders Carlsson llvm::Value *NumElements; 79021122cf6SAnders Carlsson llvm::tie(AllocatedObjectPtr, NumElements) = 79121122cf6SAnders Carlsson GetAllocatedObjectPtrAndNumElements(*this, Ptr, DeleteTy); 79221122cf6SAnders Carlsson 79321122cf6SAnders Carlsson // Multiply the size with the number of elements. 79421122cf6SAnders Carlsson if (Size) 79521122cf6SAnders Carlsson Size = Builder.CreateMul(NumElements, Size); 79621122cf6SAnders Carlsson 79721122cf6SAnders Carlsson Ptr = AllocatedObjectPtr; 79821122cf6SAnders Carlsson } 79921122cf6SAnders Carlsson 80059486a2dSAnders Carlsson QualType ArgTy = DeleteFTy->getArgType(0); 80159486a2dSAnders Carlsson llvm::Value *DeletePtr = Builder.CreateBitCast(Ptr, ConvertType(ArgTy)); 80259486a2dSAnders Carlsson DeleteArgs.push_back(std::make_pair(RValue::get(DeletePtr), ArgTy)); 80359486a2dSAnders Carlsson 80421122cf6SAnders Carlsson if (Size) 80559486a2dSAnders Carlsson DeleteArgs.push_back(std::make_pair(RValue::get(Size), SizeTy)); 80659486a2dSAnders Carlsson 80759486a2dSAnders Carlsson // Emit the call to delete. 808ab26cfa5SJohn McCall EmitCall(CGM.getTypes().getFunctionInfo(DeleteArgs, DeleteFTy), 80961a401caSAnders Carlsson CGM.GetAddrOfFunction(DeleteFD), ReturnValueSlot(), 81059486a2dSAnders Carlsson DeleteArgs, DeleteFD); 81159486a2dSAnders Carlsson } 81259486a2dSAnders Carlsson 81359486a2dSAnders Carlsson void CodeGenFunction::EmitCXXDeleteExpr(const CXXDeleteExpr *E) { 81459486a2dSAnders Carlsson 81559486a2dSAnders Carlsson // Get at the argument before we performed the implicit conversion 81659486a2dSAnders Carlsson // to void*. 81759486a2dSAnders Carlsson const Expr *Arg = E->getArgument(); 81859486a2dSAnders Carlsson while (const ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(Arg)) { 81959486a2dSAnders Carlsson if (ICE->getCastKind() != CastExpr::CK_UserDefinedConversion && 82059486a2dSAnders Carlsson ICE->getType()->isVoidPointerType()) 82159486a2dSAnders Carlsson Arg = ICE->getSubExpr(); 82259486a2dSAnders Carlsson else 82359486a2dSAnders Carlsson break; 82459486a2dSAnders Carlsson } 82559486a2dSAnders Carlsson 82659486a2dSAnders Carlsson QualType DeleteTy = Arg->getType()->getAs<PointerType>()->getPointeeType(); 82759486a2dSAnders Carlsson 82859486a2dSAnders Carlsson llvm::Value *Ptr = EmitScalarExpr(Arg); 82959486a2dSAnders Carlsson 83059486a2dSAnders Carlsson // Null check the pointer. 83159486a2dSAnders Carlsson llvm::BasicBlock *DeleteNotNull = createBasicBlock("delete.notnull"); 83259486a2dSAnders Carlsson llvm::BasicBlock *DeleteEnd = createBasicBlock("delete.end"); 83359486a2dSAnders Carlsson 83459486a2dSAnders Carlsson llvm::Value *IsNull = 83559486a2dSAnders Carlsson Builder.CreateICmpEQ(Ptr, llvm::Constant::getNullValue(Ptr->getType()), 83659486a2dSAnders Carlsson "isnull"); 83759486a2dSAnders Carlsson 83859486a2dSAnders Carlsson Builder.CreateCondBr(IsNull, DeleteEnd, DeleteNotNull); 83959486a2dSAnders Carlsson EmitBlock(DeleteNotNull); 84059486a2dSAnders Carlsson 84159486a2dSAnders Carlsson bool ShouldCallDelete = true; 84259486a2dSAnders Carlsson 84359486a2dSAnders Carlsson // Call the destructor if necessary. 84459486a2dSAnders Carlsson if (const RecordType *RT = DeleteTy->getAs<RecordType>()) { 84559486a2dSAnders Carlsson if (CXXRecordDecl *RD = dyn_cast<CXXRecordDecl>(RT->getDecl())) { 84659486a2dSAnders Carlsson if (!RD->hasTrivialDestructor()) { 847bac7490fSDouglas Gregor const CXXDestructorDecl *Dtor = RD->getDestructor(); 84859486a2dSAnders Carlsson if (E->isArrayForm()) { 84921122cf6SAnders Carlsson llvm::Value *AllocatedObjectPtr; 85021122cf6SAnders Carlsson llvm::Value *NumElements; 85121122cf6SAnders Carlsson llvm::tie(AllocatedObjectPtr, NumElements) = 85221122cf6SAnders Carlsson GetAllocatedObjectPtrAndNumElements(*this, Ptr, DeleteTy); 853f5af8deaSAnders Carlsson 85459486a2dSAnders Carlsson EmitCXXAggrDestructorCall(Dtor, NumElements, Ptr); 855f5af8deaSAnders Carlsson } else if (Dtor->isVirtual()) { 85659486a2dSAnders Carlsson const llvm::Type *Ty = 85759486a2dSAnders Carlsson CGM.getTypes().GetFunctionType(CGM.getTypes().getFunctionInfo(Dtor), 85859486a2dSAnders Carlsson /*isVariadic=*/false); 85959486a2dSAnders Carlsson 86059486a2dSAnders Carlsson llvm::Value *Callee = BuildVirtualCall(Dtor, Dtor_Deleting, Ptr, Ty); 861e36a6b3eSAnders Carlsson EmitCXXMemberCall(Dtor, Callee, ReturnValueSlot(), Ptr, /*VTT=*/0, 862e36a6b3eSAnders Carlsson 0, 0); 86359486a2dSAnders Carlsson 86459486a2dSAnders Carlsson // The dtor took care of deleting the object. 86559486a2dSAnders Carlsson ShouldCallDelete = false; 86659486a2dSAnders Carlsson } else 867f8a71f08SAnders Carlsson EmitCXXDestructorCall(Dtor, Dtor_Complete, /*ForVirtualBase=*/false, 868f8a71f08SAnders Carlsson Ptr); 86959486a2dSAnders Carlsson } 87059486a2dSAnders Carlsson } 87159486a2dSAnders Carlsson } 87259486a2dSAnders Carlsson 87359486a2dSAnders Carlsson if (ShouldCallDelete) 87459486a2dSAnders Carlsson EmitDeleteCall(E->getOperatorDelete(), Ptr, DeleteTy); 87559486a2dSAnders Carlsson 87659486a2dSAnders Carlsson EmitBlock(DeleteEnd); 87759486a2dSAnders Carlsson } 87859486a2dSAnders Carlsson 87959486a2dSAnders Carlsson llvm::Value * CodeGenFunction::EmitCXXTypeidExpr(const CXXTypeidExpr *E) { 88059486a2dSAnders Carlsson QualType Ty = E->getType(); 88159486a2dSAnders Carlsson const llvm::Type *LTy = ConvertType(Ty)->getPointerTo(); 882fd7dfeb7SAnders Carlsson 8833f4336cbSAnders Carlsson if (E->isTypeOperand()) { 8843f4336cbSAnders Carlsson llvm::Constant *TypeInfo = 8853f4336cbSAnders Carlsson CGM.GetAddrOfRTTIDescriptor(E->getTypeOperand()); 8863f4336cbSAnders Carlsson return Builder.CreateBitCast(TypeInfo, LTy); 8873f4336cbSAnders Carlsson } 888fd7dfeb7SAnders Carlsson 88959486a2dSAnders Carlsson Expr *subE = E->getExprOperand(); 89059486a2dSAnders Carlsson Ty = subE->getType(); 89159486a2dSAnders Carlsson CanQualType CanTy = CGM.getContext().getCanonicalType(Ty); 89259486a2dSAnders Carlsson Ty = CanTy.getUnqualifiedType().getNonReferenceType(); 89359486a2dSAnders Carlsson if (const RecordType *RT = Ty->getAs<RecordType>()) { 89459486a2dSAnders Carlsson const CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl()); 89559486a2dSAnders Carlsson if (RD->isPolymorphic()) { 89659486a2dSAnders Carlsson // FIXME: if subE is an lvalue do 89759486a2dSAnders Carlsson LValue Obj = EmitLValue(subE); 89859486a2dSAnders Carlsson llvm::Value *This = Obj.getAddress(); 89959486a2dSAnders Carlsson LTy = LTy->getPointerTo()->getPointerTo(); 90059486a2dSAnders Carlsson llvm::Value *V = Builder.CreateBitCast(This, LTy); 90159486a2dSAnders Carlsson // We need to do a zero check for *p, unless it has NonNullAttr. 90259486a2dSAnders Carlsson // FIXME: PointerType->hasAttr<NonNullAttr>() 90359486a2dSAnders Carlsson bool CanBeZero = false; 90459486a2dSAnders Carlsson if (UnaryOperator *UO = dyn_cast<UnaryOperator>(subE->IgnoreParens())) 90559486a2dSAnders Carlsson if (UO->getOpcode() == UnaryOperator::Deref) 90659486a2dSAnders Carlsson CanBeZero = true; 90759486a2dSAnders Carlsson if (CanBeZero) { 90859486a2dSAnders Carlsson llvm::BasicBlock *NonZeroBlock = createBasicBlock(); 90959486a2dSAnders Carlsson llvm::BasicBlock *ZeroBlock = createBasicBlock(); 91059486a2dSAnders Carlsson 91159486a2dSAnders Carlsson llvm::Value *Zero = llvm::Constant::getNullValue(LTy); 91259486a2dSAnders Carlsson Builder.CreateCondBr(Builder.CreateICmpNE(V, Zero), 91359486a2dSAnders Carlsson NonZeroBlock, ZeroBlock); 91459486a2dSAnders Carlsson EmitBlock(ZeroBlock); 91559486a2dSAnders Carlsson /// Call __cxa_bad_typeid 91659486a2dSAnders Carlsson const llvm::Type *ResultType = llvm::Type::getVoidTy(VMContext); 91759486a2dSAnders Carlsson const llvm::FunctionType *FTy; 91859486a2dSAnders Carlsson FTy = llvm::FunctionType::get(ResultType, false); 91959486a2dSAnders Carlsson llvm::Value *F = CGM.CreateRuntimeFunction(FTy, "__cxa_bad_typeid"); 92059486a2dSAnders Carlsson Builder.CreateCall(F)->setDoesNotReturn(); 92159486a2dSAnders Carlsson Builder.CreateUnreachable(); 92259486a2dSAnders Carlsson EmitBlock(NonZeroBlock); 92359486a2dSAnders Carlsson } 92459486a2dSAnders Carlsson V = Builder.CreateLoad(V, "vtable"); 92559486a2dSAnders Carlsson V = Builder.CreateConstInBoundsGEP1_64(V, -1ULL); 92659486a2dSAnders Carlsson V = Builder.CreateLoad(V); 92759486a2dSAnders Carlsson return V; 92859486a2dSAnders Carlsson } 92959486a2dSAnders Carlsson } 9303f4336cbSAnders Carlsson return Builder.CreateBitCast(CGM.GetAddrOfRTTIDescriptor(Ty), LTy); 93159486a2dSAnders Carlsson } 93259486a2dSAnders Carlsson 93359486a2dSAnders Carlsson llvm::Value *CodeGenFunction::EmitDynamicCast(llvm::Value *V, 93459486a2dSAnders Carlsson const CXXDynamicCastExpr *DCE) { 9353f4336cbSAnders Carlsson QualType SrcTy = DCE->getSubExpr()->getType(); 9363f4336cbSAnders Carlsson QualType DestTy = DCE->getTypeAsWritten(); 9373f4336cbSAnders Carlsson QualType InnerType = DestTy->getPointeeType(); 9383f4336cbSAnders Carlsson 93959486a2dSAnders Carlsson const llvm::Type *LTy = ConvertType(DCE->getType()); 94059486a2dSAnders Carlsson 94159486a2dSAnders Carlsson bool CanBeZero = false; 94259486a2dSAnders Carlsson bool ToVoid = false; 94359486a2dSAnders Carlsson bool ThrowOnBad = false; 9443f4336cbSAnders Carlsson if (DestTy->isPointerType()) { 94559486a2dSAnders Carlsson // FIXME: if PointerType->hasAttr<NonNullAttr>(), we don't set this 94659486a2dSAnders Carlsson CanBeZero = true; 94759486a2dSAnders Carlsson if (InnerType->isVoidType()) 94859486a2dSAnders Carlsson ToVoid = true; 94959486a2dSAnders Carlsson } else { 95059486a2dSAnders Carlsson LTy = LTy->getPointerTo(); 951fa8b4955SDouglas Gregor 952fa8b4955SDouglas Gregor // FIXME: What if exceptions are disabled? 95359486a2dSAnders Carlsson ThrowOnBad = true; 95459486a2dSAnders Carlsson } 95559486a2dSAnders Carlsson 9563f4336cbSAnders Carlsson if (SrcTy->isPointerType() || SrcTy->isReferenceType()) 9573f4336cbSAnders Carlsson SrcTy = SrcTy->getPointeeType(); 9583f4336cbSAnders Carlsson SrcTy = SrcTy.getUnqualifiedType(); 9593f4336cbSAnders Carlsson 9600087bc85SAnders Carlsson if (DestTy->isPointerType() || DestTy->isReferenceType()) 9613f4336cbSAnders Carlsson DestTy = DestTy->getPointeeType(); 9623f4336cbSAnders Carlsson DestTy = DestTy.getUnqualifiedType(); 96359486a2dSAnders Carlsson 96459486a2dSAnders Carlsson llvm::BasicBlock *ContBlock = createBasicBlock(); 96559486a2dSAnders Carlsson llvm::BasicBlock *NullBlock = 0; 96659486a2dSAnders Carlsson llvm::BasicBlock *NonZeroBlock = 0; 96759486a2dSAnders Carlsson if (CanBeZero) { 96859486a2dSAnders Carlsson NonZeroBlock = createBasicBlock(); 96959486a2dSAnders Carlsson NullBlock = createBasicBlock(); 9703f4336cbSAnders Carlsson Builder.CreateCondBr(Builder.CreateIsNotNull(V), NonZeroBlock, NullBlock); 97159486a2dSAnders Carlsson EmitBlock(NonZeroBlock); 97259486a2dSAnders Carlsson } 97359486a2dSAnders Carlsson 97459486a2dSAnders Carlsson llvm::BasicBlock *BadCastBlock = 0; 97559486a2dSAnders Carlsson 9763f4336cbSAnders Carlsson const llvm::Type *PtrDiffTy = ConvertType(getContext().getPointerDiffType()); 97759486a2dSAnders Carlsson 97859486a2dSAnders Carlsson // See if this is a dynamic_cast(void*) 97959486a2dSAnders Carlsson if (ToVoid) { 98059486a2dSAnders Carlsson llvm::Value *This = V; 98159486a2dSAnders Carlsson V = Builder.CreateBitCast(This, PtrDiffTy->getPointerTo()->getPointerTo()); 98259486a2dSAnders Carlsson V = Builder.CreateLoad(V, "vtable"); 98359486a2dSAnders Carlsson V = Builder.CreateConstInBoundsGEP1_64(V, -2ULL); 98459486a2dSAnders Carlsson V = Builder.CreateLoad(V, "offset to top"); 98559486a2dSAnders Carlsson This = Builder.CreateBitCast(This, llvm::Type::getInt8PtrTy(VMContext)); 98659486a2dSAnders Carlsson V = Builder.CreateInBoundsGEP(This, V); 98759486a2dSAnders Carlsson V = Builder.CreateBitCast(V, LTy); 98859486a2dSAnders Carlsson } else { 98959486a2dSAnders Carlsson /// Call __dynamic_cast 99059486a2dSAnders Carlsson const llvm::Type *ResultType = llvm::Type::getInt8PtrTy(VMContext); 99159486a2dSAnders Carlsson const llvm::FunctionType *FTy; 99259486a2dSAnders Carlsson std::vector<const llvm::Type*> ArgTys; 99359486a2dSAnders Carlsson const llvm::Type *PtrToInt8Ty 99459486a2dSAnders Carlsson = llvm::Type::getInt8Ty(VMContext)->getPointerTo(); 99559486a2dSAnders Carlsson ArgTys.push_back(PtrToInt8Ty); 99659486a2dSAnders Carlsson ArgTys.push_back(PtrToInt8Ty); 99759486a2dSAnders Carlsson ArgTys.push_back(PtrToInt8Ty); 99859486a2dSAnders Carlsson ArgTys.push_back(PtrDiffTy); 99959486a2dSAnders Carlsson FTy = llvm::FunctionType::get(ResultType, ArgTys, false); 100059486a2dSAnders Carlsson 100159486a2dSAnders Carlsson // FIXME: Calculate better hint. 100259486a2dSAnders Carlsson llvm::Value *hint = llvm::ConstantInt::get(PtrDiffTy, -1ULL); 10033f4336cbSAnders Carlsson 10043f4336cbSAnders Carlsson assert(SrcTy->isRecordType() && "Src type must be record type!"); 10053f4336cbSAnders Carlsson assert(DestTy->isRecordType() && "Dest type must be record type!"); 10063f4336cbSAnders Carlsson 1007247894b3SDouglas Gregor llvm::Value *SrcArg 1008247894b3SDouglas Gregor = CGM.GetAddrOfRTTIDescriptor(SrcTy.getUnqualifiedType()); 1009247894b3SDouglas Gregor llvm::Value *DestArg 1010247894b3SDouglas Gregor = CGM.GetAddrOfRTTIDescriptor(DestTy.getUnqualifiedType()); 10113f4336cbSAnders Carlsson 101259486a2dSAnders Carlsson V = Builder.CreateBitCast(V, PtrToInt8Ty); 101359486a2dSAnders Carlsson V = Builder.CreateCall4(CGM.CreateRuntimeFunction(FTy, "__dynamic_cast"), 10143f4336cbSAnders Carlsson V, SrcArg, DestArg, hint); 101559486a2dSAnders Carlsson V = Builder.CreateBitCast(V, LTy); 101659486a2dSAnders Carlsson 101759486a2dSAnders Carlsson if (ThrowOnBad) { 101859486a2dSAnders Carlsson BadCastBlock = createBasicBlock(); 10193f4336cbSAnders Carlsson Builder.CreateCondBr(Builder.CreateIsNotNull(V), ContBlock, BadCastBlock); 102059486a2dSAnders Carlsson EmitBlock(BadCastBlock); 1021fa8b4955SDouglas Gregor /// Invoke __cxa_bad_cast 102259486a2dSAnders Carlsson ResultType = llvm::Type::getVoidTy(VMContext); 102359486a2dSAnders Carlsson const llvm::FunctionType *FBadTy; 102459486a2dSAnders Carlsson FBadTy = llvm::FunctionType::get(ResultType, false); 102559486a2dSAnders Carlsson llvm::Value *F = CGM.CreateRuntimeFunction(FBadTy, "__cxa_bad_cast"); 1026fa8b4955SDouglas Gregor if (llvm::BasicBlock *InvokeDest = getInvokeDest()) { 1027fa8b4955SDouglas Gregor llvm::BasicBlock *Cont = createBasicBlock("invoke.cont"); 1028fa8b4955SDouglas Gregor Builder.CreateInvoke(F, Cont, InvokeDest)->setDoesNotReturn(); 1029fa8b4955SDouglas Gregor EmitBlock(Cont); 1030fa8b4955SDouglas Gregor } else { 1031fa8b4955SDouglas Gregor // FIXME: Does this ever make sense? 103259486a2dSAnders Carlsson Builder.CreateCall(F)->setDoesNotReturn(); 1033fa8b4955SDouglas Gregor } 103459486a2dSAnders Carlsson Builder.CreateUnreachable(); 103559486a2dSAnders Carlsson } 103659486a2dSAnders Carlsson } 103759486a2dSAnders Carlsson 103859486a2dSAnders Carlsson if (CanBeZero) { 103959486a2dSAnders Carlsson Builder.CreateBr(ContBlock); 104059486a2dSAnders Carlsson EmitBlock(NullBlock); 104159486a2dSAnders Carlsson Builder.CreateBr(ContBlock); 104259486a2dSAnders Carlsson } 104359486a2dSAnders Carlsson EmitBlock(ContBlock); 104459486a2dSAnders Carlsson if (CanBeZero) { 104559486a2dSAnders Carlsson llvm::PHINode *PHI = Builder.CreatePHI(LTy); 104659486a2dSAnders Carlsson PHI->reserveOperandSpace(2); 104759486a2dSAnders Carlsson PHI->addIncoming(V, NonZeroBlock); 104859486a2dSAnders Carlsson PHI->addIncoming(llvm::Constant::getNullValue(LTy), NullBlock); 104959486a2dSAnders Carlsson V = PHI; 105059486a2dSAnders Carlsson } 105159486a2dSAnders Carlsson 105259486a2dSAnders Carlsson return V; 105359486a2dSAnders Carlsson } 1054