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