blob: 15347a488fa741d3dd88a6feada6ec4f46d29b31 [file] [log] [blame]
Anders Carlssonacfde802009-02-12 00:39:25 +00001//===--- CGBlocks.cpp - Emit LLVM Code for declarations -------------------===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This contains code to emit blocks.
11//
12//===----------------------------------------------------------------------===//
13
Mike Stumpb1a6e682009-09-30 02:43:10 +000014#include "CGDebugInfo.h"
Anders Carlssonacfde802009-02-12 00:39:25 +000015#include "CodeGenFunction.h"
16#include "CodeGenModule.h"
Mike Stump6cc88f72009-03-20 21:53:12 +000017#include "clang/AST/DeclObjC.h"
Anders Carlssonacfde802009-02-12 00:39:25 +000018#include "llvm/Module.h"
Anders Carlssond5cab542009-02-12 17:55:02 +000019#include "llvm/Target/TargetData.h"
Anders Carlssonacfde802009-02-12 00:39:25 +000020#include <algorithm>
Torok Edwinf42e4a62009-08-24 13:25:12 +000021
Anders Carlssonacfde802009-02-12 00:39:25 +000022using namespace clang;
23using namespace CodeGen;
24
Mike Stumpcf62d392009-03-06 18:42:23 +000025llvm::Constant *CodeGenFunction::
Ken Dyck199c3d62010-01-11 17:06:35 +000026BuildDescriptorBlockDecl(bool BlockHasCopyDispose, CharUnits Size,
Mike Stumpa803b0e2009-03-25 17:58:24 +000027 const llvm::StructType* Ty,
Mike Stump08920992009-03-07 02:35:30 +000028 std::vector<HelperInfo> *NoteForHelper) {
Mike Stump56129b12009-02-13 16:55:51 +000029 const llvm::Type *UnsignedLongTy
30 = CGM.getTypes().ConvertType(getContext().UnsignedLongTy);
Mike Stumpe5fee252009-02-13 16:19:19 +000031 llvm::Constant *C;
32 std::vector<llvm::Constant*> Elts;
33
34 // reserved
Owen Anderson4a28d5d2009-07-24 23:12:58 +000035 C = llvm::ConstantInt::get(UnsignedLongTy, 0);
Mike Stumpe5fee252009-02-13 16:19:19 +000036 Elts.push_back(C);
37
38 // Size
Mike Stumpd6840002009-02-21 20:07:44 +000039 // FIXME: What is the right way to say this doesn't fit? We should give
40 // a user diagnostic in that case. Better fix would be to change the
41 // API to size_t.
Ken Dyck199c3d62010-01-11 17:06:35 +000042 C = llvm::ConstantInt::get(UnsignedLongTy, Size.getQuantity());
Mike Stumpe5fee252009-02-13 16:19:19 +000043 Elts.push_back(C);
44
45 if (BlockHasCopyDispose) {
46 // copy_func_helper_decl
Mike Stumpb7477cf2009-04-10 18:52:28 +000047 Elts.push_back(BuildCopyHelper(Ty, NoteForHelper));
Mike Stumpe5fee252009-02-13 16:19:19 +000048
49 // destroy_func_decl
Mike Stumpb7477cf2009-04-10 18:52:28 +000050 Elts.push_back(BuildDestroyHelper(Ty, NoteForHelper));
Mike Stumpe5fee252009-02-13 16:19:19 +000051 }
52
Nick Lewycky0d36dd22009-09-19 20:00:52 +000053 C = llvm::ConstantStruct::get(VMContext, Elts, false);
Mike Stumpe5fee252009-02-13 16:19:19 +000054
Owen Anderson1c431b32009-07-08 19:05:04 +000055 C = new llvm::GlobalVariable(CGM.getModule(), C->getType(), true,
Mike Stumpe5fee252009-02-13 16:19:19 +000056 llvm::GlobalValue::InternalLinkage,
Owen Anderson1c431b32009-07-08 19:05:04 +000057 C, "__block_descriptor_tmp");
Mike Stumpe5fee252009-02-13 16:19:19 +000058 return C;
59}
60
Mike Stump2a998142009-03-04 18:17:45 +000061llvm::Constant *BlockModule::getNSConcreteGlobalBlock() {
Chris Lattnere2f79b62009-05-13 02:50:56 +000062 if (NSConcreteGlobalBlock == 0)
Mike Stump1eb44332009-09-09 15:08:12 +000063 NSConcreteGlobalBlock = CGM.CreateRuntimeVariable(PtrToInt8Ty,
Chris Lattnere2f79b62009-05-13 02:50:56 +000064 "_NSConcreteGlobalBlock");
Mike Stumpf99f1d02009-02-13 17:23:42 +000065 return NSConcreteGlobalBlock;
66}
67
Mike Stump2a998142009-03-04 18:17:45 +000068llvm::Constant *BlockModule::getNSConcreteStackBlock() {
Chris Lattnere2f79b62009-05-13 02:50:56 +000069 if (NSConcreteStackBlock == 0)
Mike Stump1eb44332009-09-09 15:08:12 +000070 NSConcreteStackBlock = CGM.CreateRuntimeVariable(PtrToInt8Ty,
Chris Lattnere2f79b62009-05-13 02:50:56 +000071 "_NSConcreteStackBlock");
Mike Stump59c5b112009-02-13 19:29:27 +000072 return NSConcreteStackBlock;
73}
74
Mike Stump38e16272009-10-21 22:01:24 +000075static void CollectBlockDeclRefInfo(
76 const Stmt *S, CodeGenFunction::BlockInfo &Info,
77 llvm::SmallSet<const DeclContext *, 16> &InnerContexts) {
Anders Carlsson4de9fce2009-03-01 01:09:12 +000078 for (Stmt::const_child_iterator I = S->child_begin(), E = S->child_end();
79 I != E; ++I)
Daniel Dunbar82573ee2009-03-02 07:00:57 +000080 if (*I)
Mike Stump38e16272009-10-21 22:01:24 +000081 CollectBlockDeclRefInfo(*I, Info, InnerContexts);
Mike Stump00470a12009-03-05 08:32:30 +000082
Mike Stumpea26cb52009-10-21 03:49:08 +000083 // We want to ensure we walk down into block literals so we can find
84 // all nested BlockDeclRefExprs.
Mike Stump38e16272009-10-21 22:01:24 +000085 if (const BlockExpr *BE = dyn_cast<BlockExpr>(S)) {
86 InnerContexts.insert(cast<DeclContext>(BE->getBlockDecl()));
87 CollectBlockDeclRefInfo(BE->getBody(), Info, InnerContexts);
88 }
Mike Stumpea26cb52009-10-21 03:49:08 +000089
Mike Stump38e16272009-10-21 22:01:24 +000090 if (const BlockDeclRefExpr *BDRE = dyn_cast<BlockDeclRefExpr>(S)) {
Anders Carlsson4de9fce2009-03-01 01:09:12 +000091 // FIXME: Handle enums.
Mike Stump38e16272009-10-21 22:01:24 +000092 if (isa<FunctionDecl>(BDRE->getDecl()))
Anders Carlsson4de9fce2009-03-01 01:09:12 +000093 return;
Mike Stump00470a12009-03-05 08:32:30 +000094
Mike Stump38e16272009-10-21 22:01:24 +000095 // Only Decls that escape are added.
96 if (!InnerContexts.count(BDRE->getDecl()->getDeclContext()))
97 Info.DeclRefs.push_back(BDRE);
Anders Carlsson4de9fce2009-03-01 01:09:12 +000098 }
99}
100
Mike Stump58a85142009-03-04 22:48:06 +0000101/// CanBlockBeGlobal - Given a BlockInfo struct, determines if a block can be
102/// declared as a global variable instead of on the stack.
Chris Lattnere2f79b62009-05-13 02:50:56 +0000103static bool CanBlockBeGlobal(const CodeGenFunction::BlockInfo &Info) {
Mike Stumpea26cb52009-10-21 03:49:08 +0000104 return Info.DeclRefs.empty();
105}
106
107/// AllocateAllBlockDeclRefs - Preallocate all nested BlockDeclRefExprs to
108/// ensure we can generate the debug information for the parameter for the block
109/// invoke function.
110static void AllocateAllBlockDeclRefs(const CodeGenFunction::BlockInfo &Info,
111 CodeGenFunction *CGF) {
112 // Always allocate self, as it is often handy in the debugger, even if there
113 // is no codegen in the block that uses it. This is also useful to always do
114 // this as if we didn't, we'd have to figure out all code that uses a self
115 // pointer, including implicit uses.
116 if (const ObjCMethodDecl *OMD
117 = dyn_cast_or_null<ObjCMethodDecl>(CGF->CurFuncDecl)) {
118 ImplicitParamDecl *SelfDecl = OMD->getSelfDecl();
119 BlockDeclRefExpr *BDRE = new (CGF->getContext())
120 BlockDeclRefExpr(SelfDecl,
121 SelfDecl->getType(), SourceLocation(), false);
122 CGF->AllocateBlockDecl(BDRE);
123 }
124
125 // FIXME: Also always forward the this pointer in C++ as well.
126
127 for (size_t i = 0; i < Info.DeclRefs.size(); ++i)
128 CGF->AllocateBlockDecl(Info.DeclRefs[i]);
Anders Carlsson4de9fce2009-03-01 01:09:12 +0000129}
130
Mike Stump58a85142009-03-04 22:48:06 +0000131// FIXME: Push most into CGM, passing down a few bits, like current function
132// name.
Mike Stump8a2b4b12009-02-25 23:33:13 +0000133llvm::Value *CodeGenFunction::BuildBlockLiteralTmp(const BlockExpr *BE) {
Mike Stumpe5fee252009-02-13 16:19:19 +0000134
Anders Carlsson4de9fce2009-03-01 01:09:12 +0000135 std::string Name = CurFn->getName();
136 CodeGenFunction::BlockInfo Info(0, Name.c_str());
Mike Stump38e16272009-10-21 22:01:24 +0000137 llvm::SmallSet<const DeclContext *, 16> InnerContexts;
138 InnerContexts.insert(BE->getBlockDecl());
139 CollectBlockDeclRefInfo(BE->getBody(), Info, InnerContexts);
Anders Carlsson4de9fce2009-03-01 01:09:12 +0000140
141 // Check if the block can be global.
Mike Stump58a85142009-03-04 22:48:06 +0000142 // FIXME: This test doesn't work for nested blocks yet. Longer term, I'd like
143 // to just have one code path. We should move this function into CGM and pass
144 // CGF, then we can just check to see if CGF is 0.
Mike Stumpdab514f2009-03-04 03:23:46 +0000145 if (0 && CanBlockBeGlobal(Info))
Anders Carlsson4de9fce2009-03-01 01:09:12 +0000146 return CGM.GetAddrOfGlobalBlock(BE, Name.c_str());
Mike Stump00470a12009-03-05 08:32:30 +0000147
David Chisnall5e530af2009-11-17 19:33:30 +0000148 size_t BlockFields = 5;
149
150 bool hasIntrospection = CGM.getContext().getLangOptions().BlockIntrospection;
151
152 if (hasIntrospection) {
153 BlockFields++;
154 }
155 std::vector<llvm::Constant*> Elts(BlockFields);
156
157 if (hasIntrospection) {
158 std::string BlockTypeEncoding;
159 CGM.getContext().getObjCEncodingForBlock(BE, BlockTypeEncoding);
160
161 Elts[5] = llvm::ConstantExpr::getBitCast(
162 CGM.GetAddrOfConstantCString(BlockTypeEncoding), PtrToInt8Ty);
163 }
164
Mike Stumpe5fee252009-02-13 16:19:19 +0000165 llvm::Constant *C;
Mike Stump8a2b4b12009-02-25 23:33:13 +0000166 llvm::Value *V;
Mike Stumpe5fee252009-02-13 16:19:19 +0000167
Mike Stumpe5fee252009-02-13 16:19:19 +0000168 {
169 // C = BuildBlockStructInitlist();
170 unsigned int flags = BLOCK_HAS_DESCRIPTOR;
171
David Chisnall5e530af2009-11-17 19:33:30 +0000172 if (hasIntrospection)
173 flags |= BLOCK_HAS_OBJC_TYPE;
174
Mike Stump00470a12009-03-05 08:32:30 +0000175 // We run this first so that we set BlockHasCopyDispose from the entire
176 // block literal.
177 // __invoke
Ken Dyck199c3d62010-01-11 17:06:35 +0000178 CharUnits subBlockSize;
179 uint64_t subBlockAlign;
Mike Stump00470a12009-03-05 08:32:30 +0000180 llvm::SmallVector<const Expr *, 8> subBlockDeclRefDecls;
Mike Stumpa803b0e2009-03-25 17:58:24 +0000181 bool subBlockHasCopyDispose = false;
Mike Stump00470a12009-03-05 08:32:30 +0000182 llvm::Function *Fn
Mike Stumpbb304192009-09-24 22:31:14 +0000183 = CodeGenFunction(CGM).GenerateBlockFunction(BE, Info, CurFuncDecl,
184 LocalDeclMap,
Mike Stump7f28a9c2009-03-13 23:34:28 +0000185 subBlockSize,
Mike Stump00470a12009-03-05 08:32:30 +0000186 subBlockAlign,
187 subBlockDeclRefDecls,
Mike Stumpa803b0e2009-03-25 17:58:24 +0000188 subBlockHasCopyDispose);
189 BlockHasCopyDispose |= subBlockHasCopyDispose;
Mike Stump00470a12009-03-05 08:32:30 +0000190 Elts[3] = Fn;
191
Mike Stumpf5408fe2009-05-16 07:57:57 +0000192 // FIXME: Don't use BlockHasCopyDispose, it is set more often then
193 // necessary, for example: { ^{ __block int i; ^{ i = 1; }(); }(); }
Mike Stumpa803b0e2009-03-25 17:58:24 +0000194 if (subBlockHasCopyDispose)
Mike Stumpe5fee252009-02-13 16:19:19 +0000195 flags |= BLOCK_HAS_COPY_DISPOSE;
196
Mike Stump7d6dc4f2009-02-13 20:17:16 +0000197 // __isa
Mike Stump59c5b112009-02-13 19:29:27 +0000198 C = CGM.getNSConcreteStackBlock();
Owen Anderson3c4972d2009-07-29 18:54:39 +0000199 C = llvm::ConstantExpr::getBitCast(C, PtrToInt8Ty);
Mike Stump00470a12009-03-05 08:32:30 +0000200 Elts[0] = C;
Mike Stumpe5fee252009-02-13 16:19:19 +0000201
202 // __flags
203 const llvm::IntegerType *IntTy = cast<llvm::IntegerType>(
204 CGM.getTypes().ConvertType(CGM.getContext().IntTy));
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000205 C = llvm::ConstantInt::get(IntTy, flags);
Mike Stump00470a12009-03-05 08:32:30 +0000206 Elts[1] = C;
Mike Stumpe5fee252009-02-13 16:19:19 +0000207
208 // __reserved
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000209 C = llvm::ConstantInt::get(IntTy, 0);
Mike Stump00470a12009-03-05 08:32:30 +0000210 Elts[2] = C;
Mike Stumpe5fee252009-02-13 16:19:19 +0000211
Mike Stump8a2b4b12009-02-25 23:33:13 +0000212 if (subBlockDeclRefDecls.size() == 0) {
Mike Stumpcf62d392009-03-06 18:42:23 +0000213 // __descriptor
Mike Stumpea26cb52009-10-21 03:49:08 +0000214 Elts[4] = BuildDescriptorBlockDecl(subBlockHasCopyDispose, subBlockSize,
215 0, 0);
Mike Stumpcf62d392009-03-06 18:42:23 +0000216
Mike Stump5570cfe2009-03-01 20:07:53 +0000217 // Optimize to being a global block.
218 Elts[0] = CGM.getNSConcreteGlobalBlock();
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000219 Elts[1] = llvm::ConstantInt::get(IntTy, flags|BLOCK_IS_GLOBAL);
Mike Stump5570cfe2009-03-01 20:07:53 +0000220
Nick Lewycky0d36dd22009-09-19 20:00:52 +0000221 C = llvm::ConstantStruct::get(VMContext, Elts, false);
Mike Stump8a2b4b12009-02-25 23:33:13 +0000222
Owen Anderson1c431b32009-07-08 19:05:04 +0000223 C = new llvm::GlobalVariable(CGM.getModule(), C->getType(), true,
Benjamin Kramer3cf7c5d2010-01-22 13:59:13 +0000224 llvm::GlobalValue::InternalLinkage, C,
225 "__block_holder_tmp_" +
226 llvm::Twine(CGM.getGlobalUniqueCount()));
Mike Stump8a2b4b12009-02-25 23:33:13 +0000227 QualType BPT = BE->getType();
Owen Anderson3c4972d2009-07-29 18:54:39 +0000228 C = llvm::ConstantExpr::getBitCast(C, ConvertType(BPT));
Mike Stump8a2b4b12009-02-25 23:33:13 +0000229 return C;
230 }
Mike Stump00470a12009-03-05 08:32:30 +0000231
David Chisnall5e530af2009-11-17 19:33:30 +0000232 std::vector<const llvm::Type *> Types(BlockFields+subBlockDeclRefDecls.size());
Mike Stump08920992009-03-07 02:35:30 +0000233 for (int i=0; i<4; ++i)
Mike Stump8a2b4b12009-02-25 23:33:13 +0000234 Types[i] = Elts[i]->getType();
Mike Stump08920992009-03-07 02:35:30 +0000235 Types[4] = PtrToInt8Ty;
David Chisnall5e530af2009-11-17 19:33:30 +0000236 if (hasIntrospection)
237 Types[5] = PtrToInt8Ty;
Mike Stump8a2b4b12009-02-25 23:33:13 +0000238
Mike Stumpa99038c2009-02-28 09:07:16 +0000239 for (unsigned i=0; i < subBlockDeclRefDecls.size(); ++i) {
240 const Expr *E = subBlockDeclRefDecls[i];
241 const BlockDeclRefExpr *BDRE = dyn_cast<BlockDeclRefExpr>(E);
242 QualType Ty = E->getType();
Mike Stumpdab514f2009-03-04 03:23:46 +0000243 if (BDRE && BDRE->isByRef()) {
David Chisnall5e530af2009-11-17 19:33:30 +0000244 Types[i+BlockFields] = llvm::PointerType::get(BuildByRefType(BDRE->getDecl()), 0);
Mike Stumpdab514f2009-03-04 03:23:46 +0000245 } else
David Chisnall5e530af2009-11-17 19:33:30 +0000246 Types[i+BlockFields] = ConvertType(Ty);
Mike Stumpa99038c2009-02-28 09:07:16 +0000247 }
Mike Stump8a2b4b12009-02-25 23:33:13 +0000248
Owen Anderson47a434f2009-08-05 23:18:46 +0000249 llvm::StructType *Ty = llvm::StructType::get(VMContext, Types, true);
Mike Stump8a2b4b12009-02-25 23:33:13 +0000250
251 llvm::AllocaInst *A = CreateTempAlloca(Ty);
252 A->setAlignment(subBlockAlign);
253 V = A;
254
Mike Stump08920992009-03-07 02:35:30 +0000255 std::vector<HelperInfo> NoteForHelper(subBlockDeclRefDecls.size());
256 int helpersize = 0;
257
258 for (unsigned i=0; i<4; ++i)
Mike Stump8a2b4b12009-02-25 23:33:13 +0000259 Builder.CreateStore(Elts[i], Builder.CreateStructGEP(V, i, "block.tmp"));
David Chisnall5e530af2009-11-17 19:33:30 +0000260 if (hasIntrospection)
261 Builder.CreateStore(Elts[5], Builder.CreateStructGEP(V, 5, "block.tmp"));
Mike Stump00470a12009-03-05 08:32:30 +0000262
Mike Stump8a2b4b12009-02-25 23:33:13 +0000263 for (unsigned i=0; i < subBlockDeclRefDecls.size(); ++i)
264 {
Mike Stumpa99038c2009-02-28 09:07:16 +0000265 // FIXME: Push const down.
266 Expr *E = const_cast<Expr*>(subBlockDeclRefDecls[i]);
267 DeclRefExpr *DR;
268 ValueDecl *VD;
Mike Stump8a2b4b12009-02-25 23:33:13 +0000269
Mike Stumpa99038c2009-02-28 09:07:16 +0000270 DR = dyn_cast<DeclRefExpr>(E);
271 // Skip padding.
272 if (DR) continue;
273
274 BlockDeclRefExpr *BDRE = dyn_cast<BlockDeclRefExpr>(E);
275 VD = BDRE->getDecl();
276
David Chisnall5e530af2009-11-17 19:33:30 +0000277 llvm::Value* Addr = Builder.CreateStructGEP(V, i+BlockFields, "tmp");
Mike Stump08920992009-03-07 02:35:30 +0000278 NoteForHelper[helpersize].index = i+5;
Mike Stump39605b42009-09-22 02:12:52 +0000279 NoteForHelper[helpersize].RequiresCopying
280 = BlockRequiresCopying(VD->getType());
Mike Stump08920992009-03-07 02:35:30 +0000281 NoteForHelper[helpersize].flag
Mike Stump39605b42009-09-22 02:12:52 +0000282 = (VD->getType()->isBlockPointerType()
283 ? BLOCK_FIELD_IS_BLOCK
284 : BLOCK_FIELD_IS_OBJECT);
Mike Stump08920992009-03-07 02:35:30 +0000285
Mike Stumpa99038c2009-02-28 09:07:16 +0000286 if (LocalDeclMap[VD]) {
Mike Stumpdab514f2009-03-04 03:23:46 +0000287 if (BDRE->isByRef()) {
Mike Stump08920992009-03-07 02:35:30 +0000288 NoteForHelper[helpersize].flag = BLOCK_FIELD_IS_BYREF |
Mike Stumpf4bc3122009-03-07 06:04:31 +0000289 // FIXME: Someone double check this.
290 (VD->getType().isObjCGCWeak() ? BLOCK_FIELD_IS_WEAK : 0);
Mike Stumpdab514f2009-03-04 03:23:46 +0000291 llvm::Value *Loc = LocalDeclMap[VD];
292 Loc = Builder.CreateStructGEP(Loc, 1, "forwarding");
Daniel Dunbar2da84ff2009-11-29 21:23:36 +0000293 Loc = Builder.CreateLoad(Loc);
Mike Stumpdab514f2009-03-04 03:23:46 +0000294 Builder.CreateStore(Loc, Addr);
Mike Stump08920992009-03-07 02:35:30 +0000295 ++helpersize;
Mike Stumpdab514f2009-03-04 03:23:46 +0000296 continue;
297 } else
John McCalldbd872f2009-12-08 09:08:17 +0000298 E = new (getContext()) DeclRefExpr (VD,
Douglas Gregor0da76df2009-11-23 11:41:28 +0000299 VD->getType(),
300 SourceLocation());
Mike Stumpa99038c2009-02-28 09:07:16 +0000301 }
Mike Stumpdab514f2009-03-04 03:23:46 +0000302 if (BDRE->isByRef()) {
Mike Stumpa8b60c92009-03-21 21:00:35 +0000303 NoteForHelper[helpersize].flag = BLOCK_FIELD_IS_BYREF |
304 // FIXME: Someone double check this.
305 (VD->getType().isObjCGCWeak() ? BLOCK_FIELD_IS_WEAK : 0);
Mike Stumpa99038c2009-02-28 09:07:16 +0000306 E = new (getContext())
307 UnaryOperator(E, UnaryOperator::AddrOf,
308 getContext().getPointerType(E->getType()),
309 SourceLocation());
Mike Stumpdab514f2009-03-04 03:23:46 +0000310 }
Mike Stump08920992009-03-07 02:35:30 +0000311 ++helpersize;
Mike Stumpa99038c2009-02-28 09:07:16 +0000312
Mike Stumpa99038c2009-02-28 09:07:16 +0000313 RValue r = EmitAnyExpr(E, Addr, false);
Mike Stumpdab514f2009-03-04 03:23:46 +0000314 if (r.isScalar()) {
315 llvm::Value *Loc = r.getScalarVal();
David Chisnall5e530af2009-11-17 19:33:30 +0000316 const llvm::Type *Ty = Types[i+BlockFields];
Mike Stumpdab514f2009-03-04 03:23:46 +0000317 if (BDRE->isByRef()) {
Mike Stump58a85142009-03-04 22:48:06 +0000318 // E is now the address of the value field, instead, we want the
319 // address of the actual ByRef struct. We optimize this slightly
320 // compared to gcc by not grabbing the forwarding slot as this must
321 // be done during Block_copy for us, and we can postpone the work
322 // until then.
Ken Dyck199c3d62010-01-11 17:06:35 +0000323 CharUnits offset = BlockDecls[BDRE->getDecl()];
Mike Stump00470a12009-03-05 08:32:30 +0000324
Mike Stump58a85142009-03-04 22:48:06 +0000325 llvm::Value *BlockLiteral = LoadBlockStruct();
Mike Stump00470a12009-03-05 08:32:30 +0000326
Mike Stump58a85142009-03-04 22:48:06 +0000327 Loc = Builder.CreateGEP(BlockLiteral,
Mike Stumpea26cb52009-10-21 03:49:08 +0000328 llvm::ConstantInt::get(llvm::Type::getInt64Ty(VMContext),
Ken Dyck199c3d62010-01-11 17:06:35 +0000329 offset.getQuantity()),
Mike Stump58a85142009-03-04 22:48:06 +0000330 "block.literal");
Owen Anderson96e0fc72009-07-29 22:16:19 +0000331 Ty = llvm::PointerType::get(Ty, 0);
Mike Stumpdab514f2009-03-04 03:23:46 +0000332 Loc = Builder.CreateBitCast(Loc, Ty);
Daniel Dunbar2da84ff2009-11-29 21:23:36 +0000333 Loc = Builder.CreateLoad(Loc);
Mike Stump58a85142009-03-04 22:48:06 +0000334 // Loc = Builder.CreateBitCast(Loc, Ty);
Mike Stumpdab514f2009-03-04 03:23:46 +0000335 }
336 Builder.CreateStore(Loc, Addr);
337 } else if (r.isComplex())
Mike Stump8a2b4b12009-02-25 23:33:13 +0000338 // FIXME: implement
339 ErrorUnsupported(BE, "complex in block literal");
340 else if (r.isAggregate())
341 ; // Already created into the destination
342 else
343 assert (0 && "bad block variable");
344 // FIXME: Ensure that the offset created by the backend for
345 // the struct matches the previously computed offset in BlockDecls.
346 }
Mike Stump08920992009-03-07 02:35:30 +0000347 NoteForHelper.resize(helpersize);
Mike Stumpcf62d392009-03-06 18:42:23 +0000348
349 // __descriptor
Mike Stumpa803b0e2009-03-25 17:58:24 +0000350 llvm::Value *Descriptor = BuildDescriptorBlockDecl(subBlockHasCopyDispose,
351 subBlockSize, Ty,
Mike Stump08920992009-03-07 02:35:30 +0000352 &NoteForHelper);
353 Descriptor = Builder.CreateBitCast(Descriptor, PtrToInt8Ty);
354 Builder.CreateStore(Descriptor, Builder.CreateStructGEP(V, 4, "block.tmp"));
Mike Stumpe5fee252009-02-13 16:19:19 +0000355 }
Mike Stump00470a12009-03-05 08:32:30 +0000356
Mike Stumpbd65cac2009-02-19 01:01:04 +0000357 QualType BPT = BE->getType();
Mike Stump8a2b4b12009-02-25 23:33:13 +0000358 return Builder.CreateBitCast(V, ConvertType(BPT));
Mike Stumpe5fee252009-02-13 16:19:19 +0000359}
360
361
Mike Stump2a998142009-03-04 18:17:45 +0000362const llvm::Type *BlockModule::getBlockDescriptorType() {
Mike Stumpab695142009-02-13 15:16:56 +0000363 if (BlockDescriptorType)
364 return BlockDescriptorType;
365
Mike Stumpa5448542009-02-13 15:32:32 +0000366 const llvm::Type *UnsignedLongTy =
Mike Stumpab695142009-02-13 15:16:56 +0000367 getTypes().ConvertType(getContext().UnsignedLongTy);
Mike Stumpa5448542009-02-13 15:32:32 +0000368
Mike Stumpab695142009-02-13 15:16:56 +0000369 // struct __block_descriptor {
370 // unsigned long reserved;
371 // unsigned long block_size;
372 // };
Owen Anderson47a434f2009-08-05 23:18:46 +0000373 BlockDescriptorType = llvm::StructType::get(UnsignedLongTy->getContext(),
374 UnsignedLongTy,
Mike Stumpa5448542009-02-13 15:32:32 +0000375 UnsignedLongTy,
Mike Stumpab695142009-02-13 15:16:56 +0000376 NULL);
377
378 getModule().addTypeName("struct.__block_descriptor",
379 BlockDescriptorType);
380
381 return BlockDescriptorType;
Anders Carlssonacfde802009-02-12 00:39:25 +0000382}
383
Mike Stump2a998142009-03-04 18:17:45 +0000384const llvm::Type *BlockModule::getGenericBlockLiteralType() {
Mike Stump9b8a7972009-02-13 15:25:34 +0000385 if (GenericBlockLiteralType)
386 return GenericBlockLiteralType;
387
Mike Stumpa5448542009-02-13 15:32:32 +0000388 const llvm::Type *BlockDescPtrTy =
Owen Anderson96e0fc72009-07-29 22:16:19 +0000389 llvm::PointerType::getUnqual(getBlockDescriptorType());
Mike Stumpa5448542009-02-13 15:32:32 +0000390
Mike Stump7cbb3602009-02-13 16:01:35 +0000391 const llvm::IntegerType *IntTy = cast<llvm::IntegerType>(
392 getTypes().ConvertType(getContext().IntTy));
393
Mike Stump9b8a7972009-02-13 15:25:34 +0000394 // struct __block_literal_generic {
Mike Stumpbd65cac2009-02-19 01:01:04 +0000395 // void *__isa;
396 // int __flags;
397 // int __reserved;
398 // void (*__invoke)(void *);
399 // struct __block_descriptor *__descriptor;
David Chisnall5e530af2009-11-17 19:33:30 +0000400 // // GNU runtime only:
401 // const char *types;
Mike Stump9b8a7972009-02-13 15:25:34 +0000402 // };
David Chisnall5e530af2009-11-17 19:33:30 +0000403 if (CGM.getContext().getLangOptions().BlockIntrospection)
404 GenericBlockLiteralType = llvm::StructType::get(IntTy->getContext(),
405 PtrToInt8Ty,
406 IntTy,
407 IntTy,
408 PtrToInt8Ty,
409 BlockDescPtrTy,
410 PtrToInt8Ty,
411 NULL);
412 else
413 GenericBlockLiteralType = llvm::StructType::get(IntTy->getContext(),
Owen Anderson47a434f2009-08-05 23:18:46 +0000414 PtrToInt8Ty,
Mike Stump7cbb3602009-02-13 16:01:35 +0000415 IntTy,
416 IntTy,
Mike Stump797b6322009-03-05 01:23:13 +0000417 PtrToInt8Ty,
Mike Stump9b8a7972009-02-13 15:25:34 +0000418 BlockDescPtrTy,
419 NULL);
Mike Stumpa5448542009-02-13 15:32:32 +0000420
Mike Stump9b8a7972009-02-13 15:25:34 +0000421 getModule().addTypeName("struct.__block_literal_generic",
422 GenericBlockLiteralType);
Mike Stumpa5448542009-02-13 15:32:32 +0000423
Mike Stump9b8a7972009-02-13 15:25:34 +0000424 return GenericBlockLiteralType;
Anders Carlssonacfde802009-02-12 00:39:25 +0000425}
426
Mike Stump2a998142009-03-04 18:17:45 +0000427const llvm::Type *BlockModule::getGenericExtendedBlockLiteralType() {
Mike Stumpbd65cac2009-02-19 01:01:04 +0000428 if (GenericExtendedBlockLiteralType)
429 return GenericExtendedBlockLiteralType;
430
Mike Stumpbd65cac2009-02-19 01:01:04 +0000431 const llvm::Type *BlockDescPtrTy =
Owen Anderson96e0fc72009-07-29 22:16:19 +0000432 llvm::PointerType::getUnqual(getBlockDescriptorType());
Mike Stumpbd65cac2009-02-19 01:01:04 +0000433
434 const llvm::IntegerType *IntTy = cast<llvm::IntegerType>(
435 getTypes().ConvertType(getContext().IntTy));
436
437 // struct __block_literal_generic {
438 // void *__isa;
439 // int __flags;
440 // int __reserved;
441 // void (*__invoke)(void *);
442 // struct __block_descriptor *__descriptor;
443 // void *__copy_func_helper_decl;
444 // void *__destroy_func_decl;
445 // };
Owen Anderson47a434f2009-08-05 23:18:46 +0000446 GenericExtendedBlockLiteralType = llvm::StructType::get(IntTy->getContext(),
447 PtrToInt8Ty,
Mike Stumpbd65cac2009-02-19 01:01:04 +0000448 IntTy,
449 IntTy,
Mike Stump797b6322009-03-05 01:23:13 +0000450 PtrToInt8Ty,
Mike Stumpbd65cac2009-02-19 01:01:04 +0000451 BlockDescPtrTy,
Mike Stump797b6322009-03-05 01:23:13 +0000452 PtrToInt8Ty,
453 PtrToInt8Ty,
Mike Stumpbd65cac2009-02-19 01:01:04 +0000454 NULL);
455
456 getModule().addTypeName("struct.__block_literal_extended_generic",
457 GenericExtendedBlockLiteralType);
458
459 return GenericExtendedBlockLiteralType;
460}
461
Anders Carlssona1736c02009-12-24 21:13:40 +0000462RValue CodeGenFunction::EmitBlockCallExpr(const CallExpr* E,
463 ReturnValueSlot ReturnValue) {
Mike Stumpa5448542009-02-13 15:32:32 +0000464 const BlockPointerType *BPT =
Ted Kremenek6217b802009-07-29 21:53:49 +0000465 E->getCallee()->getType()->getAs<BlockPointerType>();
Mike Stumpa5448542009-02-13 15:32:32 +0000466
Anders Carlssonacfde802009-02-12 00:39:25 +0000467 llvm::Value *Callee = EmitScalarExpr(E->getCallee());
468
469 // Get a pointer to the generic block literal.
470 const llvm::Type *BlockLiteralTy =
Owen Anderson96e0fc72009-07-29 22:16:19 +0000471 llvm::PointerType::getUnqual(CGM.getGenericBlockLiteralType());
Anders Carlssonacfde802009-02-12 00:39:25 +0000472
473 // Bitcast the callee to a block literal.
Mike Stumpa5448542009-02-13 15:32:32 +0000474 llvm::Value *BlockLiteral =
Anders Carlssonacfde802009-02-12 00:39:25 +0000475 Builder.CreateBitCast(Callee, BlockLiteralTy, "block.literal");
476
477 // Get the function pointer from the literal.
478 llvm::Value *FuncPtr = Builder.CreateStructGEP(BlockLiteral, 3, "tmp");
Anders Carlssonacfde802009-02-12 00:39:25 +0000479
Mike Stumpa5448542009-02-13 15:32:32 +0000480 BlockLiteral =
481 Builder.CreateBitCast(BlockLiteral,
Benjamin Kramer3c0ef8c2009-10-13 10:07:13 +0000482 llvm::Type::getInt8PtrTy(VMContext),
Anders Carlssonacfde802009-02-12 00:39:25 +0000483 "tmp");
Mike Stumpa5448542009-02-13 15:32:32 +0000484
Anders Carlssonacfde802009-02-12 00:39:25 +0000485 // Add the block literal.
486 QualType VoidPtrTy = getContext().getPointerType(getContext().VoidTy);
487 CallArgList Args;
488 Args.push_back(std::make_pair(RValue::get(BlockLiteral), VoidPtrTy));
Mike Stumpa5448542009-02-13 15:32:32 +0000489
Anders Carlsson782f3972009-04-08 23:13:16 +0000490 QualType FnType = BPT->getPointeeType();
491
Anders Carlssonacfde802009-02-12 00:39:25 +0000492 // And the rest of the arguments.
John McCall183700f2009-09-21 23:43:11 +0000493 EmitCallArgs(Args, FnType->getAs<FunctionProtoType>(),
Anders Carlsson782f3972009-04-08 23:13:16 +0000494 E->arg_begin(), E->arg_end());
Mike Stumpa5448542009-02-13 15:32:32 +0000495
Anders Carlsson6e460ff2009-04-07 22:10:22 +0000496 // Load the function.
Daniel Dunbar2da84ff2009-11-29 21:23:36 +0000497 llvm::Value *Func = Builder.CreateLoad(FuncPtr, "tmp");
Anders Carlsson6e460ff2009-04-07 22:10:22 +0000498
John McCall183700f2009-09-21 23:43:11 +0000499 QualType ResultType = FnType->getAs<FunctionType>()->getResultType();
Anders Carlssona17d7cc2009-04-08 02:55:55 +0000500
Mike Stump1eb44332009-09-09 15:08:12 +0000501 const CGFunctionInfo &FnInfo =
Anders Carlssona17d7cc2009-04-08 02:55:55 +0000502 CGM.getTypes().getFunctionInfo(ResultType, Args);
Mike Stump1eb44332009-09-09 15:08:12 +0000503
Anders Carlsson6e460ff2009-04-07 22:10:22 +0000504 // Cast the function pointer to the right type.
Mike Stump1eb44332009-09-09 15:08:12 +0000505 const llvm::Type *BlockFTy =
Anders Carlssona17d7cc2009-04-08 02:55:55 +0000506 CGM.getTypes().GetFunctionType(FnInfo, false);
Mike Stump1eb44332009-09-09 15:08:12 +0000507
Owen Anderson96e0fc72009-07-29 22:16:19 +0000508 const llvm::Type *BlockFTyPtr = llvm::PointerType::getUnqual(BlockFTy);
Anders Carlsson6e460ff2009-04-07 22:10:22 +0000509 Func = Builder.CreateBitCast(Func, BlockFTyPtr);
Mike Stump1eb44332009-09-09 15:08:12 +0000510
Anders Carlssonacfde802009-02-12 00:39:25 +0000511 // And call the block.
Anders Carlssona1736c02009-12-24 21:13:40 +0000512 return EmitCall(FnInfo, Func, ReturnValue, Args);
Anders Carlssonacfde802009-02-12 00:39:25 +0000513}
Anders Carlssond5cab542009-02-12 17:55:02 +0000514
Ken Dyck199c3d62010-01-11 17:06:35 +0000515CharUnits CodeGenFunction::AllocateBlockDecl(const BlockDeclRefExpr *E) {
Anders Carlsson7dfa4072009-09-12 02:14:24 +0000516 const ValueDecl *VD = E->getDecl();
Ken Dyck199c3d62010-01-11 17:06:35 +0000517 CharUnits &offset = BlockDecls[VD];
Mike Stumpdab514f2009-03-04 03:23:46 +0000518
Mike Stumpdab514f2009-03-04 03:23:46 +0000519 // See if we have already allocated an offset for this variable.
Ken Dyck199c3d62010-01-11 17:06:35 +0000520 if (offset.isPositive())
Mike Stumpea26cb52009-10-21 03:49:08 +0000521 return offset;
522
523 // Don't run the expensive check, unless we have to.
Mike Stump083c25e2009-10-22 00:49:09 +0000524 if (!BlockHasCopyDispose)
525 if (E->isByRef()
526 || BlockRequiresCopying(E->getType()))
527 BlockHasCopyDispose = true;
Mike Stumpea26cb52009-10-21 03:49:08 +0000528
529 // if not, allocate one now.
530 offset = getBlockOffset(E);
531
532 return offset;
533}
534
535llvm::Value *CodeGenFunction::GetAddrOfBlockDecl(const BlockDeclRefExpr *E) {
536 const ValueDecl *VD = E->getDecl();
Ken Dyck199c3d62010-01-11 17:06:35 +0000537 CharUnits offset = AllocateBlockDecl(E);
Mike Stumpea26cb52009-10-21 03:49:08 +0000538
Mike Stumpdab514f2009-03-04 03:23:46 +0000539
540 llvm::Value *BlockLiteral = LoadBlockStruct();
541 llvm::Value *V = Builder.CreateGEP(BlockLiteral,
Mike Stumpea26cb52009-10-21 03:49:08 +0000542 llvm::ConstantInt::get(llvm::Type::getInt64Ty(VMContext),
Ken Dyck199c3d62010-01-11 17:06:35 +0000543 offset.getQuantity()),
Mike Stump58a85142009-03-04 22:48:06 +0000544 "block.literal");
Mike Stumpdab514f2009-03-04 03:23:46 +0000545 if (E->isByRef()) {
Mike Stumpdab514f2009-03-04 03:23:46 +0000546 const llvm::Type *PtrStructTy
Anders Carlsson7dfa4072009-09-12 02:14:24 +0000547 = llvm::PointerType::get(BuildByRefType(VD), 0);
Mike Stumpa8b60c92009-03-21 21:00:35 +0000548 // The block literal will need a copy/destroy helper.
549 BlockHasCopyDispose = true;
Anders Carlsson7dfa4072009-09-12 02:14:24 +0000550
551 const llvm::Type *Ty = PtrStructTy;
Owen Anderson96e0fc72009-07-29 22:16:19 +0000552 Ty = llvm::PointerType::get(Ty, 0);
Mike Stumpdab514f2009-03-04 03:23:46 +0000553 V = Builder.CreateBitCast(V, Ty);
Daniel Dunbar2da84ff2009-11-29 21:23:36 +0000554 V = Builder.CreateLoad(V);
Mike Stumpdab514f2009-03-04 03:23:46 +0000555 V = Builder.CreateStructGEP(V, 1, "forwarding");
Daniel Dunbar2da84ff2009-11-29 21:23:36 +0000556 V = Builder.CreateLoad(V);
Mike Stumpdab514f2009-03-04 03:23:46 +0000557 V = Builder.CreateBitCast(V, PtrStructTy);
Anders Carlsson7dfa4072009-09-12 02:14:24 +0000558 V = Builder.CreateStructGEP(V, getByRefValueLLVMField(VD),
559 VD->getNameAsString());
Mike Stumpdab514f2009-03-04 03:23:46 +0000560 } else {
Anders Carlsson7dfa4072009-09-12 02:14:24 +0000561 const llvm::Type *Ty = CGM.getTypes().ConvertType(VD->getType());
562
Owen Anderson96e0fc72009-07-29 22:16:19 +0000563 Ty = llvm::PointerType::get(Ty, 0);
Mike Stumpdab514f2009-03-04 03:23:46 +0000564 V = Builder.CreateBitCast(V, Ty);
565 }
566 return V;
567}
568
Mike Stump6cc88f72009-03-20 21:53:12 +0000569void CodeGenFunction::BlockForwardSelf() {
570 const ObjCMethodDecl *OMD = cast<ObjCMethodDecl>(CurFuncDecl);
571 ImplicitParamDecl *SelfDecl = OMD->getSelfDecl();
572 llvm::Value *&DMEntry = LocalDeclMap[SelfDecl];
573 if (DMEntry)
574 return;
575 // FIXME - Eliminate BlockDeclRefExprs, clients don't need/want to care
576 BlockDeclRefExpr *BDRE = new (getContext())
577 BlockDeclRefExpr(SelfDecl,
578 SelfDecl->getType(), SourceLocation(), false);
579 DMEntry = GetAddrOfBlockDecl(BDRE);
580}
581
Mike Stump67a64482009-02-14 22:16:35 +0000582llvm::Constant *
Mike Stump90a90432009-03-04 18:47:42 +0000583BlockModule::GetAddrOfGlobalBlock(const BlockExpr *BE, const char * n) {
Anders Carlssond5cab542009-02-12 17:55:02 +0000584 // Generate the block descriptor.
585 const llvm::Type *UnsignedLongTy = Types.ConvertType(Context.UnsignedLongTy);
Mike Stump7cbb3602009-02-13 16:01:35 +0000586 const llvm::IntegerType *IntTy = cast<llvm::IntegerType>(
587 getTypes().ConvertType(getContext().IntTy));
Mike Stumpa5448542009-02-13 15:32:32 +0000588
Anders Carlssond5cab542009-02-12 17:55:02 +0000589 llvm::Constant *DescriptorFields[2];
Mike Stumpa5448542009-02-13 15:32:32 +0000590
Anders Carlssond5cab542009-02-12 17:55:02 +0000591 // Reserved
Owen Andersonc9c88b42009-07-31 20:28:54 +0000592 DescriptorFields[0] = llvm::Constant::getNullValue(UnsignedLongTy);
Mike Stumpa5448542009-02-13 15:32:32 +0000593
Anders Carlssond5cab542009-02-12 17:55:02 +0000594 // Block literal size. For global blocks we just use the size of the generic
595 // block literal struct.
Ken Dyck687cc4a2010-01-26 13:48:07 +0000596 CharUnits BlockLiteralSize =
597 CGM.GetTargetTypeStoreSize(getGenericBlockLiteralType());
Owen Andersona1cf15f2009-07-14 23:10:40 +0000598 DescriptorFields[1] =
Ken Dyck199c3d62010-01-11 17:06:35 +0000599 llvm::ConstantInt::get(UnsignedLongTy,BlockLiteralSize.getQuantity());
Mike Stumpa5448542009-02-13 15:32:32 +0000600
601 llvm::Constant *DescriptorStruct =
Nick Lewycky0d36dd22009-09-19 20:00:52 +0000602 llvm::ConstantStruct::get(VMContext, &DescriptorFields[0], 2, false);
Mike Stumpa5448542009-02-13 15:32:32 +0000603
Anders Carlssond5cab542009-02-12 17:55:02 +0000604 llvm::GlobalVariable *Descriptor =
Owen Anderson1c431b32009-07-08 19:05:04 +0000605 new llvm::GlobalVariable(getModule(), DescriptorStruct->getType(), true,
Mike Stumpa5448542009-02-13 15:32:32 +0000606 llvm::GlobalVariable::InternalLinkage,
Owen Anderson1c431b32009-07-08 19:05:04 +0000607 DescriptorStruct, "__block_descriptor_global");
Mike Stumpa5448542009-02-13 15:32:32 +0000608
David Chisnall5e530af2009-11-17 19:33:30 +0000609 int FieldCount = 5;
Anders Carlssond5cab542009-02-12 17:55:02 +0000610 // Generate the constants for the block literal.
David Chisnall5e530af2009-11-17 19:33:30 +0000611 if (CGM.getContext().getLangOptions().BlockIntrospection)
612 FieldCount = 6;
613
614 std::vector<llvm::Constant*> LiteralFields(FieldCount);
Mike Stumpa5448542009-02-13 15:32:32 +0000615
Mike Stump67a64482009-02-14 22:16:35 +0000616 CodeGenFunction::BlockInfo Info(0, n);
Ken Dyck199c3d62010-01-11 17:06:35 +0000617 CharUnits subBlockSize;
618 uint64_t subBlockAlign;
Mike Stumpa99038c2009-02-28 09:07:16 +0000619 llvm::SmallVector<const Expr *, 8> subBlockDeclRefDecls;
Daniel Dunbard67b09a2009-03-12 03:07:24 +0000620 bool subBlockHasCopyDispose = false;
Mike Stump7f28a9c2009-03-13 23:34:28 +0000621 llvm::DenseMap<const Decl*, llvm::Value*> LocalDeclMap;
Mike Stump4e7a1f72009-02-21 20:00:35 +0000622 llvm::Function *Fn
Mike Stump6cc88f72009-03-20 21:53:12 +0000623 = CodeGenFunction(CGM).GenerateBlockFunction(BE, Info, 0, LocalDeclMap,
Mike Stump7f28a9c2009-03-13 23:34:28 +0000624 subBlockSize,
Mike Stump90a90432009-03-04 18:47:42 +0000625 subBlockAlign,
Mike Stump00470a12009-03-05 08:32:30 +0000626 subBlockDeclRefDecls,
627 subBlockHasCopyDispose);
Mike Stump4e7a1f72009-02-21 20:00:35 +0000628 assert(subBlockSize == BlockLiteralSize
629 && "no imports allowed for global block");
Mike Stumpa5448542009-02-13 15:32:32 +0000630
Anders Carlssond5cab542009-02-12 17:55:02 +0000631 // isa
Mike Stumpf99f1d02009-02-13 17:23:42 +0000632 LiteralFields[0] = getNSConcreteGlobalBlock();
Mike Stumpa5448542009-02-13 15:32:32 +0000633
Anders Carlssond5cab542009-02-12 17:55:02 +0000634 // Flags
David Chisnall5e530af2009-11-17 19:33:30 +0000635 LiteralFields[1] = CGM.getContext().getLangOptions().BlockIntrospection ?
636 llvm::ConstantInt::get(IntTy, BLOCK_IS_GLOBAL | BLOCK_HAS_DESCRIPTOR |
637 BLOCK_HAS_OBJC_TYPE) :
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000638 llvm::ConstantInt::get(IntTy, BLOCK_IS_GLOBAL | BLOCK_HAS_DESCRIPTOR);
Mike Stumpa5448542009-02-13 15:32:32 +0000639
Anders Carlssond5cab542009-02-12 17:55:02 +0000640 // Reserved
Owen Andersonc9c88b42009-07-31 20:28:54 +0000641 LiteralFields[2] = llvm::Constant::getNullValue(IntTy);
Mike Stumpa5448542009-02-13 15:32:32 +0000642
Anders Carlssond5cab542009-02-12 17:55:02 +0000643 // Function
644 LiteralFields[3] = Fn;
Mike Stumpa5448542009-02-13 15:32:32 +0000645
Anders Carlssond5cab542009-02-12 17:55:02 +0000646 // Descriptor
647 LiteralFields[4] = Descriptor;
David Chisnall5e530af2009-11-17 19:33:30 +0000648
649 // Type encoding
650 if (CGM.getContext().getLangOptions().BlockIntrospection) {
651 std::string BlockTypeEncoding;
652 CGM.getContext().getObjCEncodingForBlock(BE, BlockTypeEncoding);
653
654 LiteralFields[5] = CGM.GetAddrOfConstantCString(BlockTypeEncoding);
655 }
Mike Stumpa5448542009-02-13 15:32:32 +0000656
657 llvm::Constant *BlockLiteralStruct =
David Chisnall5e530af2009-11-17 19:33:30 +0000658 llvm::ConstantStruct::get(VMContext, LiteralFields, false);
Mike Stumpa5448542009-02-13 15:32:32 +0000659
660 llvm::GlobalVariable *BlockLiteral =
Owen Anderson1c431b32009-07-08 19:05:04 +0000661 new llvm::GlobalVariable(getModule(), BlockLiteralStruct->getType(), true,
Mike Stumpa5448542009-02-13 15:32:32 +0000662 llvm::GlobalVariable::InternalLinkage,
Owen Anderson1c431b32009-07-08 19:05:04 +0000663 BlockLiteralStruct, "__block_literal_global");
Mike Stumpa5448542009-02-13 15:32:32 +0000664
Anders Carlssond5cab542009-02-12 17:55:02 +0000665 return BlockLiteral;
666}
667
Mike Stump4e7a1f72009-02-21 20:00:35 +0000668llvm::Value *CodeGenFunction::LoadBlockStruct() {
Mike Stumpbf1914b2009-10-20 20:30:01 +0000669 llvm::Value *V = Builder.CreateLoad(LocalDeclMap[getBlockStructDecl()],
670 "self");
671 // For now, we codegen based upon byte offsets.
672 return Builder.CreateBitCast(V, PtrToInt8Ty);
Mike Stump4e7a1f72009-02-21 20:00:35 +0000673}
674
Mike Stump00470a12009-03-05 08:32:30 +0000675llvm::Function *
676CodeGenFunction::GenerateBlockFunction(const BlockExpr *BExpr,
677 const BlockInfo& Info,
Mike Stump6cc88f72009-03-20 21:53:12 +0000678 const Decl *OuterFuncDecl,
Mike Stump7f28a9c2009-03-13 23:34:28 +0000679 llvm::DenseMap<const Decl*, llvm::Value*> ldm,
Ken Dyck199c3d62010-01-11 17:06:35 +0000680 CharUnits &Size,
Mike Stump00470a12009-03-05 08:32:30 +0000681 uint64_t &Align,
682 llvm::SmallVector<const Expr *, 8> &subBlockDeclRefDecls,
683 bool &subBlockHasCopyDispose) {
Devang Patel963dfbd2009-04-15 21:51:44 +0000684
685 // Check if we should generate debug info for this block.
686 if (CGM.getDebugInfo())
687 DebugInfo = CGM.getDebugInfo();
Mike Stump1eb44332009-09-09 15:08:12 +0000688
Mike Stump7f28a9c2009-03-13 23:34:28 +0000689 // Arrange for local static and local extern declarations to appear
690 // to be local to this function as well, as they are directly referenced
691 // in a block.
692 for (llvm::DenseMap<const Decl *, llvm::Value*>::iterator i = ldm.begin();
693 i != ldm.end();
694 ++i) {
695 const VarDecl *VD = dyn_cast<VarDecl>(i->first);
Mike Stump1eb44332009-09-09 15:08:12 +0000696
Daniel Dunbar5466c7b2009-04-14 02:25:56 +0000697 if (VD->getStorageClass() == VarDecl::Static || VD->hasExternalStorage())
Mike Stump7f28a9c2009-03-13 23:34:28 +0000698 LocalDeclMap[VD] = i->second;
699 }
700
Ken Dyck687cc4a2010-01-26 13:48:07 +0000701 BlockOffset =
702 CGM.GetTargetTypeStoreSize(CGM.getGenericBlockLiteralType());
Eli Friedman48f91222009-03-28 03:24:54 +0000703 BlockAlign = getContext().getTypeAlign(getContext().VoidPtrTy) / 8;
704
Fariborz Jahanian140fb262009-04-11 17:55:15 +0000705 const FunctionType *BlockFunctionType = BExpr->getFunctionType();
706 QualType ResultType;
707 bool IsVariadic;
Mike Stump1eb44332009-09-09 15:08:12 +0000708 if (const FunctionProtoType *FTy =
Fariborz Jahanianda0895d2009-04-11 18:54:57 +0000709 dyn_cast<FunctionProtoType>(BlockFunctionType)) {
Fariborz Jahanian140fb262009-04-11 17:55:15 +0000710 ResultType = FTy->getResultType();
711 IsVariadic = FTy->isVariadic();
Mike Stumpb3589f42009-07-30 22:28:39 +0000712 } else {
Fariborz Jahanian140fb262009-04-11 17:55:15 +0000713 // K&R style block.
714 ResultType = BlockFunctionType->getResultType();
715 IsVariadic = false;
716 }
Mike Stumpa5448542009-02-13 15:32:32 +0000717
Anders Carlssond5cab542009-02-12 17:55:02 +0000718 FunctionArgList Args;
Mike Stumpa5448542009-02-13 15:32:32 +0000719
Mike Stumpea26cb52009-10-21 03:49:08 +0000720 CurFuncDecl = OuterFuncDecl;
721
Chris Lattner161d36d2009-02-28 19:01:03 +0000722 const BlockDecl *BD = BExpr->getBlockDecl();
Anders Carlssond5cab542009-02-12 17:55:02 +0000723
Mike Stumpea26cb52009-10-21 03:49:08 +0000724 IdentifierInfo *II = &CGM.getContext().Idents.get(".block_descriptor");
Mike Stumpadaaad32009-10-20 02:12:22 +0000725
Mike Stumpbfbd5df2009-10-21 18:24:18 +0000726 // Allocate all BlockDeclRefDecls, so we can calculate the right ParmTy below.
Mike Stump0298d382009-10-21 22:02:08 +0000727 AllocateAllBlockDeclRefs(Info, this);
Mike Stumpea26cb52009-10-21 03:49:08 +0000728
Mike Stump083c25e2009-10-22 00:49:09 +0000729 QualType ParmTy = getContext().getBlockParmType(BlockHasCopyDispose,
730 BlockDeclRefDecls);
Anders Carlssond5cab542009-02-12 17:55:02 +0000731 // FIXME: This leaks
Mike Stumpa5448542009-02-13 15:32:32 +0000732 ImplicitParamDecl *SelfDecl =
Anders Carlssond5cab542009-02-12 17:55:02 +0000733 ImplicitParamDecl::Create(getContext(), 0,
Mike Stumpadaaad32009-10-20 02:12:22 +0000734 SourceLocation(), II,
735 ParmTy);
Mike Stumpa5448542009-02-13 15:32:32 +0000736
Anders Carlssond5cab542009-02-12 17:55:02 +0000737 Args.push_back(std::make_pair(SelfDecl, SelfDecl->getType()));
Mike Stump4e7a1f72009-02-21 20:00:35 +0000738 BlockStructDecl = SelfDecl;
Mike Stumpa5448542009-02-13 15:32:32 +0000739
Steve Naroffe78b8092009-03-13 16:56:44 +0000740 for (BlockDecl::param_const_iterator i = BD->param_begin(),
Anders Carlssond5cab542009-02-12 17:55:02 +0000741 e = BD->param_end(); i != e; ++i)
Mike Stump19050612009-02-17 23:25:52 +0000742 Args.push_back(std::make_pair(*i, (*i)->getType()));
Mike Stumpa5448542009-02-13 15:32:32 +0000743
744 const CGFunctionInfo &FI =
Fariborz Jahanian140fb262009-04-11 17:55:15 +0000745 CGM.getTypes().getFunctionInfo(ResultType, Args);
Anders Carlssond5cab542009-02-12 17:55:02 +0000746
Anders Carlssond5cab542009-02-12 17:55:02 +0000747 CodeGenTypes &Types = CGM.getTypes();
Fariborz Jahanian140fb262009-04-11 17:55:15 +0000748 const llvm::FunctionType *LTy = Types.GetFunctionType(FI, IsVariadic);
Mike Stumpa5448542009-02-13 15:32:32 +0000749
750 llvm::Function *Fn =
Anders Carlssond5cab542009-02-12 17:55:02 +0000751 llvm::Function::Create(LTy, llvm::GlobalValue::InternalLinkage,
Benjamin Kramer3cf7c5d2010-01-22 13:59:13 +0000752 llvm::Twine("__") + Info.Name + "_block_invoke_",
Anders Carlssond5cab542009-02-12 17:55:02 +0000753 &CGM.getModule());
Mike Stumpa5448542009-02-13 15:32:32 +0000754
Daniel Dunbar0e4f40e2009-04-17 00:48:04 +0000755 CGM.SetInternalFunctionAttributes(BD, Fn, FI);
756
Chris Lattner4863db42009-04-23 07:18:56 +0000757 StartFunction(BD, ResultType, Fn, Args,
Chris Lattner161d36d2009-02-28 19:01:03 +0000758 BExpr->getBody()->getLocEnd());
Mike Stumpb1a6e682009-09-30 02:43:10 +0000759
Chris Lattner4863db42009-04-23 07:18:56 +0000760 CurFuncDecl = OuterFuncDecl;
Chris Lattnerb5437d22009-04-23 05:30:27 +0000761 CurCodeDecl = BD;
Mike Stumpb289b3f2009-10-01 22:29:41 +0000762
763 // Save a spot to insert the debug information for all the BlockDeclRefDecls.
764 llvm::BasicBlock *entry = Builder.GetInsertBlock();
765 llvm::BasicBlock::iterator entry_ptr = Builder.GetInsertPoint();
766 --entry_ptr;
767
Chris Lattner161d36d2009-02-28 19:01:03 +0000768 EmitStmt(BExpr->getBody());
Mike Stumpb289b3f2009-10-01 22:29:41 +0000769
Mike Stumpde8c5c72009-10-01 00:27:30 +0000770 // Remember where we were...
771 llvm::BasicBlock *resume = Builder.GetInsertBlock();
Mike Stumpb289b3f2009-10-01 22:29:41 +0000772
Mike Stumpde8c5c72009-10-01 00:27:30 +0000773 // Go back to the entry.
Mike Stumpb289b3f2009-10-01 22:29:41 +0000774 ++entry_ptr;
775 Builder.SetInsertPoint(entry, entry_ptr);
776
Mike Stumpb1a6e682009-09-30 02:43:10 +0000777 if (CGDebugInfo *DI = getDebugInfo()) {
Mike Stumpb1a6e682009-09-30 02:43:10 +0000778 // Emit debug information for all the BlockDeclRefDecls.
Chris Lattner14b1a362010-01-25 03:29:35 +0000779 for (unsigned i = 0, e = BlockDeclRefDecls.size(); i != e; ++i) {
780 if (const BlockDeclRefExpr *BDRE =
781 dyn_cast<BlockDeclRefExpr>(BlockDeclRefDecls[i])) {
Mike Stumpb1a6e682009-09-30 02:43:10 +0000782 const ValueDecl *D = BDRE->getDecl();
783 DI->setLocation(D->getLocation());
784 DI->EmitDeclareOfBlockDeclRefVariable(BDRE,
785 LocalDeclMap[getBlockStructDecl()],
786 Builder, this);
787 }
788 }
Mike Stumpb1a6e682009-09-30 02:43:10 +0000789 }
Mike Stumpde8c5c72009-10-01 00:27:30 +0000790 // And resume where we left off.
791 if (resume == 0)
792 Builder.ClearInsertionPoint();
793 else
794 Builder.SetInsertPoint(resume);
Mike Stumpb1a6e682009-09-30 02:43:10 +0000795
Chris Lattner161d36d2009-02-28 19:01:03 +0000796 FinishFunction(cast<CompoundStmt>(BExpr->getBody())->getRBracLoc());
Anders Carlssond5cab542009-02-12 17:55:02 +0000797
Mike Stump8a2b4b12009-02-25 23:33:13 +0000798 // The runtime needs a minimum alignment of a void *.
799 uint64_t MinAlign = getContext().getTypeAlign(getContext().VoidPtrTy) / 8;
Ken Dyck199c3d62010-01-11 17:06:35 +0000800 BlockOffset = CharUnits::fromQuantity(
801 llvm::RoundUpToAlignment(BlockOffset.getQuantity(), MinAlign));
Mike Stump8a2b4b12009-02-25 23:33:13 +0000802
Mike Stump4e7a1f72009-02-21 20:00:35 +0000803 Size = BlockOffset;
Mike Stump8a2b4b12009-02-25 23:33:13 +0000804 Align = BlockAlign;
805 subBlockDeclRefDecls = BlockDeclRefDecls;
Mike Stump00470a12009-03-05 08:32:30 +0000806 subBlockHasCopyDispose |= BlockHasCopyDispose;
Anders Carlssond5cab542009-02-12 17:55:02 +0000807 return Fn;
808}
Mike Stumpa99038c2009-02-28 09:07:16 +0000809
Ken Dyck199c3d62010-01-11 17:06:35 +0000810CharUnits BlockFunction::getBlockOffset(const BlockDeclRefExpr *BDRE) {
Mike Stumpa99038c2009-02-28 09:07:16 +0000811 const ValueDecl *D = dyn_cast<ValueDecl>(BDRE->getDecl());
812
Ken Dyck199c3d62010-01-11 17:06:35 +0000813 CharUnits Size = getContext().getTypeSizeInChars(D->getType());
Mike Stumpa99038c2009-02-28 09:07:16 +0000814 uint64_t Align = getContext().getDeclAlignInBytes(D);
815
816 if (BDRE->isByRef()) {
Ken Dyck199c3d62010-01-11 17:06:35 +0000817 Size = getContext().getTypeSizeInChars(getContext().VoidPtrTy);
Mike Stumpa99038c2009-02-28 09:07:16 +0000818 Align = getContext().getTypeAlign(getContext().VoidPtrTy) / 8;
819 }
820
821 assert ((Align > 0) && "alignment must be 1 byte or more");
822
Ken Dyck199c3d62010-01-11 17:06:35 +0000823 CharUnits OldOffset = BlockOffset;
Mike Stumpa99038c2009-02-28 09:07:16 +0000824
825 // Ensure proper alignment, even if it means we have to have a gap
Ken Dyck199c3d62010-01-11 17:06:35 +0000826 BlockOffset = CharUnits::fromQuantity(
827 llvm::RoundUpToAlignment(BlockOffset.getQuantity(), Align));
Mike Stumpa99038c2009-02-28 09:07:16 +0000828 BlockAlign = std::max(Align, BlockAlign);
Mike Stump00470a12009-03-05 08:32:30 +0000829
Ken Dyck199c3d62010-01-11 17:06:35 +0000830 CharUnits Pad = BlockOffset - OldOffset;
831 if (Pad.isPositive()) {
832 llvm::ArrayType::get(llvm::Type::getInt8Ty(VMContext), Pad.getQuantity());
Mike Stumpa99038c2009-02-28 09:07:16 +0000833 QualType PadTy = getContext().getConstantArrayType(getContext().CharTy,
Ken Dyck199c3d62010-01-11 17:06:35 +0000834 llvm::APInt(32,
835 Pad.getQuantity()),
Mike Stumpa99038c2009-02-28 09:07:16 +0000836 ArrayType::Normal, 0);
837 ValueDecl *PadDecl = VarDecl::Create(getContext(), 0, SourceLocation(),
Argyrios Kyrtzidisa5d82002009-08-21 00:31:54 +0000838 0, QualType(PadTy), 0, VarDecl::None);
Mike Stumpa99038c2009-02-28 09:07:16 +0000839 Expr *E;
840 E = new (getContext()) DeclRefExpr(PadDecl, PadDecl->getType(),
Douglas Gregor0da76df2009-11-23 11:41:28 +0000841 SourceLocation());
Mike Stumpa99038c2009-02-28 09:07:16 +0000842 BlockDeclRefDecls.push_back(E);
843 }
844 BlockDeclRefDecls.push_back(BDRE);
845
846 BlockOffset += Size;
847 return BlockOffset-Size;
848}
Mike Stumpdab514f2009-03-04 03:23:46 +0000849
Mike Stump08920992009-03-07 02:35:30 +0000850llvm::Constant *BlockFunction::
851GenerateCopyHelperFunction(bool BlockHasCopyDispose, const llvm::StructType *T,
Mike Stumpb7477cf2009-04-10 18:52:28 +0000852 std::vector<HelperInfo> *NoteForHelperp) {
Mike Stumpa4f668f2009-03-06 01:33:24 +0000853 QualType R = getContext().VoidTy;
854
855 FunctionArgList Args;
856 // FIXME: This leaks
Mike Stump08920992009-03-07 02:35:30 +0000857 ImplicitParamDecl *Dst =
Mike Stumpea26cb52009-10-21 03:49:08 +0000858 ImplicitParamDecl::Create(getContext(), 0,
859 SourceLocation(), 0,
Mike Stump08920992009-03-07 02:35:30 +0000860 getContext().getPointerType(getContext().VoidTy));
861 Args.push_back(std::make_pair(Dst, Dst->getType()));
Mike Stumpa4f668f2009-03-06 01:33:24 +0000862 ImplicitParamDecl *Src =
Mike Stumpea26cb52009-10-21 03:49:08 +0000863 ImplicitParamDecl::Create(getContext(), 0,
864 SourceLocation(), 0,
Mike Stumpa4f668f2009-03-06 01:33:24 +0000865 getContext().getPointerType(getContext().VoidTy));
Mike Stumpa4f668f2009-03-06 01:33:24 +0000866 Args.push_back(std::make_pair(Src, Src->getType()));
Mike Stump1eb44332009-09-09 15:08:12 +0000867
Mike Stumpa4f668f2009-03-06 01:33:24 +0000868 const CGFunctionInfo &FI =
869 CGM.getTypes().getFunctionInfo(R, Args);
870
Mike Stump3899a7f2009-06-05 23:26:36 +0000871 // FIXME: We'd like to put these into a mergable by content, with
872 // internal linkage.
Mike Stumpa4f668f2009-03-06 01:33:24 +0000873 CodeGenTypes &Types = CGM.getTypes();
874 const llvm::FunctionType *LTy = Types.GetFunctionType(FI, false);
875
876 llvm::Function *Fn =
877 llvm::Function::Create(LTy, llvm::GlobalValue::InternalLinkage,
Benjamin Kramer3cf7c5d2010-01-22 13:59:13 +0000878 "__copy_helper_block_", &CGM.getModule());
Mike Stumpa4f668f2009-03-06 01:33:24 +0000879
880 IdentifierInfo *II
881 = &CGM.getContext().Idents.get("__copy_helper_block_");
882
883 FunctionDecl *FD = FunctionDecl::Create(getContext(),
884 getContext().getTranslationUnitDecl(),
Argyrios Kyrtzidisa1d56622009-08-19 01:27:57 +0000885 SourceLocation(), II, R, 0,
Mike Stumpa4f668f2009-03-06 01:33:24 +0000886 FunctionDecl::Static, false,
887 true);
888 CGF.StartFunction(FD, R, Fn, Args, SourceLocation());
Mike Stump08920992009-03-07 02:35:30 +0000889
890 llvm::Value *SrcObj = CGF.GetAddrOfLocalVar(Src);
891 llvm::Type *PtrPtrT;
Mike Stump08920992009-03-07 02:35:30 +0000892
Mike Stumpb7477cf2009-04-10 18:52:28 +0000893 if (NoteForHelperp) {
894 std::vector<HelperInfo> &NoteForHelper = *NoteForHelperp;
Mike Stump08920992009-03-07 02:35:30 +0000895
Owen Anderson96e0fc72009-07-29 22:16:19 +0000896 PtrPtrT = llvm::PointerType::get(llvm::PointerType::get(T, 0), 0);
Mike Stumpb7477cf2009-04-10 18:52:28 +0000897 SrcObj = Builder.CreateBitCast(SrcObj, PtrPtrT);
898 SrcObj = Builder.CreateLoad(SrcObj);
Mike Stump08920992009-03-07 02:35:30 +0000899
Mike Stumpb7477cf2009-04-10 18:52:28 +0000900 llvm::Value *DstObj = CGF.GetAddrOfLocalVar(Dst);
Mike Stumpc2f4c342009-04-15 22:11:36 +0000901 llvm::Type *PtrPtrT;
Owen Anderson96e0fc72009-07-29 22:16:19 +0000902 PtrPtrT = llvm::PointerType::get(llvm::PointerType::get(T, 0), 0);
Mike Stumpc2f4c342009-04-15 22:11:36 +0000903 DstObj = Builder.CreateBitCast(DstObj, PtrPtrT);
904 DstObj = Builder.CreateLoad(DstObj);
Mike Stump08920992009-03-07 02:35:30 +0000905
Mike Stumpb7477cf2009-04-10 18:52:28 +0000906 for (unsigned i=0; i < NoteForHelper.size(); ++i) {
907 int flag = NoteForHelper[i].flag;
908 int index = NoteForHelper[i].index;
Mike Stump08920992009-03-07 02:35:30 +0000909
Mike Stumpb7477cf2009-04-10 18:52:28 +0000910 if ((NoteForHelper[i].flag & BLOCK_FIELD_IS_BYREF)
911 || NoteForHelper[i].RequiresCopying) {
912 llvm::Value *Srcv = SrcObj;
913 Srcv = Builder.CreateStructGEP(Srcv, index);
914 Srcv = Builder.CreateBitCast(Srcv,
Owen Anderson96e0fc72009-07-29 22:16:19 +0000915 llvm::PointerType::get(PtrToInt8Ty, 0));
Mike Stumpb7477cf2009-04-10 18:52:28 +0000916 Srcv = Builder.CreateLoad(Srcv);
917
918 llvm::Value *Dstv = Builder.CreateStructGEP(DstObj, index);
919 Dstv = Builder.CreateBitCast(Dstv, PtrToInt8Ty);
920
Owen Anderson0032b272009-08-13 21:57:51 +0000921 llvm::Value *N = llvm::ConstantInt::get(
922 llvm::Type::getInt32Ty(T->getContext()), flag);
Mike Stumpb7477cf2009-04-10 18:52:28 +0000923 llvm::Value *F = getBlockObjectAssign();
924 Builder.CreateCall3(F, Dstv, Srcv, N);
925 }
Mike Stump08920992009-03-07 02:35:30 +0000926 }
927 }
928
Mike Stumpa4f668f2009-03-06 01:33:24 +0000929 CGF.FinishFunction();
930
Owen Anderson3c4972d2009-07-29 18:54:39 +0000931 return llvm::ConstantExpr::getBitCast(Fn, PtrToInt8Ty);
Mike Stumpdab514f2009-03-04 03:23:46 +0000932}
933
Mike Stumpcf62d392009-03-06 18:42:23 +0000934llvm::Constant *BlockFunction::
Mike Stump08920992009-03-07 02:35:30 +0000935GenerateDestroyHelperFunction(bool BlockHasCopyDispose,
936 const llvm::StructType* T,
Mike Stumpb7477cf2009-04-10 18:52:28 +0000937 std::vector<HelperInfo> *NoteForHelperp) {
Mike Stumpa4f668f2009-03-06 01:33:24 +0000938 QualType R = getContext().VoidTy;
939
940 FunctionArgList Args;
941 // FIXME: This leaks
942 ImplicitParamDecl *Src =
Mike Stumpea26cb52009-10-21 03:49:08 +0000943 ImplicitParamDecl::Create(getContext(), 0,
944 SourceLocation(), 0,
Mike Stumpa4f668f2009-03-06 01:33:24 +0000945 getContext().getPointerType(getContext().VoidTy));
946
947 Args.push_back(std::make_pair(Src, Src->getType()));
Mike Stump1eb44332009-09-09 15:08:12 +0000948
Mike Stumpa4f668f2009-03-06 01:33:24 +0000949 const CGFunctionInfo &FI =
950 CGM.getTypes().getFunctionInfo(R, Args);
951
Mike Stump3899a7f2009-06-05 23:26:36 +0000952 // FIXME: We'd like to put these into a mergable by content, with
953 // internal linkage.
Mike Stumpa4f668f2009-03-06 01:33:24 +0000954 CodeGenTypes &Types = CGM.getTypes();
955 const llvm::FunctionType *LTy = Types.GetFunctionType(FI, false);
956
957 llvm::Function *Fn =
958 llvm::Function::Create(LTy, llvm::GlobalValue::InternalLinkage,
Benjamin Kramer3cf7c5d2010-01-22 13:59:13 +0000959 "__destroy_helper_block_", &CGM.getModule());
Mike Stumpa4f668f2009-03-06 01:33:24 +0000960
961 IdentifierInfo *II
962 = &CGM.getContext().Idents.get("__destroy_helper_block_");
963
964 FunctionDecl *FD = FunctionDecl::Create(getContext(),
965 getContext().getTranslationUnitDecl(),
Argyrios Kyrtzidisa1d56622009-08-19 01:27:57 +0000966 SourceLocation(), II, R, 0,
Mike Stumpa4f668f2009-03-06 01:33:24 +0000967 FunctionDecl::Static, false,
968 true);
969 CGF.StartFunction(FD, R, Fn, Args, SourceLocation());
Mike Stump1edf6b62009-03-07 02:53:18 +0000970
Mike Stumpb7477cf2009-04-10 18:52:28 +0000971 if (NoteForHelperp) {
972 std::vector<HelperInfo> &NoteForHelper = *NoteForHelperp;
Mike Stump1edf6b62009-03-07 02:53:18 +0000973
Mike Stumpb7477cf2009-04-10 18:52:28 +0000974 llvm::Value *SrcObj = CGF.GetAddrOfLocalVar(Src);
975 llvm::Type *PtrPtrT;
Owen Anderson96e0fc72009-07-29 22:16:19 +0000976 PtrPtrT = llvm::PointerType::get(llvm::PointerType::get(T, 0), 0);
Mike Stumpb7477cf2009-04-10 18:52:28 +0000977 SrcObj = Builder.CreateBitCast(SrcObj, PtrPtrT);
978 SrcObj = Builder.CreateLoad(SrcObj);
Mike Stump1edf6b62009-03-07 02:53:18 +0000979
Mike Stumpb7477cf2009-04-10 18:52:28 +0000980 for (unsigned i=0; i < NoteForHelper.size(); ++i) {
981 int flag = NoteForHelper[i].flag;
982 int index = NoteForHelper[i].index;
Mike Stump1edf6b62009-03-07 02:53:18 +0000983
Mike Stumpb7477cf2009-04-10 18:52:28 +0000984 if ((NoteForHelper[i].flag & BLOCK_FIELD_IS_BYREF)
985 || NoteForHelper[i].RequiresCopying) {
986 llvm::Value *Srcv = SrcObj;
987 Srcv = Builder.CreateStructGEP(Srcv, index);
988 Srcv = Builder.CreateBitCast(Srcv,
Owen Anderson96e0fc72009-07-29 22:16:19 +0000989 llvm::PointerType::get(PtrToInt8Ty, 0));
Mike Stumpb7477cf2009-04-10 18:52:28 +0000990 Srcv = Builder.CreateLoad(Srcv);
991
992 BuildBlockRelease(Srcv, flag);
993 }
Mike Stump1edf6b62009-03-07 02:53:18 +0000994 }
995 }
996
Mike Stumpa4f668f2009-03-06 01:33:24 +0000997 CGF.FinishFunction();
998
Owen Anderson3c4972d2009-07-29 18:54:39 +0000999 return llvm::ConstantExpr::getBitCast(Fn, PtrToInt8Ty);
Mike Stumpa4f668f2009-03-06 01:33:24 +00001000}
1001
Mike Stump08920992009-03-07 02:35:30 +00001002llvm::Constant *BlockFunction::BuildCopyHelper(const llvm::StructType *T,
Mike Stumpb7477cf2009-04-10 18:52:28 +00001003 std::vector<HelperInfo> *NoteForHelper) {
Mike Stump08920992009-03-07 02:35:30 +00001004 return CodeGenFunction(CGM).GenerateCopyHelperFunction(BlockHasCopyDispose,
1005 T, NoteForHelper);
Mike Stumpa4f668f2009-03-06 01:33:24 +00001006}
1007
Mike Stump08920992009-03-07 02:35:30 +00001008llvm::Constant *BlockFunction::BuildDestroyHelper(const llvm::StructType *T,
Mike Stumpb7477cf2009-04-10 18:52:28 +00001009 std::vector<HelperInfo> *NoteForHelperp) {
Mike Stump08920992009-03-07 02:35:30 +00001010 return CodeGenFunction(CGM).GenerateDestroyHelperFunction(BlockHasCopyDispose,
Mike Stumpb7477cf2009-04-10 18:52:28 +00001011 T, NoteForHelperp);
Mike Stumpdab514f2009-03-04 03:23:46 +00001012}
Mike Stump797b6322009-03-05 01:23:13 +00001013
Mike Stumpee094222009-03-06 06:12:24 +00001014llvm::Constant *BlockFunction::
1015GeneratebyrefCopyHelperFunction(const llvm::Type *T, int flag) {
Mike Stump45031c02009-03-06 02:29:21 +00001016 QualType R = getContext().VoidTy;
1017
1018 FunctionArgList Args;
1019 // FIXME: This leaks
Mike Stumpee094222009-03-06 06:12:24 +00001020 ImplicitParamDecl *Dst =
Mike Stumpea26cb52009-10-21 03:49:08 +00001021 ImplicitParamDecl::Create(getContext(), 0,
1022 SourceLocation(), 0,
Mike Stumpee094222009-03-06 06:12:24 +00001023 getContext().getPointerType(getContext().VoidTy));
1024 Args.push_back(std::make_pair(Dst, Dst->getType()));
1025
1026 // FIXME: This leaks
Mike Stump45031c02009-03-06 02:29:21 +00001027 ImplicitParamDecl *Src =
Mike Stumpea26cb52009-10-21 03:49:08 +00001028 ImplicitParamDecl::Create(getContext(), 0,
1029 SourceLocation(), 0,
Mike Stump45031c02009-03-06 02:29:21 +00001030 getContext().getPointerType(getContext().VoidTy));
Mike Stump45031c02009-03-06 02:29:21 +00001031 Args.push_back(std::make_pair(Src, Src->getType()));
Mike Stump1eb44332009-09-09 15:08:12 +00001032
Mike Stump45031c02009-03-06 02:29:21 +00001033 const CGFunctionInfo &FI =
1034 CGM.getTypes().getFunctionInfo(R, Args);
1035
Mike Stump45031c02009-03-06 02:29:21 +00001036 CodeGenTypes &Types = CGM.getTypes();
1037 const llvm::FunctionType *LTy = Types.GetFunctionType(FI, false);
1038
Mike Stump3899a7f2009-06-05 23:26:36 +00001039 // FIXME: We'd like to put these into a mergable by content, with
1040 // internal linkage.
Mike Stump45031c02009-03-06 02:29:21 +00001041 llvm::Function *Fn =
1042 llvm::Function::Create(LTy, llvm::GlobalValue::InternalLinkage,
Benjamin Kramer3cf7c5d2010-01-22 13:59:13 +00001043 "__Block_byref_id_object_copy_", &CGM.getModule());
Mike Stump45031c02009-03-06 02:29:21 +00001044
1045 IdentifierInfo *II
1046 = &CGM.getContext().Idents.get("__Block_byref_id_object_copy_");
1047
1048 FunctionDecl *FD = FunctionDecl::Create(getContext(),
1049 getContext().getTranslationUnitDecl(),
Argyrios Kyrtzidisa1d56622009-08-19 01:27:57 +00001050 SourceLocation(), II, R, 0,
Mike Stump45031c02009-03-06 02:29:21 +00001051 FunctionDecl::Static, false,
1052 true);
1053 CGF.StartFunction(FD, R, Fn, Args, SourceLocation());
Mike Stumpee094222009-03-06 06:12:24 +00001054
1055 // dst->x
1056 llvm::Value *V = CGF.GetAddrOfLocalVar(Dst);
Owen Anderson96e0fc72009-07-29 22:16:19 +00001057 V = Builder.CreateBitCast(V, llvm::PointerType::get(T, 0));
Mike Stumpc2f4c342009-04-15 22:11:36 +00001058 V = Builder.CreateLoad(V);
Mike Stumpee094222009-03-06 06:12:24 +00001059 V = Builder.CreateStructGEP(V, 6, "x");
1060 llvm::Value *DstObj = Builder.CreateBitCast(V, PtrToInt8Ty);
1061
1062 // src->x
1063 V = CGF.GetAddrOfLocalVar(Src);
1064 V = Builder.CreateLoad(V);
1065 V = Builder.CreateBitCast(V, T);
1066 V = Builder.CreateStructGEP(V, 6, "x");
Owen Anderson96e0fc72009-07-29 22:16:19 +00001067 V = Builder.CreateBitCast(V, llvm::PointerType::get(PtrToInt8Ty, 0));
Mike Stumpee094222009-03-06 06:12:24 +00001068 llvm::Value *SrcObj = Builder.CreateLoad(V);
Mike Stump1eb44332009-09-09 15:08:12 +00001069
Mike Stumpee094222009-03-06 06:12:24 +00001070 flag |= BLOCK_BYREF_CALLER;
1071
Owen Anderson0032b272009-08-13 21:57:51 +00001072 llvm::Value *N = llvm::ConstantInt::get(
1073 llvm::Type::getInt32Ty(T->getContext()), flag);
Mike Stumpee094222009-03-06 06:12:24 +00001074 llvm::Value *F = getBlockObjectAssign();
1075 Builder.CreateCall3(F, DstObj, SrcObj, N);
1076
Mike Stump45031c02009-03-06 02:29:21 +00001077 CGF.FinishFunction();
1078
Owen Anderson3c4972d2009-07-29 18:54:39 +00001079 return llvm::ConstantExpr::getBitCast(Fn, PtrToInt8Ty);
Mike Stump45031c02009-03-06 02:29:21 +00001080}
1081
Mike Stump1851b682009-03-06 04:53:30 +00001082llvm::Constant *
1083BlockFunction::GeneratebyrefDestroyHelperFunction(const llvm::Type *T,
1084 int flag) {
Mike Stump45031c02009-03-06 02:29:21 +00001085 QualType R = getContext().VoidTy;
1086
1087 FunctionArgList Args;
1088 // FIXME: This leaks
1089 ImplicitParamDecl *Src =
Mike Stumpea26cb52009-10-21 03:49:08 +00001090 ImplicitParamDecl::Create(getContext(), 0,
1091 SourceLocation(), 0,
Mike Stump45031c02009-03-06 02:29:21 +00001092 getContext().getPointerType(getContext().VoidTy));
1093
1094 Args.push_back(std::make_pair(Src, Src->getType()));
Mike Stump1eb44332009-09-09 15:08:12 +00001095
Mike Stump45031c02009-03-06 02:29:21 +00001096 const CGFunctionInfo &FI =
1097 CGM.getTypes().getFunctionInfo(R, Args);
1098
Mike Stump45031c02009-03-06 02:29:21 +00001099 CodeGenTypes &Types = CGM.getTypes();
1100 const llvm::FunctionType *LTy = Types.GetFunctionType(FI, false);
1101
Mike Stump3899a7f2009-06-05 23:26:36 +00001102 // FIXME: We'd like to put these into a mergable by content, with
1103 // internal linkage.
Mike Stump45031c02009-03-06 02:29:21 +00001104 llvm::Function *Fn =
1105 llvm::Function::Create(LTy, llvm::GlobalValue::InternalLinkage,
Benjamin Kramer3cf7c5d2010-01-22 13:59:13 +00001106 "__Block_byref_id_object_dispose_",
Mike Stump45031c02009-03-06 02:29:21 +00001107 &CGM.getModule());
1108
1109 IdentifierInfo *II
1110 = &CGM.getContext().Idents.get("__Block_byref_id_object_dispose_");
1111
1112 FunctionDecl *FD = FunctionDecl::Create(getContext(),
1113 getContext().getTranslationUnitDecl(),
Argyrios Kyrtzidisa1d56622009-08-19 01:27:57 +00001114 SourceLocation(), II, R, 0,
Mike Stump45031c02009-03-06 02:29:21 +00001115 FunctionDecl::Static, false,
1116 true);
1117 CGF.StartFunction(FD, R, Fn, Args, SourceLocation());
Mike Stump1851b682009-03-06 04:53:30 +00001118
1119 llvm::Value *V = CGF.GetAddrOfLocalVar(Src);
Owen Anderson96e0fc72009-07-29 22:16:19 +00001120 V = Builder.CreateBitCast(V, llvm::PointerType::get(T, 0));
Mike Stumpc2f4c342009-04-15 22:11:36 +00001121 V = Builder.CreateLoad(V);
Mike Stump1851b682009-03-06 04:53:30 +00001122 V = Builder.CreateStructGEP(V, 6, "x");
Owen Anderson96e0fc72009-07-29 22:16:19 +00001123 V = Builder.CreateBitCast(V, llvm::PointerType::get(PtrToInt8Ty, 0));
Mike Stumpc2f4c342009-04-15 22:11:36 +00001124 V = Builder.CreateLoad(V);
Mike Stump1851b682009-03-06 04:53:30 +00001125
Mike Stump1851b682009-03-06 04:53:30 +00001126 flag |= BLOCK_BYREF_CALLER;
1127 BuildBlockRelease(V, flag);
Mike Stump45031c02009-03-06 02:29:21 +00001128 CGF.FinishFunction();
1129
Owen Anderson3c4972d2009-07-29 18:54:39 +00001130 return llvm::ConstantExpr::getBitCast(Fn, PtrToInt8Ty);
Mike Stump45031c02009-03-06 02:29:21 +00001131}
1132
Mike Stumpee094222009-03-06 06:12:24 +00001133llvm::Constant *BlockFunction::BuildbyrefCopyHelper(const llvm::Type *T,
Chris Lattner10976d92009-12-05 08:21:30 +00001134 int Flag, unsigned Align) {
1135 // All alignments below that of pointer alignment collapse down to just
Mike Stump3899a7f2009-06-05 23:26:36 +00001136 // pointer alignment, as we always have at least that much alignment to begin
1137 // with.
1138 Align /= unsigned(CGF.Target.getPointerAlign(0)/8);
Chris Lattner10976d92009-12-05 08:21:30 +00001139
Mike Stump3899a7f2009-06-05 23:26:36 +00001140 // As an optimization, we only generate a single function of each kind we
1141 // might need. We need a different one for each alignment and for each
1142 // setting of flags. We mix Align and flag to get the kind.
Chris Lattner10976d92009-12-05 08:21:30 +00001143 uint64_t Kind = (uint64_t)Align*BLOCK_BYREF_CURRENT_MAX + Flag;
1144 llvm::Constant *&Entry = CGM.AssignCache[Kind];
Mike Stump3899a7f2009-06-05 23:26:36 +00001145 if (Entry)
1146 return Entry;
Chris Lattner10976d92009-12-05 08:21:30 +00001147 return Entry = CodeGenFunction(CGM).GeneratebyrefCopyHelperFunction(T, Flag);
Mike Stump45031c02009-03-06 02:29:21 +00001148}
1149
Mike Stump1851b682009-03-06 04:53:30 +00001150llvm::Constant *BlockFunction::BuildbyrefDestroyHelper(const llvm::Type *T,
Chris Lattner10976d92009-12-05 08:21:30 +00001151 int Flag,
Mike Stump3899a7f2009-06-05 23:26:36 +00001152 unsigned Align) {
1153 // All alignments below that of pointer alignment collpase down to just
1154 // pointer alignment, as we always have at least that much alignment to begin
1155 // with.
1156 Align /= unsigned(CGF.Target.getPointerAlign(0)/8);
Chris Lattner10976d92009-12-05 08:21:30 +00001157
Mike Stump3899a7f2009-06-05 23:26:36 +00001158 // As an optimization, we only generate a single function of each kind we
1159 // might need. We need a different one for each alignment and for each
1160 // setting of flags. We mix Align and flag to get the kind.
Chris Lattner10976d92009-12-05 08:21:30 +00001161 uint64_t Kind = (uint64_t)Align*BLOCK_BYREF_CURRENT_MAX + Flag;
1162 llvm::Constant *&Entry = CGM.DestroyCache[Kind];
Mike Stump3899a7f2009-06-05 23:26:36 +00001163 if (Entry)
1164 return Entry;
Chris Lattner10976d92009-12-05 08:21:30 +00001165 return Entry=CodeGenFunction(CGM).GeneratebyrefDestroyHelperFunction(T, Flag);
Mike Stump45031c02009-03-06 02:29:21 +00001166}
1167
Mike Stump797b6322009-03-05 01:23:13 +00001168llvm::Value *BlockFunction::getBlockObjectDispose() {
1169 if (CGM.BlockObjectDispose == 0) {
1170 const llvm::FunctionType *FTy;
1171 std::vector<const llvm::Type*> ArgTys;
Owen Anderson0032b272009-08-13 21:57:51 +00001172 const llvm::Type *ResultType = llvm::Type::getVoidTy(VMContext);
Mike Stump797b6322009-03-05 01:23:13 +00001173 ArgTys.push_back(PtrToInt8Ty);
Owen Anderson0032b272009-08-13 21:57:51 +00001174 ArgTys.push_back(llvm::Type::getInt32Ty(VMContext));
Owen Anderson96e0fc72009-07-29 22:16:19 +00001175 FTy = llvm::FunctionType::get(ResultType, ArgTys, false);
Mike Stump00470a12009-03-05 08:32:30 +00001176 CGM.BlockObjectDispose
Mike Stump797b6322009-03-05 01:23:13 +00001177 = CGM.CreateRuntimeFunction(FTy, "_Block_object_dispose");
1178 }
1179 return CGM.BlockObjectDispose;
1180}
1181
Mike Stumpee094222009-03-06 06:12:24 +00001182llvm::Value *BlockFunction::getBlockObjectAssign() {
1183 if (CGM.BlockObjectAssign == 0) {
1184 const llvm::FunctionType *FTy;
1185 std::vector<const llvm::Type*> ArgTys;
Owen Anderson0032b272009-08-13 21:57:51 +00001186 const llvm::Type *ResultType = llvm::Type::getVoidTy(VMContext);
Mike Stumpee094222009-03-06 06:12:24 +00001187 ArgTys.push_back(PtrToInt8Ty);
1188 ArgTys.push_back(PtrToInt8Ty);
Owen Anderson0032b272009-08-13 21:57:51 +00001189 ArgTys.push_back(llvm::Type::getInt32Ty(VMContext));
Owen Anderson96e0fc72009-07-29 22:16:19 +00001190 FTy = llvm::FunctionType::get(ResultType, ArgTys, false);
Mike Stumpee094222009-03-06 06:12:24 +00001191 CGM.BlockObjectAssign
1192 = CGM.CreateRuntimeFunction(FTy, "_Block_object_assign");
1193 }
1194 return CGM.BlockObjectAssign;
1195}
1196
Mike Stump1851b682009-03-06 04:53:30 +00001197void BlockFunction::BuildBlockRelease(llvm::Value *V, int flag) {
Mike Stump797b6322009-03-05 01:23:13 +00001198 llvm::Value *F = getBlockObjectDispose();
Mike Stump1851b682009-03-06 04:53:30 +00001199 llvm::Value *N;
Mike Stump797b6322009-03-05 01:23:13 +00001200 V = Builder.CreateBitCast(V, PtrToInt8Ty);
Owen Anderson0032b272009-08-13 21:57:51 +00001201 N = llvm::ConstantInt::get(llvm::Type::getInt32Ty(V->getContext()), flag);
Mike Stump797b6322009-03-05 01:23:13 +00001202 Builder.CreateCall2(F, V, N);
1203}
Mike Stump00470a12009-03-05 08:32:30 +00001204
1205ASTContext &BlockFunction::getContext() const { return CGM.getContext(); }
Mike Stump08920992009-03-07 02:35:30 +00001206
1207BlockFunction::BlockFunction(CodeGenModule &cgm, CodeGenFunction &cgf,
1208 CGBuilderTy &B)
Owen Andersona1cf15f2009-07-14 23:10:40 +00001209 : CGM(cgm), CGF(cgf), VMContext(cgm.getLLVMContext()), Builder(B) {
Owen Anderson0032b272009-08-13 21:57:51 +00001210 PtrToInt8Ty = llvm::PointerType::getUnqual(
1211 llvm::Type::getInt8Ty(VMContext));
Mike Stump08920992009-03-07 02:35:30 +00001212
1213 BlockHasCopyDispose = false;
1214}