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 } 125681c0754SFariborz 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: 128681c0754SFariborz Jahanian if (PE->isSuperReceiver()) 1299d42f0f1SJohn McCall Info.NeedsObjCSelf = true; 130681c0754SFariborz Jahanian } 131681c0754SFariborz Jahanian else if (const ObjCImplicitSetterGetterRefExpr *IE = 132681c0754SFariborz Jahanian dyn_cast<ObjCImplicitSetterGetterRefExpr>(S)) { 133681c0754SFariborz Jahanian // Getter/setter uses may also cause implicit super references, 134681c0754SFariborz Jahanian // which we can check for with: 135681c0754SFariborz Jahanian if (IE->isSuperReceiver()) 136681c0754SFariborz Jahanian Info.NeedsObjCSelf = true; 137681c0754SFariborz 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); 314*2a5deb56SFariborz Jahanian Note.RequiresCopying = BlockRequiresCopying(BDRE); 315*2a5deb56SFariborz Jahanian 31687fe5d56SJohn McCall const ValueDecl *VD = BDRE->getDecl(); 31787fe5d56SJohn McCall QualType T = VD->getType(); 31887fe5d56SJohn McCall if (BDRE->isByRef()) { 31987fe5d56SJohn McCall Note.flag = BLOCK_FIELD_IS_BYREF; 32087fe5d56SJohn McCall if (T.isObjCGCWeak()) 32187fe5d56SJohn McCall Note.flag |= BLOCK_FIELD_IS_WEAK; 32287fe5d56SJohn McCall } else if (T->isBlockPointerType()) { 32387fe5d56SJohn McCall Note.flag = BLOCK_FIELD_IS_BLOCK; 32487fe5d56SJohn McCall } else { 32587fe5d56SJohn McCall Note.flag = BLOCK_FIELD_IS_OBJECT; 32687fe5d56SJohn McCall } 327aeb0ffd5SMike Stump 3281db7d04bSMike Stump if (LocalDeclMap[VD]) { 32997d01d50SMike Stump if (BDRE->isByRef()) { 33097d01d50SMike Stump llvm::Value *Loc = LocalDeclMap[VD]; 33197d01d50SMike Stump Loc = Builder.CreateStructGEP(Loc, 1, "forwarding"); 332c76493a6SDaniel Dunbar Loc = Builder.CreateLoad(Loc); 33397d01d50SMike Stump Builder.CreateStore(Loc, Addr); 33497d01d50SMike Stump continue; 33587fe5d56SJohn McCall } else { 3363b836180SFariborz Jahanian if (BDRE->getCopyConstructorExpr()) { 337ea882cd9SFariborz Jahanian E = BDRE->getCopyConstructorExpr(); 338bd30929eSJohn McCall PushDestructorCleanup(E->getType(), Addr); 3393b836180SFariborz Jahanian } 340ea882cd9SFariborz Jahanian else { 34187fe5d56SJohn McCall E = new (getContext()) DeclRefExpr(const_cast<ValueDecl*>(VD), 342d1b378e0SFariborz Jahanian VD->getType().getNonReferenceType(), 343d1b378e0SFariborz Jahanian SourceLocation()); 34464176c2cSFariborz Jahanian if (VD->getType()->isReferenceType()) { 345d1b378e0SFariborz Jahanian E = new (getContext()) 346e302792bSJohn McCall UnaryOperator(const_cast<Expr*>(E), UO_AddrOf, 347d1b378e0SFariborz Jahanian getContext().getPointerType(E->getType()), 348ed6c7440SDouglas Gregor SourceLocation()); 3491db7d04bSMike Stump } 35064176c2cSFariborz Jahanian } 35164176c2cSFariborz Jahanian } 35287fe5d56SJohn McCall } 35387fe5d56SJohn McCall 35497d01d50SMike Stump if (BDRE->isByRef()) { 3551db7d04bSMike Stump E = new (getContext()) 356e302792bSJohn McCall UnaryOperator(const_cast<Expr*>(E), UO_AddrOf, 3571db7d04bSMike Stump getContext().getPointerType(E->getType()), 3581db7d04bSMike Stump SourceLocation()); 35997d01d50SMike Stump } 3601db7d04bSMike Stump 3617a626f63SJohn McCall RValue r = EmitAnyExpr(E, AggValueSlot::forAddr(Addr, false, true)); 36297d01d50SMike Stump if (r.isScalar()) { 36397d01d50SMike Stump llvm::Value *Loc = r.getScalarVal(); 364950a9518SDavid Chisnall const llvm::Type *Ty = Types[i+BlockFields]; 36597d01d50SMike Stump if (BDRE->isByRef()) { 3662d33d630SMike Stump // E is now the address of the value field, instead, we want the 3672d33d630SMike Stump // address of the actual ByRef struct. We optimize this slightly 3682d33d630SMike Stump // compared to gcc by not grabbing the forwarding slot as this must 3692d33d630SMike Stump // be done during Block_copy for us, and we can postpone the work 3702d33d630SMike Stump // until then. 37140775003SKen Dyck CharUnits offset = BlockDecls[BDRE->getDecl()]; 3722d33d630SMike Stump 3732d33d630SMike Stump llvm::Value *BlockLiteral = LoadBlockStruct(); 3742d33d630SMike Stump 3752d33d630SMike Stump Loc = Builder.CreateGEP(BlockLiteral, 3765e016ae9SChris Lattner llvm::ConstantInt::get(Int64Ty, offset.getQuantity()), 3772d33d630SMike Stump "block.literal"); 3789793f0e4SOwen Anderson Ty = llvm::PointerType::get(Ty, 0); 37997d01d50SMike Stump Loc = Builder.CreateBitCast(Loc, Ty); 380c76493a6SDaniel Dunbar Loc = Builder.CreateLoad(Loc); 3812d33d630SMike Stump // Loc = Builder.CreateBitCast(Loc, Ty); 38297d01d50SMike Stump } 38397d01d50SMike Stump Builder.CreateStore(Loc, Addr); 38497d01d50SMike Stump } else if (r.isComplex()) 385b750d928SMike Stump // FIXME: implement 386b750d928SMike Stump ErrorUnsupported(BE, "complex in block literal"); 387b750d928SMike Stump else if (r.isAggregate()) 388b750d928SMike Stump ; // Already created into the destination 389b750d928SMike Stump else 390b750d928SMike Stump assert (0 && "bad block variable"); 391b750d928SMike Stump // FIXME: Ensure that the offset created by the backend for 392b750d928SMike Stump // the struct matches the previously computed offset in BlockDecls. 393b750d928SMike Stump } 39487fe5d56SJohn McCall NoteForHelper.resize(NumHelpers); 395d6ef62f0SMike Stump 396d6ef62f0SMike Stump // __descriptor 397c05349e5SFariborz Jahanian llvm::Value *Descriptor = BuildDescriptorBlockDecl(BE, Info, Ty, 398afa3c0a8SFariborz Jahanian BlockVarLayout, 399aeb0ffd5SMike Stump &NoteForHelper); 400aeb0ffd5SMike Stump Descriptor = Builder.CreateBitCast(Descriptor, PtrToInt8Ty); 401aeb0ffd5SMike Stump Builder.CreateStore(Descriptor, Builder.CreateStructGEP(V, 4, "block.tmp")); 402b750d928SMike Stump } 403b750d928SMike Stump 404b750d928SMike Stump QualType BPT = BE->getType(); 405989908fbSFariborz Jahanian V = Builder.CreateBitCast(V, ConvertType(BPT)); 406989908fbSFariborz Jahanian // See if this is a __weak block variable and the must call objc_read_weak 407989908fbSFariborz Jahanian // on it. 408989908fbSFariborz Jahanian const FunctionType *ftype = BPT->getPointeeType()->getAs<FunctionType>(); 409989908fbSFariborz Jahanian QualType RES = ftype->getResultType(); 410989908fbSFariborz Jahanian if (RES.isObjCGCWeak()) { 411989908fbSFariborz Jahanian // Must cast argument to id* 412989908fbSFariborz Jahanian const llvm::Type *ObjectPtrTy = 413989908fbSFariborz Jahanian ConvertType(CGM.getContext().getObjCIdType()); 414989908fbSFariborz Jahanian const llvm::Type *PtrObjectPtrTy = 415989908fbSFariborz Jahanian llvm::PointerType::getUnqual(ObjectPtrTy); 416989908fbSFariborz Jahanian V = Builder.CreateBitCast(V, PtrObjectPtrTy); 417989908fbSFariborz Jahanian V = CGM.getObjCRuntime().EmitObjCWeakRead(*this, V); 418989908fbSFariborz Jahanian } 419989908fbSFariborz Jahanian return V; 420b750d928SMike Stump } 42185284bacSMike Stump 42285284bacSMike Stump 42395435674SMike Stump const llvm::Type *BlockModule::getBlockDescriptorType() { 424650c932dSMike Stump if (BlockDescriptorType) 425650c932dSMike Stump return BlockDescriptorType; 4262437cbfaSAnders Carlsson 4272437cbfaSAnders Carlsson const llvm::Type *UnsignedLongTy = 428650c932dSMike Stump getTypes().ConvertType(getContext().UnsignedLongTy); 4292437cbfaSAnders Carlsson 4302437cbfaSAnders Carlsson // struct __block_descriptor { 4312437cbfaSAnders Carlsson // unsigned long reserved; 4322437cbfaSAnders Carlsson // unsigned long block_size; 433fc83aa04SBlaine Garst // 434fc83aa04SBlaine Garst // // later, the following will be added 435fc83aa04SBlaine Garst // 436fc83aa04SBlaine Garst // struct { 437fc83aa04SBlaine Garst // void (*copyHelper)(); 438fc83aa04SBlaine Garst // void (*copyHelper)(); 439fc83aa04SBlaine Garst // } helpers; // !!! optional 440fc83aa04SBlaine Garst // 441fc83aa04SBlaine Garst // const char *signature; // the block signature 442fc83aa04SBlaine Garst // const char *layout; // reserved 4432437cbfaSAnders Carlsson // }; 444758428f4SOwen Anderson BlockDescriptorType = llvm::StructType::get(UnsignedLongTy->getContext(), 445758428f4SOwen Anderson UnsignedLongTy, 4462437cbfaSAnders Carlsson UnsignedLongTy, 4472437cbfaSAnders Carlsson NULL); 4482437cbfaSAnders Carlsson 449650c932dSMike Stump getModule().addTypeName("struct.__block_descriptor", 450650c932dSMike Stump BlockDescriptorType); 4512437cbfaSAnders Carlsson 452650c932dSMike Stump return BlockDescriptorType; 4532437cbfaSAnders Carlsson } 4542437cbfaSAnders Carlsson 45595435674SMike Stump const llvm::Type *BlockModule::getGenericBlockLiteralType() { 456005c9a62SMike Stump if (GenericBlockLiteralType) 457005c9a62SMike Stump return GenericBlockLiteralType; 4582437cbfaSAnders Carlsson 4592437cbfaSAnders Carlsson const llvm::Type *BlockDescPtrTy = 4609793f0e4SOwen Anderson llvm::PointerType::getUnqual(getBlockDescriptorType()); 4612437cbfaSAnders Carlsson 46292bbd6d4SMike Stump const llvm::IntegerType *IntTy = cast<llvm::IntegerType>( 46392bbd6d4SMike Stump getTypes().ConvertType(getContext().IntTy)); 46492bbd6d4SMike Stump 4652437cbfaSAnders Carlsson // struct __block_literal_generic { 4665d2534adSMike Stump // void *__isa; 4675d2534adSMike Stump // int __flags; 4685d2534adSMike Stump // int __reserved; 4695d2534adSMike Stump // void (*__invoke)(void *); 4705d2534adSMike Stump // struct __block_descriptor *__descriptor; 4712437cbfaSAnders Carlsson // }; 472758428f4SOwen Anderson GenericBlockLiteralType = llvm::StructType::get(IntTy->getContext(), 473758428f4SOwen Anderson PtrToInt8Ty, 47492bbd6d4SMike Stump IntTy, 47592bbd6d4SMike Stump IntTy, 476626aecc4SMike Stump PtrToInt8Ty, 4772437cbfaSAnders Carlsson BlockDescPtrTy, 4782437cbfaSAnders Carlsson NULL); 4792437cbfaSAnders Carlsson 480005c9a62SMike Stump getModule().addTypeName("struct.__block_literal_generic", 481005c9a62SMike Stump GenericBlockLiteralType); 4822437cbfaSAnders Carlsson 483005c9a62SMike Stump return GenericBlockLiteralType; 4842437cbfaSAnders Carlsson } 4852437cbfaSAnders Carlsson 4865d2534adSMike Stump 487bfb3671bSAnders Carlsson RValue CodeGenFunction::EmitBlockCallExpr(const CallExpr* E, 488bfb3671bSAnders Carlsson ReturnValueSlot ReturnValue) { 4892437cbfaSAnders Carlsson const BlockPointerType *BPT = 490c23c7e6aSTed Kremenek E->getCallee()->getType()->getAs<BlockPointerType>(); 4912437cbfaSAnders Carlsson 4922437cbfaSAnders Carlsson llvm::Value *Callee = EmitScalarExpr(E->getCallee()); 4932437cbfaSAnders Carlsson 4942437cbfaSAnders Carlsson // Get a pointer to the generic block literal. 4952437cbfaSAnders Carlsson const llvm::Type *BlockLiteralTy = 4969793f0e4SOwen Anderson llvm::PointerType::getUnqual(CGM.getGenericBlockLiteralType()); 4972437cbfaSAnders Carlsson 4982437cbfaSAnders Carlsson // Bitcast the callee to a block literal. 4992437cbfaSAnders Carlsson llvm::Value *BlockLiteral = 5002437cbfaSAnders Carlsson Builder.CreateBitCast(Callee, BlockLiteralTy, "block.literal"); 5012437cbfaSAnders Carlsson 5022437cbfaSAnders Carlsson // Get the function pointer from the literal. 5032437cbfaSAnders Carlsson llvm::Value *FuncPtr = Builder.CreateStructGEP(BlockLiteral, 3, "tmp"); 5042437cbfaSAnders Carlsson 5052437cbfaSAnders Carlsson BlockLiteral = 5062437cbfaSAnders Carlsson Builder.CreateBitCast(BlockLiteral, 507abd5b90eSBenjamin Kramer llvm::Type::getInt8PtrTy(VMContext), 5082437cbfaSAnders Carlsson "tmp"); 5092437cbfaSAnders Carlsson 5102437cbfaSAnders Carlsson // Add the block literal. 5112437cbfaSAnders Carlsson QualType VoidPtrTy = getContext().getPointerType(getContext().VoidTy); 5122437cbfaSAnders Carlsson CallArgList Args; 5132437cbfaSAnders Carlsson Args.push_back(std::make_pair(RValue::get(BlockLiteral), VoidPtrTy)); 5142437cbfaSAnders Carlsson 515479e6fc7SAnders Carlsson QualType FnType = BPT->getPointeeType(); 516479e6fc7SAnders Carlsson 5172437cbfaSAnders Carlsson // And the rest of the arguments. 5189dd450bbSJohn McCall EmitCallArgs(Args, FnType->getAs<FunctionProtoType>(), 519479e6fc7SAnders Carlsson E->arg_begin(), E->arg_end()); 5202437cbfaSAnders Carlsson 5215f50c651SAnders Carlsson // Load the function. 522c76493a6SDaniel Dunbar llvm::Value *Func = Builder.CreateLoad(FuncPtr, "tmp"); 5235f50c651SAnders Carlsson 524ab26cfa5SJohn McCall const FunctionType *FuncTy = FnType->getAs<FunctionType>(); 525ab26cfa5SJohn McCall QualType ResultType = FuncTy->getResultType(); 526a60cbcdfSAnders Carlsson 527a60cbcdfSAnders Carlsson const CGFunctionInfo &FnInfo = 528c50c27ccSRafael Espindola CGM.getTypes().getFunctionInfo(ResultType, Args, 529c50c27ccSRafael Espindola FuncTy->getExtInfo()); 5305f50c651SAnders Carlsson 5315f50c651SAnders Carlsson // Cast the function pointer to the right type. 5325f50c651SAnders Carlsson const llvm::Type *BlockFTy = 533a60cbcdfSAnders Carlsson CGM.getTypes().GetFunctionType(FnInfo, false); 5345f50c651SAnders Carlsson 5359793f0e4SOwen Anderson const llvm::Type *BlockFTyPtr = llvm::PointerType::getUnqual(BlockFTy); 5365f50c651SAnders Carlsson Func = Builder.CreateBitCast(Func, BlockFTyPtr); 5375f50c651SAnders Carlsson 5382437cbfaSAnders Carlsson // And call the block. 539bfb3671bSAnders Carlsson return EmitCall(FnInfo, Func, ReturnValue, Args); 5402437cbfaSAnders Carlsson } 5416a60fa24SAnders Carlsson 54287fe5d56SJohn McCall void CodeGenFunction::AllocateBlockCXXThisPointer(const CXXThisExpr *E) { 54387fe5d56SJohn McCall assert(BlockCXXThisOffset.isZero() && "already computed 'this' pointer"); 54487fe5d56SJohn McCall 54587fe5d56SJohn McCall // Figure out what the offset is. 54687fe5d56SJohn McCall QualType T = E->getType(); 54787fe5d56SJohn McCall std::pair<CharUnits,CharUnits> TypeInfo = getContext().getTypeInfoInChars(T); 54887fe5d56SJohn McCall CharUnits Offset = getBlockOffset(TypeInfo.first, TypeInfo.second); 54987fe5d56SJohn McCall 55087fe5d56SJohn McCall BlockCXXThisOffset = Offset; 55187fe5d56SJohn McCall BlockLayout.push_back(E); 55287fe5d56SJohn McCall } 55387fe5d56SJohn McCall 55487fe5d56SJohn McCall void CodeGenFunction::AllocateBlockDecl(const BlockDeclRefExpr *E) { 5550168f4b2SAnders Carlsson const ValueDecl *VD = E->getDecl(); 55687fe5d56SJohn McCall CharUnits &Offset = BlockDecls[VD]; 5570168f4b2SAnders Carlsson 55897d01d50SMike Stump // See if we have already allocated an offset for this variable. 55987fe5d56SJohn McCall if (!Offset.isZero()) 56087fe5d56SJohn McCall return; 5617fe9cc1dSMike Stump 5624446dcf0SMike Stump // Don't run the expensive check, unless we have to. 563e1b19ba0SMike Stump if (!BlockHasCopyDispose) 564e1b19ba0SMike Stump if (E->isByRef() 565*2a5deb56SFariborz Jahanian || BlockRequiresCopying(E)) 5664446dcf0SMike Stump BlockHasCopyDispose = true; 5677fe9cc1dSMike Stump 56887fe5d56SJohn McCall const ValueDecl *D = cast<ValueDecl>(E->getDecl()); 5697fe9cc1dSMike Stump 57087fe5d56SJohn McCall CharUnits Size; 57187fe5d56SJohn McCall CharUnits Align; 57287fe5d56SJohn McCall 57387fe5d56SJohn McCall if (E->isByRef()) { 57487fe5d56SJohn McCall llvm::tie(Size,Align) = 57587fe5d56SJohn McCall getContext().getTypeInfoInChars(getContext().VoidPtrTy); 57687fe5d56SJohn McCall } else { 57787fe5d56SJohn McCall Size = getContext().getTypeSizeInChars(D->getType()); 57887fe5d56SJohn McCall Align = getContext().getDeclAlign(D); 57987fe5d56SJohn McCall } 58087fe5d56SJohn McCall 58187fe5d56SJohn McCall Offset = getBlockOffset(Size, Align); 58287fe5d56SJohn McCall BlockLayout.push_back(E); 58397d01d50SMike Stump } 58497d01d50SMike Stump 5859d42f0f1SJohn McCall llvm::Value *CodeGenFunction::GetAddrOfBlockDecl(const ValueDecl *VD, 5869d42f0f1SJohn McCall bool IsByRef) { 58764176c2cSFariborz Jahanian 58887fe5d56SJohn McCall CharUnits offset = BlockDecls[VD]; 58987fe5d56SJohn McCall assert(!offset.isZero() && "getting address of unallocated decl"); 5907fe9cc1dSMike Stump 59197d01d50SMike Stump llvm::Value *BlockLiteral = LoadBlockStruct(); 59297d01d50SMike Stump llvm::Value *V = Builder.CreateGEP(BlockLiteral, 5935e016ae9SChris Lattner llvm::ConstantInt::get(Int64Ty, offset.getQuantity()), 5942d33d630SMike Stump "block.literal"); 5959d42f0f1SJohn McCall if (IsByRef) { 59697d01d50SMike Stump const llvm::Type *PtrStructTy 5970168f4b2SAnders Carlsson = llvm::PointerType::get(BuildByRefType(VD), 0); 598efd7caa8SMike Stump // The block literal will need a copy/destroy helper. 599efd7caa8SMike Stump BlockHasCopyDispose = true; 6000168f4b2SAnders Carlsson 6010168f4b2SAnders Carlsson const llvm::Type *Ty = PtrStructTy; 6029793f0e4SOwen Anderson Ty = llvm::PointerType::get(Ty, 0); 60397d01d50SMike Stump V = Builder.CreateBitCast(V, Ty); 604c76493a6SDaniel Dunbar V = Builder.CreateLoad(V); 60597d01d50SMike Stump V = Builder.CreateStructGEP(V, 1, "forwarding"); 606c76493a6SDaniel Dunbar V = Builder.CreateLoad(V); 60797d01d50SMike Stump V = Builder.CreateBitCast(V, PtrStructTy); 6080168f4b2SAnders Carlsson V = Builder.CreateStructGEP(V, getByRefValueLLVMField(VD), 6090168f4b2SAnders Carlsson VD->getNameAsString()); 610eeb23379SFariborz Jahanian if (VD->getType()->isReferenceType()) 611eeb23379SFariborz Jahanian V = Builder.CreateLoad(V); 61297d01d50SMike Stump } else { 61356514161SJohn McCall const llvm::Type *Ty = CGM.getTypes().ConvertTypeForMem(VD->getType()); 6149793f0e4SOwen Anderson Ty = llvm::PointerType::get(Ty, 0); 61597d01d50SMike Stump V = Builder.CreateBitCast(V, Ty); 616d1b378e0SFariborz Jahanian if (VD->getType()->isReferenceType()) 61764176c2cSFariborz Jahanian V = Builder.CreateLoad(V, "ref.tmp"); 61897d01d50SMike Stump } 61997d01d50SMike Stump return V; 62097d01d50SMike Stump } 62197d01d50SMike Stump 6222d5a2878SMike Stump llvm::Constant * 6236c39666aSMike Stump BlockModule::GetAddrOfGlobalBlock(const BlockExpr *BE, const char * n) { 6246a60fa24SAnders Carlsson // Generate the block descriptor. 6256a60fa24SAnders Carlsson const llvm::Type *UnsignedLongTy = Types.ConvertType(Context.UnsignedLongTy); 62692bbd6d4SMike Stump const llvm::IntegerType *IntTy = cast<llvm::IntegerType>( 62792bbd6d4SMike Stump getTypes().ConvertType(getContext().IntTy)); 6286a60fa24SAnders Carlsson 629fc83aa04SBlaine Garst llvm::Constant *DescriptorFields[4]; 6306a60fa24SAnders Carlsson 6316a60fa24SAnders Carlsson // Reserved 6320b75f23bSOwen Anderson DescriptorFields[0] = llvm::Constant::getNullValue(UnsignedLongTy); 6336a60fa24SAnders Carlsson 6346a60fa24SAnders Carlsson // Block literal size. For global blocks we just use the size of the generic 6356a60fa24SAnders Carlsson // block literal struct. 63698ca7943SKen Dyck CharUnits BlockLiteralSize = 63798ca7943SKen Dyck CGM.GetTargetTypeStoreSize(getGenericBlockLiteralType()); 638170229f6SOwen Anderson DescriptorFields[1] = 63940775003SKen Dyck llvm::ConstantInt::get(UnsignedLongTy,BlockLiteralSize.getQuantity()); 6406a60fa24SAnders Carlsson 641fc83aa04SBlaine Garst // signature. non-optional ObjC-style method descriptor @encode sequence 642fc83aa04SBlaine Garst std::string BlockTypeEncoding; 643fc83aa04SBlaine Garst CGM.getContext().getObjCEncodingForBlock(BE, BlockTypeEncoding); 644fc83aa04SBlaine Garst 645fc83aa04SBlaine Garst DescriptorFields[2] = llvm::ConstantExpr::getBitCast( 646fc83aa04SBlaine Garst CGM.GetAddrOfConstantCString(BlockTypeEncoding), PtrToInt8Ty); 647fc83aa04SBlaine Garst 648fc83aa04SBlaine Garst // layout 649fc83aa04SBlaine Garst DescriptorFields[3] = 650fc83aa04SBlaine Garst llvm::ConstantInt::get(UnsignedLongTy,0); 651fc83aa04SBlaine Garst 652fc83aa04SBlaine Garst // build the structure from the 4 elements 6536a60fa24SAnders Carlsson llvm::Constant *DescriptorStruct = 654fc83aa04SBlaine Garst llvm::ConstantStruct::get(VMContext, &DescriptorFields[0], 4, false); 6556a60fa24SAnders Carlsson 6566a60fa24SAnders Carlsson llvm::GlobalVariable *Descriptor = 657c10c8d3eSOwen Anderson new llvm::GlobalVariable(getModule(), DescriptorStruct->getType(), true, 6586a60fa24SAnders Carlsson llvm::GlobalVariable::InternalLinkage, 659c10c8d3eSOwen Anderson DescriptorStruct, "__block_descriptor_global"); 6606a60fa24SAnders Carlsson 661950a9518SDavid Chisnall int FieldCount = 5; 6626a60fa24SAnders Carlsson // Generate the constants for the block literal. 663950a9518SDavid Chisnall 664950a9518SDavid Chisnall std::vector<llvm::Constant*> LiteralFields(FieldCount); 6656a60fa24SAnders Carlsson 6669d42f0f1SJohn McCall CGBlockInfo Info(n); 667afa3c0a8SFariborz Jahanian llvm::Constant *BlockVarLayout; 6685469f29aSMike Stump llvm::DenseMap<const Decl*, llvm::Value*> LocalDeclMap; 669cb2fbcb0SMike Stump llvm::Function *Fn 670d5010898SFariborz Jahanian = CodeGenFunction(CGM).GenerateBlockFunction(GlobalDecl(), BE, 671afa3c0a8SFariborz Jahanian Info, 0, BlockVarLayout, 672afa3c0a8SFariborz Jahanian LocalDeclMap); 6739d42f0f1SJohn McCall assert(Info.BlockSize == BlockLiteralSize 674cb2fbcb0SMike Stump && "no imports allowed for global block"); 6756a60fa24SAnders Carlsson 6766a60fa24SAnders Carlsson // isa 677900546d2SDaniel Dunbar LiteralFields[0] = CGM.getNSConcreteGlobalBlock(); 6786a60fa24SAnders Carlsson 679d5010898SFariborz Jahanian // __flags 680d5010898SFariborz Jahanian unsigned flags = computeBlockFlag(CGM, BE, 681d5010898SFariborz Jahanian (BLOCK_IS_GLOBAL | BLOCK_HAS_SIGNATURE)); 682fc83aa04SBlaine Garst LiteralFields[1] = 683d5010898SFariborz Jahanian llvm::ConstantInt::get(IntTy, flags); 6846a60fa24SAnders Carlsson 6856a60fa24SAnders Carlsson // Reserved 6860b75f23bSOwen Anderson LiteralFields[2] = llvm::Constant::getNullValue(IntTy); 6876a60fa24SAnders Carlsson 6886a60fa24SAnders Carlsson // Function 6896a60fa24SAnders Carlsson LiteralFields[3] = Fn; 6906a60fa24SAnders Carlsson 6916a60fa24SAnders Carlsson // Descriptor 6926a60fa24SAnders Carlsson LiteralFields[4] = Descriptor; 6936a60fa24SAnders Carlsson 6946a60fa24SAnders Carlsson llvm::Constant *BlockLiteralStruct = 695950a9518SDavid Chisnall llvm::ConstantStruct::get(VMContext, LiteralFields, false); 6966a60fa24SAnders Carlsson 6976a60fa24SAnders Carlsson llvm::GlobalVariable *BlockLiteral = 698c10c8d3eSOwen Anderson new llvm::GlobalVariable(getModule(), BlockLiteralStruct->getType(), true, 6996a60fa24SAnders Carlsson llvm::GlobalVariable::InternalLinkage, 700c10c8d3eSOwen Anderson BlockLiteralStruct, "__block_literal_global"); 7016a60fa24SAnders Carlsson 7026a60fa24SAnders Carlsson return BlockLiteral; 7036a60fa24SAnders Carlsson } 7046a60fa24SAnders Carlsson 705cb2fbcb0SMike Stump llvm::Value *CodeGenFunction::LoadBlockStruct() { 70638382028SMike Stump llvm::Value *V = Builder.CreateLoad(LocalDeclMap[getBlockStructDecl()], 70738382028SMike Stump "self"); 70838382028SMike Stump // For now, we codegen based upon byte offsets. 70938382028SMike Stump return Builder.CreateBitCast(V, PtrToInt8Ty); 710cb2fbcb0SMike Stump } 711cb2fbcb0SMike Stump 7124446dcf0SMike Stump llvm::Function * 7139b5528d2SFariborz Jahanian CodeGenFunction::GenerateBlockFunction(GlobalDecl GD, const BlockExpr *BExpr, 7149d42f0f1SJohn McCall CGBlockInfo &Info, 715692c6e37SMike Stump const Decl *OuterFuncDecl, 716afa3c0a8SFariborz Jahanian llvm::Constant *& BlockVarLayout, 7179d42f0f1SJohn McCall llvm::DenseMap<const Decl*, llvm::Value*> ldm) { 7189074ed8cSDevang Patel 7199074ed8cSDevang Patel // Check if we should generate debug info for this block. 7209074ed8cSDevang Patel if (CGM.getDebugInfo()) 7219074ed8cSDevang Patel DebugInfo = CGM.getDebugInfo(); 7229074ed8cSDevang Patel 7235469f29aSMike Stump // Arrange for local static and local extern declarations to appear 7245469f29aSMike Stump // to be local to this function as well, as they are directly referenced 7255469f29aSMike Stump // in a block. 7265469f29aSMike Stump for (llvm::DenseMap<const Decl *, llvm::Value*>::iterator i = ldm.begin(); 7275469f29aSMike Stump i != ldm.end(); 7285469f29aSMike Stump ++i) { 7295469f29aSMike Stump const VarDecl *VD = dyn_cast<VarDecl>(i->first); 7305469f29aSMike Stump 7318e7d656aSJohn McCall if (VD->getStorageClass() == SC_Static || VD->hasExternalStorage()) 7325469f29aSMike Stump LocalDeclMap[VD] = i->second; 7335469f29aSMike Stump } 7345469f29aSMike Stump 73598ca7943SKen Dyck BlockOffset = 73698ca7943SKen Dyck CGM.GetTargetTypeStoreSize(CGM.getGenericBlockLiteralType()); 7375262b119SKen Dyck BlockAlign = getContext().getTypeAlignInChars(getContext().VoidPtrTy); 73809a9b6e3SEli Friedman 739aae4349dSFariborz Jahanian const FunctionType *BlockFunctionType = BExpr->getFunctionType(); 740aae4349dSFariborz Jahanian QualType ResultType; 741c50c27ccSRafael Espindola FunctionType::ExtInfo EInfo = getFunctionExtInfo(*BlockFunctionType); 742aae4349dSFariborz Jahanian bool IsVariadic; 743a4f6b6b1SFariborz Jahanian if (const FunctionProtoType *FTy = 744a4f6b6b1SFariborz Jahanian dyn_cast<FunctionProtoType>(BlockFunctionType)) { 745aae4349dSFariborz Jahanian ResultType = FTy->getResultType(); 746aae4349dSFariborz Jahanian IsVariadic = FTy->isVariadic(); 747658fe02dSMike Stump } else { 748aae4349dSFariborz Jahanian // K&R style block. 749aae4349dSFariborz Jahanian ResultType = BlockFunctionType->getResultType(); 750aae4349dSFariborz Jahanian IsVariadic = false; 751aae4349dSFariborz Jahanian } 7526a60fa24SAnders Carlsson 7536a60fa24SAnders Carlsson FunctionArgList Args; 7546a60fa24SAnders Carlsson 7557fe9cc1dSMike Stump CurFuncDecl = OuterFuncDecl; 7567fe9cc1dSMike Stump 7573385fe12SChris Lattner const BlockDecl *BD = BExpr->getBlockDecl(); 7586a60fa24SAnders Carlsson 7597fe9cc1dSMike Stump IdentifierInfo *II = &CGM.getContext().Idents.get(".block_descriptor"); 760d015328fSMike Stump 76187fe5d56SJohn McCall // Build the block struct now. 7629d42f0f1SJohn McCall AllocateAllBlockDeclRefs(*this, Info); 7637fe9cc1dSMike Stump 764afa3c0a8SFariborz Jahanian // Capture block layout info. here. 765afa3c0a8SFariborz Jahanian if (CGM.getContext().getLangOptions().ObjC1) 766933c6723SFariborz Jahanian BlockVarLayout = CGM.getObjCRuntime().GCBlockLayout(*this, BlockLayout); 767afa3c0a8SFariborz Jahanian else 768afa3c0a8SFariborz Jahanian BlockVarLayout = llvm::Constant::getNullValue(PtrToInt8Ty); 769afa3c0a8SFariborz Jahanian 770e1b19ba0SMike Stump QualType ParmTy = getContext().getBlockParmType(BlockHasCopyDispose, 77187fe5d56SJohn McCall BlockLayout); 77287fe5d56SJohn McCall 7736a60fa24SAnders Carlsson // FIXME: This leaks 7746a60fa24SAnders Carlsson ImplicitParamDecl *SelfDecl = 7754d4dcc8eSJohn McCall ImplicitParamDecl::Create(getContext(), const_cast<BlockDecl*>(BD), 776d015328fSMike Stump SourceLocation(), II, 777d015328fSMike Stump ParmTy); 7786a60fa24SAnders Carlsson 7796a60fa24SAnders Carlsson Args.push_back(std::make_pair(SelfDecl, SelfDecl->getType())); 780cb2fbcb0SMike Stump BlockStructDecl = SelfDecl; 7816a60fa24SAnders Carlsson 782c4b30e59SSteve Naroff for (BlockDecl::param_const_iterator i = BD->param_begin(), 7836a60fa24SAnders Carlsson e = BD->param_end(); i != e; ++i) 784eecd39f2SMike Stump Args.push_back(std::make_pair(*i, (*i)->getType())); 7856a60fa24SAnders Carlsson 7866a60fa24SAnders Carlsson const CGFunctionInfo &FI = 787c50c27ccSRafael Espindola CGM.getTypes().getFunctionInfo(ResultType, Args, EInfo); 7886a60fa24SAnders Carlsson 7896a60fa24SAnders Carlsson CodeGenTypes &Types = CGM.getTypes(); 790aae4349dSFariborz Jahanian const llvm::FunctionType *LTy = Types.GetFunctionType(FI, IsVariadic); 7916a60fa24SAnders Carlsson 7927e10f369SDouglas Gregor MangleBuffer Name; 7939b5528d2SFariborz Jahanian CGM.getMangledName(GD, Name, BD); 7946a60fa24SAnders Carlsson llvm::Function *Fn = 79502df9da2SDouglas Gregor llvm::Function::Create(LTy, llvm::GlobalValue::InternalLinkage, 79602df9da2SDouglas Gregor Name.getString(), &CGM.getModule()); 7976a60fa24SAnders Carlsson 798c3e7cff6SDaniel Dunbar CGM.SetInternalFunctionAttributes(BD, Fn, FI); 799f7f020bbSFariborz Jahanian StartFunction(BD, ResultType, Fn, Args, 800f7f020bbSFariborz Jahanian BExpr->getBody()->getLocEnd()); 801f7f020bbSFariborz Jahanian 802f7f020bbSFariborz Jahanian CurFuncDecl = OuterFuncDecl; 803c3e7cff6SDaniel Dunbar 80436ad0e99SFariborz Jahanian QualType FnType(BlockFunctionType, 0); 805c42461e1SFariborz Jahanian bool HasPrototype = isa<FunctionProtoType>(BlockFunctionType); 80636ad0e99SFariborz Jahanian 80736ad0e99SFariborz Jahanian IdentifierInfo *ID = &getContext().Idents.get(Name.getString()); 80836ad0e99SFariborz Jahanian CurCodeDecl = FunctionDecl::Create(getContext(), 80936ad0e99SFariborz Jahanian getContext().getTranslationUnitDecl(), 81036ad0e99SFariborz Jahanian SourceLocation(), ID, FnType, 81136ad0e99SFariborz Jahanian 0, 8128e7d656aSJohn McCall SC_Static, 8138e7d656aSJohn McCall SC_None, 81436ad0e99SFariborz Jahanian false, HasPrototype); 815f7f020bbSFariborz Jahanian if (FunctionProtoType *FT = dyn_cast<FunctionProtoType>(FnType)) { 816f7f020bbSFariborz Jahanian const FunctionDecl *CFD = dyn_cast<FunctionDecl>(CurCodeDecl); 817f7f020bbSFariborz Jahanian FunctionDecl *FD = const_cast<FunctionDecl *>(CFD); 818f7f020bbSFariborz Jahanian llvm::SmallVector<ParmVarDecl*, 16> Params; 819f7f020bbSFariborz Jahanian for (unsigned i = 0, e = FT->getNumArgs(); i != e; ++i) 820f7f020bbSFariborz Jahanian Params.push_back(ParmVarDecl::Create(getContext(), FD, 821f7f020bbSFariborz Jahanian SourceLocation(), 0, 822f7f020bbSFariborz Jahanian FT->getArgType(i), /*TInfo=*/0, 8238e7d656aSJohn McCall SC_None, SC_None, 0)); 824f7f020bbSFariborz Jahanian FD->setParams(Params.data(), Params.size()); 825f7f020bbSFariborz Jahanian } 82636ad0e99SFariborz Jahanian 827017460abSMike Stump 82887fe5d56SJohn McCall // If we have a C++ 'this' reference, go ahead and force it into 82987fe5d56SJohn McCall // existence now. 83087fe5d56SJohn McCall if (Info.CXXThisRef) { 83187fe5d56SJohn McCall assert(!BlockCXXThisOffset.isZero() && 83287fe5d56SJohn McCall "haven't yet allocated 'this' reference"); 83387fe5d56SJohn McCall 83487fe5d56SJohn McCall // TODO: I have a dream that one day this will be typed. 83587fe5d56SJohn McCall llvm::Value *BlockLiteral = LoadBlockStruct(); 83687fe5d56SJohn McCall llvm::Value *ThisPtrRaw = 83787fe5d56SJohn McCall Builder.CreateConstInBoundsGEP1_64(BlockLiteral, 83887fe5d56SJohn McCall BlockCXXThisOffset.getQuantity(), 83987fe5d56SJohn McCall "this.ptr.raw"); 84087fe5d56SJohn McCall 84187fe5d56SJohn McCall const llvm::Type *Ty = 84287fe5d56SJohn McCall CGM.getTypes().ConvertType(Info.CXXThisRef->getType()); 84387fe5d56SJohn McCall Ty = llvm::PointerType::get(Ty, 0); 84487fe5d56SJohn McCall llvm::Value *ThisPtr = Builder.CreateBitCast(ThisPtrRaw, Ty, "this.ptr"); 84587fe5d56SJohn McCall 84687fe5d56SJohn McCall CXXThisValue = Builder.CreateLoad(ThisPtr, "this"); 84787fe5d56SJohn McCall } 84887fe5d56SJohn McCall 8499d42f0f1SJohn McCall // If we have an Objective C 'self' reference, go ahead and force it 8509d42f0f1SJohn McCall // into existence now. 8519d42f0f1SJohn McCall if (Info.NeedsObjCSelf) { 8529d42f0f1SJohn McCall ValueDecl *Self = cast<ObjCMethodDecl>(CurFuncDecl)->getSelfDecl(); 8539d42f0f1SJohn McCall LocalDeclMap[Self] = GetAddrOfBlockDecl(Self, false); 8549d42f0f1SJohn McCall } 8559d42f0f1SJohn McCall 856017460abSMike Stump // Save a spot to insert the debug information for all the BlockDeclRefDecls. 857017460abSMike Stump llvm::BasicBlock *entry = Builder.GetInsertBlock(); 858017460abSMike Stump llvm::BasicBlock::iterator entry_ptr = Builder.GetInsertPoint(); 859017460abSMike Stump --entry_ptr; 860017460abSMike Stump 8613385fe12SChris Lattner EmitStmt(BExpr->getBody()); 862017460abSMike Stump 8637d69911aSMike Stump // Remember where we were... 8647d69911aSMike Stump llvm::BasicBlock *resume = Builder.GetInsertBlock(); 865017460abSMike Stump 8667d69911aSMike Stump // Go back to the entry. 867017460abSMike Stump ++entry_ptr; 868017460abSMike Stump Builder.SetInsertPoint(entry, entry_ptr); 869017460abSMike Stump 8702e722b91SMike Stump if (CGDebugInfo *DI = getDebugInfo()) { 8712e722b91SMike Stump // Emit debug information for all the BlockDeclRefDecls. 87287fe5d56SJohn McCall // FIXME: also for 'this' 87387fe5d56SJohn McCall for (unsigned i = 0, e = BlockLayout.size(); i != e; ++i) { 874bf78478dSChris Lattner if (const BlockDeclRefExpr *BDRE = 87587fe5d56SJohn McCall dyn_cast<BlockDeclRefExpr>(BlockLayout[i])) { 8762e722b91SMike Stump const ValueDecl *D = BDRE->getDecl(); 8772e722b91SMike Stump DI->setLocation(D->getLocation()); 8782e722b91SMike Stump DI->EmitDeclareOfBlockDeclRefVariable(BDRE, 8792e722b91SMike Stump LocalDeclMap[getBlockStructDecl()], 8802e722b91SMike Stump Builder, this); 8812e722b91SMike Stump } 8822e722b91SMike Stump } 8832e722b91SMike Stump } 8847d69911aSMike Stump // And resume where we left off. 8857d69911aSMike Stump if (resume == 0) 8867d69911aSMike Stump Builder.ClearInsertionPoint(); 8877d69911aSMike Stump else 8887d69911aSMike Stump Builder.SetInsertPoint(resume); 8892e722b91SMike Stump 8903385fe12SChris Lattner FinishFunction(cast<CompoundStmt>(BExpr->getBody())->getRBracLoc()); 8916a60fa24SAnders Carlsson 892b750d928SMike Stump // The runtime needs a minimum alignment of a void *. 8935262b119SKen Dyck CharUnits MinAlign = getContext().getTypeAlignInChars(getContext().VoidPtrTy); 89440775003SKen Dyck BlockOffset = CharUnits::fromQuantity( 8955262b119SKen Dyck llvm::RoundUpToAlignment(BlockOffset.getQuantity(), 8965262b119SKen Dyck MinAlign.getQuantity())); 897b750d928SMike Stump 8989d42f0f1SJohn McCall Info.BlockSize = BlockOffset; 8999d42f0f1SJohn McCall Info.BlockAlign = BlockAlign; 9009d42f0f1SJohn McCall Info.BlockLayout = BlockLayout; 9019d42f0f1SJohn McCall Info.BlockHasCopyDispose = BlockHasCopyDispose; 9026a60fa24SAnders Carlsson return Fn; 9036a60fa24SAnders Carlsson } 9041db7d04bSMike Stump 90587fe5d56SJohn McCall CharUnits BlockFunction::getBlockOffset(CharUnits Size, CharUnits Align) { 9065262b119SKen Dyck assert((Align.isPositive()) && "alignment must be 1 byte or more"); 9071db7d04bSMike Stump 90840775003SKen Dyck CharUnits OldOffset = BlockOffset; 9091db7d04bSMike Stump 9101db7d04bSMike Stump // Ensure proper alignment, even if it means we have to have a gap 91140775003SKen Dyck BlockOffset = CharUnits::fromQuantity( 9125262b119SKen Dyck llvm::RoundUpToAlignment(BlockOffset.getQuantity(), Align.getQuantity())); 9131db7d04bSMike Stump BlockAlign = std::max(Align, BlockAlign); 9141db7d04bSMike Stump 91540775003SKen Dyck CharUnits Pad = BlockOffset - OldOffset; 91640775003SKen Dyck if (Pad.isPositive()) { 9171db7d04bSMike Stump QualType PadTy = getContext().getConstantArrayType(getContext().CharTy, 91840775003SKen Dyck llvm::APInt(32, 91940775003SKen Dyck Pad.getQuantity()), 9201db7d04bSMike Stump ArrayType::Normal, 0); 9216a047c97SDouglas Gregor ValueDecl *PadDecl = VarDecl::Create(getContext(), 9226a047c97SDouglas Gregor getContext().getTranslationUnitDecl(), 9236a047c97SDouglas Gregor SourceLocation(), 924c4df4076SDouglas Gregor 0, QualType(PadTy), 0, 9258e7d656aSJohn McCall SC_None, SC_None); 92687fe5d56SJohn McCall Expr *E = new (getContext()) DeclRefExpr(PadDecl, PadDecl->getType(), 927ed6c7440SDouglas Gregor SourceLocation()); 92887fe5d56SJohn McCall BlockLayout.push_back(E); 9291db7d04bSMike Stump } 9301db7d04bSMike Stump 9311db7d04bSMike Stump BlockOffset += Size; 9321db7d04bSMike Stump return BlockOffset - Size; 9331db7d04bSMike Stump } 93497d01d50SMike Stump 935aeb0ffd5SMike Stump llvm::Constant *BlockFunction:: 936aeb0ffd5SMike Stump GenerateCopyHelperFunction(bool BlockHasCopyDispose, const llvm::StructType *T, 93767645930SMike Stump std::vector<HelperInfo> *NoteForHelperp) { 9380c743277SMike Stump QualType R = getContext().VoidTy; 9390c743277SMike Stump 9400c743277SMike Stump FunctionArgList Args; 9410c743277SMike Stump // FIXME: This leaks 942aeb0ffd5SMike Stump ImplicitParamDecl *Dst = 9437fe9cc1dSMike Stump ImplicitParamDecl::Create(getContext(), 0, 9447fe9cc1dSMike Stump SourceLocation(), 0, 945aeb0ffd5SMike Stump getContext().getPointerType(getContext().VoidTy)); 946aeb0ffd5SMike Stump Args.push_back(std::make_pair(Dst, Dst->getType())); 9470c743277SMike Stump ImplicitParamDecl *Src = 9487fe9cc1dSMike Stump ImplicitParamDecl::Create(getContext(), 0, 9497fe9cc1dSMike Stump SourceLocation(), 0, 9500c743277SMike Stump getContext().getPointerType(getContext().VoidTy)); 9510c743277SMike Stump Args.push_back(std::make_pair(Src, Src->getType())); 9520c743277SMike Stump 9530c743277SMike Stump const CGFunctionInfo &FI = 954c50c27ccSRafael Espindola CGM.getTypes().getFunctionInfo(R, Args, FunctionType::ExtInfo()); 9550c743277SMike Stump 956cbc2bcafSMike Stump // FIXME: We'd like to put these into a mergable by content, with 957cbc2bcafSMike Stump // internal linkage. 9580c743277SMike Stump CodeGenTypes &Types = CGM.getTypes(); 9590c743277SMike Stump const llvm::FunctionType *LTy = Types.GetFunctionType(FI, false); 9600c743277SMike Stump 9610c743277SMike Stump llvm::Function *Fn = 9620c743277SMike Stump llvm::Function::Create(LTy, llvm::GlobalValue::InternalLinkage, 963d6b28fc6SBenjamin Kramer "__copy_helper_block_", &CGM.getModule()); 9640c743277SMike Stump 9650c743277SMike Stump IdentifierInfo *II 9660c743277SMike Stump = &CGM.getContext().Idents.get("__copy_helper_block_"); 9670c743277SMike Stump 9680c743277SMike Stump FunctionDecl *FD = FunctionDecl::Create(getContext(), 9690c743277SMike Stump getContext().getTranslationUnitDecl(), 97060ed5604SArgyrios Kyrtzidis SourceLocation(), II, R, 0, 9718e7d656aSJohn McCall SC_Static, 9728e7d656aSJohn McCall SC_None, 973c4df4076SDouglas Gregor false, 9740c743277SMike Stump true); 9750c743277SMike Stump CGF.StartFunction(FD, R, Fn, Args, SourceLocation()); 976aeb0ffd5SMike Stump 977aeb0ffd5SMike Stump llvm::Value *SrcObj = CGF.GetAddrOfLocalVar(Src); 978aeb0ffd5SMike Stump llvm::Type *PtrPtrT; 97967645930SMike Stump 98067645930SMike Stump if (NoteForHelperp) { 98167645930SMike Stump std::vector<HelperInfo> &NoteForHelper = *NoteForHelperp; 98267645930SMike Stump 9839793f0e4SOwen Anderson PtrPtrT = llvm::PointerType::get(llvm::PointerType::get(T, 0), 0); 984aeb0ffd5SMike Stump SrcObj = Builder.CreateBitCast(SrcObj, PtrPtrT); 985aeb0ffd5SMike Stump SrcObj = Builder.CreateLoad(SrcObj); 986aeb0ffd5SMike Stump 987aeb0ffd5SMike Stump llvm::Value *DstObj = CGF.GetAddrOfLocalVar(Dst); 9883b65ac26SMike Stump llvm::Type *PtrPtrT; 9899793f0e4SOwen Anderson PtrPtrT = llvm::PointerType::get(llvm::PointerType::get(T, 0), 0); 9903b65ac26SMike Stump DstObj = Builder.CreateBitCast(DstObj, PtrPtrT); 9913b65ac26SMike Stump DstObj = Builder.CreateLoad(DstObj); 992aeb0ffd5SMike Stump 993aeb0ffd5SMike Stump for (unsigned i=0; i < NoteForHelper.size(); ++i) { 994aeb0ffd5SMike Stump int flag = NoteForHelper[i].flag; 995aeb0ffd5SMike Stump int index = NoteForHelper[i].index; 996aeb0ffd5SMike Stump 997aeb0ffd5SMike Stump if ((NoteForHelper[i].flag & BLOCK_FIELD_IS_BYREF) 998aeb0ffd5SMike Stump || NoteForHelper[i].RequiresCopying) { 999aeb0ffd5SMike Stump llvm::Value *Srcv = SrcObj; 1000aeb0ffd5SMike Stump Srcv = Builder.CreateStructGEP(Srcv, index); 1001aeb0ffd5SMike Stump Srcv = Builder.CreateBitCast(Srcv, 10029793f0e4SOwen Anderson llvm::PointerType::get(PtrToInt8Ty, 0)); 1003aeb0ffd5SMike Stump Srcv = Builder.CreateLoad(Srcv); 1004aeb0ffd5SMike Stump 1005aeb0ffd5SMike Stump llvm::Value *Dstv = Builder.CreateStructGEP(DstObj, index); 1006aeb0ffd5SMike Stump Dstv = Builder.CreateBitCast(Dstv, PtrToInt8Ty); 1007aeb0ffd5SMike Stump 10085e016ae9SChris Lattner llvm::Value *N = llvm::ConstantInt::get(CGF.Int32Ty, flag); 1009900546d2SDaniel Dunbar llvm::Value *F = CGM.getBlockObjectAssign(); 1010aeb0ffd5SMike Stump Builder.CreateCall3(F, Dstv, Srcv, N); 1011aeb0ffd5SMike Stump } 1012aeb0ffd5SMike Stump } 101367645930SMike Stump } 1014aeb0ffd5SMike Stump 10150c743277SMike Stump CGF.FinishFunction(); 10160c743277SMike Stump 1017ade90fd1SOwen Anderson return llvm::ConstantExpr::getBitCast(Fn, PtrToInt8Ty); 101897d01d50SMike Stump } 101997d01d50SMike Stump 1020d6ef62f0SMike Stump llvm::Constant *BlockFunction:: 1021aeb0ffd5SMike Stump GenerateDestroyHelperFunction(bool BlockHasCopyDispose, 1022aeb0ffd5SMike Stump const llvm::StructType* T, 102367645930SMike Stump std::vector<HelperInfo> *NoteForHelperp) { 10240c743277SMike Stump QualType R = getContext().VoidTy; 10250c743277SMike Stump 10260c743277SMike Stump FunctionArgList Args; 10270c743277SMike Stump // FIXME: This leaks 10280c743277SMike Stump ImplicitParamDecl *Src = 10297fe9cc1dSMike Stump ImplicitParamDecl::Create(getContext(), 0, 10307fe9cc1dSMike Stump SourceLocation(), 0, 10310c743277SMike Stump getContext().getPointerType(getContext().VoidTy)); 10320c743277SMike Stump 10330c743277SMike Stump Args.push_back(std::make_pair(Src, Src->getType())); 10340c743277SMike Stump 10350c743277SMike Stump const CGFunctionInfo &FI = 1036c50c27ccSRafael Espindola CGM.getTypes().getFunctionInfo(R, Args, FunctionType::ExtInfo()); 10370c743277SMike Stump 1038cbc2bcafSMike Stump // FIXME: We'd like to put these into a mergable by content, with 1039cbc2bcafSMike Stump // internal linkage. 10400c743277SMike Stump CodeGenTypes &Types = CGM.getTypes(); 10410c743277SMike Stump const llvm::FunctionType *LTy = Types.GetFunctionType(FI, false); 10420c743277SMike Stump 10430c743277SMike Stump llvm::Function *Fn = 10440c743277SMike Stump llvm::Function::Create(LTy, llvm::GlobalValue::InternalLinkage, 1045d6b28fc6SBenjamin Kramer "__destroy_helper_block_", &CGM.getModule()); 10460c743277SMike Stump 10470c743277SMike Stump IdentifierInfo *II 10480c743277SMike Stump = &CGM.getContext().Idents.get("__destroy_helper_block_"); 10490c743277SMike Stump 10500c743277SMike Stump FunctionDecl *FD = FunctionDecl::Create(getContext(), 10510c743277SMike Stump getContext().getTranslationUnitDecl(), 105260ed5604SArgyrios Kyrtzidis SourceLocation(), II, R, 0, 10538e7d656aSJohn McCall SC_Static, 10548e7d656aSJohn McCall SC_None, 1055c4df4076SDouglas Gregor false, true); 10560c743277SMike Stump CGF.StartFunction(FD, R, Fn, Args, SourceLocation()); 10576f7d9f87SMike Stump 105867645930SMike Stump if (NoteForHelperp) { 105967645930SMike Stump std::vector<HelperInfo> &NoteForHelper = *NoteForHelperp; 106067645930SMike Stump 10616f7d9f87SMike Stump llvm::Value *SrcObj = CGF.GetAddrOfLocalVar(Src); 10626f7d9f87SMike Stump llvm::Type *PtrPtrT; 10639793f0e4SOwen Anderson PtrPtrT = llvm::PointerType::get(llvm::PointerType::get(T, 0), 0); 10646f7d9f87SMike Stump SrcObj = Builder.CreateBitCast(SrcObj, PtrPtrT); 10656f7d9f87SMike Stump SrcObj = Builder.CreateLoad(SrcObj); 10666f7d9f87SMike Stump 10676f7d9f87SMike Stump for (unsigned i=0; i < NoteForHelper.size(); ++i) { 10686f7d9f87SMike Stump int flag = NoteForHelper[i].flag; 10696f7d9f87SMike Stump int index = NoteForHelper[i].index; 10706f7d9f87SMike Stump 10716f7d9f87SMike Stump if ((NoteForHelper[i].flag & BLOCK_FIELD_IS_BYREF) 10726f7d9f87SMike Stump || NoteForHelper[i].RequiresCopying) { 10736f7d9f87SMike Stump llvm::Value *Srcv = SrcObj; 10746f7d9f87SMike Stump Srcv = Builder.CreateStructGEP(Srcv, index); 10756f7d9f87SMike Stump Srcv = Builder.CreateBitCast(Srcv, 10769793f0e4SOwen Anderson llvm::PointerType::get(PtrToInt8Ty, 0)); 10776f7d9f87SMike Stump Srcv = Builder.CreateLoad(Srcv); 10786f7d9f87SMike Stump 10796f7d9f87SMike Stump BuildBlockRelease(Srcv, flag); 10806f7d9f87SMike Stump } 10816f7d9f87SMike Stump } 108267645930SMike Stump } 10836f7d9f87SMike Stump 10840c743277SMike Stump CGF.FinishFunction(); 10850c743277SMike Stump 1086ade90fd1SOwen Anderson return llvm::ConstantExpr::getBitCast(Fn, PtrToInt8Ty); 10870c743277SMike Stump } 10880c743277SMike Stump 1089aeb0ffd5SMike Stump llvm::Constant *BlockFunction::BuildCopyHelper(const llvm::StructType *T, 109067645930SMike Stump std::vector<HelperInfo> *NoteForHelper) { 1091aeb0ffd5SMike Stump return CodeGenFunction(CGM).GenerateCopyHelperFunction(BlockHasCopyDispose, 1092aeb0ffd5SMike Stump T, NoteForHelper); 10930c743277SMike Stump } 10940c743277SMike Stump 1095aeb0ffd5SMike Stump llvm::Constant *BlockFunction::BuildDestroyHelper(const llvm::StructType *T, 109667645930SMike Stump std::vector<HelperInfo> *NoteForHelperp) { 1097aeb0ffd5SMike Stump return CodeGenFunction(CGM).GenerateDestroyHelperFunction(BlockHasCopyDispose, 109867645930SMike Stump T, NoteForHelperp); 109997d01d50SMike Stump } 1100626aecc4SMike Stump 1101f89230d9SMike Stump llvm::Constant *BlockFunction:: 1102f89230d9SMike Stump GeneratebyrefCopyHelperFunction(const llvm::Type *T, int flag) { 1103ee2a5ee5SMike Stump QualType R = getContext().VoidTy; 1104ee2a5ee5SMike Stump 1105ee2a5ee5SMike Stump FunctionArgList Args; 1106ee2a5ee5SMike Stump // FIXME: This leaks 1107f89230d9SMike Stump ImplicitParamDecl *Dst = 11087fe9cc1dSMike Stump ImplicitParamDecl::Create(getContext(), 0, 11097fe9cc1dSMike Stump SourceLocation(), 0, 1110f89230d9SMike Stump getContext().getPointerType(getContext().VoidTy)); 1111f89230d9SMike Stump Args.push_back(std::make_pair(Dst, Dst->getType())); 1112f89230d9SMike Stump 1113f89230d9SMike Stump // FIXME: This leaks 1114ee2a5ee5SMike Stump ImplicitParamDecl *Src = 11157fe9cc1dSMike Stump ImplicitParamDecl::Create(getContext(), 0, 11167fe9cc1dSMike Stump SourceLocation(), 0, 1117ee2a5ee5SMike Stump getContext().getPointerType(getContext().VoidTy)); 1118ee2a5ee5SMike Stump Args.push_back(std::make_pair(Src, Src->getType())); 1119ee2a5ee5SMike Stump 1120ee2a5ee5SMike Stump const CGFunctionInfo &FI = 1121c50c27ccSRafael Espindola CGM.getTypes().getFunctionInfo(R, Args, FunctionType::ExtInfo()); 1122ee2a5ee5SMike Stump 1123ee2a5ee5SMike Stump CodeGenTypes &Types = CGM.getTypes(); 1124ee2a5ee5SMike Stump const llvm::FunctionType *LTy = Types.GetFunctionType(FI, false); 1125ee2a5ee5SMike Stump 1126cbc2bcafSMike Stump // FIXME: We'd like to put these into a mergable by content, with 1127cbc2bcafSMike Stump // internal linkage. 1128ee2a5ee5SMike Stump llvm::Function *Fn = 1129ee2a5ee5SMike Stump llvm::Function::Create(LTy, llvm::GlobalValue::InternalLinkage, 1130d6b28fc6SBenjamin Kramer "__Block_byref_id_object_copy_", &CGM.getModule()); 1131ee2a5ee5SMike Stump 1132ee2a5ee5SMike Stump IdentifierInfo *II 1133ee2a5ee5SMike Stump = &CGM.getContext().Idents.get("__Block_byref_id_object_copy_"); 1134ee2a5ee5SMike Stump 1135ee2a5ee5SMike Stump FunctionDecl *FD = FunctionDecl::Create(getContext(), 1136ee2a5ee5SMike Stump getContext().getTranslationUnitDecl(), 113760ed5604SArgyrios Kyrtzidis SourceLocation(), II, R, 0, 11388e7d656aSJohn McCall SC_Static, 11398e7d656aSJohn McCall SC_None, 1140c4df4076SDouglas Gregor false, true); 1141ee2a5ee5SMike Stump CGF.StartFunction(FD, R, Fn, Args, SourceLocation()); 1142f89230d9SMike Stump 1143f89230d9SMike Stump // dst->x 1144f89230d9SMike Stump llvm::Value *V = CGF.GetAddrOfLocalVar(Dst); 11459793f0e4SOwen Anderson V = Builder.CreateBitCast(V, llvm::PointerType::get(T, 0)); 11463b65ac26SMike Stump V = Builder.CreateLoad(V); 1147f89230d9SMike Stump V = Builder.CreateStructGEP(V, 6, "x"); 1148f89230d9SMike Stump llvm::Value *DstObj = Builder.CreateBitCast(V, PtrToInt8Ty); 1149f89230d9SMike Stump 1150f89230d9SMike Stump // src->x 1151f89230d9SMike Stump V = CGF.GetAddrOfLocalVar(Src); 1152f89230d9SMike Stump V = Builder.CreateLoad(V); 1153f89230d9SMike Stump V = Builder.CreateBitCast(V, T); 1154f89230d9SMike Stump V = Builder.CreateStructGEP(V, 6, "x"); 11559793f0e4SOwen Anderson V = Builder.CreateBitCast(V, llvm::PointerType::get(PtrToInt8Ty, 0)); 1156f89230d9SMike Stump llvm::Value *SrcObj = Builder.CreateLoad(V); 1157f89230d9SMike Stump 1158f89230d9SMike Stump flag |= BLOCK_BYREF_CALLER; 1159f89230d9SMike Stump 11605e016ae9SChris Lattner llvm::Value *N = llvm::ConstantInt::get(CGF.Int32Ty, flag); 1161900546d2SDaniel Dunbar llvm::Value *F = CGM.getBlockObjectAssign(); 1162f89230d9SMike Stump Builder.CreateCall3(F, DstObj, SrcObj, N); 1163f89230d9SMike Stump 1164ee2a5ee5SMike Stump CGF.FinishFunction(); 1165ee2a5ee5SMike Stump 1166ade90fd1SOwen Anderson return llvm::ConstantExpr::getBitCast(Fn, PtrToInt8Ty); 1167ee2a5ee5SMike Stump } 1168ee2a5ee5SMike Stump 1169fbe25dd1SMike Stump llvm::Constant * 1170fbe25dd1SMike Stump BlockFunction::GeneratebyrefDestroyHelperFunction(const llvm::Type *T, 1171fbe25dd1SMike Stump int flag) { 1172ee2a5ee5SMike Stump QualType R = getContext().VoidTy; 1173ee2a5ee5SMike Stump 1174ee2a5ee5SMike Stump FunctionArgList Args; 1175ee2a5ee5SMike Stump // FIXME: This leaks 1176ee2a5ee5SMike Stump ImplicitParamDecl *Src = 11777fe9cc1dSMike Stump ImplicitParamDecl::Create(getContext(), 0, 11787fe9cc1dSMike Stump SourceLocation(), 0, 1179ee2a5ee5SMike Stump getContext().getPointerType(getContext().VoidTy)); 1180ee2a5ee5SMike Stump 1181ee2a5ee5SMike Stump Args.push_back(std::make_pair(Src, Src->getType())); 1182ee2a5ee5SMike Stump 1183ee2a5ee5SMike Stump const CGFunctionInfo &FI = 1184c50c27ccSRafael Espindola CGM.getTypes().getFunctionInfo(R, Args, FunctionType::ExtInfo()); 1185ee2a5ee5SMike Stump 1186ee2a5ee5SMike Stump CodeGenTypes &Types = CGM.getTypes(); 1187ee2a5ee5SMike Stump const llvm::FunctionType *LTy = Types.GetFunctionType(FI, false); 1188ee2a5ee5SMike Stump 1189cbc2bcafSMike Stump // FIXME: We'd like to put these into a mergable by content, with 1190cbc2bcafSMike Stump // internal linkage. 1191ee2a5ee5SMike Stump llvm::Function *Fn = 1192ee2a5ee5SMike Stump llvm::Function::Create(LTy, llvm::GlobalValue::InternalLinkage, 1193d6b28fc6SBenjamin Kramer "__Block_byref_id_object_dispose_", 1194ee2a5ee5SMike Stump &CGM.getModule()); 1195ee2a5ee5SMike Stump 1196ee2a5ee5SMike Stump IdentifierInfo *II 1197ee2a5ee5SMike Stump = &CGM.getContext().Idents.get("__Block_byref_id_object_dispose_"); 1198ee2a5ee5SMike Stump 1199ee2a5ee5SMike Stump FunctionDecl *FD = FunctionDecl::Create(getContext(), 1200ee2a5ee5SMike Stump getContext().getTranslationUnitDecl(), 120160ed5604SArgyrios Kyrtzidis SourceLocation(), II, R, 0, 12028e7d656aSJohn McCall SC_Static, 12038e7d656aSJohn McCall SC_None, 1204c4df4076SDouglas Gregor false, true); 1205ee2a5ee5SMike Stump CGF.StartFunction(FD, R, Fn, Args, SourceLocation()); 1206fbe25dd1SMike Stump 1207fbe25dd1SMike Stump llvm::Value *V = CGF.GetAddrOfLocalVar(Src); 12089793f0e4SOwen Anderson V = Builder.CreateBitCast(V, llvm::PointerType::get(T, 0)); 12093b65ac26SMike Stump V = Builder.CreateLoad(V); 1210fbe25dd1SMike Stump V = Builder.CreateStructGEP(V, 6, "x"); 12119793f0e4SOwen Anderson V = Builder.CreateBitCast(V, llvm::PointerType::get(PtrToInt8Ty, 0)); 12123b65ac26SMike Stump V = Builder.CreateLoad(V); 1213fbe25dd1SMike Stump 1214fbe25dd1SMike Stump flag |= BLOCK_BYREF_CALLER; 1215fbe25dd1SMike Stump BuildBlockRelease(V, flag); 1216ee2a5ee5SMike Stump CGF.FinishFunction(); 1217ee2a5ee5SMike Stump 1218ade90fd1SOwen Anderson return llvm::ConstantExpr::getBitCast(Fn, PtrToInt8Ty); 1219ee2a5ee5SMike Stump } 1220ee2a5ee5SMike Stump 1221f89230d9SMike Stump llvm::Constant *BlockFunction::BuildbyrefCopyHelper(const llvm::Type *T, 1222e08ea787SChris Lattner int Flag, unsigned Align) { 1223e08ea787SChris Lattner // All alignments below that of pointer alignment collapse down to just 1224cbc2bcafSMike Stump // pointer alignment, as we always have at least that much alignment to begin 1225cbc2bcafSMike Stump // with. 1226cbc2bcafSMike Stump Align /= unsigned(CGF.Target.getPointerAlign(0)/8); 1227e08ea787SChris Lattner 1228cbc2bcafSMike Stump // As an optimization, we only generate a single function of each kind we 1229cbc2bcafSMike Stump // might need. We need a different one for each alignment and for each 1230cbc2bcafSMike Stump // setting of flags. We mix Align and flag to get the kind. 1231e08ea787SChris Lattner uint64_t Kind = (uint64_t)Align*BLOCK_BYREF_CURRENT_MAX + Flag; 1232e08ea787SChris Lattner llvm::Constant *&Entry = CGM.AssignCache[Kind]; 1233cbc2bcafSMike Stump if (Entry) 1234cbc2bcafSMike Stump return Entry; 1235e08ea787SChris Lattner return Entry = CodeGenFunction(CGM).GeneratebyrefCopyHelperFunction(T, Flag); 1236ee2a5ee5SMike Stump } 1237ee2a5ee5SMike Stump 1238fbe25dd1SMike Stump llvm::Constant *BlockFunction::BuildbyrefDestroyHelper(const llvm::Type *T, 1239e08ea787SChris Lattner int Flag, 1240cbc2bcafSMike Stump unsigned Align) { 1241cbc2bcafSMike Stump // All alignments below that of pointer alignment collpase down to just 1242cbc2bcafSMike Stump // pointer alignment, as we always have at least that much alignment to begin 1243cbc2bcafSMike Stump // with. 1244cbc2bcafSMike Stump Align /= unsigned(CGF.Target.getPointerAlign(0)/8); 1245e08ea787SChris Lattner 1246cbc2bcafSMike Stump // As an optimization, we only generate a single function of each kind we 1247cbc2bcafSMike Stump // might need. We need a different one for each alignment and for each 1248cbc2bcafSMike Stump // setting of flags. We mix Align and flag to get the kind. 1249e08ea787SChris Lattner uint64_t Kind = (uint64_t)Align*BLOCK_BYREF_CURRENT_MAX + Flag; 1250e08ea787SChris Lattner llvm::Constant *&Entry = CGM.DestroyCache[Kind]; 1251cbc2bcafSMike Stump if (Entry) 1252cbc2bcafSMike Stump return Entry; 1253e08ea787SChris Lattner return Entry=CodeGenFunction(CGM).GeneratebyrefDestroyHelperFunction(T, Flag); 1254ee2a5ee5SMike Stump } 1255ee2a5ee5SMike Stump 1256fbe25dd1SMike Stump void BlockFunction::BuildBlockRelease(llvm::Value *V, int flag) { 1257900546d2SDaniel Dunbar llvm::Value *F = CGM.getBlockObjectDispose(); 1258fbe25dd1SMike Stump llvm::Value *N; 1259626aecc4SMike Stump V = Builder.CreateBitCast(V, PtrToInt8Ty); 12605e016ae9SChris Lattner N = llvm::ConstantInt::get(CGF.Int32Ty, flag); 1261626aecc4SMike Stump Builder.CreateCall2(F, V, N); 1262626aecc4SMike Stump } 12634446dcf0SMike Stump 12644446dcf0SMike Stump ASTContext &BlockFunction::getContext() const { return CGM.getContext(); } 1265aeb0ffd5SMike Stump 1266aeb0ffd5SMike Stump BlockFunction::BlockFunction(CodeGenModule &cgm, CodeGenFunction &cgf, 1267aeb0ffd5SMike Stump CGBuilderTy &B) 12685e016ae9SChris Lattner : CGM(cgm), VMContext(cgm.getLLVMContext()), CGF(cgf), Builder(B) { 126941a75027SOwen Anderson PtrToInt8Ty = llvm::PointerType::getUnqual( 127041a75027SOwen Anderson llvm::Type::getInt8Ty(VMContext)); 1271aeb0ffd5SMike Stump 1272aeb0ffd5SMike Stump BlockHasCopyDispose = false; 1273aeb0ffd5SMike Stump } 1274