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