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 168e302792bSJohn McCall if (BO->getOpcode() == BO_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) { 2997648fb46SArgyrios Kyrtzidis ElementType = Ctx.getBaseElementType(ElementType); 30021122cf6SAnders Carlsson const RecordType *RT = ElementType->getAs<RecordType>(); 30159486a2dSAnders Carlsson if (!RT) 3023eb55cfeSKen Dyck return CharUnits::Zero(); 30359486a2dSAnders Carlsson 30459486a2dSAnders Carlsson const CXXRecordDecl *RD = dyn_cast<CXXRecordDecl>(RT->getDecl()); 30559486a2dSAnders Carlsson if (!RD) 3063eb55cfeSKen Dyck return CharUnits::Zero(); 30759486a2dSAnders Carlsson 30859486a2dSAnders Carlsson // Check if the class has a trivial destructor. 30959486a2dSAnders Carlsson if (RD->hasTrivialDestructor()) { 31021122cf6SAnders Carlsson // Check if the usual deallocation function takes two arguments. 311adbe4249SAnders Carlsson const CXXMethodDecl *UsualDeallocationFunction = 0; 312adbe4249SAnders Carlsson 31321122cf6SAnders Carlsson DeclarationName OpName = 31421122cf6SAnders Carlsson Ctx.DeclarationNames.getCXXOperatorName(OO_Array_Delete); 31521122cf6SAnders Carlsson DeclContext::lookup_const_iterator Op, OpEnd; 31621122cf6SAnders Carlsson for (llvm::tie(Op, OpEnd) = RD->lookup(OpName); 31721122cf6SAnders Carlsson Op != OpEnd; ++Op) { 318adbe4249SAnders Carlsson const CXXMethodDecl *Delete = cast<CXXMethodDecl>(*Op); 31921122cf6SAnders Carlsson 32021122cf6SAnders Carlsson if (Delete->isUsualDeallocationFunction()) { 321adbe4249SAnders Carlsson UsualDeallocationFunction = Delete; 32221122cf6SAnders Carlsson break; 32321122cf6SAnders Carlsson } 32421122cf6SAnders Carlsson } 325adbe4249SAnders Carlsson 326adbe4249SAnders Carlsson // No usual deallocation function, we don't need a cookie. 327adbe4249SAnders Carlsson if (!UsualDeallocationFunction) 3283eb55cfeSKen Dyck return CharUnits::Zero(); 329adbe4249SAnders Carlsson 330adbe4249SAnders Carlsson // The usual deallocation function doesn't take a size_t argument, so we 331adbe4249SAnders Carlsson // don't need a cookie. 332adbe4249SAnders Carlsson if (UsualDeallocationFunction->getNumParams() == 1) 3333eb55cfeSKen Dyck return CharUnits::Zero(); 334adbe4249SAnders Carlsson 335adbe4249SAnders Carlsson assert(UsualDeallocationFunction->getNumParams() == 2 && 336adbe4249SAnders Carlsson "Unexpected deallocation function type!"); 33759486a2dSAnders Carlsson } 33859486a2dSAnders Carlsson 33921122cf6SAnders Carlsson // Padding is the maximum of sizeof(size_t) and alignof(ElementType) 3403eb55cfeSKen Dyck return std::max(Ctx.getTypeSizeInChars(Ctx.getSizeType()), 3413eb55cfeSKen Dyck Ctx.getTypeAlignInChars(ElementType)); 34221122cf6SAnders Carlsson } 34321122cf6SAnders Carlsson 344aa4149a2SJohn McCall /// Check whether the given operator new[] is the global placement 345aa4149a2SJohn McCall /// operator new[]. 346aa4149a2SJohn McCall static bool IsPlacementOperatorNewArray(ASTContext &Ctx, 347aa4149a2SJohn McCall const FunctionDecl *Fn) { 348aa4149a2SJohn McCall // Must be in global scope. Note that allocation functions can't be 349aa4149a2SJohn McCall // declared in namespaces. 350*50c68258SSebastian Redl if (!Fn->getDeclContext()->getRedeclContext()->isFileContext()) 351aa4149a2SJohn McCall return false; 352aa4149a2SJohn McCall 353aa4149a2SJohn McCall // Signature must be void *operator new[](size_t, void*). 354aa4149a2SJohn McCall // The size_t is common to all operator new[]s. 355aa4149a2SJohn McCall if (Fn->getNumParams() != 2) 356aa4149a2SJohn McCall return false; 357aa4149a2SJohn McCall 358aa4149a2SJohn McCall CanQualType ParamType = Ctx.getCanonicalType(Fn->getParamDecl(1)->getType()); 359aa4149a2SJohn McCall return (ParamType == Ctx.VoidPtrTy); 360aa4149a2SJohn McCall } 361aa4149a2SJohn McCall 3623eb55cfeSKen Dyck static CharUnits CalculateCookiePadding(ASTContext &Ctx, const CXXNewExpr *E) { 36321122cf6SAnders Carlsson if (!E->isArray()) 3643eb55cfeSKen Dyck return CharUnits::Zero(); 36521122cf6SAnders Carlsson 366399f499fSAnders Carlsson // No cookie is required if the new operator being used is 367399f499fSAnders Carlsson // ::operator new[](size_t, void*). 368399f499fSAnders Carlsson const FunctionDecl *OperatorNew = E->getOperatorNew(); 369aa4149a2SJohn McCall if (IsPlacementOperatorNewArray(Ctx, OperatorNew)) 3703eb55cfeSKen Dyck return CharUnits::Zero(); 371399f499fSAnders Carlsson 37221122cf6SAnders Carlsson return CalculateCookiePadding(Ctx, E->getAllocatedType()); 37359486a2dSAnders Carlsson } 37459486a2dSAnders Carlsson 37547b4629bSFariborz Jahanian static llvm::Value *EmitCXXNewAllocSize(ASTContext &Context, 37647b4629bSFariborz Jahanian CodeGenFunction &CGF, 37759486a2dSAnders Carlsson const CXXNewExpr *E, 37805fc5be3SDouglas Gregor llvm::Value *&NumElements, 37905fc5be3SDouglas Gregor llvm::Value *&SizeWithoutCookie) { 3807648fb46SArgyrios Kyrtzidis QualType ElemType = E->getAllocatedType(); 38159486a2dSAnders Carlsson 38205fc5be3SDouglas Gregor if (!E->isArray()) { 3837648fb46SArgyrios Kyrtzidis CharUnits TypeSize = CGF.getContext().getTypeSizeInChars(ElemType); 3847648fb46SArgyrios Kyrtzidis const llvm::Type *SizeTy = CGF.ConvertType(CGF.getContext().getSizeType()); 38505fc5be3SDouglas Gregor SizeWithoutCookie = llvm::ConstantInt::get(SizeTy, TypeSize.getQuantity()); 38605fc5be3SDouglas Gregor return SizeWithoutCookie; 38705fc5be3SDouglas Gregor } 38859486a2dSAnders Carlsson 38959486a2dSAnders Carlsson // Emit the array size expression. 3907648fb46SArgyrios Kyrtzidis // We multiply the size of all dimensions for NumElements. 3917648fb46SArgyrios Kyrtzidis // e.g for 'int[2][3]', ElemType is 'int' and NumElements is 6. 39259486a2dSAnders Carlsson NumElements = CGF.EmitScalarExpr(E->getArraySize()); 3937648fb46SArgyrios Kyrtzidis while (const ConstantArrayType *CAT 3947648fb46SArgyrios Kyrtzidis = CGF.getContext().getAsConstantArrayType(ElemType)) { 3957648fb46SArgyrios Kyrtzidis ElemType = CAT->getElementType(); 3967648fb46SArgyrios Kyrtzidis llvm::Value *ArraySize 3977648fb46SArgyrios Kyrtzidis = llvm::ConstantInt::get(CGF.CGM.getLLVMContext(), CAT->getSize()); 3987648fb46SArgyrios Kyrtzidis NumElements = CGF.Builder.CreateMul(NumElements, ArraySize); 3997648fb46SArgyrios Kyrtzidis } 40059486a2dSAnders Carlsson 4017648fb46SArgyrios Kyrtzidis CharUnits TypeSize = CGF.getContext().getTypeSizeInChars(ElemType); 4027648fb46SArgyrios Kyrtzidis const llvm::Type *SizeTy = CGF.ConvertType(CGF.getContext().getSizeType()); 40332ac583dSChris Lattner llvm::Value *Size = llvm::ConstantInt::get(SizeTy, TypeSize.getQuantity()); 40432ac583dSChris Lattner 40532ac583dSChris Lattner // If someone is doing 'new int[42]' there is no need to do a dynamic check. 40632ac583dSChris Lattner // Don't bloat the -O0 code. 40732ac583dSChris Lattner if (llvm::ConstantInt *NumElementsC = 40832ac583dSChris Lattner dyn_cast<llvm::ConstantInt>(NumElements)) { 40932ac583dSChris Lattner // Determine if there is an overflow here by doing an extended multiply. 41032ac583dSChris Lattner llvm::APInt NEC = NumElementsC->getValue(); 41132ac583dSChris Lattner NEC.zext(NEC.getBitWidth()*2); 41232ac583dSChris Lattner 41332ac583dSChris Lattner llvm::APInt SC = cast<llvm::ConstantInt>(Size)->getValue(); 41432ac583dSChris Lattner SC.zext(SC.getBitWidth()*2); 41532ac583dSChris Lattner SC *= NEC; 41632ac583dSChris Lattner 41732ac583dSChris Lattner if (SC.countLeadingZeros() >= NumElementsC->getValue().getBitWidth()) { 41832ac583dSChris Lattner SC.trunc(NumElementsC->getValue().getBitWidth()); 41932ac583dSChris Lattner Size = llvm::ConstantInt::get(Size->getContext(), SC); 42032ac583dSChris Lattner } else { 42132ac583dSChris Lattner // On overflow, produce a -1 so operator new throws. 42232ac583dSChris Lattner Size = llvm::Constant::getAllOnesValue(Size->getType()); 42332ac583dSChris Lattner } 42432ac583dSChris Lattner 42532ac583dSChris Lattner } else { 42626008e07SChris Lattner // Multiply with the type size. This multiply can overflow, e.g. in: 42726008e07SChris Lattner // new double[n] 42826008e07SChris Lattner // where n is 2^30 on a 32-bit machine or 2^62 on a 64-bit machine. Because 42926008e07SChris Lattner // of this, we need to detect the overflow and ensure that an exception is 430f2f38701SChris Lattner // called by forcing the size to -1 on overflow. 43132ac583dSChris Lattner llvm::Value *UMulF = 43232ac583dSChris Lattner CGF.CGM.getIntrinsic(llvm::Intrinsic::umul_with_overflow, &SizeTy, 1); 43332ac583dSChris Lattner llvm::Value *MulRes = CGF.Builder.CreateCall2(UMulF, NumElements, Size); 43432ac583dSChris Lattner // Branch on the overflow bit to the overflow block, which is lazily 43532ac583dSChris Lattner // created. 43626008e07SChris Lattner llvm::Value *DidOverflow = CGF.Builder.CreateExtractValue(MulRes, 1); 43726008e07SChris Lattner // Get the normal result of the multiplication. 43826008e07SChris Lattner llvm::Value *V = CGF.Builder.CreateExtractValue(MulRes, 0); 43947b4629bSFariborz Jahanian 440f2f38701SChris Lattner llvm::BasicBlock *NormalBB = CGF.createBasicBlock("no_overflow"); 441f2f38701SChris Lattner llvm::BasicBlock *OverflowBB = CGF.createBasicBlock("overflow"); 442f2f38701SChris Lattner 443f2f38701SChris Lattner CGF.Builder.CreateCondBr(DidOverflow, OverflowBB, NormalBB); 444f2f38701SChris Lattner 445f2f38701SChris Lattner llvm::BasicBlock *PrevBB = CGF.Builder.GetInsertBlock(); 446f2f38701SChris Lattner 447f2f38701SChris Lattner // We just need the overflow block to build a PHI node. 448f2f38701SChris Lattner CGF.EmitBlock(OverflowBB); 449f2f38701SChris Lattner CGF.EmitBlock(NormalBB); 450f2f38701SChris Lattner 451f2f38701SChris Lattner llvm::PHINode *PN = CGF.Builder.CreatePHI(V->getType()); 452f2f38701SChris Lattner 453f2f38701SChris Lattner PN->addIncoming(V, PrevBB); 454f2f38701SChris Lattner PN->addIncoming(llvm::Constant::getAllOnesValue(V->getType()), OverflowBB); 45532ac583dSChris Lattner Size = PN; 45632ac583dSChris Lattner } 45705fc5be3SDouglas Gregor SizeWithoutCookie = Size; 458f2f38701SChris Lattner 45932ac583dSChris Lattner // Add the cookie padding if necessary. 46032ac583dSChris Lattner CharUnits CookiePadding = CalculateCookiePadding(CGF.getContext(), E); 4613eb55cfeSKen Dyck if (!CookiePadding.isZero()) 46232ac583dSChris Lattner Size = CGF.Builder.CreateAdd(Size, 4633eb55cfeSKen Dyck llvm::ConstantInt::get(SizeTy, CookiePadding.getQuantity())); 46459486a2dSAnders Carlsson 46532ac583dSChris Lattner return Size; 46659486a2dSAnders Carlsson } 46759486a2dSAnders Carlsson 468d5202e09SFariborz Jahanian static void StoreAnyExprIntoOneUnit(CodeGenFunction &CGF, const CXXNewExpr *E, 469d5202e09SFariborz Jahanian llvm::Value *NewPtr) { 470d5202e09SFariborz Jahanian 471d5202e09SFariborz Jahanian assert(E->getNumConstructorArgs() == 1 && 472d5202e09SFariborz Jahanian "Can only have one argument to initializer of POD type."); 473d5202e09SFariborz Jahanian 474d5202e09SFariborz Jahanian const Expr *Init = E->getConstructorArg(0); 475d5202e09SFariborz Jahanian QualType AllocType = E->getAllocatedType(); 476d5202e09SFariborz Jahanian 4770381634aSDaniel Dunbar unsigned Alignment = 4780381634aSDaniel Dunbar CGF.getContext().getTypeAlignInChars(AllocType).getQuantity(); 479d5202e09SFariborz Jahanian if (!CGF.hasAggregateLLVMType(AllocType)) 480d5202e09SFariborz Jahanian CGF.EmitStoreOfScalar(CGF.EmitScalarExpr(Init), NewPtr, 4810381634aSDaniel Dunbar AllocType.isVolatileQualified(), Alignment, 4820381634aSDaniel Dunbar AllocType); 483d5202e09SFariborz Jahanian else if (AllocType->isAnyComplexType()) 484d5202e09SFariborz Jahanian CGF.EmitComplexExprIntoAddr(Init, NewPtr, 485d5202e09SFariborz Jahanian AllocType.isVolatileQualified()); 486d5202e09SFariborz Jahanian else 487d5202e09SFariborz Jahanian CGF.EmitAggExpr(Init, NewPtr, AllocType.isVolatileQualified()); 488d5202e09SFariborz Jahanian } 489d5202e09SFariborz Jahanian 490d5202e09SFariborz Jahanian void 491d5202e09SFariborz Jahanian CodeGenFunction::EmitNewArrayInitializer(const CXXNewExpr *E, 492d5202e09SFariborz Jahanian llvm::Value *NewPtr, 493d5202e09SFariborz Jahanian llvm::Value *NumElements) { 494b66b08efSFariborz Jahanian // We have a POD type. 495b66b08efSFariborz Jahanian if (E->getNumConstructorArgs() == 0) 496b66b08efSFariborz Jahanian return; 497b66b08efSFariborz Jahanian 498d5202e09SFariborz Jahanian const llvm::Type *SizeTy = ConvertType(getContext().getSizeType()); 499d5202e09SFariborz Jahanian 500d5202e09SFariborz Jahanian // Create a temporary for the loop index and initialize it with 0. 501d5202e09SFariborz Jahanian llvm::Value *IndexPtr = CreateTempAlloca(SizeTy, "loop.index"); 502d5202e09SFariborz Jahanian llvm::Value *Zero = llvm::Constant::getNullValue(SizeTy); 503d5202e09SFariborz Jahanian Builder.CreateStore(Zero, IndexPtr); 504d5202e09SFariborz Jahanian 505d5202e09SFariborz Jahanian // Start the loop with a block that tests the condition. 506d5202e09SFariborz Jahanian llvm::BasicBlock *CondBlock = createBasicBlock("for.cond"); 507d5202e09SFariborz Jahanian llvm::BasicBlock *AfterFor = createBasicBlock("for.end"); 508d5202e09SFariborz Jahanian 509d5202e09SFariborz Jahanian EmitBlock(CondBlock); 510d5202e09SFariborz Jahanian 511d5202e09SFariborz Jahanian llvm::BasicBlock *ForBody = createBasicBlock("for.body"); 512d5202e09SFariborz Jahanian 513d5202e09SFariborz Jahanian // Generate: if (loop-index < number-of-elements fall to the loop body, 514d5202e09SFariborz Jahanian // otherwise, go to the block after the for-loop. 515d5202e09SFariborz Jahanian llvm::Value *Counter = Builder.CreateLoad(IndexPtr); 516d5202e09SFariborz Jahanian llvm::Value *IsLess = Builder.CreateICmpULT(Counter, NumElements, "isless"); 517d5202e09SFariborz Jahanian // If the condition is true, execute the body. 518d5202e09SFariborz Jahanian Builder.CreateCondBr(IsLess, ForBody, AfterFor); 519d5202e09SFariborz Jahanian 520d5202e09SFariborz Jahanian EmitBlock(ForBody); 521d5202e09SFariborz Jahanian 522d5202e09SFariborz Jahanian llvm::BasicBlock *ContinueBlock = createBasicBlock("for.inc"); 523d5202e09SFariborz Jahanian // Inside the loop body, emit the constructor call on the array element. 524d5202e09SFariborz Jahanian Counter = Builder.CreateLoad(IndexPtr); 525d5202e09SFariborz Jahanian llvm::Value *Address = Builder.CreateInBoundsGEP(NewPtr, Counter, 526d5202e09SFariborz Jahanian "arrayidx"); 527d5202e09SFariborz Jahanian StoreAnyExprIntoOneUnit(*this, E, Address); 528d5202e09SFariborz Jahanian 529d5202e09SFariborz Jahanian EmitBlock(ContinueBlock); 530d5202e09SFariborz Jahanian 531d5202e09SFariborz Jahanian // Emit the increment of the loop counter. 532d5202e09SFariborz Jahanian llvm::Value *NextVal = llvm::ConstantInt::get(SizeTy, 1); 533d5202e09SFariborz Jahanian Counter = Builder.CreateLoad(IndexPtr); 534d5202e09SFariborz Jahanian NextVal = Builder.CreateAdd(Counter, NextVal, "inc"); 535d5202e09SFariborz Jahanian Builder.CreateStore(NextVal, IndexPtr); 536d5202e09SFariborz Jahanian 537d5202e09SFariborz Jahanian // Finally, branch back up to the condition for the next iteration. 538d5202e09SFariborz Jahanian EmitBranch(CondBlock); 539d5202e09SFariborz Jahanian 540d5202e09SFariborz Jahanian // Emit the fall-through block. 541d5202e09SFariborz Jahanian EmitBlock(AfterFor, true); 542d5202e09SFariborz Jahanian } 543d5202e09SFariborz Jahanian 54405fc5be3SDouglas Gregor static void EmitZeroMemSet(CodeGenFunction &CGF, QualType T, 54505fc5be3SDouglas Gregor llvm::Value *NewPtr, llvm::Value *Size) { 54605fc5be3SDouglas Gregor llvm::LLVMContext &VMContext = CGF.CGM.getLLVMContext(); 54705fc5be3SDouglas Gregor const llvm::Type *BP = llvm::Type::getInt8PtrTy(VMContext); 54805fc5be3SDouglas Gregor if (NewPtr->getType() != BP) 54905fc5be3SDouglas Gregor NewPtr = CGF.Builder.CreateBitCast(NewPtr, BP, "tmp"); 55005fc5be3SDouglas Gregor 55105fc5be3SDouglas Gregor CGF.Builder.CreateCall5(CGF.CGM.getMemSetFn(BP, CGF.IntPtrTy), NewPtr, 55205fc5be3SDouglas Gregor llvm::Constant::getNullValue(llvm::Type::getInt8Ty(VMContext)), 55305fc5be3SDouglas Gregor Size, 55405fc5be3SDouglas Gregor llvm::ConstantInt::get(CGF.Int32Ty, 55505fc5be3SDouglas Gregor CGF.getContext().getTypeAlign(T)/8), 55605fc5be3SDouglas Gregor llvm::ConstantInt::get(llvm::Type::getInt1Ty(VMContext), 55705fc5be3SDouglas Gregor 0)); 55805fc5be3SDouglas Gregor } 55905fc5be3SDouglas Gregor 56059486a2dSAnders Carlsson static void EmitNewInitializer(CodeGenFunction &CGF, const CXXNewExpr *E, 56159486a2dSAnders Carlsson llvm::Value *NewPtr, 56205fc5be3SDouglas Gregor llvm::Value *NumElements, 56305fc5be3SDouglas Gregor llvm::Value *AllocSizeWithoutCookie) { 5643a202f60SAnders Carlsson if (E->isArray()) { 565d040e6b2SAnders Carlsson if (CXXConstructorDecl *Ctor = E->getConstructor()) { 56605fc5be3SDouglas Gregor bool RequiresZeroInitialization = false; 56705fc5be3SDouglas Gregor if (Ctor->getParent()->hasTrivialConstructor()) { 56805fc5be3SDouglas Gregor // If new expression did not specify value-initialization, then there 56905fc5be3SDouglas Gregor // is no initialization. 57005fc5be3SDouglas Gregor if (!E->hasInitializer() || Ctor->getParent()->isEmpty()) 57105fc5be3SDouglas Gregor return; 57205fc5be3SDouglas Gregor 573614dbdcdSJohn McCall if (CGF.CGM.getTypes().isZeroInitializable(E->getAllocatedType())) { 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); 5783a202f60SAnders Carlsson return; 5793a202f60SAnders Carlsson } 58005fc5be3SDouglas Gregor 58105fc5be3SDouglas Gregor RequiresZeroInitialization = true; 58205fc5be3SDouglas Gregor } 58305fc5be3SDouglas Gregor 58405fc5be3SDouglas Gregor CGF.EmitCXXAggrConstructorCall(Ctor, NumElements, NewPtr, 58505fc5be3SDouglas Gregor E->constructor_arg_begin(), 58605fc5be3SDouglas Gregor E->constructor_arg_end(), 58705fc5be3SDouglas Gregor RequiresZeroInitialization); 58805fc5be3SDouglas Gregor return; 58905fc5be3SDouglas Gregor } else if (E->getNumConstructorArgs() == 1 && 59005fc5be3SDouglas Gregor isa<ImplicitValueInitExpr>(E->getConstructorArg(0))) { 59105fc5be3SDouglas Gregor // Optimization: since zero initialization will just set the memory 59205fc5be3SDouglas Gregor // to all zeroes, generate a single memset to do it in one shot. 59305fc5be3SDouglas Gregor EmitZeroMemSet(CGF, E->getAllocatedType(), NewPtr, 59405fc5be3SDouglas Gregor AllocSizeWithoutCookie); 59505fc5be3SDouglas Gregor return; 59605fc5be3SDouglas Gregor } else { 597d5202e09SFariborz Jahanian CGF.EmitNewArrayInitializer(E, NewPtr, NumElements); 598d5202e09SFariborz Jahanian return; 599d040e6b2SAnders Carlsson } 600d5202e09SFariborz Jahanian } 60159486a2dSAnders Carlsson 60259486a2dSAnders Carlsson if (CXXConstructorDecl *Ctor = E->getConstructor()) { 603747eb784SDouglas Gregor // Per C++ [expr.new]p15, if we have an initializer, then we're performing 604747eb784SDouglas Gregor // direct initialization. C++ [dcl.init]p5 requires that we 605747eb784SDouglas Gregor // zero-initialize storage if there are no user-declared constructors. 606747eb784SDouglas Gregor if (E->hasInitializer() && 607747eb784SDouglas Gregor !Ctor->getParent()->hasUserDeclaredConstructor() && 608747eb784SDouglas Gregor !Ctor->getParent()->isEmpty()) 609747eb784SDouglas Gregor CGF.EmitNullInitialization(NewPtr, E->getAllocatedType()); 610747eb784SDouglas Gregor 611e11f9ce9SAnders Carlsson CGF.EmitCXXConstructorCall(Ctor, Ctor_Complete, /*ForVirtualBase=*/false, 612e11f9ce9SAnders Carlsson NewPtr, E->constructor_arg_begin(), 61359486a2dSAnders Carlsson E->constructor_arg_end()); 61459486a2dSAnders Carlsson 61559486a2dSAnders Carlsson return; 61659486a2dSAnders Carlsson } 617b66b08efSFariborz Jahanian // We have a POD type. 618b66b08efSFariborz Jahanian if (E->getNumConstructorArgs() == 0) 619b66b08efSFariborz Jahanian return; 62059486a2dSAnders Carlsson 621d5202e09SFariborz Jahanian StoreAnyExprIntoOneUnit(CGF, E, NewPtr); 62259486a2dSAnders Carlsson } 62359486a2dSAnders Carlsson 62459486a2dSAnders Carlsson llvm::Value *CodeGenFunction::EmitCXXNewExpr(const CXXNewExpr *E) { 62559486a2dSAnders Carlsson QualType AllocType = E->getAllocatedType(); 62659486a2dSAnders Carlsson FunctionDecl *NewFD = E->getOperatorNew(); 62759486a2dSAnders Carlsson const FunctionProtoType *NewFTy = NewFD->getType()->getAs<FunctionProtoType>(); 62859486a2dSAnders Carlsson 62959486a2dSAnders Carlsson CallArgList NewArgs; 63059486a2dSAnders Carlsson 63159486a2dSAnders Carlsson // The allocation size is the first argument. 63259486a2dSAnders Carlsson QualType SizeTy = getContext().getSizeType(); 63359486a2dSAnders Carlsson 63459486a2dSAnders Carlsson llvm::Value *NumElements = 0; 63505fc5be3SDouglas Gregor llvm::Value *AllocSizeWithoutCookie = 0; 63647b4629bSFariborz Jahanian llvm::Value *AllocSize = EmitCXXNewAllocSize(getContext(), 63705fc5be3SDouglas Gregor *this, E, NumElements, 63805fc5be3SDouglas Gregor AllocSizeWithoutCookie); 63959486a2dSAnders Carlsson 64059486a2dSAnders Carlsson NewArgs.push_back(std::make_pair(RValue::get(AllocSize), SizeTy)); 64159486a2dSAnders Carlsson 64259486a2dSAnders Carlsson // Emit the rest of the arguments. 64359486a2dSAnders Carlsson // FIXME: Ideally, this should just use EmitCallArgs. 64459486a2dSAnders Carlsson CXXNewExpr::const_arg_iterator NewArg = E->placement_arg_begin(); 64559486a2dSAnders Carlsson 64659486a2dSAnders Carlsson // First, use the types from the function type. 64759486a2dSAnders Carlsson // We start at 1 here because the first argument (the allocation size) 64859486a2dSAnders Carlsson // has already been emitted. 64959486a2dSAnders Carlsson for (unsigned i = 1, e = NewFTy->getNumArgs(); i != e; ++i, ++NewArg) { 65059486a2dSAnders Carlsson QualType ArgType = NewFTy->getArgType(i); 65159486a2dSAnders Carlsson 65259486a2dSAnders Carlsson assert(getContext().getCanonicalType(ArgType.getNonReferenceType()). 65359486a2dSAnders Carlsson getTypePtr() == 65459486a2dSAnders Carlsson getContext().getCanonicalType(NewArg->getType()).getTypePtr() && 65559486a2dSAnders Carlsson "type mismatch in call argument!"); 65659486a2dSAnders Carlsson 65759486a2dSAnders Carlsson NewArgs.push_back(std::make_pair(EmitCallArg(*NewArg, ArgType), 65859486a2dSAnders Carlsson ArgType)); 65959486a2dSAnders Carlsson 66059486a2dSAnders Carlsson } 66159486a2dSAnders Carlsson 66259486a2dSAnders Carlsson // Either we've emitted all the call args, or we have a call to a 66359486a2dSAnders Carlsson // variadic function. 66459486a2dSAnders Carlsson assert((NewArg == E->placement_arg_end() || NewFTy->isVariadic()) && 66559486a2dSAnders Carlsson "Extra arguments in non-variadic function!"); 66659486a2dSAnders Carlsson 66759486a2dSAnders Carlsson // If we still have any arguments, emit them using the type of the argument. 66859486a2dSAnders Carlsson for (CXXNewExpr::const_arg_iterator NewArgEnd = E->placement_arg_end(); 66959486a2dSAnders Carlsson NewArg != NewArgEnd; ++NewArg) { 67059486a2dSAnders Carlsson QualType ArgType = NewArg->getType(); 67159486a2dSAnders Carlsson NewArgs.push_back(std::make_pair(EmitCallArg(*NewArg, ArgType), 67259486a2dSAnders Carlsson ArgType)); 67359486a2dSAnders Carlsson } 67459486a2dSAnders Carlsson 67559486a2dSAnders Carlsson // Emit the call to new. 67659486a2dSAnders Carlsson RValue RV = 677ab26cfa5SJohn McCall EmitCall(CGM.getTypes().getFunctionInfo(NewArgs, NewFTy), 67861a401caSAnders Carlsson CGM.GetAddrOfFunction(NewFD), ReturnValueSlot(), NewArgs, NewFD); 67959486a2dSAnders Carlsson 68059486a2dSAnders Carlsson // If an allocation function is declared with an empty exception specification 68159486a2dSAnders Carlsson // it returns null to indicate failure to allocate storage. [expr.new]p13. 68259486a2dSAnders Carlsson // (We don't need to check for null when there's no new initializer and 68359486a2dSAnders Carlsson // we're allocating a POD type). 68459486a2dSAnders Carlsson bool NullCheckResult = NewFTy->hasEmptyExceptionSpec() && 68559486a2dSAnders Carlsson !(AllocType->isPODType() && !E->hasInitializer()); 68659486a2dSAnders Carlsson 68759486a2dSAnders Carlsson llvm::BasicBlock *NewNull = 0; 68859486a2dSAnders Carlsson llvm::BasicBlock *NewNotNull = 0; 68959486a2dSAnders Carlsson llvm::BasicBlock *NewEnd = 0; 69059486a2dSAnders Carlsson 69159486a2dSAnders Carlsson llvm::Value *NewPtr = RV.getScalarVal(); 69259486a2dSAnders Carlsson 69359486a2dSAnders Carlsson if (NullCheckResult) { 69459486a2dSAnders Carlsson NewNull = createBasicBlock("new.null"); 69559486a2dSAnders Carlsson NewNotNull = createBasicBlock("new.notnull"); 69659486a2dSAnders Carlsson NewEnd = createBasicBlock("new.end"); 69759486a2dSAnders Carlsson 69859486a2dSAnders Carlsson llvm::Value *IsNull = 69959486a2dSAnders Carlsson Builder.CreateICmpEQ(NewPtr, 70059486a2dSAnders Carlsson llvm::Constant::getNullValue(NewPtr->getType()), 70159486a2dSAnders Carlsson "isnull"); 70259486a2dSAnders Carlsson 70359486a2dSAnders Carlsson Builder.CreateCondBr(IsNull, NewNull, NewNotNull); 70459486a2dSAnders Carlsson EmitBlock(NewNotNull); 70559486a2dSAnders Carlsson } 70659486a2dSAnders Carlsson 7073eb55cfeSKen Dyck CharUnits CookiePadding = CalculateCookiePadding(getContext(), E); 7083eb55cfeSKen Dyck if (!CookiePadding.isZero()) { 7093eb55cfeSKen Dyck CharUnits CookieOffset = 7103eb55cfeSKen Dyck CookiePadding - getContext().getTypeSizeInChars(SizeTy); 71159486a2dSAnders Carlsson 71259486a2dSAnders Carlsson llvm::Value *NumElementsPtr = 7133eb55cfeSKen Dyck Builder.CreateConstInBoundsGEP1_64(NewPtr, CookieOffset.getQuantity()); 71459486a2dSAnders Carlsson 71559486a2dSAnders Carlsson NumElementsPtr = Builder.CreateBitCast(NumElementsPtr, 71659486a2dSAnders Carlsson ConvertType(SizeTy)->getPointerTo()); 71759486a2dSAnders Carlsson Builder.CreateStore(NumElements, NumElementsPtr); 71859486a2dSAnders Carlsson 71959486a2dSAnders Carlsson // Now add the padding to the new ptr. 7203eb55cfeSKen Dyck NewPtr = Builder.CreateConstInBoundsGEP1_64(NewPtr, 7213eb55cfeSKen Dyck CookiePadding.getQuantity()); 72259486a2dSAnders Carlsson } 72359486a2dSAnders Carlsson 72447b4629bSFariborz Jahanian if (AllocType->isArrayType()) { 72547b4629bSFariborz Jahanian while (const ArrayType *AType = getContext().getAsArrayType(AllocType)) 72647b4629bSFariborz Jahanian AllocType = AType->getElementType(); 72747b4629bSFariborz Jahanian NewPtr = 72847b4629bSFariborz Jahanian Builder.CreateBitCast(NewPtr, 72947b4629bSFariborz Jahanian ConvertType(getContext().getPointerType(AllocType))); 73005fc5be3SDouglas Gregor EmitNewInitializer(*this, E, NewPtr, NumElements, AllocSizeWithoutCookie); 73147b4629bSFariborz Jahanian NewPtr = Builder.CreateBitCast(NewPtr, ConvertType(E->getType())); 73247b4629bSFariborz Jahanian } 73347b4629bSFariborz Jahanian else { 73447b4629bSFariborz Jahanian NewPtr = Builder.CreateBitCast(NewPtr, ConvertType(E->getType())); 73505fc5be3SDouglas Gregor EmitNewInitializer(*this, E, NewPtr, NumElements, AllocSizeWithoutCookie); 73647b4629bSFariborz Jahanian } 73759486a2dSAnders Carlsson 73859486a2dSAnders Carlsson if (NullCheckResult) { 73959486a2dSAnders Carlsson Builder.CreateBr(NewEnd); 74059486a2dSAnders Carlsson NewNotNull = Builder.GetInsertBlock(); 74159486a2dSAnders Carlsson EmitBlock(NewNull); 74259486a2dSAnders Carlsson Builder.CreateBr(NewEnd); 74359486a2dSAnders Carlsson EmitBlock(NewEnd); 74459486a2dSAnders Carlsson 74559486a2dSAnders Carlsson llvm::PHINode *PHI = Builder.CreatePHI(NewPtr->getType()); 74659486a2dSAnders Carlsson PHI->reserveOperandSpace(2); 74759486a2dSAnders Carlsson PHI->addIncoming(NewPtr, NewNotNull); 74859486a2dSAnders Carlsson PHI->addIncoming(llvm::Constant::getNullValue(NewPtr->getType()), NewNull); 74959486a2dSAnders Carlsson 75059486a2dSAnders Carlsson NewPtr = PHI; 75159486a2dSAnders Carlsson } 75259486a2dSAnders Carlsson 75359486a2dSAnders Carlsson return NewPtr; 75459486a2dSAnders Carlsson } 75559486a2dSAnders Carlsson 75621122cf6SAnders Carlsson static std::pair<llvm::Value *, llvm::Value *> 75721122cf6SAnders Carlsson GetAllocatedObjectPtrAndNumElements(CodeGenFunction &CGF, 75821122cf6SAnders Carlsson llvm::Value *Ptr, QualType DeleteTy) { 75921122cf6SAnders Carlsson QualType SizeTy = CGF.getContext().getSizeType(); 76021122cf6SAnders Carlsson const llvm::Type *SizeLTy = CGF.ConvertType(SizeTy); 76121122cf6SAnders Carlsson 7623eb55cfeSKen Dyck CharUnits DeleteTypeAlign = CGF.getContext().getTypeAlignInChars(DeleteTy); 7633eb55cfeSKen Dyck CharUnits CookiePadding = 7643eb55cfeSKen Dyck std::max(CGF.getContext().getTypeSizeInChars(SizeTy), 7653eb55cfeSKen Dyck DeleteTypeAlign); 7663eb55cfeSKen Dyck assert(!CookiePadding.isZero() && "CookiePadding should not be 0."); 76721122cf6SAnders Carlsson 76821122cf6SAnders Carlsson const llvm::Type *Int8PtrTy = llvm::Type::getInt8PtrTy(CGF.getLLVMContext()); 7693eb55cfeSKen Dyck CharUnits CookieOffset = 7703eb55cfeSKen Dyck CookiePadding - CGF.getContext().getTypeSizeInChars(SizeTy); 77121122cf6SAnders Carlsson 77221122cf6SAnders Carlsson llvm::Value *AllocatedObjectPtr = CGF.Builder.CreateBitCast(Ptr, Int8PtrTy); 77321122cf6SAnders Carlsson AllocatedObjectPtr = 77421122cf6SAnders Carlsson CGF.Builder.CreateConstInBoundsGEP1_64(AllocatedObjectPtr, 7753eb55cfeSKen Dyck -CookiePadding.getQuantity()); 77621122cf6SAnders Carlsson 77721122cf6SAnders Carlsson llvm::Value *NumElementsPtr = 77821122cf6SAnders Carlsson CGF.Builder.CreateConstInBoundsGEP1_64(AllocatedObjectPtr, 7793eb55cfeSKen Dyck CookieOffset.getQuantity()); 78021122cf6SAnders Carlsson NumElementsPtr = 78121122cf6SAnders Carlsson CGF.Builder.CreateBitCast(NumElementsPtr, SizeLTy->getPointerTo()); 78221122cf6SAnders Carlsson 78321122cf6SAnders Carlsson llvm::Value *NumElements = CGF.Builder.CreateLoad(NumElementsPtr); 78421122cf6SAnders Carlsson NumElements = 78521122cf6SAnders Carlsson CGF.Builder.CreateIntCast(NumElements, SizeLTy, /*isSigned=*/false); 78621122cf6SAnders Carlsson 78721122cf6SAnders Carlsson return std::make_pair(AllocatedObjectPtr, NumElements); 78821122cf6SAnders Carlsson } 78921122cf6SAnders Carlsson 79059486a2dSAnders Carlsson void CodeGenFunction::EmitDeleteCall(const FunctionDecl *DeleteFD, 79159486a2dSAnders Carlsson llvm::Value *Ptr, 79259486a2dSAnders Carlsson QualType DeleteTy) { 79359486a2dSAnders Carlsson const FunctionProtoType *DeleteFTy = 79459486a2dSAnders Carlsson DeleteFD->getType()->getAs<FunctionProtoType>(); 79559486a2dSAnders Carlsson 79659486a2dSAnders Carlsson CallArgList DeleteArgs; 79759486a2dSAnders Carlsson 79821122cf6SAnders Carlsson // Check if we need to pass the size to the delete operator. 79921122cf6SAnders Carlsson llvm::Value *Size = 0; 80021122cf6SAnders Carlsson QualType SizeTy; 80121122cf6SAnders Carlsson if (DeleteFTy->getNumArgs() == 2) { 80221122cf6SAnders Carlsson SizeTy = DeleteFTy->getArgType(1); 8037df3cbebSKen Dyck CharUnits DeleteTypeSize = getContext().getTypeSizeInChars(DeleteTy); 8047df3cbebSKen Dyck Size = llvm::ConstantInt::get(ConvertType(SizeTy), 8057df3cbebSKen Dyck DeleteTypeSize.getQuantity()); 80621122cf6SAnders Carlsson } 80721122cf6SAnders Carlsson 80821122cf6SAnders Carlsson if (DeleteFD->getOverloadedOperator() == OO_Array_Delete && 8093eb55cfeSKen Dyck !CalculateCookiePadding(getContext(), DeleteTy).isZero()) { 81021122cf6SAnders Carlsson // We need to get the number of elements in the array from the cookie. 81121122cf6SAnders Carlsson llvm::Value *AllocatedObjectPtr; 81221122cf6SAnders Carlsson llvm::Value *NumElements; 81321122cf6SAnders Carlsson llvm::tie(AllocatedObjectPtr, NumElements) = 81421122cf6SAnders Carlsson GetAllocatedObjectPtrAndNumElements(*this, Ptr, DeleteTy); 81521122cf6SAnders Carlsson 81621122cf6SAnders Carlsson // Multiply the size with the number of elements. 81721122cf6SAnders Carlsson if (Size) 81821122cf6SAnders Carlsson Size = Builder.CreateMul(NumElements, Size); 81921122cf6SAnders Carlsson 82021122cf6SAnders Carlsson Ptr = AllocatedObjectPtr; 82121122cf6SAnders Carlsson } 82221122cf6SAnders Carlsson 82359486a2dSAnders Carlsson QualType ArgTy = DeleteFTy->getArgType(0); 82459486a2dSAnders Carlsson llvm::Value *DeletePtr = Builder.CreateBitCast(Ptr, ConvertType(ArgTy)); 82559486a2dSAnders Carlsson DeleteArgs.push_back(std::make_pair(RValue::get(DeletePtr), ArgTy)); 82659486a2dSAnders Carlsson 82721122cf6SAnders Carlsson if (Size) 82859486a2dSAnders Carlsson DeleteArgs.push_back(std::make_pair(RValue::get(Size), SizeTy)); 82959486a2dSAnders Carlsson 83059486a2dSAnders Carlsson // Emit the call to delete. 831ab26cfa5SJohn McCall EmitCall(CGM.getTypes().getFunctionInfo(DeleteArgs, DeleteFTy), 83261a401caSAnders Carlsson CGM.GetAddrOfFunction(DeleteFD), ReturnValueSlot(), 83359486a2dSAnders Carlsson DeleteArgs, DeleteFD); 83459486a2dSAnders Carlsson } 83559486a2dSAnders Carlsson 83659486a2dSAnders Carlsson void CodeGenFunction::EmitCXXDeleteExpr(const CXXDeleteExpr *E) { 83759486a2dSAnders Carlsson 83859486a2dSAnders Carlsson // Get at the argument before we performed the implicit conversion 83959486a2dSAnders Carlsson // to void*. 84059486a2dSAnders Carlsson const Expr *Arg = E->getArgument(); 84159486a2dSAnders Carlsson while (const ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(Arg)) { 842e302792bSJohn McCall if (ICE->getCastKind() != CK_UserDefinedConversion && 84359486a2dSAnders Carlsson ICE->getType()->isVoidPointerType()) 84459486a2dSAnders Carlsson Arg = ICE->getSubExpr(); 84559486a2dSAnders Carlsson else 84659486a2dSAnders Carlsson break; 84759486a2dSAnders Carlsson } 84859486a2dSAnders Carlsson 84959486a2dSAnders Carlsson QualType DeleteTy = Arg->getType()->getAs<PointerType>()->getPointeeType(); 85059486a2dSAnders Carlsson 85159486a2dSAnders Carlsson llvm::Value *Ptr = EmitScalarExpr(Arg); 85259486a2dSAnders Carlsson 85359486a2dSAnders Carlsson // Null check the pointer. 85459486a2dSAnders Carlsson llvm::BasicBlock *DeleteNotNull = createBasicBlock("delete.notnull"); 85559486a2dSAnders Carlsson llvm::BasicBlock *DeleteEnd = createBasicBlock("delete.end"); 85659486a2dSAnders Carlsson 85759486a2dSAnders Carlsson llvm::Value *IsNull = 85859486a2dSAnders Carlsson Builder.CreateICmpEQ(Ptr, llvm::Constant::getNullValue(Ptr->getType()), 85959486a2dSAnders Carlsson "isnull"); 86059486a2dSAnders Carlsson 86159486a2dSAnders Carlsson Builder.CreateCondBr(IsNull, DeleteEnd, DeleteNotNull); 86259486a2dSAnders Carlsson EmitBlock(DeleteNotNull); 86359486a2dSAnders Carlsson 86459486a2dSAnders Carlsson bool ShouldCallDelete = true; 86559486a2dSAnders Carlsson 86659486a2dSAnders Carlsson // Call the destructor if necessary. 86759486a2dSAnders Carlsson if (const RecordType *RT = DeleteTy->getAs<RecordType>()) { 86859486a2dSAnders Carlsson if (CXXRecordDecl *RD = dyn_cast<CXXRecordDecl>(RT->getDecl())) { 86959486a2dSAnders Carlsson if (!RD->hasTrivialDestructor()) { 870bac7490fSDouglas Gregor const CXXDestructorDecl *Dtor = RD->getDestructor(); 87159486a2dSAnders Carlsson if (E->isArrayForm()) { 87221122cf6SAnders Carlsson llvm::Value *AllocatedObjectPtr; 87321122cf6SAnders Carlsson llvm::Value *NumElements; 87421122cf6SAnders Carlsson llvm::tie(AllocatedObjectPtr, NumElements) = 87521122cf6SAnders Carlsson GetAllocatedObjectPtrAndNumElements(*this, Ptr, DeleteTy); 876f5af8deaSAnders Carlsson 87759486a2dSAnders Carlsson EmitCXXAggrDestructorCall(Dtor, NumElements, Ptr); 878f5af8deaSAnders Carlsson } else if (Dtor->isVirtual()) { 87959486a2dSAnders Carlsson const llvm::Type *Ty = 88059486a2dSAnders Carlsson CGM.getTypes().GetFunctionType(CGM.getTypes().getFunctionInfo(Dtor), 88159486a2dSAnders Carlsson /*isVariadic=*/false); 88259486a2dSAnders Carlsson 88359486a2dSAnders Carlsson llvm::Value *Callee = BuildVirtualCall(Dtor, Dtor_Deleting, Ptr, Ty); 884e36a6b3eSAnders Carlsson EmitCXXMemberCall(Dtor, Callee, ReturnValueSlot(), Ptr, /*VTT=*/0, 885e36a6b3eSAnders Carlsson 0, 0); 88659486a2dSAnders Carlsson 88759486a2dSAnders Carlsson // The dtor took care of deleting the object. 88859486a2dSAnders Carlsson ShouldCallDelete = false; 88959486a2dSAnders Carlsson } else 890f8a71f08SAnders Carlsson EmitCXXDestructorCall(Dtor, Dtor_Complete, /*ForVirtualBase=*/false, 891f8a71f08SAnders Carlsson Ptr); 89259486a2dSAnders Carlsson } 89359486a2dSAnders Carlsson } 89459486a2dSAnders Carlsson } 89559486a2dSAnders Carlsson 89659486a2dSAnders Carlsson if (ShouldCallDelete) 89759486a2dSAnders Carlsson EmitDeleteCall(E->getOperatorDelete(), Ptr, DeleteTy); 89859486a2dSAnders Carlsson 89959486a2dSAnders Carlsson EmitBlock(DeleteEnd); 90059486a2dSAnders Carlsson } 90159486a2dSAnders Carlsson 90259486a2dSAnders Carlsson llvm::Value * CodeGenFunction::EmitCXXTypeidExpr(const CXXTypeidExpr *E) { 90359486a2dSAnders Carlsson QualType Ty = E->getType(); 90459486a2dSAnders Carlsson const llvm::Type *LTy = ConvertType(Ty)->getPointerTo(); 905fd7dfeb7SAnders Carlsson 9063f4336cbSAnders Carlsson if (E->isTypeOperand()) { 9073f4336cbSAnders Carlsson llvm::Constant *TypeInfo = 9083f4336cbSAnders Carlsson CGM.GetAddrOfRTTIDescriptor(E->getTypeOperand()); 9093f4336cbSAnders Carlsson return Builder.CreateBitCast(TypeInfo, LTy); 9103f4336cbSAnders Carlsson } 911fd7dfeb7SAnders Carlsson 91259486a2dSAnders Carlsson Expr *subE = E->getExprOperand(); 91359486a2dSAnders Carlsson Ty = subE->getType(); 91459486a2dSAnders Carlsson CanQualType CanTy = CGM.getContext().getCanonicalType(Ty); 91559486a2dSAnders Carlsson Ty = CanTy.getUnqualifiedType().getNonReferenceType(); 91659486a2dSAnders Carlsson if (const RecordType *RT = Ty->getAs<RecordType>()) { 91759486a2dSAnders Carlsson const CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl()); 91859486a2dSAnders Carlsson if (RD->isPolymorphic()) { 91959486a2dSAnders Carlsson // FIXME: if subE is an lvalue do 92059486a2dSAnders Carlsson LValue Obj = EmitLValue(subE); 92159486a2dSAnders Carlsson llvm::Value *This = Obj.getAddress(); 92259486a2dSAnders Carlsson LTy = LTy->getPointerTo()->getPointerTo(); 92359486a2dSAnders Carlsson llvm::Value *V = Builder.CreateBitCast(This, LTy); 92459486a2dSAnders Carlsson // We need to do a zero check for *p, unless it has NonNullAttr. 92559486a2dSAnders Carlsson // FIXME: PointerType->hasAttr<NonNullAttr>() 92659486a2dSAnders Carlsson bool CanBeZero = false; 92759486a2dSAnders Carlsson if (UnaryOperator *UO = dyn_cast<UnaryOperator>(subE->IgnoreParens())) 928e302792bSJohn McCall if (UO->getOpcode() == UO_Deref) 92959486a2dSAnders Carlsson CanBeZero = true; 93059486a2dSAnders Carlsson if (CanBeZero) { 93159486a2dSAnders Carlsson llvm::BasicBlock *NonZeroBlock = createBasicBlock(); 93259486a2dSAnders Carlsson llvm::BasicBlock *ZeroBlock = createBasicBlock(); 93359486a2dSAnders Carlsson 93459486a2dSAnders Carlsson llvm::Value *Zero = llvm::Constant::getNullValue(LTy); 93559486a2dSAnders Carlsson Builder.CreateCondBr(Builder.CreateICmpNE(V, Zero), 93659486a2dSAnders Carlsson NonZeroBlock, ZeroBlock); 93759486a2dSAnders Carlsson EmitBlock(ZeroBlock); 93859486a2dSAnders Carlsson /// Call __cxa_bad_typeid 93959486a2dSAnders Carlsson const llvm::Type *ResultType = llvm::Type::getVoidTy(VMContext); 94059486a2dSAnders Carlsson const llvm::FunctionType *FTy; 94159486a2dSAnders Carlsson FTy = llvm::FunctionType::get(ResultType, false); 94259486a2dSAnders Carlsson llvm::Value *F = CGM.CreateRuntimeFunction(FTy, "__cxa_bad_typeid"); 94359486a2dSAnders Carlsson Builder.CreateCall(F)->setDoesNotReturn(); 94459486a2dSAnders Carlsson Builder.CreateUnreachable(); 94559486a2dSAnders Carlsson EmitBlock(NonZeroBlock); 94659486a2dSAnders Carlsson } 94759486a2dSAnders Carlsson V = Builder.CreateLoad(V, "vtable"); 94859486a2dSAnders Carlsson V = Builder.CreateConstInBoundsGEP1_64(V, -1ULL); 94959486a2dSAnders Carlsson V = Builder.CreateLoad(V); 95059486a2dSAnders Carlsson return V; 95159486a2dSAnders Carlsson } 95259486a2dSAnders Carlsson } 9533f4336cbSAnders Carlsson return Builder.CreateBitCast(CGM.GetAddrOfRTTIDescriptor(Ty), LTy); 95459486a2dSAnders Carlsson } 95559486a2dSAnders Carlsson 95659486a2dSAnders Carlsson llvm::Value *CodeGenFunction::EmitDynamicCast(llvm::Value *V, 95759486a2dSAnders Carlsson const CXXDynamicCastExpr *DCE) { 9583f4336cbSAnders Carlsson QualType SrcTy = DCE->getSubExpr()->getType(); 9593f4336cbSAnders Carlsson QualType DestTy = DCE->getTypeAsWritten(); 9603f4336cbSAnders Carlsson QualType InnerType = DestTy->getPointeeType(); 9613f4336cbSAnders Carlsson 96259486a2dSAnders Carlsson const llvm::Type *LTy = ConvertType(DCE->getType()); 96359486a2dSAnders Carlsson 96459486a2dSAnders Carlsson bool CanBeZero = false; 96559486a2dSAnders Carlsson bool ToVoid = false; 96659486a2dSAnders Carlsson bool ThrowOnBad = false; 9673f4336cbSAnders Carlsson if (DestTy->isPointerType()) { 96859486a2dSAnders Carlsson // FIXME: if PointerType->hasAttr<NonNullAttr>(), we don't set this 96959486a2dSAnders Carlsson CanBeZero = true; 97059486a2dSAnders Carlsson if (InnerType->isVoidType()) 97159486a2dSAnders Carlsson ToVoid = true; 97259486a2dSAnders Carlsson } else { 97359486a2dSAnders Carlsson LTy = LTy->getPointerTo(); 974fa8b4955SDouglas Gregor 975fa8b4955SDouglas Gregor // FIXME: What if exceptions are disabled? 97659486a2dSAnders Carlsson ThrowOnBad = true; 97759486a2dSAnders Carlsson } 97859486a2dSAnders Carlsson 9793f4336cbSAnders Carlsson if (SrcTy->isPointerType() || SrcTy->isReferenceType()) 9803f4336cbSAnders Carlsson SrcTy = SrcTy->getPointeeType(); 9813f4336cbSAnders Carlsson SrcTy = SrcTy.getUnqualifiedType(); 9823f4336cbSAnders Carlsson 9830087bc85SAnders Carlsson if (DestTy->isPointerType() || DestTy->isReferenceType()) 9843f4336cbSAnders Carlsson DestTy = DestTy->getPointeeType(); 9853f4336cbSAnders Carlsson DestTy = DestTy.getUnqualifiedType(); 98659486a2dSAnders Carlsson 98759486a2dSAnders Carlsson llvm::BasicBlock *ContBlock = createBasicBlock(); 98859486a2dSAnders Carlsson llvm::BasicBlock *NullBlock = 0; 98959486a2dSAnders Carlsson llvm::BasicBlock *NonZeroBlock = 0; 99059486a2dSAnders Carlsson if (CanBeZero) { 99159486a2dSAnders Carlsson NonZeroBlock = createBasicBlock(); 99259486a2dSAnders Carlsson NullBlock = createBasicBlock(); 9933f4336cbSAnders Carlsson Builder.CreateCondBr(Builder.CreateIsNotNull(V), NonZeroBlock, NullBlock); 99459486a2dSAnders Carlsson EmitBlock(NonZeroBlock); 99559486a2dSAnders Carlsson } 99659486a2dSAnders Carlsson 99759486a2dSAnders Carlsson llvm::BasicBlock *BadCastBlock = 0; 99859486a2dSAnders Carlsson 9993f4336cbSAnders Carlsson const llvm::Type *PtrDiffTy = ConvertType(getContext().getPointerDiffType()); 100059486a2dSAnders Carlsson 100159486a2dSAnders Carlsson // See if this is a dynamic_cast(void*) 100259486a2dSAnders Carlsson if (ToVoid) { 100359486a2dSAnders Carlsson llvm::Value *This = V; 100459486a2dSAnders Carlsson V = Builder.CreateBitCast(This, PtrDiffTy->getPointerTo()->getPointerTo()); 100559486a2dSAnders Carlsson V = Builder.CreateLoad(V, "vtable"); 100659486a2dSAnders Carlsson V = Builder.CreateConstInBoundsGEP1_64(V, -2ULL); 100759486a2dSAnders Carlsson V = Builder.CreateLoad(V, "offset to top"); 100859486a2dSAnders Carlsson This = Builder.CreateBitCast(This, llvm::Type::getInt8PtrTy(VMContext)); 100959486a2dSAnders Carlsson V = Builder.CreateInBoundsGEP(This, V); 101059486a2dSAnders Carlsson V = Builder.CreateBitCast(V, LTy); 101159486a2dSAnders Carlsson } else { 101259486a2dSAnders Carlsson /// Call __dynamic_cast 101359486a2dSAnders Carlsson const llvm::Type *ResultType = llvm::Type::getInt8PtrTy(VMContext); 101459486a2dSAnders Carlsson const llvm::FunctionType *FTy; 101559486a2dSAnders Carlsson std::vector<const llvm::Type*> ArgTys; 101659486a2dSAnders Carlsson const llvm::Type *PtrToInt8Ty 101759486a2dSAnders Carlsson = llvm::Type::getInt8Ty(VMContext)->getPointerTo(); 101859486a2dSAnders Carlsson ArgTys.push_back(PtrToInt8Ty); 101959486a2dSAnders Carlsson ArgTys.push_back(PtrToInt8Ty); 102059486a2dSAnders Carlsson ArgTys.push_back(PtrToInt8Ty); 102159486a2dSAnders Carlsson ArgTys.push_back(PtrDiffTy); 102259486a2dSAnders Carlsson FTy = llvm::FunctionType::get(ResultType, ArgTys, false); 102359486a2dSAnders Carlsson 102459486a2dSAnders Carlsson // FIXME: Calculate better hint. 102559486a2dSAnders Carlsson llvm::Value *hint = llvm::ConstantInt::get(PtrDiffTy, -1ULL); 10263f4336cbSAnders Carlsson 10273f4336cbSAnders Carlsson assert(SrcTy->isRecordType() && "Src type must be record type!"); 10283f4336cbSAnders Carlsson assert(DestTy->isRecordType() && "Dest type must be record type!"); 10293f4336cbSAnders Carlsson 1030247894b3SDouglas Gregor llvm::Value *SrcArg 1031247894b3SDouglas Gregor = CGM.GetAddrOfRTTIDescriptor(SrcTy.getUnqualifiedType()); 1032247894b3SDouglas Gregor llvm::Value *DestArg 1033247894b3SDouglas Gregor = CGM.GetAddrOfRTTIDescriptor(DestTy.getUnqualifiedType()); 10343f4336cbSAnders Carlsson 103559486a2dSAnders Carlsson V = Builder.CreateBitCast(V, PtrToInt8Ty); 103659486a2dSAnders Carlsson V = Builder.CreateCall4(CGM.CreateRuntimeFunction(FTy, "__dynamic_cast"), 10373f4336cbSAnders Carlsson V, SrcArg, DestArg, hint); 103859486a2dSAnders Carlsson V = Builder.CreateBitCast(V, LTy); 103959486a2dSAnders Carlsson 104059486a2dSAnders Carlsson if (ThrowOnBad) { 104159486a2dSAnders Carlsson BadCastBlock = createBasicBlock(); 10423f4336cbSAnders Carlsson Builder.CreateCondBr(Builder.CreateIsNotNull(V), ContBlock, BadCastBlock); 104359486a2dSAnders Carlsson EmitBlock(BadCastBlock); 1044fa8b4955SDouglas Gregor /// Invoke __cxa_bad_cast 104559486a2dSAnders Carlsson ResultType = llvm::Type::getVoidTy(VMContext); 104659486a2dSAnders Carlsson const llvm::FunctionType *FBadTy; 104759486a2dSAnders Carlsson FBadTy = llvm::FunctionType::get(ResultType, false); 104859486a2dSAnders Carlsson llvm::Value *F = CGM.CreateRuntimeFunction(FBadTy, "__cxa_bad_cast"); 1049fa8b4955SDouglas Gregor if (llvm::BasicBlock *InvokeDest = getInvokeDest()) { 1050fa8b4955SDouglas Gregor llvm::BasicBlock *Cont = createBasicBlock("invoke.cont"); 1051fa8b4955SDouglas Gregor Builder.CreateInvoke(F, Cont, InvokeDest)->setDoesNotReturn(); 1052fa8b4955SDouglas Gregor EmitBlock(Cont); 1053fa8b4955SDouglas Gregor } else { 1054fa8b4955SDouglas Gregor // FIXME: Does this ever make sense? 105559486a2dSAnders Carlsson Builder.CreateCall(F)->setDoesNotReturn(); 1056fa8b4955SDouglas Gregor } 105759486a2dSAnders Carlsson Builder.CreateUnreachable(); 105859486a2dSAnders Carlsson } 105959486a2dSAnders Carlsson } 106059486a2dSAnders Carlsson 106159486a2dSAnders Carlsson if (CanBeZero) { 106259486a2dSAnders Carlsson Builder.CreateBr(ContBlock); 106359486a2dSAnders Carlsson EmitBlock(NullBlock); 106459486a2dSAnders Carlsson Builder.CreateBr(ContBlock); 106559486a2dSAnders Carlsson } 106659486a2dSAnders Carlsson EmitBlock(ContBlock); 106759486a2dSAnders Carlsson if (CanBeZero) { 106859486a2dSAnders Carlsson llvm::PHINode *PHI = Builder.CreatePHI(LTy); 106959486a2dSAnders Carlsson PHI->reserveOperandSpace(2); 107059486a2dSAnders Carlsson PHI->addIncoming(V, NonZeroBlock); 107159486a2dSAnders Carlsson PHI->addIncoming(llvm::Constant::getNullValue(LTy), NullBlock); 107259486a2dSAnders Carlsson V = PHI; 107359486a2dSAnders Carlsson } 107459486a2dSAnders Carlsson 107559486a2dSAnders Carlsson return V; 107659486a2dSAnders Carlsson } 1077