12437cbfaSAnders Carlsson //===--- CGBlocks.cpp - Emit LLVM Code for declarations -------------------===// 22437cbfaSAnders Carlsson // 32437cbfaSAnders Carlsson // The LLVM Compiler Infrastructure 42437cbfaSAnders Carlsson // 52437cbfaSAnders Carlsson // This file is distributed under the University of Illinois Open Source 62437cbfaSAnders Carlsson // License. See LICENSE.TXT for details. 72437cbfaSAnders Carlsson // 82437cbfaSAnders Carlsson //===----------------------------------------------------------------------===// 92437cbfaSAnders Carlsson // 102437cbfaSAnders Carlsson // This contains code to emit blocks. 112437cbfaSAnders Carlsson // 122437cbfaSAnders Carlsson //===----------------------------------------------------------------------===// 132437cbfaSAnders Carlsson 142e722b91SMike Stump #include "CGDebugInfo.h" 152437cbfaSAnders Carlsson #include "CodeGenFunction.h" 16989908fbSFariborz Jahanian #include "CGObjCRuntime.h" 172437cbfaSAnders Carlsson #include "CodeGenModule.h" 18692c6e37SMike Stump #include "clang/AST/DeclObjC.h" 192437cbfaSAnders Carlsson #include "llvm/Module.h" 209e2e1c90SBenjamin Kramer #include "llvm/ADT/SmallSet.h" 216a60fa24SAnders Carlsson #include "llvm/Target/TargetData.h" 222437cbfaSAnders Carlsson #include <algorithm> 23db714924STorok Edwin 242437cbfaSAnders Carlsson using namespace clang; 252437cbfaSAnders Carlsson using namespace CodeGen; 262437cbfaSAnders Carlsson 279d42f0f1SJohn McCall CGBlockInfo::CGBlockInfo(const char *N) 289d42f0f1SJohn McCall : Name(N), CXXThisRef(0), NeedsObjCSelf(false) { 299d42f0f1SJohn McCall 309d42f0f1SJohn McCall // Skip asm prefix, if any. 319d42f0f1SJohn McCall if (Name && Name[0] == '\01') 329d42f0f1SJohn McCall ++Name; 339d42f0f1SJohn McCall } 349d42f0f1SJohn McCall 359d42f0f1SJohn McCall 36d6ef62f0SMike Stump llvm::Constant *CodeGenFunction:: 37c05349e5SFariborz Jahanian BuildDescriptorBlockDecl(const BlockExpr *BE, const CGBlockInfo &Info, 38d2142cffSMike Stump const llvm::StructType* Ty, 39afa3c0a8SFariborz Jahanian llvm::Constant *BlockVarLayout, 40aeb0ffd5SMike Stump std::vector<HelperInfo> *NoteForHelper) { 41c05349e5SFariborz Jahanian bool BlockHasCopyDispose = Info.BlockHasCopyDispose; 42c05349e5SFariborz Jahanian CharUnits Size = Info.BlockSize; 43c2c38331SMike Stump const llvm::Type *UnsignedLongTy 44c2c38331SMike Stump = CGM.getTypes().ConvertType(getContext().UnsignedLongTy); 4585284bacSMike Stump llvm::Constant *C; 4685284bacSMike Stump std::vector<llvm::Constant*> Elts; 4785284bacSMike Stump 4885284bacSMike Stump // reserved 49b7a2fe6fSOwen Anderson C = llvm::ConstantInt::get(UnsignedLongTy, 0); 5085284bacSMike Stump Elts.push_back(C); 5185284bacSMike Stump 5285284bacSMike Stump // Size 532ac40a9fSMike Stump // FIXME: What is the right way to say this doesn't fit? We should give 542ac40a9fSMike Stump // a user diagnostic in that case. Better fix would be to change the 552ac40a9fSMike Stump // API to size_t. 5640775003SKen Dyck C = llvm::ConstantInt::get(UnsignedLongTy, Size.getQuantity()); 5785284bacSMike Stump Elts.push_back(C); 5885284bacSMike Stump 59fc83aa04SBlaine Garst // optional copy/dispose helpers 6085284bacSMike Stump if (BlockHasCopyDispose) { 6185284bacSMike Stump // copy_func_helper_decl 6267645930SMike Stump Elts.push_back(BuildCopyHelper(Ty, NoteForHelper)); 6385284bacSMike Stump 6485284bacSMike Stump // destroy_func_decl 6567645930SMike Stump Elts.push_back(BuildDestroyHelper(Ty, NoteForHelper)); 6685284bacSMike Stump } 6785284bacSMike Stump 68fc83aa04SBlaine Garst // Signature. non-optional ObjC-style method descriptor @encode sequence 69fc83aa04SBlaine Garst std::string BlockTypeEncoding; 70fc83aa04SBlaine Garst CGM.getContext().getObjCEncodingForBlock(BE, BlockTypeEncoding); 71fc83aa04SBlaine Garst 72fc83aa04SBlaine Garst Elts.push_back(llvm::ConstantExpr::getBitCast( 73fc83aa04SBlaine Garst CGM.GetAddrOfConstantCString(BlockTypeEncoding), PtrToInt8Ty)); 74fc83aa04SBlaine Garst 75fc83aa04SBlaine Garst // Layout. 76afa3c0a8SFariborz Jahanian C = BlockVarLayout; 77c05349e5SFariborz Jahanian 78fc83aa04SBlaine Garst Elts.push_back(C); 79fc83aa04SBlaine Garst 8041eaf0acSNick Lewycky C = llvm::ConstantStruct::get(VMContext, Elts, false); 8185284bacSMike Stump 82c10c8d3eSOwen Anderson C = new llvm::GlobalVariable(CGM.getModule(), C->getType(), true, 8385284bacSMike Stump llvm::GlobalValue::InternalLinkage, 84c10c8d3eSOwen Anderson C, "__block_descriptor_tmp"); 8585284bacSMike Stump return C; 8685284bacSMike Stump } 8785284bacSMike Stump 889d42f0f1SJohn McCall static void CollectBlockDeclRefInfo(const Stmt *S, CGBlockInfo &Info) { 89ed5e69feSAnders Carlsson for (Stmt::const_child_iterator I = S->child_begin(), E = S->child_end(); 90ed5e69feSAnders Carlsson I != E; ++I) 91dd942713SDaniel Dunbar if (*I) 929d42f0f1SJohn McCall CollectBlockDeclRefInfo(*I, Info); 93ed5e69feSAnders Carlsson 947fe9cc1dSMike Stump // We want to ensure we walk down into block literals so we can find 957fe9cc1dSMike Stump // all nested BlockDeclRefExprs. 96066b6166SMike Stump if (const BlockExpr *BE = dyn_cast<BlockExpr>(S)) { 979d42f0f1SJohn McCall Info.InnerBlocks.insert(BE->getBlockDecl()); 989d42f0f1SJohn McCall CollectBlockDeclRefInfo(BE->getBody(), Info); 99066b6166SMike Stump } 1007fe9cc1dSMike Stump 1019d42f0f1SJohn McCall else if (const BlockDeclRefExpr *BDRE = dyn_cast<BlockDeclRefExpr>(S)) { 1029d42f0f1SJohn McCall const ValueDecl *D = BDRE->getDecl(); 103ed5e69feSAnders Carlsson // FIXME: Handle enums. 1049d42f0f1SJohn McCall if (isa<FunctionDecl>(D)) 105ed5e69feSAnders Carlsson return; 106ed5e69feSAnders Carlsson 1079d42f0f1SJohn McCall if (isa<ImplicitParamDecl>(D) && 1089d42f0f1SJohn McCall isa<ObjCMethodDecl>(D->getDeclContext()) && 1099d42f0f1SJohn McCall cast<ObjCMethodDecl>(D->getDeclContext())->getSelfDecl() == D) { 1109d42f0f1SJohn McCall Info.NeedsObjCSelf = true; 1119d42f0f1SJohn McCall return; 1129d42f0f1SJohn McCall } 1139d42f0f1SJohn McCall 114066b6166SMike Stump // Only Decls that escape are added. 1159d42f0f1SJohn McCall if (!Info.InnerBlocks.count(D->getDeclContext())) 116066b6166SMike Stump Info.DeclRefs.push_back(BDRE); 117ed5e69feSAnders Carlsson } 11887fe5d56SJohn McCall 1199d42f0f1SJohn McCall // Make sure to capture implicit 'self' references due to super calls. 1207ea01c3fSChandler Carruth else if (const ObjCMessageExpr *E = dyn_cast<ObjCMessageExpr>(S)) { 1219d42f0f1SJohn McCall if (E->getReceiverKind() == ObjCMessageExpr::SuperClass || 1229d42f0f1SJohn McCall E->getReceiverKind() == ObjCMessageExpr::SuperInstance) 1239d42f0f1SJohn McCall Info.NeedsObjCSelf = true; 1247ea01c3fSChandler Carruth } 125*681c0754SFariborz Jahanian else if (const ObjCPropertyRefExpr *PE = dyn_cast<ObjCPropertyRefExpr>(S)) { 1269d42f0f1SJohn McCall // Getter/setter uses may also cause implicit super references, 1279d42f0f1SJohn McCall // which we can check for with: 128*681c0754SFariborz Jahanian if (PE->isSuperReceiver()) 1299d42f0f1SJohn McCall Info.NeedsObjCSelf = true; 130*681c0754SFariborz Jahanian } 131*681c0754SFariborz Jahanian else if (const ObjCImplicitSetterGetterRefExpr *IE = 132*681c0754SFariborz Jahanian dyn_cast<ObjCImplicitSetterGetterRefExpr>(S)) { 133*681c0754SFariborz Jahanian // Getter/setter uses may also cause implicit super references, 134*681c0754SFariborz Jahanian // which we can check for with: 135*681c0754SFariborz Jahanian if (IE->isSuperReceiver()) 136*681c0754SFariborz Jahanian Info.NeedsObjCSelf = true; 137*681c0754SFariborz Jahanian } 1389d42f0f1SJohn McCall else if (isa<CXXThisExpr>(S)) 13987fe5d56SJohn McCall Info.CXXThisRef = cast<CXXThisExpr>(S); 140ed5e69feSAnders Carlsson } 141ed5e69feSAnders Carlsson 1429d42f0f1SJohn McCall /// CanBlockBeGlobal - Given a CGBlockInfo struct, determines if a block can be 1432d33d630SMike Stump /// declared as a global variable instead of on the stack. 1449d42f0f1SJohn McCall static bool CanBlockBeGlobal(const CGBlockInfo &Info) { 1457fe9cc1dSMike Stump return Info.DeclRefs.empty(); 1467fe9cc1dSMike Stump } 1477fe9cc1dSMike Stump 1487fe9cc1dSMike Stump /// AllocateAllBlockDeclRefs - Preallocate all nested BlockDeclRefExprs to 1497fe9cc1dSMike Stump /// ensure we can generate the debug information for the parameter for the block 1507fe9cc1dSMike Stump /// invoke function. 1519d42f0f1SJohn McCall static void AllocateAllBlockDeclRefs(CodeGenFunction &CGF, CGBlockInfo &Info) { 15287fe5d56SJohn McCall if (Info.CXXThisRef) 1539d42f0f1SJohn McCall CGF.AllocateBlockCXXThisPointer(Info.CXXThisRef); 1547fe9cc1dSMike Stump 1557fe9cc1dSMike Stump for (size_t i = 0; i < Info.DeclRefs.size(); ++i) 1569d42f0f1SJohn McCall CGF.AllocateBlockDecl(Info.DeclRefs[i]); 1579d42f0f1SJohn McCall 1589d42f0f1SJohn McCall if (Info.NeedsObjCSelf) { 1599d42f0f1SJohn McCall ValueDecl *Self = cast<ObjCMethodDecl>(CGF.CurFuncDecl)->getSelfDecl(); 1609d42f0f1SJohn McCall BlockDeclRefExpr *BDRE = 1619d42f0f1SJohn McCall new (CGF.getContext()) BlockDeclRefExpr(Self, Self->getType(), 1629d42f0f1SJohn McCall SourceLocation(), false); 1639d42f0f1SJohn McCall Info.DeclRefs.push_back(BDRE); 1649d42f0f1SJohn McCall CGF.AllocateBlockDecl(BDRE); 1659d42f0f1SJohn McCall } 166ed5e69feSAnders Carlsson } 167ed5e69feSAnders Carlsson 168d5010898SFariborz Jahanian static unsigned computeBlockFlag(CodeGenModule &CGM, 169d5010898SFariborz Jahanian const BlockExpr *BE, unsigned flags) { 170d5010898SFariborz Jahanian QualType BPT = BE->getType(); 171d5010898SFariborz Jahanian const FunctionType *ftype = BPT->getPointeeType()->getAs<FunctionType>(); 172d5010898SFariborz Jahanian QualType ResultType = ftype->getResultType(); 173d5010898SFariborz Jahanian 174d5010898SFariborz Jahanian CallArgList Args; 175d5010898SFariborz Jahanian CodeGenTypes &Types = CGM.getTypes(); 176d5010898SFariborz Jahanian const CGFunctionInfo &FnInfo = Types.getFunctionInfo(ResultType, Args, 177d5010898SFariborz Jahanian FunctionType::ExtInfo()); 178d5010898SFariborz Jahanian if (CGM.ReturnTypeUsesSRet(FnInfo)) 179d5010898SFariborz Jahanian flags |= CodeGenFunction::BLOCK_USE_STRET; 180d5010898SFariborz Jahanian return flags; 181d5010898SFariborz Jahanian } 182d5010898SFariborz Jahanian 1832d33d630SMike Stump // FIXME: Push most into CGM, passing down a few bits, like current function 1842d33d630SMike Stump // name. 185b750d928SMike Stump llvm::Value *CodeGenFunction::BuildBlockLiteralTmp(const BlockExpr *BE) { 186ed5e69feSAnders Carlsson std::string Name = CurFn->getName(); 1879d42f0f1SJohn McCall CGBlockInfo Info(Name.c_str()); 1889d42f0f1SJohn McCall Info.InnerBlocks.insert(BE->getBlockDecl()); 1899d42f0f1SJohn McCall CollectBlockDeclRefInfo(BE->getBody(), Info); 190ed5e69feSAnders Carlsson 191ed5e69feSAnders Carlsson // Check if the block can be global. 1922d33d630SMike Stump // FIXME: This test doesn't work for nested blocks yet. Longer term, I'd like 1932d33d630SMike Stump // to just have one code path. We should move this function into CGM and pass 1942d33d630SMike Stump // CGF, then we can just check to see if CGF is 0. 19597d01d50SMike Stump if (0 && CanBlockBeGlobal(Info)) 196ed5e69feSAnders Carlsson return CGM.GetAddrOfGlobalBlock(BE, Name.c_str()); 19785284bacSMike Stump 198950a9518SDavid Chisnall size_t BlockFields = 5; 199950a9518SDavid Chisnall 200950a9518SDavid Chisnall std::vector<llvm::Constant*> Elts(BlockFields); 201950a9518SDavid Chisnall 20285284bacSMike Stump llvm::Constant *C; 203b750d928SMike Stump llvm::Value *V; 20485284bacSMike Stump 20585284bacSMike Stump { 206afa3c0a8SFariborz Jahanian llvm::Constant *BlockVarLayout; 20785284bacSMike Stump // C = BuildBlockStructInitlist(); 208f27ab71dSBlaine Garst unsigned int flags = BLOCK_HAS_SIGNATURE; 209950a9518SDavid Chisnall 2104446dcf0SMike Stump // We run this first so that we set BlockHasCopyDispose from the entire 2114446dcf0SMike Stump // block literal. 2124446dcf0SMike Stump // __invoke 2134446dcf0SMike Stump llvm::Function *Fn 2149b5528d2SFariborz Jahanian = CodeGenFunction(CGM).GenerateBlockFunction(CurGD, BE, Info, CurFuncDecl, 215afa3c0a8SFariborz Jahanian BlockVarLayout, 2169d42f0f1SJohn McCall LocalDeclMap); 2179d42f0f1SJohn McCall BlockHasCopyDispose |= Info.BlockHasCopyDispose; 2184446dcf0SMike Stump Elts[3] = Fn; 2194446dcf0SMike Stump 22018bb9284SMike Stump // FIXME: Don't use BlockHasCopyDispose, it is set more often then 22118bb9284SMike Stump // necessary, for example: { ^{ __block int i; ^{ i = 1; }(); }(); } 2229d42f0f1SJohn McCall if (Info.BlockHasCopyDispose) 22385284bacSMike Stump flags |= BLOCK_HAS_COPY_DISPOSE; 22485284bacSMike Stump 225499ae7ecSMike Stump // __isa 2267ab278dbSMike Stump C = CGM.getNSConcreteStackBlock(); 227ade90fd1SOwen Anderson C = llvm::ConstantExpr::getBitCast(C, PtrToInt8Ty); 2284446dcf0SMike Stump Elts[0] = C; 22985284bacSMike Stump 23085284bacSMike Stump // __flags 231d5010898SFariborz Jahanian flags = computeBlockFlag(CGM, BE, flags); 23285284bacSMike Stump const llvm::IntegerType *IntTy = cast<llvm::IntegerType>( 23385284bacSMike Stump CGM.getTypes().ConvertType(CGM.getContext().IntTy)); 234b7a2fe6fSOwen Anderson C = llvm::ConstantInt::get(IntTy, flags); 2354446dcf0SMike Stump Elts[1] = C; 23685284bacSMike Stump 23785284bacSMike Stump // __reserved 238b7a2fe6fSOwen Anderson C = llvm::ConstantInt::get(IntTy, 0); 2394446dcf0SMike Stump Elts[2] = C; 24085284bacSMike Stump 2419d42f0f1SJohn McCall if (Info.BlockLayout.empty()) { 242d6ef62f0SMike Stump // __descriptor 243afa3c0a8SFariborz Jahanian C = llvm::Constant::getNullValue(PtrToInt8Ty); 244afa3c0a8SFariborz Jahanian Elts[4] = BuildDescriptorBlockDecl(BE, Info, 0, C, 0); 245d6ef62f0SMike Stump 2463a139f37SMike Stump // Optimize to being a global block. 2473a139f37SMike Stump Elts[0] = CGM.getNSConcreteGlobalBlock(); 248f27ab71dSBlaine Garst 249b7a2fe6fSOwen Anderson Elts[1] = llvm::ConstantInt::get(IntTy, flags|BLOCK_IS_GLOBAL); 2503a139f37SMike Stump 25141eaf0acSNick Lewycky C = llvm::ConstantStruct::get(VMContext, Elts, false); 25285284bacSMike Stump 253c10c8d3eSOwen Anderson C = new llvm::GlobalVariable(CGM.getModule(), C->getType(), true, 254d6b28fc6SBenjamin Kramer llvm::GlobalValue::InternalLinkage, C, 255d6b28fc6SBenjamin Kramer "__block_holder_tmp_" + 256d6b28fc6SBenjamin Kramer llvm::Twine(CGM.getGlobalUniqueCount())); 2575d2534adSMike Stump QualType BPT = BE->getType(); 258ade90fd1SOwen Anderson C = llvm::ConstantExpr::getBitCast(C, ConvertType(BPT)); 25985284bacSMike Stump return C; 26085284bacSMike Stump } 26185284bacSMike Stump 2629d42f0f1SJohn McCall std::vector<const llvm::Type *> Types(BlockFields+Info.BlockLayout.size()); 263aeb0ffd5SMike Stump for (int i=0; i<4; ++i) 264b750d928SMike Stump Types[i] = Elts[i]->getType(); 265aeb0ffd5SMike Stump Types[4] = PtrToInt8Ty; 26685284bacSMike Stump 2679d42f0f1SJohn McCall for (unsigned i = 0, n = Info.BlockLayout.size(); i != n; ++i) { 2689d42f0f1SJohn McCall const Expr *E = Info.BlockLayout[i]; 2691db7d04bSMike Stump const BlockDeclRefExpr *BDRE = dyn_cast<BlockDeclRefExpr>(E); 2701db7d04bSMike Stump QualType Ty = E->getType(); 27197d01d50SMike Stump if (BDRE && BDRE->isByRef()) { 272d1b378e0SFariborz Jahanian Types[i+BlockFields] = 273d1b378e0SFariborz Jahanian llvm::PointerType::get(BuildByRefType(BDRE->getDecl()), 0); 274d1b378e0SFariborz Jahanian } else if (BDRE && BDRE->getDecl()->getType()->isReferenceType()) { 275d1b378e0SFariborz Jahanian Types[i+BlockFields] = llvm::PointerType::get(ConvertType(Ty), 0); 27697d01d50SMike Stump } else 277950a9518SDavid Chisnall Types[i+BlockFields] = ConvertType(Ty); 2781db7d04bSMike Stump } 279b750d928SMike Stump 280758428f4SOwen Anderson llvm::StructType *Ty = llvm::StructType::get(VMContext, Types, true); 281b750d928SMike Stump 282b750d928SMike Stump llvm::AllocaInst *A = CreateTempAlloca(Ty); 2839d42f0f1SJohn McCall A->setAlignment(Info.BlockAlign.getQuantity()); 284b750d928SMike Stump V = A; 285b750d928SMike Stump 28687fe5d56SJohn McCall // Build layout / cleanup information for all the data entries in the 28787fe5d56SJohn McCall // layout, and write the enclosing fields into the type. 2889d42f0f1SJohn McCall std::vector<HelperInfo> NoteForHelper(Info.BlockLayout.size()); 28987fe5d56SJohn McCall unsigned NumHelpers = 0; 290aeb0ffd5SMike Stump 291aeb0ffd5SMike Stump for (unsigned i=0; i<4; ++i) 292b750d928SMike Stump Builder.CreateStore(Elts[i], Builder.CreateStructGEP(V, i, "block.tmp")); 293b750d928SMike Stump 2949d42f0f1SJohn McCall for (unsigned i=0; i < Info.BlockLayout.size(); ++i) { 2959d42f0f1SJohn McCall const Expr *E = Info.BlockLayout[i]; 296b750d928SMike Stump 2971db7d04bSMike Stump // Skip padding. 29887fe5d56SJohn McCall if (isa<DeclRefExpr>(E)) continue; 2991db7d04bSMike Stump 300950a9518SDavid Chisnall llvm::Value* Addr = Builder.CreateStructGEP(V, i+BlockFields, "tmp"); 30187fe5d56SJohn McCall HelperInfo &Note = NoteForHelper[NumHelpers++]; 30287fe5d56SJohn McCall 30387fe5d56SJohn McCall Note.index = i+5; 30487fe5d56SJohn McCall 30587fe5d56SJohn McCall if (isa<CXXThisExpr>(E)) { 30687fe5d56SJohn McCall Note.RequiresCopying = false; 30787fe5d56SJohn McCall Note.flag = BLOCK_FIELD_IS_OBJECT; 30887fe5d56SJohn McCall 30987fe5d56SJohn McCall Builder.CreateStore(LoadCXXThis(), Addr); 31087fe5d56SJohn McCall continue; 31187fe5d56SJohn McCall } 31287fe5d56SJohn McCall 31387fe5d56SJohn McCall const BlockDeclRefExpr *BDRE = cast<BlockDeclRefExpr>(E); 31487fe5d56SJohn McCall const ValueDecl *VD = BDRE->getDecl(); 31587fe5d56SJohn McCall QualType T = VD->getType(); 31687fe5d56SJohn McCall 31787fe5d56SJohn McCall Note.RequiresCopying = BlockRequiresCopying(T); 31887fe5d56SJohn McCall 31987fe5d56SJohn McCall if (BDRE->isByRef()) { 32087fe5d56SJohn McCall Note.flag = BLOCK_FIELD_IS_BYREF; 32187fe5d56SJohn McCall if (T.isObjCGCWeak()) 32287fe5d56SJohn McCall Note.flag |= BLOCK_FIELD_IS_WEAK; 32387fe5d56SJohn McCall } else if (T->isBlockPointerType()) { 32487fe5d56SJohn McCall Note.flag = BLOCK_FIELD_IS_BLOCK; 32587fe5d56SJohn McCall } else { 32687fe5d56SJohn McCall Note.flag = BLOCK_FIELD_IS_OBJECT; 32787fe5d56SJohn McCall } 328aeb0ffd5SMike Stump 3291db7d04bSMike Stump if (LocalDeclMap[VD]) { 33097d01d50SMike Stump if (BDRE->isByRef()) { 33197d01d50SMike Stump llvm::Value *Loc = LocalDeclMap[VD]; 33297d01d50SMike Stump Loc = Builder.CreateStructGEP(Loc, 1, "forwarding"); 333c76493a6SDaniel Dunbar Loc = Builder.CreateLoad(Loc); 33497d01d50SMike Stump Builder.CreateStore(Loc, Addr); 33597d01d50SMike Stump continue; 33687fe5d56SJohn McCall } else { 3373b836180SFariborz Jahanian if (BDRE->getCopyConstructorExpr()) { 338ea882cd9SFariborz Jahanian E = BDRE->getCopyConstructorExpr(); 339bd30929eSJohn McCall PushDestructorCleanup(E->getType(), Addr); 3403b836180SFariborz Jahanian } 341ea882cd9SFariborz Jahanian else { 34287fe5d56SJohn McCall E = new (getContext()) DeclRefExpr(const_cast<ValueDecl*>(VD), 343d1b378e0SFariborz Jahanian VD->getType().getNonReferenceType(), 344d1b378e0SFariborz Jahanian SourceLocation()); 34564176c2cSFariborz Jahanian if (VD->getType()->isReferenceType()) { 346d1b378e0SFariborz Jahanian E = new (getContext()) 347e302792bSJohn McCall UnaryOperator(const_cast<Expr*>(E), UO_AddrOf, 348d1b378e0SFariborz Jahanian getContext().getPointerType(E->getType()), 349ed6c7440SDouglas Gregor SourceLocation()); 3501db7d04bSMike Stump } 35164176c2cSFariborz Jahanian } 35264176c2cSFariborz Jahanian } 35387fe5d56SJohn McCall } 35487fe5d56SJohn McCall 35597d01d50SMike Stump if (BDRE->isByRef()) { 3561db7d04bSMike Stump E = new (getContext()) 357e302792bSJohn McCall UnaryOperator(const_cast<Expr*>(E), UO_AddrOf, 3581db7d04bSMike Stump getContext().getPointerType(E->getType()), 3591db7d04bSMike Stump SourceLocation()); 36097d01d50SMike Stump } 3611db7d04bSMike Stump 3627a626f63SJohn McCall RValue r = EmitAnyExpr(E, AggValueSlot::forAddr(Addr, false, true)); 36397d01d50SMike Stump if (r.isScalar()) { 36497d01d50SMike Stump llvm::Value *Loc = r.getScalarVal(); 365950a9518SDavid Chisnall const llvm::Type *Ty = Types[i+BlockFields]; 36697d01d50SMike Stump if (BDRE->isByRef()) { 3672d33d630SMike Stump // E is now the address of the value field, instead, we want the 3682d33d630SMike Stump // address of the actual ByRef struct. We optimize this slightly 3692d33d630SMike Stump // compared to gcc by not grabbing the forwarding slot as this must 3702d33d630SMike Stump // be done during Block_copy for us, and we can postpone the work 3712d33d630SMike Stump // until then. 37240775003SKen Dyck CharUnits offset = BlockDecls[BDRE->getDecl()]; 3732d33d630SMike Stump 3742d33d630SMike Stump llvm::Value *BlockLiteral = LoadBlockStruct(); 3752d33d630SMike Stump 3762d33d630SMike Stump Loc = Builder.CreateGEP(BlockLiteral, 3775e016ae9SChris Lattner llvm::ConstantInt::get(Int64Ty, offset.getQuantity()), 3782d33d630SMike Stump "block.literal"); 3799793f0e4SOwen Anderson Ty = llvm::PointerType::get(Ty, 0); 38097d01d50SMike Stump Loc = Builder.CreateBitCast(Loc, Ty); 381c76493a6SDaniel Dunbar Loc = Builder.CreateLoad(Loc); 3822d33d630SMike Stump // Loc = Builder.CreateBitCast(Loc, Ty); 38397d01d50SMike Stump } 38497d01d50SMike Stump Builder.CreateStore(Loc, Addr); 38597d01d50SMike Stump } else if (r.isComplex()) 386b750d928SMike Stump // FIXME: implement 387b750d928SMike Stump ErrorUnsupported(BE, "complex in block literal"); 388b750d928SMike Stump else if (r.isAggregate()) 389b750d928SMike Stump ; // Already created into the destination 390b750d928SMike Stump else 391b750d928SMike Stump assert (0 && "bad block variable"); 392b750d928SMike Stump // FIXME: Ensure that the offset created by the backend for 393b750d928SMike Stump // the struct matches the previously computed offset in BlockDecls. 394b750d928SMike Stump } 39587fe5d56SJohn McCall NoteForHelper.resize(NumHelpers); 396d6ef62f0SMike Stump 397d6ef62f0SMike Stump // __descriptor 398c05349e5SFariborz Jahanian llvm::Value *Descriptor = BuildDescriptorBlockDecl(BE, Info, Ty, 399afa3c0a8SFariborz Jahanian BlockVarLayout, 400aeb0ffd5SMike Stump &NoteForHelper); 401aeb0ffd5SMike Stump Descriptor = Builder.CreateBitCast(Descriptor, PtrToInt8Ty); 402aeb0ffd5SMike Stump Builder.CreateStore(Descriptor, Builder.CreateStructGEP(V, 4, "block.tmp")); 403b750d928SMike Stump } 404b750d928SMike Stump 405b750d928SMike Stump QualType BPT = BE->getType(); 406989908fbSFariborz Jahanian V = Builder.CreateBitCast(V, ConvertType(BPT)); 407989908fbSFariborz Jahanian // See if this is a __weak block variable and the must call objc_read_weak 408989908fbSFariborz Jahanian // on it. 409989908fbSFariborz Jahanian const FunctionType *ftype = BPT->getPointeeType()->getAs<FunctionType>(); 410989908fbSFariborz Jahanian QualType RES = ftype->getResultType(); 411989908fbSFariborz Jahanian if (RES.isObjCGCWeak()) { 412989908fbSFariborz Jahanian // Must cast argument to id* 413989908fbSFariborz Jahanian const llvm::Type *ObjectPtrTy = 414989908fbSFariborz Jahanian ConvertType(CGM.getContext().getObjCIdType()); 415989908fbSFariborz Jahanian const llvm::Type *PtrObjectPtrTy = 416989908fbSFariborz Jahanian llvm::PointerType::getUnqual(ObjectPtrTy); 417989908fbSFariborz Jahanian V = Builder.CreateBitCast(V, PtrObjectPtrTy); 418989908fbSFariborz Jahanian V = CGM.getObjCRuntime().EmitObjCWeakRead(*this, V); 419989908fbSFariborz Jahanian } 420989908fbSFariborz Jahanian return V; 421b750d928SMike Stump } 42285284bacSMike Stump 42385284bacSMike Stump 42495435674SMike Stump const llvm::Type *BlockModule::getBlockDescriptorType() { 425650c932dSMike Stump if (BlockDescriptorType) 426650c932dSMike Stump return BlockDescriptorType; 4272437cbfaSAnders Carlsson 4282437cbfaSAnders Carlsson const llvm::Type *UnsignedLongTy = 429650c932dSMike Stump getTypes().ConvertType(getContext().UnsignedLongTy); 4302437cbfaSAnders Carlsson 4312437cbfaSAnders Carlsson // struct __block_descriptor { 4322437cbfaSAnders Carlsson // unsigned long reserved; 4332437cbfaSAnders Carlsson // unsigned long block_size; 434fc83aa04SBlaine Garst // 435fc83aa04SBlaine Garst // // later, the following will be added 436fc83aa04SBlaine Garst // 437fc83aa04SBlaine Garst // struct { 438fc83aa04SBlaine Garst // void (*copyHelper)(); 439fc83aa04SBlaine Garst // void (*copyHelper)(); 440fc83aa04SBlaine Garst // } helpers; // !!! optional 441fc83aa04SBlaine Garst // 442fc83aa04SBlaine Garst // const char *signature; // the block signature 443fc83aa04SBlaine Garst // const char *layout; // reserved 4442437cbfaSAnders Carlsson // }; 445758428f4SOwen Anderson BlockDescriptorType = llvm::StructType::get(UnsignedLongTy->getContext(), 446758428f4SOwen Anderson UnsignedLongTy, 4472437cbfaSAnders Carlsson UnsignedLongTy, 4482437cbfaSAnders Carlsson NULL); 4492437cbfaSAnders Carlsson 450650c932dSMike Stump getModule().addTypeName("struct.__block_descriptor", 451650c932dSMike Stump BlockDescriptorType); 4522437cbfaSAnders Carlsson 453650c932dSMike Stump return BlockDescriptorType; 4542437cbfaSAnders Carlsson } 4552437cbfaSAnders Carlsson 45695435674SMike Stump const llvm::Type *BlockModule::getGenericBlockLiteralType() { 457005c9a62SMike Stump if (GenericBlockLiteralType) 458005c9a62SMike Stump return GenericBlockLiteralType; 4592437cbfaSAnders Carlsson 4602437cbfaSAnders Carlsson const llvm::Type *BlockDescPtrTy = 4619793f0e4SOwen Anderson llvm::PointerType::getUnqual(getBlockDescriptorType()); 4622437cbfaSAnders Carlsson 46392bbd6d4SMike Stump const llvm::IntegerType *IntTy = cast<llvm::IntegerType>( 46492bbd6d4SMike Stump getTypes().ConvertType(getContext().IntTy)); 46592bbd6d4SMike Stump 4662437cbfaSAnders Carlsson // struct __block_literal_generic { 4675d2534adSMike Stump // void *__isa; 4685d2534adSMike Stump // int __flags; 4695d2534adSMike Stump // int __reserved; 4705d2534adSMike Stump // void (*__invoke)(void *); 4715d2534adSMike Stump // struct __block_descriptor *__descriptor; 4722437cbfaSAnders Carlsson // }; 473758428f4SOwen Anderson GenericBlockLiteralType = llvm::StructType::get(IntTy->getContext(), 474758428f4SOwen Anderson PtrToInt8Ty, 47592bbd6d4SMike Stump IntTy, 47692bbd6d4SMike Stump IntTy, 477626aecc4SMike Stump PtrToInt8Ty, 4782437cbfaSAnders Carlsson BlockDescPtrTy, 4792437cbfaSAnders Carlsson NULL); 4802437cbfaSAnders Carlsson 481005c9a62SMike Stump getModule().addTypeName("struct.__block_literal_generic", 482005c9a62SMike Stump GenericBlockLiteralType); 4832437cbfaSAnders Carlsson 484005c9a62SMike Stump return GenericBlockLiteralType; 4852437cbfaSAnders Carlsson } 4862437cbfaSAnders Carlsson 4875d2534adSMike Stump 488bfb3671bSAnders Carlsson RValue CodeGenFunction::EmitBlockCallExpr(const CallExpr* E, 489bfb3671bSAnders Carlsson ReturnValueSlot ReturnValue) { 4902437cbfaSAnders Carlsson const BlockPointerType *BPT = 491c23c7e6aSTed Kremenek E->getCallee()->getType()->getAs<BlockPointerType>(); 4922437cbfaSAnders Carlsson 4932437cbfaSAnders Carlsson llvm::Value *Callee = EmitScalarExpr(E->getCallee()); 4942437cbfaSAnders Carlsson 4952437cbfaSAnders Carlsson // Get a pointer to the generic block literal. 4962437cbfaSAnders Carlsson const llvm::Type *BlockLiteralTy = 4979793f0e4SOwen Anderson llvm::PointerType::getUnqual(CGM.getGenericBlockLiteralType()); 4982437cbfaSAnders Carlsson 4992437cbfaSAnders Carlsson // Bitcast the callee to a block literal. 5002437cbfaSAnders Carlsson llvm::Value *BlockLiteral = 5012437cbfaSAnders Carlsson Builder.CreateBitCast(Callee, BlockLiteralTy, "block.literal"); 5022437cbfaSAnders Carlsson 5032437cbfaSAnders Carlsson // Get the function pointer from the literal. 5042437cbfaSAnders Carlsson llvm::Value *FuncPtr = Builder.CreateStructGEP(BlockLiteral, 3, "tmp"); 5052437cbfaSAnders Carlsson 5062437cbfaSAnders Carlsson BlockLiteral = 5072437cbfaSAnders Carlsson Builder.CreateBitCast(BlockLiteral, 508abd5b90eSBenjamin Kramer llvm::Type::getInt8PtrTy(VMContext), 5092437cbfaSAnders Carlsson "tmp"); 5102437cbfaSAnders Carlsson 5112437cbfaSAnders Carlsson // Add the block literal. 5122437cbfaSAnders Carlsson QualType VoidPtrTy = getContext().getPointerType(getContext().VoidTy); 5132437cbfaSAnders Carlsson CallArgList Args; 5142437cbfaSAnders Carlsson Args.push_back(std::make_pair(RValue::get(BlockLiteral), VoidPtrTy)); 5152437cbfaSAnders Carlsson 516479e6fc7SAnders Carlsson QualType FnType = BPT->getPointeeType(); 517479e6fc7SAnders Carlsson 5182437cbfaSAnders Carlsson // And the rest of the arguments. 5199dd450bbSJohn McCall EmitCallArgs(Args, FnType->getAs<FunctionProtoType>(), 520479e6fc7SAnders Carlsson E->arg_begin(), E->arg_end()); 5212437cbfaSAnders Carlsson 5225f50c651SAnders Carlsson // Load the function. 523c76493a6SDaniel Dunbar llvm::Value *Func = Builder.CreateLoad(FuncPtr, "tmp"); 5245f50c651SAnders Carlsson 525ab26cfa5SJohn McCall const FunctionType *FuncTy = FnType->getAs<FunctionType>(); 526ab26cfa5SJohn McCall QualType ResultType = FuncTy->getResultType(); 527a60cbcdfSAnders Carlsson 528a60cbcdfSAnders Carlsson const CGFunctionInfo &FnInfo = 529c50c27ccSRafael Espindola CGM.getTypes().getFunctionInfo(ResultType, Args, 530c50c27ccSRafael Espindola FuncTy->getExtInfo()); 5315f50c651SAnders Carlsson 5325f50c651SAnders Carlsson // Cast the function pointer to the right type. 5335f50c651SAnders Carlsson const llvm::Type *BlockFTy = 534a60cbcdfSAnders Carlsson CGM.getTypes().GetFunctionType(FnInfo, false); 5355f50c651SAnders Carlsson 5369793f0e4SOwen Anderson const llvm::Type *BlockFTyPtr = llvm::PointerType::getUnqual(BlockFTy); 5375f50c651SAnders Carlsson Func = Builder.CreateBitCast(Func, BlockFTyPtr); 5385f50c651SAnders Carlsson 5392437cbfaSAnders Carlsson // And call the block. 540bfb3671bSAnders Carlsson return EmitCall(FnInfo, Func, ReturnValue, Args); 5412437cbfaSAnders Carlsson } 5426a60fa24SAnders Carlsson 54387fe5d56SJohn McCall void CodeGenFunction::AllocateBlockCXXThisPointer(const CXXThisExpr *E) { 54487fe5d56SJohn McCall assert(BlockCXXThisOffset.isZero() && "already computed 'this' pointer"); 54587fe5d56SJohn McCall 54687fe5d56SJohn McCall // Figure out what the offset is. 54787fe5d56SJohn McCall QualType T = E->getType(); 54887fe5d56SJohn McCall std::pair<CharUnits,CharUnits> TypeInfo = getContext().getTypeInfoInChars(T); 54987fe5d56SJohn McCall CharUnits Offset = getBlockOffset(TypeInfo.first, TypeInfo.second); 55087fe5d56SJohn McCall 55187fe5d56SJohn McCall BlockCXXThisOffset = Offset; 55287fe5d56SJohn McCall BlockLayout.push_back(E); 55387fe5d56SJohn McCall } 55487fe5d56SJohn McCall 55587fe5d56SJohn McCall void CodeGenFunction::AllocateBlockDecl(const BlockDeclRefExpr *E) { 5560168f4b2SAnders Carlsson const ValueDecl *VD = E->getDecl(); 55787fe5d56SJohn McCall CharUnits &Offset = BlockDecls[VD]; 5580168f4b2SAnders Carlsson 55997d01d50SMike Stump // See if we have already allocated an offset for this variable. 56087fe5d56SJohn McCall if (!Offset.isZero()) 56187fe5d56SJohn McCall return; 5627fe9cc1dSMike Stump 5634446dcf0SMike Stump // Don't run the expensive check, unless we have to. 564e1b19ba0SMike Stump if (!BlockHasCopyDispose) 565e1b19ba0SMike Stump if (E->isByRef() 566e1b19ba0SMike Stump || BlockRequiresCopying(E->getType())) 5674446dcf0SMike Stump BlockHasCopyDispose = true; 5687fe9cc1dSMike Stump 56987fe5d56SJohn McCall const ValueDecl *D = cast<ValueDecl>(E->getDecl()); 5707fe9cc1dSMike Stump 57187fe5d56SJohn McCall CharUnits Size; 57287fe5d56SJohn McCall CharUnits Align; 57387fe5d56SJohn McCall 57487fe5d56SJohn McCall if (E->isByRef()) { 57587fe5d56SJohn McCall llvm::tie(Size,Align) = 57687fe5d56SJohn McCall getContext().getTypeInfoInChars(getContext().VoidPtrTy); 57787fe5d56SJohn McCall } else { 57887fe5d56SJohn McCall Size = getContext().getTypeSizeInChars(D->getType()); 57987fe5d56SJohn McCall Align = getContext().getDeclAlign(D); 58087fe5d56SJohn McCall } 58187fe5d56SJohn McCall 58287fe5d56SJohn McCall Offset = getBlockOffset(Size, Align); 58387fe5d56SJohn McCall BlockLayout.push_back(E); 58497d01d50SMike Stump } 58597d01d50SMike Stump 5869d42f0f1SJohn McCall llvm::Value *CodeGenFunction::GetAddrOfBlockDecl(const ValueDecl *VD, 5879d42f0f1SJohn McCall bool IsByRef) { 58864176c2cSFariborz Jahanian 58987fe5d56SJohn McCall CharUnits offset = BlockDecls[VD]; 59087fe5d56SJohn McCall assert(!offset.isZero() && "getting address of unallocated decl"); 5917fe9cc1dSMike Stump 59297d01d50SMike Stump llvm::Value *BlockLiteral = LoadBlockStruct(); 59397d01d50SMike Stump llvm::Value *V = Builder.CreateGEP(BlockLiteral, 5945e016ae9SChris Lattner llvm::ConstantInt::get(Int64Ty, offset.getQuantity()), 5952d33d630SMike Stump "block.literal"); 5969d42f0f1SJohn McCall if (IsByRef) { 59797d01d50SMike Stump const llvm::Type *PtrStructTy 5980168f4b2SAnders Carlsson = llvm::PointerType::get(BuildByRefType(VD), 0); 599efd7caa8SMike Stump // The block literal will need a copy/destroy helper. 600efd7caa8SMike Stump BlockHasCopyDispose = true; 6010168f4b2SAnders Carlsson 6020168f4b2SAnders Carlsson const llvm::Type *Ty = PtrStructTy; 6039793f0e4SOwen Anderson Ty = llvm::PointerType::get(Ty, 0); 60497d01d50SMike Stump V = Builder.CreateBitCast(V, Ty); 605c76493a6SDaniel Dunbar V = Builder.CreateLoad(V); 60697d01d50SMike Stump V = Builder.CreateStructGEP(V, 1, "forwarding"); 607c76493a6SDaniel Dunbar V = Builder.CreateLoad(V); 60897d01d50SMike Stump V = Builder.CreateBitCast(V, PtrStructTy); 6090168f4b2SAnders Carlsson V = Builder.CreateStructGEP(V, getByRefValueLLVMField(VD), 6100168f4b2SAnders Carlsson VD->getNameAsString()); 611eeb23379SFariborz Jahanian if (VD->getType()->isReferenceType()) 612eeb23379SFariborz Jahanian V = Builder.CreateLoad(V); 61397d01d50SMike Stump } else { 6140168f4b2SAnders Carlsson const llvm::Type *Ty = CGM.getTypes().ConvertType(VD->getType()); 6159793f0e4SOwen Anderson Ty = llvm::PointerType::get(Ty, 0); 61697d01d50SMike Stump V = Builder.CreateBitCast(V, Ty); 617d1b378e0SFariborz Jahanian if (VD->getType()->isReferenceType()) 61864176c2cSFariborz Jahanian V = Builder.CreateLoad(V, "ref.tmp"); 61997d01d50SMike Stump } 62097d01d50SMike Stump return V; 62197d01d50SMike Stump } 62297d01d50SMike Stump 6232d5a2878SMike Stump llvm::Constant * 6246c39666aSMike Stump BlockModule::GetAddrOfGlobalBlock(const BlockExpr *BE, const char * n) { 6256a60fa24SAnders Carlsson // Generate the block descriptor. 6266a60fa24SAnders Carlsson const llvm::Type *UnsignedLongTy = Types.ConvertType(Context.UnsignedLongTy); 62792bbd6d4SMike Stump const llvm::IntegerType *IntTy = cast<llvm::IntegerType>( 62892bbd6d4SMike Stump getTypes().ConvertType(getContext().IntTy)); 6296a60fa24SAnders Carlsson 630fc83aa04SBlaine Garst llvm::Constant *DescriptorFields[4]; 6316a60fa24SAnders Carlsson 6326a60fa24SAnders Carlsson // Reserved 6330b75f23bSOwen Anderson DescriptorFields[0] = llvm::Constant::getNullValue(UnsignedLongTy); 6346a60fa24SAnders Carlsson 6356a60fa24SAnders Carlsson // Block literal size. For global blocks we just use the size of the generic 6366a60fa24SAnders Carlsson // block literal struct. 63798ca7943SKen Dyck CharUnits BlockLiteralSize = 63898ca7943SKen Dyck CGM.GetTargetTypeStoreSize(getGenericBlockLiteralType()); 639170229f6SOwen Anderson DescriptorFields[1] = 64040775003SKen Dyck llvm::ConstantInt::get(UnsignedLongTy,BlockLiteralSize.getQuantity()); 6416a60fa24SAnders Carlsson 642fc83aa04SBlaine Garst // signature. non-optional ObjC-style method descriptor @encode sequence 643fc83aa04SBlaine Garst std::string BlockTypeEncoding; 644fc83aa04SBlaine Garst CGM.getContext().getObjCEncodingForBlock(BE, BlockTypeEncoding); 645fc83aa04SBlaine Garst 646fc83aa04SBlaine Garst DescriptorFields[2] = llvm::ConstantExpr::getBitCast( 647fc83aa04SBlaine Garst CGM.GetAddrOfConstantCString(BlockTypeEncoding), PtrToInt8Ty); 648fc83aa04SBlaine Garst 649fc83aa04SBlaine Garst // layout 650fc83aa04SBlaine Garst DescriptorFields[3] = 651fc83aa04SBlaine Garst llvm::ConstantInt::get(UnsignedLongTy,0); 652fc83aa04SBlaine Garst 653fc83aa04SBlaine Garst // build the structure from the 4 elements 6546a60fa24SAnders Carlsson llvm::Constant *DescriptorStruct = 655fc83aa04SBlaine Garst llvm::ConstantStruct::get(VMContext, &DescriptorFields[0], 4, false); 6566a60fa24SAnders Carlsson 6576a60fa24SAnders Carlsson llvm::GlobalVariable *Descriptor = 658c10c8d3eSOwen Anderson new llvm::GlobalVariable(getModule(), DescriptorStruct->getType(), true, 6596a60fa24SAnders Carlsson llvm::GlobalVariable::InternalLinkage, 660c10c8d3eSOwen Anderson DescriptorStruct, "__block_descriptor_global"); 6616a60fa24SAnders Carlsson 662950a9518SDavid Chisnall int FieldCount = 5; 6636a60fa24SAnders Carlsson // Generate the constants for the block literal. 664950a9518SDavid Chisnall 665950a9518SDavid Chisnall std::vector<llvm::Constant*> LiteralFields(FieldCount); 6666a60fa24SAnders Carlsson 6679d42f0f1SJohn McCall CGBlockInfo Info(n); 668afa3c0a8SFariborz Jahanian llvm::Constant *BlockVarLayout; 6695469f29aSMike Stump llvm::DenseMap<const Decl*, llvm::Value*> LocalDeclMap; 670cb2fbcb0SMike Stump llvm::Function *Fn 671d5010898SFariborz Jahanian = CodeGenFunction(CGM).GenerateBlockFunction(GlobalDecl(), BE, 672afa3c0a8SFariborz Jahanian Info, 0, BlockVarLayout, 673afa3c0a8SFariborz Jahanian LocalDeclMap); 6749d42f0f1SJohn McCall assert(Info.BlockSize == BlockLiteralSize 675cb2fbcb0SMike Stump && "no imports allowed for global block"); 6766a60fa24SAnders Carlsson 6776a60fa24SAnders Carlsson // isa 678900546d2SDaniel Dunbar LiteralFields[0] = CGM.getNSConcreteGlobalBlock(); 6796a60fa24SAnders Carlsson 680d5010898SFariborz Jahanian // __flags 681d5010898SFariborz Jahanian unsigned flags = computeBlockFlag(CGM, BE, 682d5010898SFariborz Jahanian (BLOCK_IS_GLOBAL | BLOCK_HAS_SIGNATURE)); 683fc83aa04SBlaine Garst LiteralFields[1] = 684d5010898SFariborz Jahanian llvm::ConstantInt::get(IntTy, flags); 6856a60fa24SAnders Carlsson 6866a60fa24SAnders Carlsson // Reserved 6870b75f23bSOwen Anderson LiteralFields[2] = llvm::Constant::getNullValue(IntTy); 6886a60fa24SAnders Carlsson 6896a60fa24SAnders Carlsson // Function 6906a60fa24SAnders Carlsson LiteralFields[3] = Fn; 6916a60fa24SAnders Carlsson 6926a60fa24SAnders Carlsson // Descriptor 6936a60fa24SAnders Carlsson LiteralFields[4] = Descriptor; 6946a60fa24SAnders Carlsson 6956a60fa24SAnders Carlsson llvm::Constant *BlockLiteralStruct = 696950a9518SDavid Chisnall llvm::ConstantStruct::get(VMContext, LiteralFields, false); 6976a60fa24SAnders Carlsson 6986a60fa24SAnders Carlsson llvm::GlobalVariable *BlockLiteral = 699c10c8d3eSOwen Anderson new llvm::GlobalVariable(getModule(), BlockLiteralStruct->getType(), true, 7006a60fa24SAnders Carlsson llvm::GlobalVariable::InternalLinkage, 701c10c8d3eSOwen Anderson BlockLiteralStruct, "__block_literal_global"); 7026a60fa24SAnders Carlsson 7036a60fa24SAnders Carlsson return BlockLiteral; 7046a60fa24SAnders Carlsson } 7056a60fa24SAnders Carlsson 706cb2fbcb0SMike Stump llvm::Value *CodeGenFunction::LoadBlockStruct() { 70738382028SMike Stump llvm::Value *V = Builder.CreateLoad(LocalDeclMap[getBlockStructDecl()], 70838382028SMike Stump "self"); 70938382028SMike Stump // For now, we codegen based upon byte offsets. 71038382028SMike Stump return Builder.CreateBitCast(V, PtrToInt8Ty); 711cb2fbcb0SMike Stump } 712cb2fbcb0SMike Stump 7134446dcf0SMike Stump llvm::Function * 7149b5528d2SFariborz Jahanian CodeGenFunction::GenerateBlockFunction(GlobalDecl GD, const BlockExpr *BExpr, 7159d42f0f1SJohn McCall CGBlockInfo &Info, 716692c6e37SMike Stump const Decl *OuterFuncDecl, 717afa3c0a8SFariborz Jahanian llvm::Constant *& BlockVarLayout, 7189d42f0f1SJohn McCall llvm::DenseMap<const Decl*, llvm::Value*> ldm) { 7199074ed8cSDevang Patel 7209074ed8cSDevang Patel // Check if we should generate debug info for this block. 7219074ed8cSDevang Patel if (CGM.getDebugInfo()) 7229074ed8cSDevang Patel DebugInfo = CGM.getDebugInfo(); 7239074ed8cSDevang Patel 7245469f29aSMike Stump // Arrange for local static and local extern declarations to appear 7255469f29aSMike Stump // to be local to this function as well, as they are directly referenced 7265469f29aSMike Stump // in a block. 7275469f29aSMike Stump for (llvm::DenseMap<const Decl *, llvm::Value*>::iterator i = ldm.begin(); 7285469f29aSMike Stump i != ldm.end(); 7295469f29aSMike Stump ++i) { 7305469f29aSMike Stump const VarDecl *VD = dyn_cast<VarDecl>(i->first); 7315469f29aSMike Stump 7328e7d656aSJohn McCall if (VD->getStorageClass() == SC_Static || VD->hasExternalStorage()) 7335469f29aSMike Stump LocalDeclMap[VD] = i->second; 7345469f29aSMike Stump } 7355469f29aSMike Stump 73698ca7943SKen Dyck BlockOffset = 73798ca7943SKen Dyck CGM.GetTargetTypeStoreSize(CGM.getGenericBlockLiteralType()); 7385262b119SKen Dyck BlockAlign = getContext().getTypeAlignInChars(getContext().VoidPtrTy); 73909a9b6e3SEli Friedman 740aae4349dSFariborz Jahanian const FunctionType *BlockFunctionType = BExpr->getFunctionType(); 741aae4349dSFariborz Jahanian QualType ResultType; 742c50c27ccSRafael Espindola FunctionType::ExtInfo EInfo = getFunctionExtInfo(*BlockFunctionType); 743aae4349dSFariborz Jahanian bool IsVariadic; 744a4f6b6b1SFariborz Jahanian if (const FunctionProtoType *FTy = 745a4f6b6b1SFariborz Jahanian dyn_cast<FunctionProtoType>(BlockFunctionType)) { 746aae4349dSFariborz Jahanian ResultType = FTy->getResultType(); 747aae4349dSFariborz Jahanian IsVariadic = FTy->isVariadic(); 748658fe02dSMike Stump } else { 749aae4349dSFariborz Jahanian // K&R style block. 750aae4349dSFariborz Jahanian ResultType = BlockFunctionType->getResultType(); 751aae4349dSFariborz Jahanian IsVariadic = false; 752aae4349dSFariborz Jahanian } 7536a60fa24SAnders Carlsson 7546a60fa24SAnders Carlsson FunctionArgList Args; 7556a60fa24SAnders Carlsson 7567fe9cc1dSMike Stump CurFuncDecl = OuterFuncDecl; 7577fe9cc1dSMike Stump 7583385fe12SChris Lattner const BlockDecl *BD = BExpr->getBlockDecl(); 7596a60fa24SAnders Carlsson 7607fe9cc1dSMike Stump IdentifierInfo *II = &CGM.getContext().Idents.get(".block_descriptor"); 761d015328fSMike Stump 76287fe5d56SJohn McCall // Build the block struct now. 7639d42f0f1SJohn McCall AllocateAllBlockDeclRefs(*this, Info); 7647fe9cc1dSMike Stump 765afa3c0a8SFariborz Jahanian // Capture block layout info. here. 766afa3c0a8SFariborz Jahanian if (CGM.getContext().getLangOptions().ObjC1) 767933c6723SFariborz Jahanian BlockVarLayout = CGM.getObjCRuntime().GCBlockLayout(*this, BlockLayout); 768afa3c0a8SFariborz Jahanian else 769afa3c0a8SFariborz Jahanian BlockVarLayout = llvm::Constant::getNullValue(PtrToInt8Ty); 770afa3c0a8SFariborz Jahanian 771e1b19ba0SMike Stump QualType ParmTy = getContext().getBlockParmType(BlockHasCopyDispose, 77287fe5d56SJohn McCall BlockLayout); 77387fe5d56SJohn McCall 7746a60fa24SAnders Carlsson // FIXME: This leaks 7756a60fa24SAnders Carlsson ImplicitParamDecl *SelfDecl = 7764d4dcc8eSJohn McCall ImplicitParamDecl::Create(getContext(), const_cast<BlockDecl*>(BD), 777d015328fSMike Stump SourceLocation(), II, 778d015328fSMike Stump ParmTy); 7796a60fa24SAnders Carlsson 7806a60fa24SAnders Carlsson Args.push_back(std::make_pair(SelfDecl, SelfDecl->getType())); 781cb2fbcb0SMike Stump BlockStructDecl = SelfDecl; 7826a60fa24SAnders Carlsson 783c4b30e59SSteve Naroff for (BlockDecl::param_const_iterator i = BD->param_begin(), 7846a60fa24SAnders Carlsson e = BD->param_end(); i != e; ++i) 785eecd39f2SMike Stump Args.push_back(std::make_pair(*i, (*i)->getType())); 7866a60fa24SAnders Carlsson 7876a60fa24SAnders Carlsson const CGFunctionInfo &FI = 788c50c27ccSRafael Espindola CGM.getTypes().getFunctionInfo(ResultType, Args, EInfo); 7896a60fa24SAnders Carlsson 7906a60fa24SAnders Carlsson CodeGenTypes &Types = CGM.getTypes(); 791aae4349dSFariborz Jahanian const llvm::FunctionType *LTy = Types.GetFunctionType(FI, IsVariadic); 7926a60fa24SAnders Carlsson 7937e10f369SDouglas Gregor MangleBuffer Name; 7949b5528d2SFariborz Jahanian CGM.getMangledName(GD, Name, BD); 7956a60fa24SAnders Carlsson llvm::Function *Fn = 79602df9da2SDouglas Gregor llvm::Function::Create(LTy, llvm::GlobalValue::InternalLinkage, 79702df9da2SDouglas Gregor Name.getString(), &CGM.getModule()); 7986a60fa24SAnders Carlsson 799c3e7cff6SDaniel Dunbar CGM.SetInternalFunctionAttributes(BD, Fn, FI); 800f7f020bbSFariborz Jahanian StartFunction(BD, ResultType, Fn, Args, 801f7f020bbSFariborz Jahanian BExpr->getBody()->getLocEnd()); 802f7f020bbSFariborz Jahanian 803f7f020bbSFariborz Jahanian CurFuncDecl = OuterFuncDecl; 804c3e7cff6SDaniel Dunbar 80536ad0e99SFariborz Jahanian QualType FnType(BlockFunctionType, 0); 806c42461e1SFariborz Jahanian bool HasPrototype = isa<FunctionProtoType>(BlockFunctionType); 80736ad0e99SFariborz Jahanian 80836ad0e99SFariborz Jahanian IdentifierInfo *ID = &getContext().Idents.get(Name.getString()); 80936ad0e99SFariborz Jahanian CurCodeDecl = FunctionDecl::Create(getContext(), 81036ad0e99SFariborz Jahanian getContext().getTranslationUnitDecl(), 81136ad0e99SFariborz Jahanian SourceLocation(), ID, FnType, 81236ad0e99SFariborz Jahanian 0, 8138e7d656aSJohn McCall SC_Static, 8148e7d656aSJohn McCall SC_None, 81536ad0e99SFariborz Jahanian false, HasPrototype); 816f7f020bbSFariborz Jahanian if (FunctionProtoType *FT = dyn_cast<FunctionProtoType>(FnType)) { 817f7f020bbSFariborz Jahanian const FunctionDecl *CFD = dyn_cast<FunctionDecl>(CurCodeDecl); 818f7f020bbSFariborz Jahanian FunctionDecl *FD = const_cast<FunctionDecl *>(CFD); 819f7f020bbSFariborz Jahanian llvm::SmallVector<ParmVarDecl*, 16> Params; 820f7f020bbSFariborz Jahanian for (unsigned i = 0, e = FT->getNumArgs(); i != e; ++i) 821f7f020bbSFariborz Jahanian Params.push_back(ParmVarDecl::Create(getContext(), FD, 822f7f020bbSFariborz Jahanian SourceLocation(), 0, 823f7f020bbSFariborz Jahanian FT->getArgType(i), /*TInfo=*/0, 8248e7d656aSJohn McCall SC_None, SC_None, 0)); 825f7f020bbSFariborz Jahanian FD->setParams(Params.data(), Params.size()); 826f7f020bbSFariborz Jahanian } 82736ad0e99SFariborz Jahanian 828017460abSMike Stump 82987fe5d56SJohn McCall // If we have a C++ 'this' reference, go ahead and force it into 83087fe5d56SJohn McCall // existence now. 83187fe5d56SJohn McCall if (Info.CXXThisRef) { 83287fe5d56SJohn McCall assert(!BlockCXXThisOffset.isZero() && 83387fe5d56SJohn McCall "haven't yet allocated 'this' reference"); 83487fe5d56SJohn McCall 83587fe5d56SJohn McCall // TODO: I have a dream that one day this will be typed. 83687fe5d56SJohn McCall llvm::Value *BlockLiteral = LoadBlockStruct(); 83787fe5d56SJohn McCall llvm::Value *ThisPtrRaw = 83887fe5d56SJohn McCall Builder.CreateConstInBoundsGEP1_64(BlockLiteral, 83987fe5d56SJohn McCall BlockCXXThisOffset.getQuantity(), 84087fe5d56SJohn McCall "this.ptr.raw"); 84187fe5d56SJohn McCall 84287fe5d56SJohn McCall const llvm::Type *Ty = 84387fe5d56SJohn McCall CGM.getTypes().ConvertType(Info.CXXThisRef->getType()); 84487fe5d56SJohn McCall Ty = llvm::PointerType::get(Ty, 0); 84587fe5d56SJohn McCall llvm::Value *ThisPtr = Builder.CreateBitCast(ThisPtrRaw, Ty, "this.ptr"); 84687fe5d56SJohn McCall 84787fe5d56SJohn McCall CXXThisValue = Builder.CreateLoad(ThisPtr, "this"); 84887fe5d56SJohn McCall } 84987fe5d56SJohn McCall 8509d42f0f1SJohn McCall // If we have an Objective C 'self' reference, go ahead and force it 8519d42f0f1SJohn McCall // into existence now. 8529d42f0f1SJohn McCall if (Info.NeedsObjCSelf) { 8539d42f0f1SJohn McCall ValueDecl *Self = cast<ObjCMethodDecl>(CurFuncDecl)->getSelfDecl(); 8549d42f0f1SJohn McCall LocalDeclMap[Self] = GetAddrOfBlockDecl(Self, false); 8559d42f0f1SJohn McCall } 8569d42f0f1SJohn McCall 857017460abSMike Stump // Save a spot to insert the debug information for all the BlockDeclRefDecls. 858017460abSMike Stump llvm::BasicBlock *entry = Builder.GetInsertBlock(); 859017460abSMike Stump llvm::BasicBlock::iterator entry_ptr = Builder.GetInsertPoint(); 860017460abSMike Stump --entry_ptr; 861017460abSMike Stump 8623385fe12SChris Lattner EmitStmt(BExpr->getBody()); 863017460abSMike Stump 8647d69911aSMike Stump // Remember where we were... 8657d69911aSMike Stump llvm::BasicBlock *resume = Builder.GetInsertBlock(); 866017460abSMike Stump 8677d69911aSMike Stump // Go back to the entry. 868017460abSMike Stump ++entry_ptr; 869017460abSMike Stump Builder.SetInsertPoint(entry, entry_ptr); 870017460abSMike Stump 8712e722b91SMike Stump if (CGDebugInfo *DI = getDebugInfo()) { 8722e722b91SMike Stump // Emit debug information for all the BlockDeclRefDecls. 87387fe5d56SJohn McCall // FIXME: also for 'this' 87487fe5d56SJohn McCall for (unsigned i = 0, e = BlockLayout.size(); i != e; ++i) { 875bf78478dSChris Lattner if (const BlockDeclRefExpr *BDRE = 87687fe5d56SJohn McCall dyn_cast<BlockDeclRefExpr>(BlockLayout[i])) { 8772e722b91SMike Stump const ValueDecl *D = BDRE->getDecl(); 8782e722b91SMike Stump DI->setLocation(D->getLocation()); 8792e722b91SMike Stump DI->EmitDeclareOfBlockDeclRefVariable(BDRE, 8802e722b91SMike Stump LocalDeclMap[getBlockStructDecl()], 8812e722b91SMike Stump Builder, this); 8822e722b91SMike Stump } 8832e722b91SMike Stump } 8842e722b91SMike Stump } 8857d69911aSMike Stump // And resume where we left off. 8867d69911aSMike Stump if (resume == 0) 8877d69911aSMike Stump Builder.ClearInsertionPoint(); 8887d69911aSMike Stump else 8897d69911aSMike Stump Builder.SetInsertPoint(resume); 8902e722b91SMike Stump 8913385fe12SChris Lattner FinishFunction(cast<CompoundStmt>(BExpr->getBody())->getRBracLoc()); 8926a60fa24SAnders Carlsson 893b750d928SMike Stump // The runtime needs a minimum alignment of a void *. 8945262b119SKen Dyck CharUnits MinAlign = getContext().getTypeAlignInChars(getContext().VoidPtrTy); 89540775003SKen Dyck BlockOffset = CharUnits::fromQuantity( 8965262b119SKen Dyck llvm::RoundUpToAlignment(BlockOffset.getQuantity(), 8975262b119SKen Dyck MinAlign.getQuantity())); 898b750d928SMike Stump 8999d42f0f1SJohn McCall Info.BlockSize = BlockOffset; 9009d42f0f1SJohn McCall Info.BlockAlign = BlockAlign; 9019d42f0f1SJohn McCall Info.BlockLayout = BlockLayout; 9029d42f0f1SJohn McCall Info.BlockHasCopyDispose = BlockHasCopyDispose; 9036a60fa24SAnders Carlsson return Fn; 9046a60fa24SAnders Carlsson } 9051db7d04bSMike Stump 90687fe5d56SJohn McCall CharUnits BlockFunction::getBlockOffset(CharUnits Size, CharUnits Align) { 9075262b119SKen Dyck assert((Align.isPositive()) && "alignment must be 1 byte or more"); 9081db7d04bSMike Stump 90940775003SKen Dyck CharUnits OldOffset = BlockOffset; 9101db7d04bSMike Stump 9111db7d04bSMike Stump // Ensure proper alignment, even if it means we have to have a gap 91240775003SKen Dyck BlockOffset = CharUnits::fromQuantity( 9135262b119SKen Dyck llvm::RoundUpToAlignment(BlockOffset.getQuantity(), Align.getQuantity())); 9141db7d04bSMike Stump BlockAlign = std::max(Align, BlockAlign); 9151db7d04bSMike Stump 91640775003SKen Dyck CharUnits Pad = BlockOffset - OldOffset; 91740775003SKen Dyck if (Pad.isPositive()) { 9181db7d04bSMike Stump QualType PadTy = getContext().getConstantArrayType(getContext().CharTy, 91940775003SKen Dyck llvm::APInt(32, 92040775003SKen Dyck Pad.getQuantity()), 9211db7d04bSMike Stump ArrayType::Normal, 0); 9226a047c97SDouglas Gregor ValueDecl *PadDecl = VarDecl::Create(getContext(), 9236a047c97SDouglas Gregor getContext().getTranslationUnitDecl(), 9246a047c97SDouglas Gregor SourceLocation(), 925c4df4076SDouglas Gregor 0, QualType(PadTy), 0, 9268e7d656aSJohn McCall SC_None, SC_None); 92787fe5d56SJohn McCall Expr *E = new (getContext()) DeclRefExpr(PadDecl, PadDecl->getType(), 928ed6c7440SDouglas Gregor SourceLocation()); 92987fe5d56SJohn McCall BlockLayout.push_back(E); 9301db7d04bSMike Stump } 9311db7d04bSMike Stump 9321db7d04bSMike Stump BlockOffset += Size; 9331db7d04bSMike Stump return BlockOffset - Size; 9341db7d04bSMike Stump } 93597d01d50SMike Stump 936aeb0ffd5SMike Stump llvm::Constant *BlockFunction:: 937aeb0ffd5SMike Stump GenerateCopyHelperFunction(bool BlockHasCopyDispose, const llvm::StructType *T, 93867645930SMike Stump std::vector<HelperInfo> *NoteForHelperp) { 9390c743277SMike Stump QualType R = getContext().VoidTy; 9400c743277SMike Stump 9410c743277SMike Stump FunctionArgList Args; 9420c743277SMike Stump // FIXME: This leaks 943aeb0ffd5SMike Stump ImplicitParamDecl *Dst = 9447fe9cc1dSMike Stump ImplicitParamDecl::Create(getContext(), 0, 9457fe9cc1dSMike Stump SourceLocation(), 0, 946aeb0ffd5SMike Stump getContext().getPointerType(getContext().VoidTy)); 947aeb0ffd5SMike Stump Args.push_back(std::make_pair(Dst, Dst->getType())); 9480c743277SMike Stump ImplicitParamDecl *Src = 9497fe9cc1dSMike Stump ImplicitParamDecl::Create(getContext(), 0, 9507fe9cc1dSMike Stump SourceLocation(), 0, 9510c743277SMike Stump getContext().getPointerType(getContext().VoidTy)); 9520c743277SMike Stump Args.push_back(std::make_pair(Src, Src->getType())); 9530c743277SMike Stump 9540c743277SMike Stump const CGFunctionInfo &FI = 955c50c27ccSRafael Espindola CGM.getTypes().getFunctionInfo(R, Args, FunctionType::ExtInfo()); 9560c743277SMike Stump 957cbc2bcafSMike Stump // FIXME: We'd like to put these into a mergable by content, with 958cbc2bcafSMike Stump // internal linkage. 9590c743277SMike Stump CodeGenTypes &Types = CGM.getTypes(); 9600c743277SMike Stump const llvm::FunctionType *LTy = Types.GetFunctionType(FI, false); 9610c743277SMike Stump 9620c743277SMike Stump llvm::Function *Fn = 9630c743277SMike Stump llvm::Function::Create(LTy, llvm::GlobalValue::InternalLinkage, 964d6b28fc6SBenjamin Kramer "__copy_helper_block_", &CGM.getModule()); 9650c743277SMike Stump 9660c743277SMike Stump IdentifierInfo *II 9670c743277SMike Stump = &CGM.getContext().Idents.get("__copy_helper_block_"); 9680c743277SMike Stump 9690c743277SMike Stump FunctionDecl *FD = FunctionDecl::Create(getContext(), 9700c743277SMike Stump getContext().getTranslationUnitDecl(), 97160ed5604SArgyrios Kyrtzidis SourceLocation(), II, R, 0, 9728e7d656aSJohn McCall SC_Static, 9738e7d656aSJohn McCall SC_None, 974c4df4076SDouglas Gregor false, 9750c743277SMike Stump true); 9760c743277SMike Stump CGF.StartFunction(FD, R, Fn, Args, SourceLocation()); 977aeb0ffd5SMike Stump 978aeb0ffd5SMike Stump llvm::Value *SrcObj = CGF.GetAddrOfLocalVar(Src); 979aeb0ffd5SMike Stump llvm::Type *PtrPtrT; 98067645930SMike Stump 98167645930SMike Stump if (NoteForHelperp) { 98267645930SMike Stump std::vector<HelperInfo> &NoteForHelper = *NoteForHelperp; 98367645930SMike Stump 9849793f0e4SOwen Anderson PtrPtrT = llvm::PointerType::get(llvm::PointerType::get(T, 0), 0); 985aeb0ffd5SMike Stump SrcObj = Builder.CreateBitCast(SrcObj, PtrPtrT); 986aeb0ffd5SMike Stump SrcObj = Builder.CreateLoad(SrcObj); 987aeb0ffd5SMike Stump 988aeb0ffd5SMike Stump llvm::Value *DstObj = CGF.GetAddrOfLocalVar(Dst); 9893b65ac26SMike Stump llvm::Type *PtrPtrT; 9909793f0e4SOwen Anderson PtrPtrT = llvm::PointerType::get(llvm::PointerType::get(T, 0), 0); 9913b65ac26SMike Stump DstObj = Builder.CreateBitCast(DstObj, PtrPtrT); 9923b65ac26SMike Stump DstObj = Builder.CreateLoad(DstObj); 993aeb0ffd5SMike Stump 994aeb0ffd5SMike Stump for (unsigned i=0; i < NoteForHelper.size(); ++i) { 995aeb0ffd5SMike Stump int flag = NoteForHelper[i].flag; 996aeb0ffd5SMike Stump int index = NoteForHelper[i].index; 997aeb0ffd5SMike Stump 998aeb0ffd5SMike Stump if ((NoteForHelper[i].flag & BLOCK_FIELD_IS_BYREF) 999aeb0ffd5SMike Stump || NoteForHelper[i].RequiresCopying) { 1000aeb0ffd5SMike Stump llvm::Value *Srcv = SrcObj; 1001aeb0ffd5SMike Stump Srcv = Builder.CreateStructGEP(Srcv, index); 1002aeb0ffd5SMike Stump Srcv = Builder.CreateBitCast(Srcv, 10039793f0e4SOwen Anderson llvm::PointerType::get(PtrToInt8Ty, 0)); 1004aeb0ffd5SMike Stump Srcv = Builder.CreateLoad(Srcv); 1005aeb0ffd5SMike Stump 1006aeb0ffd5SMike Stump llvm::Value *Dstv = Builder.CreateStructGEP(DstObj, index); 1007aeb0ffd5SMike Stump Dstv = Builder.CreateBitCast(Dstv, PtrToInt8Ty); 1008aeb0ffd5SMike Stump 10095e016ae9SChris Lattner llvm::Value *N = llvm::ConstantInt::get(CGF.Int32Ty, flag); 1010900546d2SDaniel Dunbar llvm::Value *F = CGM.getBlockObjectAssign(); 1011aeb0ffd5SMike Stump Builder.CreateCall3(F, Dstv, Srcv, N); 1012aeb0ffd5SMike Stump } 1013aeb0ffd5SMike Stump } 101467645930SMike Stump } 1015aeb0ffd5SMike Stump 10160c743277SMike Stump CGF.FinishFunction(); 10170c743277SMike Stump 1018ade90fd1SOwen Anderson return llvm::ConstantExpr::getBitCast(Fn, PtrToInt8Ty); 101997d01d50SMike Stump } 102097d01d50SMike Stump 1021d6ef62f0SMike Stump llvm::Constant *BlockFunction:: 1022aeb0ffd5SMike Stump GenerateDestroyHelperFunction(bool BlockHasCopyDispose, 1023aeb0ffd5SMike Stump const llvm::StructType* T, 102467645930SMike Stump std::vector<HelperInfo> *NoteForHelperp) { 10250c743277SMike Stump QualType R = getContext().VoidTy; 10260c743277SMike Stump 10270c743277SMike Stump FunctionArgList Args; 10280c743277SMike Stump // FIXME: This leaks 10290c743277SMike Stump ImplicitParamDecl *Src = 10307fe9cc1dSMike Stump ImplicitParamDecl::Create(getContext(), 0, 10317fe9cc1dSMike Stump SourceLocation(), 0, 10320c743277SMike Stump getContext().getPointerType(getContext().VoidTy)); 10330c743277SMike Stump 10340c743277SMike Stump Args.push_back(std::make_pair(Src, Src->getType())); 10350c743277SMike Stump 10360c743277SMike Stump const CGFunctionInfo &FI = 1037c50c27ccSRafael Espindola CGM.getTypes().getFunctionInfo(R, Args, FunctionType::ExtInfo()); 10380c743277SMike Stump 1039cbc2bcafSMike Stump // FIXME: We'd like to put these into a mergable by content, with 1040cbc2bcafSMike Stump // internal linkage. 10410c743277SMike Stump CodeGenTypes &Types = CGM.getTypes(); 10420c743277SMike Stump const llvm::FunctionType *LTy = Types.GetFunctionType(FI, false); 10430c743277SMike Stump 10440c743277SMike Stump llvm::Function *Fn = 10450c743277SMike Stump llvm::Function::Create(LTy, llvm::GlobalValue::InternalLinkage, 1046d6b28fc6SBenjamin Kramer "__destroy_helper_block_", &CGM.getModule()); 10470c743277SMike Stump 10480c743277SMike Stump IdentifierInfo *II 10490c743277SMike Stump = &CGM.getContext().Idents.get("__destroy_helper_block_"); 10500c743277SMike Stump 10510c743277SMike Stump FunctionDecl *FD = FunctionDecl::Create(getContext(), 10520c743277SMike Stump getContext().getTranslationUnitDecl(), 105360ed5604SArgyrios Kyrtzidis SourceLocation(), II, R, 0, 10548e7d656aSJohn McCall SC_Static, 10558e7d656aSJohn McCall SC_None, 1056c4df4076SDouglas Gregor false, true); 10570c743277SMike Stump CGF.StartFunction(FD, R, Fn, Args, SourceLocation()); 10586f7d9f87SMike Stump 105967645930SMike Stump if (NoteForHelperp) { 106067645930SMike Stump std::vector<HelperInfo> &NoteForHelper = *NoteForHelperp; 106167645930SMike Stump 10626f7d9f87SMike Stump llvm::Value *SrcObj = CGF.GetAddrOfLocalVar(Src); 10636f7d9f87SMike Stump llvm::Type *PtrPtrT; 10649793f0e4SOwen Anderson PtrPtrT = llvm::PointerType::get(llvm::PointerType::get(T, 0), 0); 10656f7d9f87SMike Stump SrcObj = Builder.CreateBitCast(SrcObj, PtrPtrT); 10666f7d9f87SMike Stump SrcObj = Builder.CreateLoad(SrcObj); 10676f7d9f87SMike Stump 10686f7d9f87SMike Stump for (unsigned i=0; i < NoteForHelper.size(); ++i) { 10696f7d9f87SMike Stump int flag = NoteForHelper[i].flag; 10706f7d9f87SMike Stump int index = NoteForHelper[i].index; 10716f7d9f87SMike Stump 10726f7d9f87SMike Stump if ((NoteForHelper[i].flag & BLOCK_FIELD_IS_BYREF) 10736f7d9f87SMike Stump || NoteForHelper[i].RequiresCopying) { 10746f7d9f87SMike Stump llvm::Value *Srcv = SrcObj; 10756f7d9f87SMike Stump Srcv = Builder.CreateStructGEP(Srcv, index); 10766f7d9f87SMike Stump Srcv = Builder.CreateBitCast(Srcv, 10779793f0e4SOwen Anderson llvm::PointerType::get(PtrToInt8Ty, 0)); 10786f7d9f87SMike Stump Srcv = Builder.CreateLoad(Srcv); 10796f7d9f87SMike Stump 10806f7d9f87SMike Stump BuildBlockRelease(Srcv, flag); 10816f7d9f87SMike Stump } 10826f7d9f87SMike Stump } 108367645930SMike Stump } 10846f7d9f87SMike Stump 10850c743277SMike Stump CGF.FinishFunction(); 10860c743277SMike Stump 1087ade90fd1SOwen Anderson return llvm::ConstantExpr::getBitCast(Fn, PtrToInt8Ty); 10880c743277SMike Stump } 10890c743277SMike Stump 1090aeb0ffd5SMike Stump llvm::Constant *BlockFunction::BuildCopyHelper(const llvm::StructType *T, 109167645930SMike Stump std::vector<HelperInfo> *NoteForHelper) { 1092aeb0ffd5SMike Stump return CodeGenFunction(CGM).GenerateCopyHelperFunction(BlockHasCopyDispose, 1093aeb0ffd5SMike Stump T, NoteForHelper); 10940c743277SMike Stump } 10950c743277SMike Stump 1096aeb0ffd5SMike Stump llvm::Constant *BlockFunction::BuildDestroyHelper(const llvm::StructType *T, 109767645930SMike Stump std::vector<HelperInfo> *NoteForHelperp) { 1098aeb0ffd5SMike Stump return CodeGenFunction(CGM).GenerateDestroyHelperFunction(BlockHasCopyDispose, 109967645930SMike Stump T, NoteForHelperp); 110097d01d50SMike Stump } 1101626aecc4SMike Stump 1102f89230d9SMike Stump llvm::Constant *BlockFunction:: 1103f89230d9SMike Stump GeneratebyrefCopyHelperFunction(const llvm::Type *T, int flag) { 1104ee2a5ee5SMike Stump QualType R = getContext().VoidTy; 1105ee2a5ee5SMike Stump 1106ee2a5ee5SMike Stump FunctionArgList Args; 1107ee2a5ee5SMike Stump // FIXME: This leaks 1108f89230d9SMike Stump ImplicitParamDecl *Dst = 11097fe9cc1dSMike Stump ImplicitParamDecl::Create(getContext(), 0, 11107fe9cc1dSMike Stump SourceLocation(), 0, 1111f89230d9SMike Stump getContext().getPointerType(getContext().VoidTy)); 1112f89230d9SMike Stump Args.push_back(std::make_pair(Dst, Dst->getType())); 1113f89230d9SMike Stump 1114f89230d9SMike Stump // FIXME: This leaks 1115ee2a5ee5SMike Stump ImplicitParamDecl *Src = 11167fe9cc1dSMike Stump ImplicitParamDecl::Create(getContext(), 0, 11177fe9cc1dSMike Stump SourceLocation(), 0, 1118ee2a5ee5SMike Stump getContext().getPointerType(getContext().VoidTy)); 1119ee2a5ee5SMike Stump Args.push_back(std::make_pair(Src, Src->getType())); 1120ee2a5ee5SMike Stump 1121ee2a5ee5SMike Stump const CGFunctionInfo &FI = 1122c50c27ccSRafael Espindola CGM.getTypes().getFunctionInfo(R, Args, FunctionType::ExtInfo()); 1123ee2a5ee5SMike Stump 1124ee2a5ee5SMike Stump CodeGenTypes &Types = CGM.getTypes(); 1125ee2a5ee5SMike Stump const llvm::FunctionType *LTy = Types.GetFunctionType(FI, false); 1126ee2a5ee5SMike Stump 1127cbc2bcafSMike Stump // FIXME: We'd like to put these into a mergable by content, with 1128cbc2bcafSMike Stump // internal linkage. 1129ee2a5ee5SMike Stump llvm::Function *Fn = 1130ee2a5ee5SMike Stump llvm::Function::Create(LTy, llvm::GlobalValue::InternalLinkage, 1131d6b28fc6SBenjamin Kramer "__Block_byref_id_object_copy_", &CGM.getModule()); 1132ee2a5ee5SMike Stump 1133ee2a5ee5SMike Stump IdentifierInfo *II 1134ee2a5ee5SMike Stump = &CGM.getContext().Idents.get("__Block_byref_id_object_copy_"); 1135ee2a5ee5SMike Stump 1136ee2a5ee5SMike Stump FunctionDecl *FD = FunctionDecl::Create(getContext(), 1137ee2a5ee5SMike Stump getContext().getTranslationUnitDecl(), 113860ed5604SArgyrios Kyrtzidis SourceLocation(), II, R, 0, 11398e7d656aSJohn McCall SC_Static, 11408e7d656aSJohn McCall SC_None, 1141c4df4076SDouglas Gregor false, true); 1142ee2a5ee5SMike Stump CGF.StartFunction(FD, R, Fn, Args, SourceLocation()); 1143f89230d9SMike Stump 1144f89230d9SMike Stump // dst->x 1145f89230d9SMike Stump llvm::Value *V = CGF.GetAddrOfLocalVar(Dst); 11469793f0e4SOwen Anderson V = Builder.CreateBitCast(V, llvm::PointerType::get(T, 0)); 11473b65ac26SMike Stump V = Builder.CreateLoad(V); 1148f89230d9SMike Stump V = Builder.CreateStructGEP(V, 6, "x"); 1149f89230d9SMike Stump llvm::Value *DstObj = Builder.CreateBitCast(V, PtrToInt8Ty); 1150f89230d9SMike Stump 1151f89230d9SMike Stump // src->x 1152f89230d9SMike Stump V = CGF.GetAddrOfLocalVar(Src); 1153f89230d9SMike Stump V = Builder.CreateLoad(V); 1154f89230d9SMike Stump V = Builder.CreateBitCast(V, T); 1155f89230d9SMike Stump V = Builder.CreateStructGEP(V, 6, "x"); 11569793f0e4SOwen Anderson V = Builder.CreateBitCast(V, llvm::PointerType::get(PtrToInt8Ty, 0)); 1157f89230d9SMike Stump llvm::Value *SrcObj = Builder.CreateLoad(V); 1158f89230d9SMike Stump 1159f89230d9SMike Stump flag |= BLOCK_BYREF_CALLER; 1160f89230d9SMike Stump 11615e016ae9SChris Lattner llvm::Value *N = llvm::ConstantInt::get(CGF.Int32Ty, flag); 1162900546d2SDaniel Dunbar llvm::Value *F = CGM.getBlockObjectAssign(); 1163f89230d9SMike Stump Builder.CreateCall3(F, DstObj, SrcObj, N); 1164f89230d9SMike Stump 1165ee2a5ee5SMike Stump CGF.FinishFunction(); 1166ee2a5ee5SMike Stump 1167ade90fd1SOwen Anderson return llvm::ConstantExpr::getBitCast(Fn, PtrToInt8Ty); 1168ee2a5ee5SMike Stump } 1169ee2a5ee5SMike Stump 1170fbe25dd1SMike Stump llvm::Constant * 1171fbe25dd1SMike Stump BlockFunction::GeneratebyrefDestroyHelperFunction(const llvm::Type *T, 1172fbe25dd1SMike Stump int flag) { 1173ee2a5ee5SMike Stump QualType R = getContext().VoidTy; 1174ee2a5ee5SMike Stump 1175ee2a5ee5SMike Stump FunctionArgList Args; 1176ee2a5ee5SMike Stump // FIXME: This leaks 1177ee2a5ee5SMike Stump ImplicitParamDecl *Src = 11787fe9cc1dSMike Stump ImplicitParamDecl::Create(getContext(), 0, 11797fe9cc1dSMike Stump SourceLocation(), 0, 1180ee2a5ee5SMike Stump getContext().getPointerType(getContext().VoidTy)); 1181ee2a5ee5SMike Stump 1182ee2a5ee5SMike Stump Args.push_back(std::make_pair(Src, Src->getType())); 1183ee2a5ee5SMike Stump 1184ee2a5ee5SMike Stump const CGFunctionInfo &FI = 1185c50c27ccSRafael Espindola CGM.getTypes().getFunctionInfo(R, Args, FunctionType::ExtInfo()); 1186ee2a5ee5SMike Stump 1187ee2a5ee5SMike Stump CodeGenTypes &Types = CGM.getTypes(); 1188ee2a5ee5SMike Stump const llvm::FunctionType *LTy = Types.GetFunctionType(FI, false); 1189ee2a5ee5SMike Stump 1190cbc2bcafSMike Stump // FIXME: We'd like to put these into a mergable by content, with 1191cbc2bcafSMike Stump // internal linkage. 1192ee2a5ee5SMike Stump llvm::Function *Fn = 1193ee2a5ee5SMike Stump llvm::Function::Create(LTy, llvm::GlobalValue::InternalLinkage, 1194d6b28fc6SBenjamin Kramer "__Block_byref_id_object_dispose_", 1195ee2a5ee5SMike Stump &CGM.getModule()); 1196ee2a5ee5SMike Stump 1197ee2a5ee5SMike Stump IdentifierInfo *II 1198ee2a5ee5SMike Stump = &CGM.getContext().Idents.get("__Block_byref_id_object_dispose_"); 1199ee2a5ee5SMike Stump 1200ee2a5ee5SMike Stump FunctionDecl *FD = FunctionDecl::Create(getContext(), 1201ee2a5ee5SMike Stump getContext().getTranslationUnitDecl(), 120260ed5604SArgyrios Kyrtzidis SourceLocation(), II, R, 0, 12038e7d656aSJohn McCall SC_Static, 12048e7d656aSJohn McCall SC_None, 1205c4df4076SDouglas Gregor false, true); 1206ee2a5ee5SMike Stump CGF.StartFunction(FD, R, Fn, Args, SourceLocation()); 1207fbe25dd1SMike Stump 1208fbe25dd1SMike Stump llvm::Value *V = CGF.GetAddrOfLocalVar(Src); 12099793f0e4SOwen Anderson V = Builder.CreateBitCast(V, llvm::PointerType::get(T, 0)); 12103b65ac26SMike Stump V = Builder.CreateLoad(V); 1211fbe25dd1SMike Stump V = Builder.CreateStructGEP(V, 6, "x"); 12129793f0e4SOwen Anderson V = Builder.CreateBitCast(V, llvm::PointerType::get(PtrToInt8Ty, 0)); 12133b65ac26SMike Stump V = Builder.CreateLoad(V); 1214fbe25dd1SMike Stump 1215fbe25dd1SMike Stump flag |= BLOCK_BYREF_CALLER; 1216fbe25dd1SMike Stump BuildBlockRelease(V, flag); 1217ee2a5ee5SMike Stump CGF.FinishFunction(); 1218ee2a5ee5SMike Stump 1219ade90fd1SOwen Anderson return llvm::ConstantExpr::getBitCast(Fn, PtrToInt8Ty); 1220ee2a5ee5SMike Stump } 1221ee2a5ee5SMike Stump 1222f89230d9SMike Stump llvm::Constant *BlockFunction::BuildbyrefCopyHelper(const llvm::Type *T, 1223e08ea787SChris Lattner int Flag, unsigned Align) { 1224e08ea787SChris Lattner // All alignments below that of pointer alignment collapse down to just 1225cbc2bcafSMike Stump // pointer alignment, as we always have at least that much alignment to begin 1226cbc2bcafSMike Stump // with. 1227cbc2bcafSMike Stump Align /= unsigned(CGF.Target.getPointerAlign(0)/8); 1228e08ea787SChris Lattner 1229cbc2bcafSMike Stump // As an optimization, we only generate a single function of each kind we 1230cbc2bcafSMike Stump // might need. We need a different one for each alignment and for each 1231cbc2bcafSMike Stump // setting of flags. We mix Align and flag to get the kind. 1232e08ea787SChris Lattner uint64_t Kind = (uint64_t)Align*BLOCK_BYREF_CURRENT_MAX + Flag; 1233e08ea787SChris Lattner llvm::Constant *&Entry = CGM.AssignCache[Kind]; 1234cbc2bcafSMike Stump if (Entry) 1235cbc2bcafSMike Stump return Entry; 1236e08ea787SChris Lattner return Entry = CodeGenFunction(CGM).GeneratebyrefCopyHelperFunction(T, Flag); 1237ee2a5ee5SMike Stump } 1238ee2a5ee5SMike Stump 1239fbe25dd1SMike Stump llvm::Constant *BlockFunction::BuildbyrefDestroyHelper(const llvm::Type *T, 1240e08ea787SChris Lattner int Flag, 1241cbc2bcafSMike Stump unsigned Align) { 1242cbc2bcafSMike Stump // All alignments below that of pointer alignment collpase down to just 1243cbc2bcafSMike Stump // pointer alignment, as we always have at least that much alignment to begin 1244cbc2bcafSMike Stump // with. 1245cbc2bcafSMike Stump Align /= unsigned(CGF.Target.getPointerAlign(0)/8); 1246e08ea787SChris Lattner 1247cbc2bcafSMike Stump // As an optimization, we only generate a single function of each kind we 1248cbc2bcafSMike Stump // might need. We need a different one for each alignment and for each 1249cbc2bcafSMike Stump // setting of flags. We mix Align and flag to get the kind. 1250e08ea787SChris Lattner uint64_t Kind = (uint64_t)Align*BLOCK_BYREF_CURRENT_MAX + Flag; 1251e08ea787SChris Lattner llvm::Constant *&Entry = CGM.DestroyCache[Kind]; 1252cbc2bcafSMike Stump if (Entry) 1253cbc2bcafSMike Stump return Entry; 1254e08ea787SChris Lattner return Entry=CodeGenFunction(CGM).GeneratebyrefDestroyHelperFunction(T, Flag); 1255ee2a5ee5SMike Stump } 1256ee2a5ee5SMike Stump 1257fbe25dd1SMike Stump void BlockFunction::BuildBlockRelease(llvm::Value *V, int flag) { 1258900546d2SDaniel Dunbar llvm::Value *F = CGM.getBlockObjectDispose(); 1259fbe25dd1SMike Stump llvm::Value *N; 1260626aecc4SMike Stump V = Builder.CreateBitCast(V, PtrToInt8Ty); 12615e016ae9SChris Lattner N = llvm::ConstantInt::get(CGF.Int32Ty, flag); 1262626aecc4SMike Stump Builder.CreateCall2(F, V, N); 1263626aecc4SMike Stump } 12644446dcf0SMike Stump 12654446dcf0SMike Stump ASTContext &BlockFunction::getContext() const { return CGM.getContext(); } 1266aeb0ffd5SMike Stump 1267aeb0ffd5SMike Stump BlockFunction::BlockFunction(CodeGenModule &cgm, CodeGenFunction &cgf, 1268aeb0ffd5SMike Stump CGBuilderTy &B) 12695e016ae9SChris Lattner : CGM(cgm), VMContext(cgm.getLLVMContext()), CGF(cgf), Builder(B) { 127041a75027SOwen Anderson PtrToInt8Ty = llvm::PointerType::getUnqual( 127141a75027SOwen Anderson llvm::Type::getInt8Ty(VMContext)); 1272aeb0ffd5SMike Stump 1273aeb0ffd5SMike Stump BlockHasCopyDispose = false; 1274aeb0ffd5SMike Stump } 1275