blob: b7ad9207f29dca1e9de633c263590781522d3eac [file] [log] [blame]
Hans Wennborgdcfba332015-10-06 23:40:43 +00001//===--- CGBlocks.cpp - Emit LLVM Code for declarations ---------*- C++ -*-===//
Anders Carlsson2437cbf2009-02-12 00:39:25 +00002//
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
John McCallad7c5c12011-02-08 08:22:06 +000014#include "CGBlocks.h"
Akira Hatanaka9978da32018-08-10 15:09:24 +000015#include "CGCXXABI.h"
Chandler Carruth3a022472012-12-04 09:13:33 +000016#include "CGDebugInfo.h"
17#include "CGObjCRuntime.h"
Yaxun Liu10712d92017-10-04 20:32:17 +000018#include "CGOpenCLRuntime.h"
Chandler Carruth3a022472012-12-04 09:13:33 +000019#include "CodeGenFunction.h"
20#include "CodeGenModule.h"
John McCallde0fe072017-08-15 21:42:52 +000021#include "ConstantEmitter.h"
Yaxun Liu10712d92017-10-04 20:32:17 +000022#include "TargetInfo.h"
Mike Stump692c6e32009-03-20 21:53:12 +000023#include "clang/AST/DeclObjC.h"
Yaxun Liu10712d92017-10-04 20:32:17 +000024#include "clang/CodeGen/ConstantInitBuilder.h"
Benjamin Kramer9e2e1c92010-03-31 15:04:05 +000025#include "llvm/ADT/SmallSet.h"
Chandler Carruthc80ceea2014-03-04 11:02:08 +000026#include "llvm/IR/CallSite.h"
Chandler Carruthffd55512013-01-02 11:45:17 +000027#include "llvm/IR/DataLayout.h"
28#include "llvm/IR/Module.h"
Akira Hatanaka9978da32018-08-10 15:09:24 +000029#include "llvm/Support/ScopedPrinter.h"
Anders Carlsson2437cbf2009-02-12 00:39:25 +000030#include <algorithm>
Fariborz Jahanian983ae492012-11-14 17:43:08 +000031#include <cstdio>
Torok Edwindb714922009-08-24 13:25:12 +000032
Anders Carlsson2437cbf2009-02-12 00:39:25 +000033using namespace clang;
34using namespace CodeGen;
35
John McCall08ef4662011-11-10 08:15:53 +000036CGBlockInfo::CGBlockInfo(const BlockDecl *block, StringRef name)
37 : Name(name), CXXThisIndex(0), CanBeGlobal(false), NeedsCopyDispose(false),
Fariborz Jahanian23290b02012-11-01 18:32:55 +000038 HasCXXObject(false), UsesStret(false), HasCapturedVariableLayout(false),
Akira Hatanaka9978da32018-08-10 15:09:24 +000039 CapturesNonExternalType(false), LocalAddress(Address::invalid()),
40 StructureType(nullptr), Block(block), DominatingIP(nullptr) {
Craig Topper8a13c412014-05-21 05:09:00 +000041
John McCall08ef4662011-11-10 08:15:53 +000042 // Skip asm prefix, if any. 'name' is usually taken directly from
43 // the mangled name of the enclosing function.
44 if (!name.empty() && name[0] == '\01')
45 name = name.substr(1);
John McCall9d42f0f2010-05-21 04:11:14 +000046}
47
John McCallf9b056b2011-03-31 08:03:29 +000048// Anchor the vtable to this translation unit.
Angel Garcia Gomez637d1e62015-10-20 13:23:58 +000049BlockByrefHelpers::~BlockByrefHelpers() {}
John McCallf9b056b2011-03-31 08:03:29 +000050
John McCall351762c2011-02-07 10:33:21 +000051/// Build the given block as a global block.
52static llvm::Constant *buildGlobalBlock(CodeGenModule &CGM,
53 const CGBlockInfo &blockInfo,
54 llvm::Constant *blockFn);
John McCall9d42f0f2010-05-21 04:11:14 +000055
John McCall351762c2011-02-07 10:33:21 +000056/// Build the helper function to copy a block.
57static llvm::Constant *buildCopyHelper(CodeGenModule &CGM,
58 const CGBlockInfo &blockInfo) {
59 return CodeGenFunction(CGM).GenerateCopyHelperFunction(blockInfo);
60}
61
Alp Tokerf6a24ce2013-12-05 16:25:25 +000062/// Build the helper function to dispose of a block.
John McCall351762c2011-02-07 10:33:21 +000063static llvm::Constant *buildDisposeHelper(CodeGenModule &CGM,
64 const CGBlockInfo &blockInfo) {
65 return CodeGenFunction(CGM).GenerateDestroyHelperFunction(blockInfo);
66}
67
Akira Hatanaka2ec36f02018-08-17 15:46:07 +000068namespace {
69
70/// Represents a type of copy/destroy operation that should be performed for an
71/// entity that's captured by a block.
72enum class BlockCaptureEntityKind {
73 CXXRecord, // Copy or destroy
74 ARCWeak,
75 ARCStrong,
76 NonTrivialCStruct,
77 BlockObject, // Assign or release
78 None
79};
80
81/// Represents a captured entity that requires extra operations in order for
82/// this entity to be copied or destroyed correctly.
83struct BlockCaptureManagedEntity {
84 BlockCaptureEntityKind CopyKind, DisposeKind;
85 BlockFieldFlags CopyFlags, DisposeFlags;
86 const BlockDecl::Capture *CI;
87 const CGBlockInfo::Capture *Capture;
88
89 BlockCaptureManagedEntity(BlockCaptureEntityKind CopyType,
90 BlockCaptureEntityKind DisposeType,
91 BlockFieldFlags CopyFlags,
92 BlockFieldFlags DisposeFlags,
93 const BlockDecl::Capture &CI,
94 const CGBlockInfo::Capture &Capture)
95 : CopyKind(CopyType), DisposeKind(DisposeType), CopyFlags(CopyFlags),
96 DisposeFlags(DisposeFlags), CI(&CI), Capture(&Capture) {}
97
98 bool operator<(const BlockCaptureManagedEntity &Other) const {
99 return Capture->getOffset() < Other.Capture->getOffset();
100 }
101};
102
103enum class CaptureStrKind {
104 // String for the copy helper.
105 CopyHelper,
106 // String for the dispose helper.
107 DisposeHelper,
108 // Merge the strings for the copy helper and dispose helper.
109 Merged
110};
111
112} // end anonymous namespace
113
114static void findBlockCapturedManagedEntities(
115 const CGBlockInfo &BlockInfo, const LangOptions &LangOpts,
116 SmallVectorImpl<BlockCaptureManagedEntity> &ManagedCaptures);
117
118static std::string getBlockCaptureStr(const BlockCaptureManagedEntity &E,
119 CaptureStrKind StrKind,
120 CharUnits BlockAlignment,
121 CodeGenModule &CGM);
122
123static std::string getBlockDescriptorName(const CGBlockInfo &BlockInfo,
124 CodeGenModule &CGM) {
125 std::string Name = "__block_descriptor_";
126 Name += llvm::to_string(BlockInfo.BlockSize.getQuantity()) + "_";
127
128 if (BlockInfo.needsCopyDisposeHelpers()) {
129 if (CGM.getLangOpts().Exceptions)
130 Name += "e";
131 if (CGM.getCodeGenOpts().ObjCAutoRefCountExceptions)
132 Name += "a";
133 Name += llvm::to_string(BlockInfo.BlockAlign.getQuantity()) + "_";
134
135 SmallVector<BlockCaptureManagedEntity, 4> ManagedCaptures;
136 findBlockCapturedManagedEntities(BlockInfo, CGM.getContext().getLangOpts(),
137 ManagedCaptures);
138
139 for (const BlockCaptureManagedEntity &E : ManagedCaptures) {
140 Name += llvm::to_string(E.Capture->getOffset().getQuantity());
141
142 if (E.CopyKind == E.DisposeKind) {
143 // If CopyKind and DisposeKind are the same, merge the capture
144 // information.
145 assert(E.CopyKind != BlockCaptureEntityKind::None &&
146 "shouldn't see BlockCaptureManagedEntity that is None");
147 Name += getBlockCaptureStr(E, CaptureStrKind::Merged,
148 BlockInfo.BlockAlign, CGM);
149 } else {
150 // If CopyKind and DisposeKind are not the same, which can happen when
151 // either Kind is None or the captured object is a __strong block,
152 // concatenate the copy and dispose strings.
153 Name += getBlockCaptureStr(E, CaptureStrKind::CopyHelper,
154 BlockInfo.BlockAlign, CGM);
155 Name += getBlockCaptureStr(E, CaptureStrKind::DisposeHelper,
156 BlockInfo.BlockAlign, CGM);
157 }
158 }
159 Name += "_";
160 }
161
162 std::string TypeAtEncoding =
163 CGM.getContext().getObjCEncodingForBlock(BlockInfo.getBlockExpr());
164 Name += "e" + llvm::to_string(TypeAtEncoding.size()) + "_" + TypeAtEncoding;
165 Name += "l" + CGM.getObjCRuntime().getRCBlockLayoutStr(CGM, BlockInfo);
166 return Name;
167}
168
Fariborz Jahanianbf7bf292012-10-25 18:06:53 +0000169/// buildBlockDescriptor - Build the block descriptor meta-data for a block.
170/// buildBlockDescriptor is accessed from 5th field of the Block_literal
171/// meta-data and contains stationary information about the block literal.
Alexander Kornienko2a8c18d2018-04-06 15:14:32 +0000172/// Its definition will have 4 (or optionally 6) words.
Dmitri Gribenko6c96ba22013-05-08 23:09:44 +0000173/// \code
Fariborz Jahanianbf7bf292012-10-25 18:06:53 +0000174/// struct Block_descriptor {
175/// unsigned long reserved;
176/// unsigned long size; // size of Block_literal metadata in bytes.
177/// void *copy_func_helper_decl; // optional copy helper.
178/// void *destroy_func_decl; // optioanl destructor helper.
Dmitri Gribenko6c96ba22013-05-08 23:09:44 +0000179/// void *block_method_encoding_address; // @encode for block literal signature.
Fariborz Jahanianbf7bf292012-10-25 18:06:53 +0000180/// void *block_layout_info; // encoding of captured block variables.
181/// };
Dmitri Gribenko6c96ba22013-05-08 23:09:44 +0000182/// \endcode
John McCall351762c2011-02-07 10:33:21 +0000183static llvm::Constant *buildBlockDescriptor(CodeGenModule &CGM,
184 const CGBlockInfo &blockInfo) {
185 ASTContext &C = CGM.getContext();
186
John McCall6c9f1fdb2016-11-19 08:17:24 +0000187 llvm::IntegerType *ulong =
188 cast<llvm::IntegerType>(CGM.getTypes().ConvertType(C.UnsignedLongTy));
189 llvm::PointerType *i8p = nullptr;
Pekka Jaaskelainenab751a82014-08-14 09:37:50 +0000190 if (CGM.getLangOpts().OpenCL)
Fangrui Song6907ce22018-07-30 19:24:48 +0000191 i8p =
Pekka Jaaskelainenab751a82014-08-14 09:37:50 +0000192 llvm::Type::getInt8PtrTy(
193 CGM.getLLVMContext(), C.getTargetAddressSpace(LangAS::opencl_constant));
194 else
John McCall6c9f1fdb2016-11-19 08:17:24 +0000195 i8p = CGM.VoidPtrTy;
John McCall351762c2011-02-07 10:33:21 +0000196
Akira Hatanaka2ec36f02018-08-17 15:46:07 +0000197 std::string descName;
198
199 // If an equivalent block descriptor global variable exists, return it.
200 if (C.getLangOpts().ObjC1 &&
201 CGM.getLangOpts().getGC() == LangOptions::NonGC) {
202 descName = getBlockDescriptorName(blockInfo, CGM);
203 if (llvm::GlobalValue *desc = CGM.getModule().getNamedValue(descName))
204 return llvm::ConstantExpr::getBitCast(desc,
205 CGM.getBlockDescriptorType());
206 }
207
208 // If there isn't an equivalent block descriptor global variable, create a new
209 // one.
John McCall23c9dc62016-11-28 22:18:27 +0000210 ConstantInitBuilder builder(CGM);
John McCall6c9f1fdb2016-11-19 08:17:24 +0000211 auto elements = builder.beginStruct();
Mike Stump85284ba2009-02-13 16:19:19 +0000212
213 // reserved
John McCall6c9f1fdb2016-11-19 08:17:24 +0000214 elements.addInt(ulong, 0);
Mike Stump85284ba2009-02-13 16:19:19 +0000215
216 // Size
Mike Stump2ac40a92009-02-21 20:07:44 +0000217 // FIXME: What is the right way to say this doesn't fit? We should give
218 // a user diagnostic in that case. Better fix would be to change the
219 // API to size_t.
John McCall6c9f1fdb2016-11-19 08:17:24 +0000220 elements.addInt(ulong, blockInfo.BlockSize.getQuantity());
Mike Stump85284ba2009-02-13 16:19:19 +0000221
John McCall351762c2011-02-07 10:33:21 +0000222 // Optional copy/dispose helpers.
Akira Hatanaka2ec36f02018-08-17 15:46:07 +0000223 bool hasInternalHelper = false;
Akira Hatanakadbfa4532018-07-20 17:10:32 +0000224 if (blockInfo.needsCopyDisposeHelpers()) {
Mike Stump85284ba2009-02-13 16:19:19 +0000225 // copy_func_helper_decl
Akira Hatanaka2ec36f02018-08-17 15:46:07 +0000226 llvm::Constant *copyHelper = buildCopyHelper(CGM, blockInfo);
227 elements.add(copyHelper);
Mike Stump85284ba2009-02-13 16:19:19 +0000228
229 // destroy_func_decl
Akira Hatanaka2ec36f02018-08-17 15:46:07 +0000230 llvm::Constant *disposeHelper = buildDisposeHelper(CGM, blockInfo);
231 elements.add(disposeHelper);
232
233 if (cast<llvm::Function>(copyHelper->getOperand(0))->hasInternalLinkage() ||
234 cast<llvm::Function>(disposeHelper->getOperand(0))
235 ->hasInternalLinkage())
236 hasInternalHelper = true;
Mike Stump85284ba2009-02-13 16:19:19 +0000237 }
238
John McCall351762c2011-02-07 10:33:21 +0000239 // Signature. Mandatory ObjC-style method descriptor @encode sequence.
240 std::string typeAtEncoding =
241 CGM.getContext().getObjCEncodingForBlock(blockInfo.getBlockExpr());
John McCall6c9f1fdb2016-11-19 08:17:24 +0000242 elements.add(llvm::ConstantExpr::getBitCast(
John McCall7f416cc2015-09-08 08:05:57 +0000243 CGM.GetAddrOfConstantCString(typeAtEncoding).getPointer(), i8p));
Fangrui Song6907ce22018-07-30 19:24:48 +0000244
John McCall351762c2011-02-07 10:33:21 +0000245 // GC layout.
Fariborz Jahanian0c58ce92012-10-27 21:10:38 +0000246 if (C.getLangOpts().ObjC1) {
247 if (CGM.getLangOpts().getGC() != LangOptions::NonGC)
John McCall6c9f1fdb2016-11-19 08:17:24 +0000248 elements.add(CGM.getObjCRuntime().BuildGCBlockLayout(CGM, blockInfo));
Fariborz Jahanian0c58ce92012-10-27 21:10:38 +0000249 else
John McCall6c9f1fdb2016-11-19 08:17:24 +0000250 elements.add(CGM.getObjCRuntime().BuildRCBlockLayout(CGM, blockInfo));
Fariborz Jahanian0c58ce92012-10-27 21:10:38 +0000251 }
John McCall351762c2011-02-07 10:33:21 +0000252 else
John McCall6c9f1fdb2016-11-19 08:17:24 +0000253 elements.addNullPointer(i8p);
Mike Stump85284ba2009-02-13 16:19:19 +0000254
Joey Goulyddbda402016-08-10 15:57:02 +0000255 unsigned AddrSpace = 0;
256 if (C.getLangOpts().OpenCL)
257 AddrSpace = C.getTargetAddressSpace(LangAS::opencl_constant);
John McCall6c9f1fdb2016-11-19 08:17:24 +0000258
Akira Hatanaka2ec36f02018-08-17 15:46:07 +0000259 llvm::GlobalValue::LinkageTypes linkage;
260 if (descName.empty()) {
261 linkage = llvm::GlobalValue::InternalLinkage;
262 descName = "__block_descriptor_tmp";
263 } else if (hasInternalHelper) {
264 // If either the copy helper or the dispose helper has internal linkage,
265 // the block descriptor must have internal linkage too.
266 linkage = llvm::GlobalValue::InternalLinkage;
267 } else {
268 linkage = llvm::GlobalValue::LinkOnceODRLinkage;
269 }
270
John McCall351762c2011-02-07 10:33:21 +0000271 llvm::GlobalVariable *global =
Akira Hatanaka2ec36f02018-08-17 15:46:07 +0000272 elements.finishAndCreateGlobal(descName, CGM.getPointerAlign(),
273 /*constant*/ true, linkage, AddrSpace);
274
275 if (linkage == llvm::GlobalValue::LinkOnceODRLinkage) {
276 global->setVisibility(llvm::GlobalValue::HiddenVisibility);
277 global->setUnnamedAddr(llvm::GlobalValue::UnnamedAddr::Global);
278 }
Mike Stump85284ba2009-02-13 16:19:19 +0000279
John McCall351762c2011-02-07 10:33:21 +0000280 return llvm::ConstantExpr::getBitCast(global, CGM.getBlockDescriptorType());
Anders Carlssoned5e69f2009-03-01 01:09:12 +0000281}
282
John McCall351762c2011-02-07 10:33:21 +0000283/*
284 Purely notional variadic template describing the layout of a block.
Anders Carlssoned5e69f2009-03-01 01:09:12 +0000285
John McCall351762c2011-02-07 10:33:21 +0000286 template <class _ResultType, class... _ParamTypes, class... _CaptureTypes>
287 struct Block_literal {
288 /// Initialized to one of:
289 /// extern void *_NSConcreteStackBlock[];
290 /// extern void *_NSConcreteGlobalBlock[];
291 ///
292 /// In theory, we could start one off malloc'ed by setting
293 /// BLOCK_NEEDS_FREE, giving it a refcount of 1, and using
294 /// this isa:
295 /// extern void *_NSConcreteMallocBlock[];
296 struct objc_class *isa;
Mike Stump4446dcf2009-03-05 08:32:30 +0000297
John McCall351762c2011-02-07 10:33:21 +0000298 /// These are the flags (with corresponding bit number) that the
299 /// compiler is actually supposed to know about.
Akira Hatanakadbfa4532018-07-20 17:10:32 +0000300 /// 23. BLOCK_IS_NOESCAPE - indicates that the block is non-escaping
John McCall351762c2011-02-07 10:33:21 +0000301 /// 25. BLOCK_HAS_COPY_DISPOSE - indicates that the block
302 /// descriptor provides copy and dispose helper functions
303 /// 26. BLOCK_HAS_CXX_OBJ - indicates that there's a captured
304 /// object with a nontrivial destructor or copy constructor
305 /// 28. BLOCK_IS_GLOBAL - indicates that the block is allocated
306 /// as global memory
307 /// 29. BLOCK_USE_STRET - indicates that the block function
308 /// uses stret, which objc_msgSend needs to know about
309 /// 30. BLOCK_HAS_SIGNATURE - indicates that the block has an
310 /// @encoded signature string
311 /// And we're not supposed to manipulate these:
312 /// 24. BLOCK_NEEDS_FREE - indicates that the block has been moved
313 /// to malloc'ed memory
314 /// 27. BLOCK_IS_GC - indicates that the block has been moved to
315 /// to GC-allocated memory
316 /// Additionally, the bottom 16 bits are a reference count which
317 /// should be zero on the stack.
318 int flags;
David Chisnall950a9512009-11-17 19:33:30 +0000319
John McCall351762c2011-02-07 10:33:21 +0000320 /// Reserved; should be zero-initialized.
321 int reserved;
David Chisnall950a9512009-11-17 19:33:30 +0000322
John McCall351762c2011-02-07 10:33:21 +0000323 /// Function pointer generated from block literal.
324 _ResultType (*invoke)(Block_literal *, _ParamTypes...);
Mike Stump85284ba2009-02-13 16:19:19 +0000325
John McCall351762c2011-02-07 10:33:21 +0000326 /// Block description metadata generated from block literal.
327 struct Block_descriptor *block_descriptor;
John McCall3882ace2011-01-05 12:14:39 +0000328
John McCall351762c2011-02-07 10:33:21 +0000329 /// Captured values follow.
330 _CapturesTypes captures...;
331 };
332 */
David Chisnall950a9512009-11-17 19:33:30 +0000333
John McCall351762c2011-02-07 10:33:21 +0000334namespace {
335 /// A chunk of data that we actually have to capture in the block.
336 struct BlockLayoutChunk {
337 CharUnits Alignment;
338 CharUnits Size;
Fariborz Jahanianc8892052013-01-17 00:25:06 +0000339 Qualifiers::ObjCLifetime Lifetime;
John McCall351762c2011-02-07 10:33:21 +0000340 const BlockDecl::Capture *Capture; // null for 'this'
Jay Foad7c57be32011-07-11 09:56:20 +0000341 llvm::Type *Type;
Akira Hatanakad542ccf2016-09-16 00:02:06 +0000342 QualType FieldType;
Mike Stump85284ba2009-02-13 16:19:19 +0000343
John McCall351762c2011-02-07 10:33:21 +0000344 BlockLayoutChunk(CharUnits align, CharUnits size,
Fariborz Jahanianc8892052013-01-17 00:25:06 +0000345 Qualifiers::ObjCLifetime lifetime,
John McCall351762c2011-02-07 10:33:21 +0000346 const BlockDecl::Capture *capture,
Akira Hatanakad542ccf2016-09-16 00:02:06 +0000347 llvm::Type *type, QualType fieldType)
Fariborz Jahanianc8892052013-01-17 00:25:06 +0000348 : Alignment(align), Size(size), Lifetime(lifetime),
Akira Hatanakad542ccf2016-09-16 00:02:06 +0000349 Capture(capture), Type(type), FieldType(fieldType) {}
Mike Stump85284ba2009-02-13 16:19:19 +0000350
John McCall351762c2011-02-07 10:33:21 +0000351 /// Tell the block info that this chunk has the given field index.
John McCall7f416cc2015-09-08 08:05:57 +0000352 void setIndex(CGBlockInfo &info, unsigned index, CharUnits offset) {
353 if (!Capture) {
John McCall351762c2011-02-07 10:33:21 +0000354 info.CXXThisIndex = index;
John McCall7f416cc2015-09-08 08:05:57 +0000355 info.CXXThisOffset = offset;
356 } else {
Akira Hatanakad542ccf2016-09-16 00:02:06 +0000357 auto C = CGBlockInfo::Capture::makeIndex(index, offset, FieldType);
358 info.Captures.insert({Capture->getVariable(), C});
John McCall7f416cc2015-09-08 08:05:57 +0000359 }
John McCall87fe5d52010-05-20 01:18:31 +0000360 }
John McCall351762c2011-02-07 10:33:21 +0000361 };
Mike Stumpd6ef62f2009-03-06 18:42:23 +0000362
Fariborz Jahanianc8892052013-01-17 00:25:06 +0000363 /// Order by 1) all __strong together 2) next, all byfref together 3) next,
364 /// all __weak together. Preserve descending alignment in all situations.
John McCall351762c2011-02-07 10:33:21 +0000365 bool operator<(const BlockLayoutChunk &left, const BlockLayoutChunk &right) {
John McCall7f416cc2015-09-08 08:05:57 +0000366 if (left.Alignment != right.Alignment)
367 return left.Alignment > right.Alignment;
368
369 auto getPrefOrder = [](const BlockLayoutChunk &chunk) {
John McCall9c52b282015-09-11 22:00:51 +0000370 if (chunk.Capture && chunk.Capture->isByRef())
John McCall7f416cc2015-09-08 08:05:57 +0000371 return 1;
372 if (chunk.Lifetime == Qualifiers::OCL_Strong)
373 return 0;
374 if (chunk.Lifetime == Qualifiers::OCL_Weak)
375 return 2;
376 return 3;
377 };
378
379 return getPrefOrder(left) < getPrefOrder(right);
John McCall351762c2011-02-07 10:33:21 +0000380 }
Hans Wennborgdcfba332015-10-06 23:40:43 +0000381} // end anonymous namespace
John McCall351762c2011-02-07 10:33:21 +0000382
John McCallb0a3ecb2011-02-08 03:07:00 +0000383/// Determines if the given type is safe for constant capture in C++.
384static bool isSafeForCXXConstantCapture(QualType type) {
385 const RecordType *recordType =
386 type->getBaseElementTypeUnsafe()->getAs<RecordType>();
387
388 // Only records can be unsafe.
389 if (!recordType) return true;
390
Rafael Espindola2ae250c2014-05-09 00:08:36 +0000391 const auto *record = cast<CXXRecordDecl>(recordType->getDecl());
John McCallb0a3ecb2011-02-08 03:07:00 +0000392
393 // Maintain semantics for classes with non-trivial dtors or copy ctors.
394 if (!record->hasTrivialDestructor()) return false;
Richard Smith16488472012-11-16 00:53:38 +0000395 if (record->hasNonTrivialCopyConstructor()) return false;
John McCallb0a3ecb2011-02-08 03:07:00 +0000396
397 // Otherwise, we just have to make sure there aren't any mutable
398 // fields that might have changed since initialization.
Douglas Gregor61226d32011-05-13 01:05:07 +0000399 return !record->hasMutableFields();
John McCallb0a3ecb2011-02-08 03:07:00 +0000400}
401
John McCall351762c2011-02-07 10:33:21 +0000402/// It is illegal to modify a const object after initialization.
403/// Therefore, if a const object has a constant initializer, we don't
404/// actually need to keep storage for it in the block; we'll just
405/// rematerialize it at the start of the block function. This is
406/// acceptable because we make no promises about address stability of
407/// captured variables.
408static llvm::Constant *tryCaptureAsConstant(CodeGenModule &CGM,
Richard Smithdafff942012-01-14 04:30:29 +0000409 CodeGenFunction *CGF,
John McCall351762c2011-02-07 10:33:21 +0000410 const VarDecl *var) {
Simon Pilgrim2c518802017-03-30 14:13:19 +0000411 // Return if this is a function parameter. We shouldn't try to
Akira Hatanaka1cfa2732016-05-02 22:29:40 +0000412 // rematerialize default arguments of function parameters.
413 if (isa<ParmVarDecl>(var))
414 return nullptr;
Akira Hatanaka3ba65352016-05-02 21:52:57 +0000415
John McCall351762c2011-02-07 10:33:21 +0000416 QualType type = var->getType();
417
418 // We can only do this if the variable is const.
Craig Topper8a13c412014-05-21 05:09:00 +0000419 if (!type.isConstQualified()) return nullptr;
John McCall351762c2011-02-07 10:33:21 +0000420
John McCallb0a3ecb2011-02-08 03:07:00 +0000421 // Furthermore, in C++ we have to worry about mutable fields:
422 // C++ [dcl.type.cv]p4:
423 // Except that any class member declared mutable can be
424 // modified, any attempt to modify a const object during its
425 // lifetime results in undefined behavior.
David Blaikiebbafb8a2012-03-11 07:00:24 +0000426 if (CGM.getLangOpts().CPlusPlus && !isSafeForCXXConstantCapture(type))
Craig Topper8a13c412014-05-21 05:09:00 +0000427 return nullptr;
John McCall351762c2011-02-07 10:33:21 +0000428
429 // If the variable doesn't have any initializer (shouldn't this be
430 // invalid?), it's not clear what we should do. Maybe capture as
431 // zero?
432 const Expr *init = var->getInit();
Craig Topper8a13c412014-05-21 05:09:00 +0000433 if (!init) return nullptr;
John McCall351762c2011-02-07 10:33:21 +0000434
John McCallde0fe072017-08-15 21:42:52 +0000435 return ConstantEmitter(CGM, CGF).tryEmitAbstractForInitializer(*var);
John McCall351762c2011-02-07 10:33:21 +0000436}
437
438/// Get the low bit of a nonzero character count. This is the
439/// alignment of the nth byte if the 0th byte is universally aligned.
440static CharUnits getLowBit(CharUnits v) {
441 return CharUnits::fromQuantity(v.getQuantity() & (~v.getQuantity() + 1));
442}
443
444static void initializeForBlockHeader(CodeGenModule &CGM, CGBlockInfo &info,
Chris Lattner0e62c1c2011-07-23 10:55:15 +0000445 SmallVectorImpl<llvm::Type*> &elementTypes) {
John McCall351762c2011-02-07 10:33:21 +0000446
447 assert(elementTypes.empty());
Yaxun Liu10712d92017-10-04 20:32:17 +0000448 if (CGM.getLangOpts().OpenCL) {
Sven van Haastregtda3b6322018-10-02 13:02:24 +0000449 // The header is basically 'struct { int; int; generic void *;
Yaxun Liu10712d92017-10-04 20:32:17 +0000450 // custom_fields; }'. Assert that struct is packed.
Sven van Haastregtda3b6322018-10-02 13:02:24 +0000451 auto GenericAS =
452 CGM.getContext().getTargetAddressSpace(LangAS::opencl_generic);
453 auto GenPtrAlign =
454 CharUnits::fromQuantity(CGM.getTarget().getPointerAlign(GenericAS) / 8);
455 auto GenPtrSize =
456 CharUnits::fromQuantity(CGM.getTarget().getPointerWidth(GenericAS) / 8);
457 assert(CGM.getIntSize() <= GenPtrSize);
458 assert(CGM.getIntAlign() <= GenPtrAlign);
459 assert((2 * CGM.getIntSize()).isMultipleOf(GenPtrAlign));
Yaxun Liu10712d92017-10-04 20:32:17 +0000460 elementTypes.push_back(CGM.IntTy); /* total size */
461 elementTypes.push_back(CGM.IntTy); /* align */
Sven van Haastregtda3b6322018-10-02 13:02:24 +0000462 elementTypes.push_back(
463 CGM.getOpenCLRuntime()
464 .getGenericVoidPointerType()); /* invoke function */
465 unsigned Offset =
466 2 * CGM.getIntSize().getQuantity() + GenPtrSize.getQuantity();
467 unsigned BlockAlign = GenPtrAlign.getQuantity();
Yaxun Liu10712d92017-10-04 20:32:17 +0000468 if (auto *Helper =
469 CGM.getTargetCodeGenInfo().getTargetOpenCLBlockHelper()) {
470 for (auto I : Helper->getCustomFieldTypes()) /* custom fields */ {
471 // TargetOpenCLBlockHelp needs to make sure the struct is packed.
472 // If necessary, add padding fields to the custom fields.
473 unsigned Align = CGM.getDataLayout().getABITypeAlignment(I);
474 if (BlockAlign < Align)
475 BlockAlign = Align;
476 assert(Offset % Align == 0);
477 Offset += CGM.getDataLayout().getTypeAllocSize(I);
478 elementTypes.push_back(I);
479 }
480 }
481 info.BlockAlign = CharUnits::fromQuantity(BlockAlign);
482 info.BlockSize = CharUnits::fromQuantity(Offset);
483 } else {
484 // The header is basically 'struct { void *; int; int; void *; void *; }'.
Alexander Kornienko2a8c18d2018-04-06 15:14:32 +0000485 // Assert that the struct is packed.
Yaxun Liu10712d92017-10-04 20:32:17 +0000486 assert(CGM.getIntSize() <= CGM.getPointerSize());
487 assert(CGM.getIntAlign() <= CGM.getPointerAlign());
488 assert((2 * CGM.getIntSize()).isMultipleOf(CGM.getPointerAlign()));
489 info.BlockAlign = CGM.getPointerAlign();
490 info.BlockSize = 3 * CGM.getPointerSize() + 2 * CGM.getIntSize();
491 elementTypes.push_back(CGM.VoidPtrTy);
492 elementTypes.push_back(CGM.IntTy);
493 elementTypes.push_back(CGM.IntTy);
494 elementTypes.push_back(CGM.VoidPtrTy);
495 elementTypes.push_back(CGM.getBlockDescriptorType());
496 }
John McCall351762c2011-02-07 10:33:21 +0000497}
498
Akira Hatanakaf1b3fc72017-02-14 06:46:55 +0000499static QualType getCaptureFieldType(const CodeGenFunction &CGF,
500 const BlockDecl::Capture &CI) {
501 const VarDecl *VD = CI.getVariable();
502
503 // If the variable is captured by an enclosing block or lambda expression,
504 // use the type of the capture field.
505 if (CGF.BlockInfo && CI.isNested())
506 return CGF.BlockInfo->getCapture(VD).fieldType();
507 if (auto *FD = CGF.LambdaCaptureFields.lookup(VD))
508 return FD->getType();
Akira Hatanaka8e57b072018-10-01 21:51:28 +0000509 // If the captured variable is a non-escaping __block variable, the field
510 // type is the reference type. If the variable is a __block variable that
511 // already has a reference type, the field type is the variable's type.
512 return VD->isNonEscapingByref() ?
513 CGF.getContext().getLValueReferenceType(VD->getType()) : VD->getType();
Akira Hatanakaf1b3fc72017-02-14 06:46:55 +0000514}
515
John McCall351762c2011-02-07 10:33:21 +0000516/// Compute the layout of the given block. Attempts to lay the block
517/// out with minimal space requirements.
Richard Smithdafff942012-01-14 04:30:29 +0000518static void computeBlockInfo(CodeGenModule &CGM, CodeGenFunction *CGF,
519 CGBlockInfo &info) {
John McCall351762c2011-02-07 10:33:21 +0000520 ASTContext &C = CGM.getContext();
521 const BlockDecl *block = info.getBlockDecl();
522
Chris Lattner0e62c1c2011-07-23 10:55:15 +0000523 SmallVector<llvm::Type*, 8> elementTypes;
John McCall351762c2011-02-07 10:33:21 +0000524 initializeForBlockHeader(CGM, info, elementTypes);
Yaxun Liu10712d92017-10-04 20:32:17 +0000525 bool hasNonConstantCustomFields = false;
526 if (auto *OpenCLHelper =
527 CGM.getTargetCodeGenInfo().getTargetOpenCLBlockHelper())
528 hasNonConstantCustomFields =
529 !OpenCLHelper->areAllCustomFieldValuesConstant(info);
530 if (!block->hasCaptures() && !hasNonConstantCustomFields) {
John McCall351762c2011-02-07 10:33:21 +0000531 info.StructureType =
532 llvm::StructType::get(CGM.getLLVMContext(), elementTypes, true);
533 info.CanBeGlobal = true;
534 return;
Mike Stump85284ba2009-02-13 16:19:19 +0000535 }
Fariborz Jahanian23290b02012-11-01 18:32:55 +0000536 else if (C.getLangOpts().ObjC1 &&
537 CGM.getLangOpts().getGC() == LangOptions::NonGC)
538 info.HasCapturedVariableLayout = true;
Fangrui Song6907ce22018-07-30 19:24:48 +0000539
John McCall351762c2011-02-07 10:33:21 +0000540 // Collect the layout chunks.
Chris Lattner0e62c1c2011-07-23 10:55:15 +0000541 SmallVector<BlockLayoutChunk, 16> layout;
John McCall351762c2011-02-07 10:33:21 +0000542 layout.reserve(block->capturesCXXThis() +
543 (block->capture_end() - block->capture_begin()));
544
545 CharUnits maxFieldAlign;
546
547 // First, 'this'.
548 if (block->capturesCXXThis()) {
Eli Friedmanc6036aa2013-07-12 22:05:26 +0000549 assert(CGF && CGF->CurFuncDecl && isa<CXXMethodDecl>(CGF->CurFuncDecl) &&
550 "Can't capture 'this' outside a method");
551 QualType thisType = cast<CXXMethodDecl>(CGF->CurFuncDecl)->getThisType(C);
John McCall351762c2011-02-07 10:33:21 +0000552
John McCall7f416cc2015-09-08 08:05:57 +0000553 // Theoretically, this could be in a different address space, so
554 // don't assume standard pointer size/align.
Jay Foad7c57be32011-07-11 09:56:20 +0000555 llvm::Type *llvmType = CGM.getTypes().ConvertType(thisType);
John McCall351762c2011-02-07 10:33:21 +0000556 std::pair<CharUnits,CharUnits> tinfo
557 = CGM.getContext().getTypeInfoInChars(thisType);
558 maxFieldAlign = std::max(maxFieldAlign, tinfo.second);
559
Fariborz Jahanianc8892052013-01-17 00:25:06 +0000560 layout.push_back(BlockLayoutChunk(tinfo.second, tinfo.first,
561 Qualifiers::OCL_None,
Akira Hatanakad542ccf2016-09-16 00:02:06 +0000562 nullptr, llvmType, thisType));
John McCall351762c2011-02-07 10:33:21 +0000563 }
564
565 // Next, all the block captures.
Aaron Ballman9371dd22014-03-14 18:34:04 +0000566 for (const auto &CI : block->captures()) {
567 const VarDecl *variable = CI.getVariable();
John McCall351762c2011-02-07 10:33:21 +0000568
Akira Hatanaka8e57b072018-10-01 21:51:28 +0000569 if (CI.isEscapingByref()) {
John McCall351762c2011-02-07 10:33:21 +0000570 // We have to copy/dispose of the __block reference.
571 info.NeedsCopyDispose = true;
572
John McCall351762c2011-02-07 10:33:21 +0000573 // Just use void* instead of a pointer to the byref type.
John McCall7f416cc2015-09-08 08:05:57 +0000574 CharUnits align = CGM.getPointerAlign();
575 maxFieldAlign = std::max(maxFieldAlign, align);
John McCall351762c2011-02-07 10:33:21 +0000576
Akira Hatanaka2a5e4632018-08-22 13:41:19 +0000577 // Since a __block variable cannot be captured by lambdas, its type and
578 // the capture field type should always match.
579 assert(getCaptureFieldType(*CGF, CI) == variable->getType() &&
580 "capture type differs from the variable type");
John McCall7f416cc2015-09-08 08:05:57 +0000581 layout.push_back(BlockLayoutChunk(align, CGM.getPointerSize(),
582 Qualifiers::OCL_None, &CI,
Akira Hatanakad542ccf2016-09-16 00:02:06 +0000583 CGM.VoidPtrTy, variable->getType()));
John McCall351762c2011-02-07 10:33:21 +0000584 continue;
585 }
586
587 // Otherwise, build a layout chunk with the size and alignment of
588 // the declaration.
Richard Smithdafff942012-01-14 04:30:29 +0000589 if (llvm::Constant *constant = tryCaptureAsConstant(CGM, CGF, variable)) {
John McCall351762c2011-02-07 10:33:21 +0000590 info.Captures[variable] = CGBlockInfo::Capture::makeConstant(constant);
591 continue;
592 }
593
Akira Hatanaka2a5e4632018-08-22 13:41:19 +0000594 QualType VT = getCaptureFieldType(*CGF, CI);
595
John McCall31168b02011-06-15 23:02:42 +0000596 // If we have a lifetime qualifier, honor it for capture purposes.
597 // That includes *not* copying it if it's __unsafe_unretained.
Akira Hatanaka2a5e4632018-08-22 13:41:19 +0000598 Qualifiers::ObjCLifetime lifetime = VT.getObjCLifetime();
Fariborz Jahanianc8892052013-01-17 00:25:06 +0000599 if (lifetime) {
John McCall31168b02011-06-15 23:02:42 +0000600 switch (lifetime) {
601 case Qualifiers::OCL_None: llvm_unreachable("impossible");
602 case Qualifiers::OCL_ExplicitNone:
603 case Qualifiers::OCL_Autoreleasing:
604 break;
John McCall351762c2011-02-07 10:33:21 +0000605
John McCall31168b02011-06-15 23:02:42 +0000606 case Qualifiers::OCL_Strong:
607 case Qualifiers::OCL_Weak:
608 info.NeedsCopyDispose = true;
609 }
610
611 // Block pointers require copy/dispose. So do Objective-C pointers.
Akira Hatanaka2a5e4632018-08-22 13:41:19 +0000612 } else if (VT->isObjCRetainableType()) {
John McCall00b2bbb2015-11-19 02:28:03 +0000613 // But honor the inert __unsafe_unretained qualifier, which doesn't
614 // actually make it into the type system.
Akira Hatanaka2a5e4632018-08-22 13:41:19 +0000615 if (VT->isObjCInertUnsafeUnretainedType()) {
John McCall00b2bbb2015-11-19 02:28:03 +0000616 lifetime = Qualifiers::OCL_ExplicitNone;
617 } else {
618 info.NeedsCopyDispose = true;
619 // used for mrr below.
620 lifetime = Qualifiers::OCL_Strong;
621 }
John McCall351762c2011-02-07 10:33:21 +0000622
623 // So do types that require non-trivial copy construction.
Aaron Ballman9371dd22014-03-14 18:34:04 +0000624 } else if (CI.hasCopyExpr()) {
John McCall351762c2011-02-07 10:33:21 +0000625 info.NeedsCopyDispose = true;
626 info.HasCXXObject = true;
Akira Hatanaka2a5e4632018-08-22 13:41:19 +0000627 if (!VT->getAsCXXRecordDecl()->isExternallyVisible())
Akira Hatanaka9978da32018-08-10 15:09:24 +0000628 info.CapturesNonExternalType = true;
John McCall351762c2011-02-07 10:33:21 +0000629
Akira Hatanaka7275da02018-02-28 07:15:55 +0000630 // So do C structs that require non-trivial copy construction or
631 // destruction.
Akira Hatanaka2a5e4632018-08-22 13:41:19 +0000632 } else if (VT.isNonTrivialToPrimitiveCopy() == QualType::PCK_Struct ||
633 VT.isDestructedType() == QualType::DK_nontrivial_c_struct) {
Akira Hatanaka7275da02018-02-28 07:15:55 +0000634 info.NeedsCopyDispose = true;
635
John McCall351762c2011-02-07 10:33:21 +0000636 // And so do types with destructors.
David Blaikiebbafb8a2012-03-11 07:00:24 +0000637 } else if (CGM.getLangOpts().CPlusPlus) {
Akira Hatanaka2a5e4632018-08-22 13:41:19 +0000638 if (const CXXRecordDecl *record = VT->getAsCXXRecordDecl()) {
John McCall351762c2011-02-07 10:33:21 +0000639 if (!record->hasTrivialDestructor()) {
640 info.HasCXXObject = true;
641 info.NeedsCopyDispose = true;
Akira Hatanaka9978da32018-08-10 15:09:24 +0000642 if (!record->isExternallyVisible())
643 info.CapturesNonExternalType = true;
John McCall351762c2011-02-07 10:33:21 +0000644 }
645 }
646 }
647
Fariborz Jahanianf0cda632011-10-31 23:44:33 +0000648 CharUnits size = C.getTypeSizeInChars(VT);
Fariborz Jahanian10317ea2011-11-02 22:53:43 +0000649 CharUnits align = C.getDeclAlign(variable);
Fangrui Song6907ce22018-07-30 19:24:48 +0000650
John McCall351762c2011-02-07 10:33:21 +0000651 maxFieldAlign = std::max(maxFieldAlign, align);
652
Jay Foad7c57be32011-07-11 09:56:20 +0000653 llvm::Type *llvmType =
Fariborz Jahanianf0cda632011-10-31 23:44:33 +0000654 CGM.getTypes().ConvertTypeForMem(VT);
Fangrui Song6907ce22018-07-30 19:24:48 +0000655
Akira Hatanakad542ccf2016-09-16 00:02:06 +0000656 layout.push_back(
657 BlockLayoutChunk(align, size, lifetime, &CI, llvmType, VT));
John McCall351762c2011-02-07 10:33:21 +0000658 }
659
660 // If that was everything, we're done here.
661 if (layout.empty()) {
662 info.StructureType =
663 llvm::StructType::get(CGM.getLLVMContext(), elementTypes, true);
664 info.CanBeGlobal = true;
665 return;
666 }
667
668 // Sort the layout by alignment. We have to use a stable sort here
669 // to get reproducible results. There should probably be an
670 // llvm::array_pod_stable_sort.
671 std::stable_sort(layout.begin(), layout.end());
Fangrui Song6907ce22018-07-30 19:24:48 +0000672
Fariborz Jahanian4cf177e2012-12-04 17:20:57 +0000673 // Needed for blocks layout info.
674 info.BlockHeaderForcedGapOffset = info.BlockSize;
675 info.BlockHeaderForcedGapSize = CharUnits::Zero();
Fangrui Song6907ce22018-07-30 19:24:48 +0000676
John McCall351762c2011-02-07 10:33:21 +0000677 CharUnits &blockSize = info.BlockSize;
678 info.BlockAlign = std::max(maxFieldAlign, info.BlockAlign);
679
680 // Assuming that the first byte in the header is maximally aligned,
681 // get the alignment of the first byte following the header.
682 CharUnits endAlign = getLowBit(blockSize);
683
684 // If the end of the header isn't satisfactorily aligned for the
685 // maximum thing, look for things that are okay with the header-end
686 // alignment, and keep appending them until we get something that's
687 // aligned right. This algorithm is only guaranteed optimal if
688 // that condition is satisfied at some point; otherwise we can get
689 // things like:
690 // header // next byte has alignment 4
691 // something_with_size_5; // next byte has alignment 1
692 // something_with_alignment_8;
693 // which has 7 bytes of padding, as opposed to the naive solution
694 // which might have less (?).
695 if (endAlign < maxFieldAlign) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +0000696 SmallVectorImpl<BlockLayoutChunk>::iterator
John McCall351762c2011-02-07 10:33:21 +0000697 li = layout.begin() + 1, le = layout.end();
698
699 // Look for something that the header end is already
700 // satisfactorily aligned for.
701 for (; li != le && endAlign < li->Alignment; ++li)
702 ;
703
704 // If we found something that's naturally aligned for the end of
705 // the header, keep adding things...
706 if (li != le) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +0000707 SmallVectorImpl<BlockLayoutChunk>::iterator first = li;
John McCall351762c2011-02-07 10:33:21 +0000708 for (; li != le; ++li) {
709 assert(endAlign >= li->Alignment);
710
John McCall7f416cc2015-09-08 08:05:57 +0000711 li->setIndex(info, elementTypes.size(), blockSize);
John McCall351762c2011-02-07 10:33:21 +0000712 elementTypes.push_back(li->Type);
713 blockSize += li->Size;
714 endAlign = getLowBit(blockSize);
715
716 // ...until we get to the alignment of the maximum field.
Fariborz Jahanian4cf177e2012-12-04 17:20:57 +0000717 if (endAlign >= maxFieldAlign) {
John McCall351762c2011-02-07 10:33:21 +0000718 break;
Fariborz Jahanian4cf177e2012-12-04 17:20:57 +0000719 }
John McCall351762c2011-02-07 10:33:21 +0000720 }
John McCall351762c2011-02-07 10:33:21 +0000721 // Don't re-append everything we just appended.
722 layout.erase(first, li);
723 }
724 }
725
John McCallac0350a2012-04-26 21:14:42 +0000726 assert(endAlign == getLowBit(blockSize));
Fangrui Song6907ce22018-07-30 19:24:48 +0000727
John McCall351762c2011-02-07 10:33:21 +0000728 // At this point, we just have to add padding if the end align still
729 // isn't aligned right.
730 if (endAlign < maxFieldAlign) {
Rui Ueyama83aa9792016-01-14 21:00:27 +0000731 CharUnits newBlockSize = blockSize.alignTo(maxFieldAlign);
John McCallac0350a2012-04-26 21:14:42 +0000732 CharUnits padding = newBlockSize - blockSize;
John McCall351762c2011-02-07 10:33:21 +0000733
John McCall7f416cc2015-09-08 08:05:57 +0000734 // If we haven't yet added any fields, remember that there was an
735 // initial gap; this need to go into the block layout bit map.
736 if (blockSize == info.BlockHeaderForcedGapOffset) {
737 info.BlockHeaderForcedGapSize = padding;
738 }
739
John McCalle3dc1702011-02-15 09:22:45 +0000740 elementTypes.push_back(llvm::ArrayType::get(CGM.Int8Ty,
741 padding.getQuantity()));
John McCallac0350a2012-04-26 21:14:42 +0000742 blockSize = newBlockSize;
John McCall1db0a2f2012-05-01 20:28:00 +0000743 endAlign = getLowBit(blockSize); // might be > maxFieldAlign
John McCall351762c2011-02-07 10:33:21 +0000744 }
745
John McCall1db0a2f2012-05-01 20:28:00 +0000746 assert(endAlign >= maxFieldAlign);
John McCallac0350a2012-04-26 21:14:42 +0000747 assert(endAlign == getLowBit(blockSize));
John McCall351762c2011-02-07 10:33:21 +0000748 // Slam everything else on now. This works because they have
749 // strictly decreasing alignment and we expect that size is always a
750 // multiple of alignment.
Chris Lattner0e62c1c2011-07-23 10:55:15 +0000751 for (SmallVectorImpl<BlockLayoutChunk>::iterator
John McCall351762c2011-02-07 10:33:21 +0000752 li = layout.begin(), le = layout.end(); li != le; ++li) {
Fariborz Jahanian9c56fc92014-08-12 15:51:49 +0000753 if (endAlign < li->Alignment) {
754 // size may not be multiple of alignment. This can only happen with
755 // an over-aligned variable. We will be adding a padding field to
756 // make the size be multiple of alignment.
757 CharUnits padding = li->Alignment - endAlign;
758 elementTypes.push_back(llvm::ArrayType::get(CGM.Int8Ty,
759 padding.getQuantity()));
760 blockSize += padding;
761 endAlign = getLowBit(blockSize);
762 }
John McCall351762c2011-02-07 10:33:21 +0000763 assert(endAlign >= li->Alignment);
John McCall7f416cc2015-09-08 08:05:57 +0000764 li->setIndex(info, elementTypes.size(), blockSize);
John McCall351762c2011-02-07 10:33:21 +0000765 elementTypes.push_back(li->Type);
766 blockSize += li->Size;
767 endAlign = getLowBit(blockSize);
768 }
769
770 info.StructureType =
771 llvm::StructType::get(CGM.getLLVMContext(), elementTypes, true);
772}
773
John McCall08ef4662011-11-10 08:15:53 +0000774/// Enter the scope of a block. This should be run at the entrance to
775/// a full-expression so that the block's cleanups are pushed at the
776/// right place in the stack.
777static void enterBlockScope(CodeGenFunction &CGF, BlockDecl *block) {
John McCall8c38d352012-04-13 18:44:05 +0000778 assert(CGF.HaveInsertPoint());
779
John McCall08ef4662011-11-10 08:15:53 +0000780 // Allocate the block info and place it at the head of the list.
781 CGBlockInfo &blockInfo =
782 *new CGBlockInfo(block, CGF.CurFn->getName());
783 blockInfo.NextBlockInfo = CGF.FirstBlockInfo;
784 CGF.FirstBlockInfo = &blockInfo;
785
786 // Compute information about the layout, etc., of this block,
787 // pushing cleanups as necessary.
Richard Smithdafff942012-01-14 04:30:29 +0000788 computeBlockInfo(CGF.CGM, &CGF, blockInfo);
John McCall08ef4662011-11-10 08:15:53 +0000789
790 // Nothing else to do if it can be global.
791 if (blockInfo.CanBeGlobal) return;
792
793 // Make the allocation for the block.
John McCall7f416cc2015-09-08 08:05:57 +0000794 blockInfo.LocalAddress = CGF.CreateTempAlloca(blockInfo.StructureType,
795 blockInfo.BlockAlign, "block");
John McCall08ef4662011-11-10 08:15:53 +0000796
797 // If there are cleanups to emit, enter them (but inactive).
798 if (!blockInfo.NeedsCopyDispose) return;
799
800 // Walk through the captures (in order) and find the ones not
801 // captured by constant.
Aaron Ballman9371dd22014-03-14 18:34:04 +0000802 for (const auto &CI : block->captures()) {
John McCall08ef4662011-11-10 08:15:53 +0000803 // Ignore __block captures; there's nothing special in the
804 // on-stack block that we need to do for them.
Aaron Ballman9371dd22014-03-14 18:34:04 +0000805 if (CI.isByRef()) continue;
John McCall08ef4662011-11-10 08:15:53 +0000806
807 // Ignore variables that are constant-captured.
Aaron Ballman9371dd22014-03-14 18:34:04 +0000808 const VarDecl *variable = CI.getVariable();
John McCall08ef4662011-11-10 08:15:53 +0000809 CGBlockInfo::Capture &capture = blockInfo.getCapture(variable);
810 if (capture.isConstant()) continue;
811
812 // Ignore objects that aren't destructed.
Akira Hatanakaf1b3fc72017-02-14 06:46:55 +0000813 QualType VT = getCaptureFieldType(CGF, CI);
814 QualType::DestructionKind dtorKind = VT.isDestructedType();
John McCall08ef4662011-11-10 08:15:53 +0000815 if (dtorKind == QualType::DK_none) continue;
816
817 CodeGenFunction::Destroyer *destroyer;
818
819 // Block captures count as local values and have imprecise semantics.
820 // They also can't be arrays, so need to worry about that.
Akira Hatanakaa6b6dcc2017-04-28 18:50:57 +0000821 //
822 // For const-qualified captures, emit clang.arc.use to ensure the captured
823 // object doesn't get released while we are still depending on its validity
824 // within the block.
Saleem Abdulrasoold95f6252017-05-05 18:39:06 +0000825 if (VT.isConstQualified() &&
826 VT.getObjCLifetime() == Qualifiers::OCL_Strong &&
827 CGF.CGM.getCodeGenOpts().OptimizationLevel != 0) {
828 assert(CGF.CGM.getLangOpts().ObjCAutoRefCount &&
829 "expected ObjC ARC to be enabled");
Akira Hatanakaa6b6dcc2017-04-28 18:50:57 +0000830 destroyer = CodeGenFunction::emitARCIntrinsicUse;
Saleem Abdulrasoold95f6252017-05-05 18:39:06 +0000831 } else if (dtorKind == QualType::DK_objc_strong_lifetime) {
Peter Collingbourne1425b452012-01-26 03:33:36 +0000832 destroyer = CodeGenFunction::destroyARCStrongImprecise;
John McCall08ef4662011-11-10 08:15:53 +0000833 } else {
Peter Collingbourne1425b452012-01-26 03:33:36 +0000834 destroyer = CGF.getDestroyer(dtorKind);
John McCall08ef4662011-11-10 08:15:53 +0000835 }
836
837 // GEP down to the address.
John McCall7f416cc2015-09-08 08:05:57 +0000838 Address addr = CGF.Builder.CreateStructGEP(blockInfo.LocalAddress,
839 capture.getIndex(),
840 capture.getOffset());
John McCall08ef4662011-11-10 08:15:53 +0000841
John McCallf4beacd2011-11-10 10:43:54 +0000842 // We can use that GEP as the dominating IP.
843 if (!blockInfo.DominatingIP)
John McCall7f416cc2015-09-08 08:05:57 +0000844 blockInfo.DominatingIP = cast<llvm::Instruction>(addr.getPointer());
John McCallf4beacd2011-11-10 10:43:54 +0000845
John McCall08ef4662011-11-10 08:15:53 +0000846 CleanupKind cleanupKind = InactiveNormalCleanup;
847 bool useArrayEHCleanup = CGF.needsEHCleanup(dtorKind);
Fangrui Song6907ce22018-07-30 19:24:48 +0000848 if (useArrayEHCleanup)
John McCall08ef4662011-11-10 08:15:53 +0000849 cleanupKind = InactiveNormalAndEHCleanup;
850
Akira Hatanakaf1b3fc72017-02-14 06:46:55 +0000851 CGF.pushDestroy(cleanupKind, addr, VT,
Peter Collingbourne1425b452012-01-26 03:33:36 +0000852 destroyer, useArrayEHCleanup);
John McCall08ef4662011-11-10 08:15:53 +0000853
854 // Remember where that cleanup was.
855 capture.setCleanup(CGF.EHStack.stable_begin());
856 }
857}
858
859/// Enter a full-expression with a non-trivial number of objects to
860/// clean up. This is in this file because, at the moment, the only
861/// kind of cleanup object is a BlockDecl*.
862void CodeGenFunction::enterNonTrivialFullExpression(const ExprWithCleanups *E) {
863 assert(E->getNumObjects() != 0);
George Burgess IV8b93a592018-03-02 20:10:38 +0000864 for (const ExprWithCleanups::CleanupObject &C : E->getObjects())
865 enterBlockScope(*this, C);
John McCall08ef4662011-11-10 08:15:53 +0000866}
867
868/// Find the layout for the given block in a linked list and remove it.
869static CGBlockInfo *findAndRemoveBlockInfo(CGBlockInfo **head,
870 const BlockDecl *block) {
871 while (true) {
872 assert(head && *head);
873 CGBlockInfo *cur = *head;
874
875 // If this is the block we're looking for, splice it out of the list.
876 if (cur->getBlockDecl() == block) {
877 *head = cur->NextBlockInfo;
878 return cur;
879 }
880
881 head = &cur->NextBlockInfo;
882 }
883}
884
885/// Destroy a chain of block layouts.
886void CodeGenFunction::destroyBlockInfos(CGBlockInfo *head) {
887 assert(head && "destroying an empty chain");
888 do {
889 CGBlockInfo *cur = head;
890 head = cur->NextBlockInfo;
891 delete cur;
Craig Topper8a13c412014-05-21 05:09:00 +0000892 } while (head != nullptr);
John McCall08ef4662011-11-10 08:15:53 +0000893}
894
John McCall351762c2011-02-07 10:33:21 +0000895/// Emit a block literal expression in the current function.
Yaxun Liufa13d012018-02-15 16:39:19 +0000896llvm::Value *CodeGenFunction::EmitBlockLiteral(const BlockExpr *blockExpr) {
John McCall08ef4662011-11-10 08:15:53 +0000897 // If the block has no captures, we won't have a pre-computed
898 // layout for it.
899 if (!blockExpr->getBlockDecl()->hasCaptures()) {
Yaxun Liuc2a87a02017-10-14 12:23:50 +0000900 // The block literal is emitted as a global variable, and the block invoke
901 // function has to be extracted from its initializer.
902 if (llvm::Constant *Block = CGM.getAddrOfGlobalBlockIfEmitted(blockExpr)) {
George Burgess IVe3763372016-12-22 02:50:20 +0000903 return Block;
Yaxun Liuc2a87a02017-10-14 12:23:50 +0000904 }
John McCall08ef4662011-11-10 08:15:53 +0000905 CGBlockInfo blockInfo(blockExpr->getBlockDecl(), CurFn->getName());
Richard Smithdafff942012-01-14 04:30:29 +0000906 computeBlockInfo(CGM, this, blockInfo);
John McCall08ef4662011-11-10 08:15:53 +0000907 blockInfo.BlockExpression = blockExpr;
Yaxun Liufa13d012018-02-15 16:39:19 +0000908 return EmitBlockLiteral(blockInfo);
John McCall08ef4662011-11-10 08:15:53 +0000909 }
John McCall351762c2011-02-07 10:33:21 +0000910
John McCall08ef4662011-11-10 08:15:53 +0000911 // Find the block info for this block and take ownership of it.
Ahmed Charlesb8984322014-03-07 20:03:18 +0000912 std::unique_ptr<CGBlockInfo> blockInfo;
John McCall08ef4662011-11-10 08:15:53 +0000913 blockInfo.reset(findAndRemoveBlockInfo(&FirstBlockInfo,
914 blockExpr->getBlockDecl()));
John McCall351762c2011-02-07 10:33:21 +0000915
John McCall08ef4662011-11-10 08:15:53 +0000916 blockInfo->BlockExpression = blockExpr;
Yaxun Liufa13d012018-02-15 16:39:19 +0000917 return EmitBlockLiteral(*blockInfo);
John McCall08ef4662011-11-10 08:15:53 +0000918}
919
Yaxun Liufa13d012018-02-15 16:39:19 +0000920llvm::Value *CodeGenFunction::EmitBlockLiteral(const CGBlockInfo &blockInfo) {
Yaxun Liu10712d92017-10-04 20:32:17 +0000921 bool IsOpenCL = CGM.getContext().getLangOpts().OpenCL;
Sven van Haastregtda3b6322018-10-02 13:02:24 +0000922 auto GenVoidPtrTy =
923 IsOpenCL ? CGM.getOpenCLRuntime().getGenericVoidPointerType() : VoidPtrTy;
924 LangAS GenVoidPtrAddr = IsOpenCL ? LangAS::opencl_generic : LangAS::Default;
925 auto GenVoidPtrSize = CharUnits::fromQuantity(
926 CGM.getTarget().getPointerWidth(
927 CGM.getContext().getTargetAddressSpace(GenVoidPtrAddr)) /
928 8);
John McCall08ef4662011-11-10 08:15:53 +0000929 // Using the computed layout, generate the actual block function.
Eli Friedman98b01ed2012-03-01 04:01:32 +0000930 bool isLambdaConv = blockInfo.getBlockDecl()->isConversionFromLambda();
Vedant Kumar29477dc2017-12-08 02:47:58 +0000931 CodeGenFunction BlockCGF{CGM, true};
932 BlockCGF.SanOpts = SanOpts;
933 auto *InvokeFn = BlockCGF.GenerateBlockFunction(
Yaxun Liu10712d92017-10-04 20:32:17 +0000934 CurGD, blockInfo, LocalDeclMap, isLambdaConv, blockInfo.CanBeGlobal);
Sven van Haastregtda3b6322018-10-02 13:02:24 +0000935 auto *blockFn = llvm::ConstantExpr::getPointerCast(InvokeFn, GenVoidPtrTy);
John McCall351762c2011-02-07 10:33:21 +0000936
937 // If there is nothing to capture, we can emit this as a global block.
938 if (blockInfo.CanBeGlobal)
Akira Hatanakaba0367a2017-09-22 21:32:06 +0000939 return CGM.getAddrOfGlobalBlockIfEmitted(blockInfo.BlockExpression);
John McCall351762c2011-02-07 10:33:21 +0000940
941 // Otherwise, we have to emit this as a local block.
942
John McCall7f416cc2015-09-08 08:05:57 +0000943 Address blockAddr = blockInfo.LocalAddress;
944 assert(blockAddr.isValid() && "block has no address!");
John McCall351762c2011-02-07 10:33:21 +0000945
Yaxun Liu10712d92017-10-04 20:32:17 +0000946 llvm::Constant *isa;
947 llvm::Constant *descriptor;
948 BlockFlags flags;
949 if (!IsOpenCL) {
Akira Hatanakadbfa4532018-07-20 17:10:32 +0000950 // If the block is non-escaping, set field 'isa 'to NSConcreteGlobalBlock
951 // and set the BLOCK_IS_GLOBAL bit of field 'flags'. Copying a non-escaping
952 // block just returns the original block and releasing it is a no-op.
953 llvm::Constant *blockISA = blockInfo.getBlockDecl()->doesNotEscape()
954 ? CGM.getNSConcreteGlobalBlock()
955 : CGM.getNSConcreteStackBlock();
956 isa = llvm::ConstantExpr::getBitCast(blockISA, VoidPtrTy);
Yaxun Liu10712d92017-10-04 20:32:17 +0000957
958 // Build the block descriptor.
959 descriptor = buildBlockDescriptor(CGM, blockInfo);
960
961 // Compute the initial on-stack block flags.
962 flags = BLOCK_HAS_SIGNATURE;
963 if (blockInfo.HasCapturedVariableLayout)
964 flags |= BLOCK_HAS_EXTENDED_LAYOUT;
Akira Hatanakadbfa4532018-07-20 17:10:32 +0000965 if (blockInfo.needsCopyDisposeHelpers())
Yaxun Liu10712d92017-10-04 20:32:17 +0000966 flags |= BLOCK_HAS_COPY_DISPOSE;
967 if (blockInfo.HasCXXObject)
968 flags |= BLOCK_HAS_CXX_OBJ;
969 if (blockInfo.UsesStret)
970 flags |= BLOCK_USE_STRET;
Akira Hatanakadbfa4532018-07-20 17:10:32 +0000971 if (blockInfo.getBlockDecl()->doesNotEscape())
972 flags |= BLOCK_IS_NOESCAPE | BLOCK_IS_GLOBAL;
Yaxun Liu10712d92017-10-04 20:32:17 +0000973 }
John McCall351762c2011-02-07 10:33:21 +0000974
John McCall7f416cc2015-09-08 08:05:57 +0000975 auto projectField =
976 [&](unsigned index, CharUnits offset, const Twine &name) -> Address {
977 return Builder.CreateStructGEP(blockAddr, index, offset, name);
978 };
979 auto storeField =
980 [&](llvm::Value *value, unsigned index, CharUnits offset,
981 const Twine &name) {
982 Builder.CreateStore(value, projectField(index, offset, name));
983 };
984
985 // Initialize the block header.
986 {
987 // We assume all the header fields are densely packed.
988 unsigned index = 0;
989 CharUnits offset;
990 auto addHeaderField =
991 [&](llvm::Value *value, CharUnits size, const Twine &name) {
992 storeField(value, index, offset, name);
993 offset += size;
994 index++;
995 };
996
Yaxun Liu10712d92017-10-04 20:32:17 +0000997 if (!IsOpenCL) {
998 addHeaderField(isa, getPointerSize(), "block.isa");
999 addHeaderField(llvm::ConstantInt::get(IntTy, flags.getBitMask()),
1000 getIntSize(), "block.flags");
1001 addHeaderField(llvm::ConstantInt::get(IntTy, 0), getIntSize(),
1002 "block.reserved");
1003 } else {
1004 addHeaderField(
1005 llvm::ConstantInt::get(IntTy, blockInfo.BlockSize.getQuantity()),
1006 getIntSize(), "block.size");
1007 addHeaderField(
1008 llvm::ConstantInt::get(IntTy, blockInfo.BlockAlign.getQuantity()),
1009 getIntSize(), "block.align");
1010 }
Sven van Haastregtda3b6322018-10-02 13:02:24 +00001011 addHeaderField(blockFn, GenVoidPtrSize, "block.invoke");
1012 if (!IsOpenCL)
Yaxun Liu10712d92017-10-04 20:32:17 +00001013 addHeaderField(descriptor, getPointerSize(), "block.descriptor");
Sven van Haastregtda3b6322018-10-02 13:02:24 +00001014 else if (auto *Helper =
1015 CGM.getTargetCodeGenInfo().getTargetOpenCLBlockHelper()) {
Yaxun Liu10712d92017-10-04 20:32:17 +00001016 for (auto I : Helper->getCustomFieldValues(*this, blockInfo)) {
1017 addHeaderField(
1018 I.first,
1019 CharUnits::fromQuantity(
1020 CGM.getDataLayout().getTypeAllocSize(I.first->getType())),
1021 I.second);
1022 }
1023 }
John McCall7f416cc2015-09-08 08:05:57 +00001024 }
John McCall351762c2011-02-07 10:33:21 +00001025
1026 // Finally, capture all the values into the block.
1027 const BlockDecl *blockDecl = blockInfo.getBlockDecl();
1028
1029 // First, 'this'.
1030 if (blockDecl->capturesCXXThis()) {
John McCall7f416cc2015-09-08 08:05:57 +00001031 Address addr = projectField(blockInfo.CXXThisIndex, blockInfo.CXXThisOffset,
1032 "block.captured-this.addr");
John McCall351762c2011-02-07 10:33:21 +00001033 Builder.CreateStore(LoadCXXThis(), addr);
1034 }
1035
1036 // Next, captured variables.
Aaron Ballman9371dd22014-03-14 18:34:04 +00001037 for (const auto &CI : blockDecl->captures()) {
1038 const VarDecl *variable = CI.getVariable();
John McCall351762c2011-02-07 10:33:21 +00001039 const CGBlockInfo::Capture &capture = blockInfo.getCapture(variable);
1040
1041 // Ignore constant captures.
1042 if (capture.isConstant()) continue;
1043
Akira Hatanakad542ccf2016-09-16 00:02:06 +00001044 QualType type = capture.fieldType();
John McCall351762c2011-02-07 10:33:21 +00001045
1046 // This will be a [[type]]*, except that a byref entry will just be
1047 // an i8**.
John McCall7f416cc2015-09-08 08:05:57 +00001048 Address blockField =
1049 projectField(capture.getIndex(), capture.getOffset(), "block.captured");
John McCall351762c2011-02-07 10:33:21 +00001050
1051 // Compute the address of the thing we're going to move into the
1052 // block literal.
John McCall7f416cc2015-09-08 08:05:57 +00001053 Address src = Address::invalid();
John McCall351762c2011-02-07 10:33:21 +00001054
Akira Hatanaka1cce6e12016-05-04 18:40:33 +00001055 if (blockDecl->isConversionFromLambda()) {
Eli Friedman2495ab02012-02-25 02:48:22 +00001056 // The lambda capture in a lambda's conversion-to-block-pointer is
Eli Friedman98b01ed2012-03-01 04:01:32 +00001057 // special; we'll simply emit it directly.
John McCall7f416cc2015-09-08 08:05:57 +00001058 src = Address::invalid();
Akira Hatanaka8e57b072018-10-01 21:51:28 +00001059 } else if (CI.isEscapingByref()) {
Akira Hatanaka1cce6e12016-05-04 18:40:33 +00001060 if (BlockInfo && CI.isNested()) {
1061 // We need to use the capture from the enclosing block.
1062 const CGBlockInfo::Capture &enclosingCapture =
1063 BlockInfo->getCapture(variable);
1064
Alexander Kornienko2a8c18d2018-04-06 15:14:32 +00001065 // This is a [[type]]*, except that a byref entry will just be an i8**.
Akira Hatanaka1cce6e12016-05-04 18:40:33 +00001066 src = Builder.CreateStructGEP(LoadBlockStruct(),
1067 enclosingCapture.getIndex(),
1068 enclosingCapture.getOffset(),
1069 "block.capture.addr");
John McCall7f416cc2015-09-08 08:05:57 +00001070 } else {
Akira Hatanaka1cce6e12016-05-04 18:40:33 +00001071 auto I = LocalDeclMap.find(variable);
1072 assert(I != LocalDeclMap.end());
1073 src = I->second;
John McCalla37c2fa2013-03-04 06:32:36 +00001074 }
Akira Hatanaka1cce6e12016-05-04 18:40:33 +00001075 } else {
1076 DeclRefExpr declRef(const_cast<VarDecl *>(variable),
1077 /*RefersToEnclosingVariableOrCapture*/ CI.isNested(),
1078 type.getNonReferenceType(), VK_LValue,
1079 SourceLocation());
1080 src = EmitDeclRefLValue(&declRef).getAddress();
1081 };
John McCall351762c2011-02-07 10:33:21 +00001082
1083 // For byrefs, we just write the pointer to the byref struct into
1084 // the block field. There's no need to chase the forwarding
1085 // pointer at this point, since we're building something that will
1086 // live a shorter life than the stack byref anyway.
Akira Hatanaka8e57b072018-10-01 21:51:28 +00001087 if (CI.isEscapingByref()) {
John McCalle3dc1702011-02-15 09:22:45 +00001088 // Get a void* that points to the byref struct.
John McCall7f416cc2015-09-08 08:05:57 +00001089 llvm::Value *byrefPointer;
Aaron Ballman9371dd22014-03-14 18:34:04 +00001090 if (CI.isNested())
John McCall7f416cc2015-09-08 08:05:57 +00001091 byrefPointer = Builder.CreateLoad(src, "byref.capture");
John McCall351762c2011-02-07 10:33:21 +00001092 else
John McCall7f416cc2015-09-08 08:05:57 +00001093 byrefPointer = Builder.CreateBitCast(src.getPointer(), VoidPtrTy);
John McCall351762c2011-02-07 10:33:21 +00001094
John McCalle3dc1702011-02-15 09:22:45 +00001095 // Write that void* into the capture field.
John McCall7f416cc2015-09-08 08:05:57 +00001096 Builder.CreateStore(byrefPointer, blockField);
John McCall351762c2011-02-07 10:33:21 +00001097
1098 // If we have a copy constructor, evaluate that into the block field.
Aaron Ballman9371dd22014-03-14 18:34:04 +00001099 } else if (const Expr *copyExpr = CI.getCopyExpr()) {
Eli Friedman98b01ed2012-03-01 04:01:32 +00001100 if (blockDecl->isConversionFromLambda()) {
1101 // If we have a lambda conversion, emit the expression
1102 // directly into the block instead.
Eli Friedman98b01ed2012-03-01 04:01:32 +00001103 AggValueSlot Slot =
John McCall7f416cc2015-09-08 08:05:57 +00001104 AggValueSlot::forAddr(blockField, Qualifiers(),
Eli Friedman98b01ed2012-03-01 04:01:32 +00001105 AggValueSlot::IsDestructed,
1106 AggValueSlot::DoesNotNeedGCBarriers,
Richard Smithe78fac52018-04-05 20:52:58 +00001107 AggValueSlot::IsNotAliased,
1108 AggValueSlot::DoesNotOverlap);
Eli Friedman98b01ed2012-03-01 04:01:32 +00001109 EmitAggExpr(copyExpr, Slot);
1110 } else {
1111 EmitSynthesizedCXXCopyCtor(blockField, src, copyExpr);
1112 }
John McCall351762c2011-02-07 10:33:21 +00001113
1114 // If it's a reference variable, copy the reference into the block field.
Fariborz Jahanian10317ea2011-11-02 22:53:43 +00001115 } else if (type->isReferenceType()) {
Akira Hatanaka1cce6e12016-05-04 18:40:33 +00001116 Builder.CreateStore(src.getPointer(), blockField);
John McCall4d14a902013-04-08 23:27:49 +00001117
Akira Hatanakaa6b6dcc2017-04-28 18:50:57 +00001118 // If type is const-qualified, copy the value into the block field.
1119 } else if (type.isConstQualified() &&
Akira Hatanaka855d70c2017-05-09 01:20:05 +00001120 type.getObjCLifetime() == Qualifiers::OCL_Strong &&
1121 CGM.getCodeGenOpts().OptimizationLevel != 0) {
Akira Hatanakaa6b6dcc2017-04-28 18:50:57 +00001122 llvm::Value *value = Builder.CreateLoad(src, "captured");
1123 Builder.CreateStore(value, blockField);
1124
John McCall4d14a902013-04-08 23:27:49 +00001125 // If this is an ARC __strong block-pointer variable, don't do a
1126 // block copy.
1127 //
1128 // TODO: this can be generalized into the normal initialization logic:
1129 // we should never need to do a block-copy when initializing a local
1130 // variable, because the local variable's lifetime should be strictly
1131 // contained within the stack block's.
1132 } else if (type.getObjCLifetime() == Qualifiers::OCL_Strong &&
1133 type->isBlockPointerType()) {
1134 // Load the block and do a simple retain.
John McCall7f416cc2015-09-08 08:05:57 +00001135 llvm::Value *value = Builder.CreateLoad(src, "block.captured_block");
John McCall4d14a902013-04-08 23:27:49 +00001136 value = EmitARCRetainNonBlock(value);
1137
1138 // Do a primitive store to the block field.
John McCall7f416cc2015-09-08 08:05:57 +00001139 Builder.CreateStore(value, blockField);
John McCall351762c2011-02-07 10:33:21 +00001140
1141 // Otherwise, fake up a POD copy into the block field.
1142 } else {
John McCall31168b02011-06-15 23:02:42 +00001143 // Fake up a new variable so that EmitScalarInit doesn't think
1144 // we're referring to the variable in its own initializer.
Alexey Bataev56223232017-06-09 13:40:18 +00001145 ImplicitParamDecl BlockFieldPseudoVar(getContext(), type,
1146 ImplicitParamDecl::Other);
John McCall31168b02011-06-15 23:02:42 +00001147
John McCall93be3f72011-02-07 18:37:40 +00001148 // We use one of these or the other depending on whether the
1149 // reference is nested.
Alexey Bataev19acc3d2015-01-12 10:17:46 +00001150 DeclRefExpr declRef(const_cast<VarDecl *>(variable),
1151 /*RefersToEnclosingVariableOrCapture*/ CI.isNested(),
1152 type, VK_LValue, SourceLocation());
John McCall93be3f72011-02-07 18:37:40 +00001153
Fariborz Jahanian10317ea2011-11-02 22:53:43 +00001154 ImplicitCastExpr l2r(ImplicitCastExpr::OnStack, type, CK_LValueToRValue,
John McCall113bee02012-03-10 09:33:50 +00001155 &declRef, VK_RValue);
David Blaikie7f138812014-12-09 22:04:13 +00001156 // FIXME: Pass a specific location for the expr init so that the store is
1157 // attributed to a reasonable location - otherwise it may be attributed to
1158 // locations of subexpressions in the initialization.
Alexey Bataev56223232017-06-09 13:40:18 +00001159 EmitExprAsInit(&l2r, &BlockFieldPseudoVar,
Ivan A. Kosarev5f8c0ca2017-10-10 09:39:32 +00001160 MakeAddrLValue(blockField, type, AlignmentSource::Decl),
David Blaikie66e41972015-01-14 07:38:27 +00001161 /*captured by init*/ false);
John McCall351762c2011-02-07 10:33:21 +00001162 }
1163
John McCall08ef4662011-11-10 08:15:53 +00001164 // Activate the cleanup if layout pushed one.
Aaron Ballman9371dd22014-03-14 18:34:04 +00001165 if (!CI.isByRef()) {
John McCall08ef4662011-11-10 08:15:53 +00001166 EHScopeStack::stable_iterator cleanup = capture.getCleanup();
1167 if (cleanup.isValid())
John McCallf4beacd2011-11-10 10:43:54 +00001168 ActivateCleanupBlock(cleanup, blockInfo.DominatingIP);
John McCall31168b02011-06-15 23:02:42 +00001169 }
John McCall351762c2011-02-07 10:33:21 +00001170 }
1171
1172 // Cast to the converted block-pointer type, which happens (somewhat
1173 // unfortunately) to be a pointer to function type.
Anastasia Stulovaaf0a7bb2017-01-27 15:11:34 +00001174 llvm::Value *result = Builder.CreatePointerCast(
1175 blockAddr.getPointer(), ConvertType(blockInfo.getBlockExpr()->getType()));
John McCall3882ace2011-01-05 12:14:39 +00001176
Yaxun Liufa13d012018-02-15 16:39:19 +00001177 if (IsOpenCL) {
1178 CGM.getOpenCLRuntime().recordBlockInfo(blockInfo.BlockExpression, InvokeFn,
1179 result);
1180 }
1181
John McCall351762c2011-02-07 10:33:21 +00001182 return result;
Mike Stump85284ba2009-02-13 16:19:19 +00001183}
1184
1185
Chris Lattnera5f58b02011-07-09 17:41:47 +00001186llvm::Type *CodeGenModule::getBlockDescriptorType() {
Mike Stump650c9322009-02-13 15:16:56 +00001187 if (BlockDescriptorType)
1188 return BlockDescriptorType;
1189
Chris Lattnera5f58b02011-07-09 17:41:47 +00001190 llvm::Type *UnsignedLongTy =
Mike Stump650c9322009-02-13 15:16:56 +00001191 getTypes().ConvertType(getContext().UnsignedLongTy);
Mike Stumpb7074c02009-02-13 15:32:32 +00001192
Mike Stump650c9322009-02-13 15:16:56 +00001193 // struct __block_descriptor {
1194 // unsigned long reserved;
1195 // unsigned long block_size;
Blaine Garstfc83aa02010-02-23 21:51:17 +00001196 //
1197 // // later, the following will be added
1198 //
1199 // struct {
1200 // void (*copyHelper)();
1201 // void (*copyHelper)();
1202 // } helpers; // !!! optional
1203 //
1204 // const char *signature; // the block signature
1205 // const char *layout; // reserved
Mike Stump650c9322009-02-13 15:16:56 +00001206 // };
Serge Guelton1d993272017-05-09 19:31:30 +00001207 BlockDescriptorType = llvm::StructType::create(
1208 "struct.__block_descriptor", UnsignedLongTy, UnsignedLongTy);
Mike Stump650c9322009-02-13 15:16:56 +00001209
John McCall351762c2011-02-07 10:33:21 +00001210 // Now form a pointer to that.
Joey Goulyddbda402016-08-10 15:57:02 +00001211 unsigned AddrSpace = 0;
1212 if (getLangOpts().OpenCL)
1213 AddrSpace = getContext().getTargetAddressSpace(LangAS::opencl_constant);
1214 BlockDescriptorType = llvm::PointerType::get(BlockDescriptorType, AddrSpace);
Mike Stump650c9322009-02-13 15:16:56 +00001215 return BlockDescriptorType;
Anders Carlsson2437cbf2009-02-12 00:39:25 +00001216}
1217
Chris Lattnera5f58b02011-07-09 17:41:47 +00001218llvm::Type *CodeGenModule::getGenericBlockLiteralType() {
Mike Stump005c9a62009-02-13 15:25:34 +00001219 if (GenericBlockLiteralType)
1220 return GenericBlockLiteralType;
1221
Chris Lattnera5f58b02011-07-09 17:41:47 +00001222 llvm::Type *BlockDescPtrTy = getBlockDescriptorType();
Mike Stumpb7074c02009-02-13 15:32:32 +00001223
Sven van Haastregtda3b6322018-10-02 13:02:24 +00001224 if (getLangOpts().OpenCL) {
1225 // struct __opencl_block_literal_generic {
1226 // int __size;
1227 // int __align;
1228 // __generic void *__invoke;
1229 // /* custom fields */
1230 // };
1231 SmallVector<llvm::Type *, 8> StructFields(
1232 {IntTy, IntTy, getOpenCLRuntime().getGenericVoidPointerType()});
1233 if (auto *Helper = getTargetCodeGenInfo().getTargetOpenCLBlockHelper()) {
1234 for (auto I : Helper->getCustomFieldTypes())
1235 StructFields.push_back(I);
1236 }
1237 GenericBlockLiteralType = llvm::StructType::create(
1238 StructFields, "struct.__opencl_block_literal_generic");
1239 } else {
1240 // struct __block_literal_generic {
1241 // void *__isa;
1242 // int __flags;
1243 // int __reserved;
1244 // void (*__invoke)(void *);
1245 // struct __block_descriptor *__descriptor;
1246 // };
1247 GenericBlockLiteralType =
1248 llvm::StructType::create("struct.__block_literal_generic", VoidPtrTy,
1249 IntTy, IntTy, VoidPtrTy, BlockDescPtrTy);
1250 }
Mike Stumpb7074c02009-02-13 15:32:32 +00001251
Mike Stump005c9a62009-02-13 15:25:34 +00001252 return GenericBlockLiteralType;
Anders Carlsson2437cbf2009-02-12 00:39:25 +00001253}
1254
Yaxun Liu10712d92017-10-04 20:32:17 +00001255RValue CodeGenFunction::EmitBlockCallExpr(const CallExpr *E,
Anders Carlssonbfb36712009-12-24 21:13:40 +00001256 ReturnValueSlot ReturnValue) {
Mike Stumpb7074c02009-02-13 15:32:32 +00001257 const BlockPointerType *BPT =
Ted Kremenekc23c7e62009-07-29 21:53:49 +00001258 E->getCallee()->getType()->getAs<BlockPointerType>();
Mike Stumpb7074c02009-02-13 15:32:32 +00001259
John McCallb92ab1a2016-10-26 23:46:34 +00001260 llvm::Value *BlockPtr = EmitScalarExpr(E->getCallee());
Anders Carlsson2437cbf2009-02-12 00:39:25 +00001261
Sven van Haastregtda3b6322018-10-02 13:02:24 +00001262 // Get a pointer to the generic block literal.
1263 // For OpenCL we generate generic AS void ptr to be able to reuse the same
1264 // block definition for blocks with captures generated as private AS local
1265 // variables and without captures generated as global AS program scope
1266 // variables.
1267 unsigned AddrSpace = 0;
1268 if (getLangOpts().OpenCL)
1269 AddrSpace = getContext().getTargetAddressSpace(LangAS::opencl_generic);
Anastasia Stulovaaf0a7bb2017-01-27 15:11:34 +00001270
Sven van Haastregtda3b6322018-10-02 13:02:24 +00001271 llvm::Type *BlockLiteralTy =
1272 llvm::PointerType::get(CGM.getGenericBlockLiteralType(), AddrSpace);
Anders Carlsson2437cbf2009-02-12 00:39:25 +00001273
Sven van Haastregtda3b6322018-10-02 13:02:24 +00001274 // Bitcast the callee to a block literal.
1275 BlockPtr =
1276 Builder.CreatePointerCast(BlockPtr, BlockLiteralTy, "block.literal");
1277
1278 // Get the function pointer from the literal.
1279 llvm::Value *FuncPtr =
1280 Builder.CreateStructGEP(CGM.getGenericBlockLiteralType(), BlockPtr,
1281 CGM.getLangOpts().OpenCL ? 2 : 3);
Mike Stumpb7074c02009-02-13 15:32:32 +00001282
Anders Carlsson2437cbf2009-02-12 00:39:25 +00001283 // Add the block literal.
Anders Carlsson2437cbf2009-02-12 00:39:25 +00001284 CallArgList Args;
Anastasia Stulovaaf0a7bb2017-01-27 15:11:34 +00001285
1286 QualType VoidPtrQualTy = getContext().VoidPtrTy;
1287 llvm::Type *GenericVoidPtrTy = VoidPtrTy;
1288 if (getLangOpts().OpenCL) {
Yaxun Liu10712d92017-10-04 20:32:17 +00001289 GenericVoidPtrTy = CGM.getOpenCLRuntime().getGenericVoidPointerType();
Anastasia Stulovaaf0a7bb2017-01-27 15:11:34 +00001290 VoidPtrQualTy =
1291 getContext().getPointerType(getContext().getAddrSpaceQualType(
1292 getContext().VoidTy, LangAS::opencl_generic));
1293 }
1294
1295 BlockPtr = Builder.CreatePointerCast(BlockPtr, GenericVoidPtrTy);
1296 Args.add(RValue::get(BlockPtr), VoidPtrQualTy);
Mike Stumpb7074c02009-02-13 15:32:32 +00001297
Anders Carlsson479e6fc2009-04-08 23:13:16 +00001298 QualType FnType = BPT->getPointeeType();
1299
Anders Carlsson2437cbf2009-02-12 00:39:25 +00001300 // And the rest of the arguments.
David Blaikief05779e2015-07-21 18:37:18 +00001301 EmitCallArgs(Args, FnType->getAs<FunctionProtoType>(), E->arguments());
Mike Stumpb7074c02009-02-13 15:32:32 +00001302
Anders Carlsson5f50c652009-04-07 22:10:22 +00001303 // Load the function.
Sven van Haastregtda3b6322018-10-02 13:02:24 +00001304 llvm::Value *Func = Builder.CreateAlignedLoad(FuncPtr, getPointerAlign());
Anders Carlsson5f50c652009-04-07 22:10:22 +00001305
John McCall85915252011-03-09 08:39:33 +00001306 const FunctionType *FuncTy = FnType->castAs<FunctionType>();
John McCalla729c622012-02-17 03:33:10 +00001307 const CGFunctionInfo &FnInfo =
John McCallc818bbb2012-12-07 07:03:17 +00001308 CGM.getTypes().arrangeBlockFunctionCall(Args, FuncTy);
Mike Stump11289f42009-09-09 15:08:12 +00001309
Anders Carlsson5f50c652009-04-07 22:10:22 +00001310 // Cast the function pointer to the right type.
John McCalla729c622012-02-17 03:33:10 +00001311 llvm::Type *BlockFTy = CGM.getTypes().GetFunctionType(FnInfo);
Mike Stump11289f42009-09-09 15:08:12 +00001312
Chris Lattner2192fe52011-07-18 04:24:23 +00001313 llvm::Type *BlockFTyPtr = llvm::PointerType::getUnqual(BlockFTy);
Yaxun Liu10712d92017-10-04 20:32:17 +00001314 Func = Builder.CreatePointerCast(Func, BlockFTyPtr);
Mike Stump11289f42009-09-09 15:08:12 +00001315
John McCallb92ab1a2016-10-26 23:46:34 +00001316 // Prepare the callee.
1317 CGCallee Callee(CGCalleeInfo(), Func);
1318
Anders Carlsson2437cbf2009-02-12 00:39:25 +00001319 // And call the block.
John McCallb92ab1a2016-10-26 23:46:34 +00001320 return EmitCall(FnInfo, Callee, ReturnValue, Args);
Anders Carlsson2437cbf2009-02-12 00:39:25 +00001321}
Anders Carlsson6a60fa22009-02-12 17:55:02 +00001322
Akira Hatanaka8e57b072018-10-01 21:51:28 +00001323Address CodeGenFunction::GetAddrOfBlockDecl(const VarDecl *variable) {
John McCall351762c2011-02-07 10:33:21 +00001324 assert(BlockInfo && "evaluating block ref without block information?");
1325 const CGBlockInfo::Capture &capture = BlockInfo->getCapture(variable);
John McCall87fe5d52010-05-20 01:18:31 +00001326
John McCall351762c2011-02-07 10:33:21 +00001327 // Handle constant captures.
John McCall7f416cc2015-09-08 08:05:57 +00001328 if (capture.isConstant()) return LocalDeclMap.find(variable)->second;
John McCall87fe5d52010-05-20 01:18:31 +00001329
John McCall7f416cc2015-09-08 08:05:57 +00001330 Address addr =
1331 Builder.CreateStructGEP(LoadBlockStruct(), capture.getIndex(),
1332 capture.getOffset(), "block.capture.addr");
John McCall87fe5d52010-05-20 01:18:31 +00001333
Akira Hatanaka8e57b072018-10-01 21:51:28 +00001334 if (variable->isEscapingByref()) {
John McCall351762c2011-02-07 10:33:21 +00001335 // addr should be a void** right now. Load, then cast the result
1336 // to byref*.
Mike Stump97d01d52009-03-04 03:23:46 +00001337
John McCall7f416cc2015-09-08 08:05:57 +00001338 auto &byrefInfo = getBlockByrefInfo(variable);
1339 addr = Address(Builder.CreateLoad(addr), byrefInfo.ByrefAlignment);
Mike Stump7fe9cc12009-10-21 03:49:08 +00001340
John McCall7f416cc2015-09-08 08:05:57 +00001341 auto byrefPointerType = llvm::PointerType::get(byrefInfo.Type, 0);
1342 addr = Builder.CreateBitCast(addr, byrefPointerType, "byref.addr");
Mike Stump7fe9cc12009-10-21 03:49:08 +00001343
John McCall7f416cc2015-09-08 08:05:57 +00001344 addr = emitBlockByrefAddress(addr, byrefInfo, /*follow*/ true,
1345 variable->getName());
John McCall87fe5d52010-05-20 01:18:31 +00001346 }
1347
Akira Hatanaka8e57b072018-10-01 21:51:28 +00001348 assert((!variable->isNonEscapingByref() ||
1349 capture.fieldType()->isReferenceType()) &&
1350 "the capture field of a non-escaping variable should have a "
1351 "reference type");
Ivan A. Kosarev9f9d1572017-10-30 11:49:31 +00001352 if (capture.fieldType()->isReferenceType())
1353 addr = EmitLoadOfReference(MakeAddrLValue(addr, capture.fieldType()));
Mike Stump7fe9cc12009-10-21 03:49:08 +00001354
John McCall351762c2011-02-07 10:33:21 +00001355 return addr;
Mike Stump97d01d52009-03-04 03:23:46 +00001356}
1357
George Burgess IVe3763372016-12-22 02:50:20 +00001358void CodeGenModule::setAddrOfGlobalBlock(const BlockExpr *BE,
1359 llvm::Constant *Addr) {
1360 bool Ok = EmittedGlobalBlocks.insert(std::make_pair(BE, Addr)).second;
1361 (void)Ok;
1362 assert(Ok && "Trying to replace an already-existing global block!");
1363}
1364
Mike Stump2d5a2872009-02-14 22:16:35 +00001365llvm::Constant *
George Burgess IV70d15b32016-11-03 02:21:43 +00001366CodeGenModule::GetAddrOfGlobalBlock(const BlockExpr *BE,
1367 StringRef Name) {
George Burgess IVe3763372016-12-22 02:50:20 +00001368 if (llvm::Constant *Block = getAddrOfGlobalBlockIfEmitted(BE))
1369 return Block;
1370
George Burgess IV70d15b32016-11-03 02:21:43 +00001371 CGBlockInfo blockInfo(BE->getBlockDecl(), Name);
1372 blockInfo.BlockExpression = BE;
Mike Stumpb7074c02009-02-13 15:32:32 +00001373
John McCall351762c2011-02-07 10:33:21 +00001374 // Compute information about the layout, etc., of this block.
Craig Topper8a13c412014-05-21 05:09:00 +00001375 computeBlockInfo(*this, nullptr, blockInfo);
Mike Stumpb7074c02009-02-13 15:32:32 +00001376
John McCall351762c2011-02-07 10:33:21 +00001377 // Using that metadata, generate the actual block function.
John McCall351762c2011-02-07 10:33:21 +00001378 {
John McCall7f416cc2015-09-08 08:05:57 +00001379 CodeGenFunction::DeclMapTy LocalDeclMap;
Akira Hatanakaba0367a2017-09-22 21:32:06 +00001380 CodeGenFunction(*this).GenerateBlockFunction(
1381 GlobalDecl(), blockInfo, LocalDeclMap,
1382 /*IsLambdaConversionToBlock*/ false, /*BuildGlobalBlock*/ true);
John McCall351762c2011-02-07 10:33:21 +00001383 }
Mike Stumpb7074c02009-02-13 15:32:32 +00001384
Akira Hatanakaba0367a2017-09-22 21:32:06 +00001385 return getAddrOfGlobalBlockIfEmitted(BE);
Anders Carlsson6a60fa22009-02-12 17:55:02 +00001386}
1387
John McCall351762c2011-02-07 10:33:21 +00001388static llvm::Constant *buildGlobalBlock(CodeGenModule &CGM,
1389 const CGBlockInfo &blockInfo,
1390 llvm::Constant *blockFn) {
1391 assert(blockInfo.CanBeGlobal);
George Burgess IVe3763372016-12-22 02:50:20 +00001392 // Callers should detect this case on their own: calling this function
1393 // generally requires computing layout information, which is a waste of time
1394 // if we've already emitted this block.
1395 assert(!CGM.getAddrOfGlobalBlockIfEmitted(blockInfo.BlockExpression) &&
1396 "Refusing to re-emit a global block.");
John McCall351762c2011-02-07 10:33:21 +00001397
1398 // Generate the constants for the block literal initializer.
John McCall23c9dc62016-11-28 22:18:27 +00001399 ConstantInitBuilder builder(CGM);
John McCall6c9f1fdb2016-11-19 08:17:24 +00001400 auto fields = builder.beginStruct();
John McCall351762c2011-02-07 10:33:21 +00001401
Yaxun Liu10712d92017-10-04 20:32:17 +00001402 bool IsOpenCL = CGM.getLangOpts().OpenCL;
David Chisnallc5a458c2018-08-09 08:02:42 +00001403 bool IsWindows = CGM.getTarget().getTriple().isOSWindows();
Yaxun Liu10712d92017-10-04 20:32:17 +00001404 if (!IsOpenCL) {
1405 // isa
David Chisnallc5a458c2018-08-09 08:02:42 +00001406 if (IsWindows)
1407 fields.addNullPointer(CGM.Int8PtrPtrTy);
1408 else
1409 fields.add(CGM.getNSConcreteGlobalBlock());
John McCall351762c2011-02-07 10:33:21 +00001410
Yaxun Liu10712d92017-10-04 20:32:17 +00001411 // __flags
1412 BlockFlags flags = BLOCK_IS_GLOBAL | BLOCK_HAS_SIGNATURE;
1413 if (blockInfo.UsesStret)
1414 flags |= BLOCK_USE_STRET;
John McCall351762c2011-02-07 10:33:21 +00001415
Yaxun Liu10712d92017-10-04 20:32:17 +00001416 fields.addInt(CGM.IntTy, flags.getBitMask());
1417
1418 // Reserved
1419 fields.addInt(CGM.IntTy, 0);
1420 } else {
1421 fields.addInt(CGM.IntTy, blockInfo.BlockSize.getQuantity());
1422 fields.addInt(CGM.IntTy, blockInfo.BlockAlign.getQuantity());
1423 }
John McCall351762c2011-02-07 10:33:21 +00001424
Sven van Haastregtda3b6322018-10-02 13:02:24 +00001425 // Function
1426 fields.add(blockFn);
1427
Yaxun Liu10712d92017-10-04 20:32:17 +00001428 if (!IsOpenCL) {
1429 // Descriptor
1430 fields.add(buildBlockDescriptor(CGM, blockInfo));
1431 } else if (auto *Helper =
1432 CGM.getTargetCodeGenInfo().getTargetOpenCLBlockHelper()) {
1433 for (auto I : Helper->getCustomFieldValues(CGM, blockInfo)) {
1434 fields.add(I);
1435 }
1436 }
John McCall351762c2011-02-07 10:33:21 +00001437
Anastasia Stulovaaf0a7bb2017-01-27 15:11:34 +00001438 unsigned AddrSpace = 0;
1439 if (CGM.getContext().getLangOpts().OpenCL)
1440 AddrSpace = CGM.getContext().getTargetAddressSpace(LangAS::opencl_global);
1441
1442 llvm::Constant *literal = fields.finishAndCreateGlobal(
1443 "__block_literal_global", blockInfo.BlockAlign,
David Chisnallc5a458c2018-08-09 08:02:42 +00001444 /*constant*/ !IsWindows, llvm::GlobalVariable::InternalLinkage, AddrSpace);
1445
1446 // Windows does not allow globals to be initialised to point to globals in
1447 // different DLLs. Any such variables must run code to initialise them.
1448 if (IsWindows) {
1449 auto *Init = llvm::Function::Create(llvm::FunctionType::get(CGM.VoidTy,
1450 {}), llvm::GlobalValue::InternalLinkage, ".block_isa_init",
1451 &CGM.getModule());
1452 llvm::IRBuilder<> b(llvm::BasicBlock::Create(CGM.getLLVMContext(), "entry",
1453 Init));
1454 b.CreateAlignedStore(CGM.getNSConcreteGlobalBlock(),
1455 b.CreateStructGEP(literal, 0), CGM.getPointerAlign().getQuantity());
1456 b.CreateRetVoid();
1457 // We can't use the normal LLVM global initialisation array, because we
1458 // need to specify that this runs early in library initialisation.
1459 auto *InitVar = new llvm::GlobalVariable(CGM.getModule(), Init->getType(),
1460 /*isConstant*/true, llvm::GlobalValue::InternalLinkage,
1461 Init, ".block_isa_init_ptr");
1462 InitVar->setSection(".CRT$XCLa");
1463 CGM.addUsedGlobal(InitVar);
1464 }
John McCall351762c2011-02-07 10:33:21 +00001465
1466 // Return a constant of the appropriately-casted type.
George Burgess IVe3763372016-12-22 02:50:20 +00001467 llvm::Type *RequiredType =
John McCall351762c2011-02-07 10:33:21 +00001468 CGM.getTypes().ConvertType(blockInfo.getBlockExpr()->getType());
George Burgess IVe3763372016-12-22 02:50:20 +00001469 llvm::Constant *Result =
Anastasia Stulovaaf0a7bb2017-01-27 15:11:34 +00001470 llvm::ConstantExpr::getPointerCast(literal, RequiredType);
George Burgess IVe3763372016-12-22 02:50:20 +00001471 CGM.setAddrOfGlobalBlock(blockInfo.BlockExpression, Result);
Yaxun Liufa13d012018-02-15 16:39:19 +00001472 if (CGM.getContext().getLangOpts().OpenCL)
1473 CGM.getOpenCLRuntime().recordBlockInfo(
1474 blockInfo.BlockExpression,
1475 cast<llvm::Function>(blockFn->stripPointerCasts()), Result);
George Burgess IVe3763372016-12-22 02:50:20 +00001476 return Result;
Mike Stumpcb2fbcb2009-02-21 20:00:35 +00001477}
1478
John McCall7f416cc2015-09-08 08:05:57 +00001479void CodeGenFunction::setBlockContextParameter(const ImplicitParamDecl *D,
1480 unsigned argNum,
1481 llvm::Value *arg) {
1482 assert(BlockInfo && "not emitting prologue of block invocation function?!");
1483
Adrian Prantl356347b2017-10-26 20:08:52 +00001484 // Allocate a stack slot like for any local variable to guarantee optimal
1485 // debug info at -O0. The mem2reg pass will eliminate it when optimizing.
1486 Address alloc = CreateMemTemp(D->getType(), D->getName() + ".addr");
1487 Builder.CreateStore(arg, alloc);
John McCall7f416cc2015-09-08 08:05:57 +00001488 if (CGDebugInfo *DI = getDebugInfo()) {
Benjamin Kramer8c305922016-02-02 11:06:51 +00001489 if (CGM.getCodeGenOpts().getDebugInfo() >=
1490 codegenoptions::LimitedDebugInfo) {
John McCall7f416cc2015-09-08 08:05:57 +00001491 DI->setLocation(D->getLocation());
Adrian Prantl356347b2017-10-26 20:08:52 +00001492 DI->EmitDeclareOfBlockLiteralArgVariable(
1493 *BlockInfo, D->getName(), argNum,
1494 cast<llvm::AllocaInst>(alloc.getPointer()), Builder);
John McCall7f416cc2015-09-08 08:05:57 +00001495 }
1496 }
1497
Stephen Kellyf2ceec42018-08-09 21:08:08 +00001498 SourceLocation StartLoc = BlockInfo->getBlockExpr()->getBody()->getBeginLoc();
John McCall7f416cc2015-09-08 08:05:57 +00001499 ApplyDebugLocation Scope(*this, StartLoc);
1500
1501 // Instead of messing around with LocalDeclMap, just set the value
1502 // directly as BlockPointer.
Anastasia Stulovaaf0a7bb2017-01-27 15:11:34 +00001503 BlockPointer = Builder.CreatePointerCast(
1504 arg,
1505 BlockInfo->StructureType->getPointerTo(
1506 getContext().getLangOpts().OpenCL
1507 ? getContext().getTargetAddressSpace(LangAS::opencl_generic)
1508 : 0),
1509 "block");
John McCall7f416cc2015-09-08 08:05:57 +00001510}
1511
1512Address CodeGenFunction::LoadBlockStruct() {
1513 assert(BlockInfo && "not in a block invocation function!");
1514 assert(BlockPointer && "no block pointer set!");
1515 return Address(BlockPointer, BlockInfo->BlockAlign);
1516}
1517
Mike Stump4446dcf2009-03-05 08:32:30 +00001518llvm::Function *
John McCall351762c2011-02-07 10:33:21 +00001519CodeGenFunction::GenerateBlockFunction(GlobalDecl GD,
1520 const CGBlockInfo &blockInfo,
Eli Friedman2495ab02012-02-25 02:48:22 +00001521 const DeclMapTy &ldm,
Akira Hatanakaba0367a2017-09-22 21:32:06 +00001522 bool IsLambdaConversionToBlock,
1523 bool BuildGlobalBlock) {
John McCall351762c2011-02-07 10:33:21 +00001524 const BlockDecl *blockDecl = blockInfo.getBlockDecl();
Devang Patel9074ed82009-04-15 21:51:44 +00001525
Fariborz Jahanian63628032012-06-26 16:06:38 +00001526 CurGD = GD;
David Blaikie1ae04912015-01-13 23:06:27 +00001527
Stephen Kelly1c301dc2018-08-09 21:09:38 +00001528 CurEHLocation = blockInfo.getBlockExpr()->getEndLoc();
Fangrui Song6907ce22018-07-30 19:24:48 +00001529
John McCall351762c2011-02-07 10:33:21 +00001530 BlockInfo = &blockInfo;
Mike Stump11289f42009-09-09 15:08:12 +00001531
Mike Stump5469f292009-03-13 23:34:28 +00001532 // Arrange for local static and local extern declarations to appear
John McCall351762c2011-02-07 10:33:21 +00001533 // to be local to this function as well, in case they're directly
1534 // referenced in a block.
1535 for (DeclMapTy::const_iterator i = ldm.begin(), e = ldm.end(); i != e; ++i) {
Rafael Espindola2ae250c2014-05-09 00:08:36 +00001536 const auto *var = dyn_cast<VarDecl>(i->first);
John McCall351762c2011-02-07 10:33:21 +00001537 if (var && !var->hasLocalStorage())
John McCall7f416cc2015-09-08 08:05:57 +00001538 setAddrOfLocalVar(var, i->second);
Mike Stump5469f292009-03-13 23:34:28 +00001539 }
1540
John McCall351762c2011-02-07 10:33:21 +00001541 // Begin building the function declaration.
Eli Friedman09a9b6e2009-03-28 03:24:54 +00001542
John McCall351762c2011-02-07 10:33:21 +00001543 // Build the argument list.
1544 FunctionArgList args;
Mike Stumpb7074c02009-02-13 15:32:32 +00001545
John McCall351762c2011-02-07 10:33:21 +00001546 // The first argument is the block pointer. Just take it as a void*
1547 // and cast it later.
1548 QualType selfTy = getContext().VoidPtrTy;
Anastasia Stulovaaf0a7bb2017-01-27 15:11:34 +00001549
1550 // For OpenCL passed block pointer can be private AS local variable or
1551 // global AS program scope variable (for the case with and without captures).
Hiroshi Inouec5e54dd2017-07-03 08:49:44 +00001552 // Generic AS is used therefore to be able to accommodate both private and
Anastasia Stulovaaf0a7bb2017-01-27 15:11:34 +00001553 // generic AS in one implementation.
1554 if (getLangOpts().OpenCL)
1555 selfTy = getContext().getPointerType(getContext().getAddrSpaceQualType(
1556 getContext().VoidTy, LangAS::opencl_generic));
1557
Mike Stump7fe9cc12009-10-21 03:49:08 +00001558 IdentifierInfo *II = &CGM.getContext().Idents.get(".block_descriptor");
Mike Stumpd0153282009-10-20 02:12:22 +00001559
Alexey Bataev56223232017-06-09 13:40:18 +00001560 ImplicitParamDecl SelfDecl(getContext(), const_cast<BlockDecl *>(blockDecl),
1561 SourceLocation(), II, selfTy,
1562 ImplicitParamDecl::ObjCSelf);
1563 args.push_back(&SelfDecl);
Mike Stump7fe9cc12009-10-21 03:49:08 +00001564
John McCall351762c2011-02-07 10:33:21 +00001565 // Now add the rest of the parameters.
Benjamin Kramerf9890422015-02-17 16:48:30 +00001566 args.append(blockDecl->param_begin(), blockDecl->param_end());
John McCall87fe5d52010-05-20 01:18:31 +00001567
John McCall351762c2011-02-07 10:33:21 +00001568 // Create the function declaration.
John McCalla729c622012-02-17 03:33:10 +00001569 const FunctionProtoType *fnType = blockInfo.getBlockExpr()->getFunctionType();
John McCallc56a8b32016-03-11 04:30:31 +00001570 const CGFunctionInfo &fnInfo =
1571 CGM.getTypes().arrangeBlockFunctionDeclaration(fnType, args);
Tim Northovere77cc392014-03-29 13:28:05 +00001572 if (CGM.ReturnSlotInterferesWithArgs(fnInfo))
John McCall85915252011-03-09 08:39:33 +00001573 blockInfo.UsesStret = true;
1574
John McCalla729c622012-02-17 03:33:10 +00001575 llvm::FunctionType *fnLLVMType = CGM.getTypes().GetFunctionType(fnInfo);
Mike Stumpb7074c02009-02-13 15:32:32 +00001576
Alp Tokerfb8d02b2014-06-05 22:10:59 +00001577 StringRef name = CGM.getBlockMangledName(GD, blockDecl);
Alp Toker0e64e0d2014-06-03 02:13:57 +00001578 llvm::Function *fn = llvm::Function::Create(
1579 fnLLVMType, llvm::GlobalValue::InternalLinkage, name, &CGM.getModule());
John McCall351762c2011-02-07 10:33:21 +00001580 CGM.SetInternalFunctionAttributes(blockDecl, fn, fnInfo);
Mike Stumpb7074c02009-02-13 15:32:32 +00001581
Yaxun Liu10712d92017-10-04 20:32:17 +00001582 if (BuildGlobalBlock) {
1583 auto GenVoidPtrTy = getContext().getLangOpts().OpenCL
1584 ? CGM.getOpenCLRuntime().getGenericVoidPointerType()
1585 : VoidPtrTy;
Akira Hatanakaba0367a2017-09-22 21:32:06 +00001586 buildGlobalBlock(CGM, blockInfo,
Yaxun Liu10712d92017-10-04 20:32:17 +00001587 llvm::ConstantExpr::getPointerCast(fn, GenVoidPtrTy));
1588 }
Akira Hatanakaba0367a2017-09-22 21:32:06 +00001589
John McCall351762c2011-02-07 10:33:21 +00001590 // Begin generating the function.
Alp Toker314cc812014-01-25 16:55:45 +00001591 StartFunction(blockDecl, fnType->getReturnType(), fn, fnInfo, args,
Adrian Prantl42d71b92014-04-10 23:21:53 +00001592 blockDecl->getLocation(),
Stephen Kellyf2ceec42018-08-09 21:08:08 +00001593 blockInfo.getBlockExpr()->getBody()->getBeginLoc());
Mike Stumpb7074c02009-02-13 15:32:32 +00001594
John McCall147d0212011-02-22 22:38:33 +00001595 // Okay. Undo some of what StartFunction did.
John McCall7f416cc2015-09-08 08:05:57 +00001596
Adrian Prantl0f6df002013-03-29 19:20:35 +00001597 // At -O0 we generate an explicit alloca for the BlockPointer, so the RA
1598 // won't delete the dbg.declare intrinsics for captured variables.
1599 llvm::Value *BlockPointerDbgLoc = BlockPointer;
1600 if (CGM.getCodeGenOpts().OptimizationLevel == 0) {
1601 // Allocate a stack slot for it, so we can point the debugger to it
John McCall7f416cc2015-09-08 08:05:57 +00001602 Address Alloca = CreateTempAlloca(BlockPointer->getType(),
1603 getPointerAlign(),
1604 "block.addr");
Adrian Prantl2832b4e2013-04-02 01:00:48 +00001605 // Set the DebugLocation to empty, so the store is recognized as a
1606 // frame setup instruction by llvm::DwarfDebug::beginFunction().
Adrian Prantl95b24e92015-02-03 20:00:54 +00001607 auto NL = ApplyDebugLocation::CreateEmpty(*this);
John McCall7f416cc2015-09-08 08:05:57 +00001608 Builder.CreateStore(BlockPointer, Alloca);
1609 BlockPointerDbgLoc = Alloca.getPointer();
Adrian Prantl0f6df002013-03-29 19:20:35 +00001610 }
Anders Carlsson6a60fa22009-02-12 17:55:02 +00001611
John McCall87fe5d52010-05-20 01:18:31 +00001612 // If we have a C++ 'this' reference, go ahead and force it into
1613 // existence now.
John McCall351762c2011-02-07 10:33:21 +00001614 if (blockDecl->capturesCXXThis()) {
John McCall7f416cc2015-09-08 08:05:57 +00001615 Address addr =
1616 Builder.CreateStructGEP(LoadBlockStruct(), blockInfo.CXXThisIndex,
1617 blockInfo.CXXThisOffset, "block.captured-this");
John McCall351762c2011-02-07 10:33:21 +00001618 CXXThisValue = Builder.CreateLoad(addr, "this");
John McCall87fe5d52010-05-20 01:18:31 +00001619 }
1620
John McCall351762c2011-02-07 10:33:21 +00001621 // Also force all the constant captures.
Aaron Ballman9371dd22014-03-14 18:34:04 +00001622 for (const auto &CI : blockDecl->captures()) {
1623 const VarDecl *variable = CI.getVariable();
John McCall351762c2011-02-07 10:33:21 +00001624 const CGBlockInfo::Capture &capture = blockInfo.getCapture(variable);
1625 if (!capture.isConstant()) continue;
1626
John McCall7f416cc2015-09-08 08:05:57 +00001627 CharUnits align = getContext().getDeclAlign(variable);
1628 Address alloca =
1629 CreateMemTemp(variable->getType(), align, "block.captured-const");
John McCall351762c2011-02-07 10:33:21 +00001630
John McCall7f416cc2015-09-08 08:05:57 +00001631 Builder.CreateStore(capture.getConstant(), alloca);
John McCall351762c2011-02-07 10:33:21 +00001632
John McCall7f416cc2015-09-08 08:05:57 +00001633 setAddrOfLocalVar(variable, alloca);
John McCall9d42f0f2010-05-21 04:11:14 +00001634 }
1635
John McCall113bee02012-03-10 09:33:50 +00001636 // Save a spot to insert the debug information for all the DeclRefExprs.
Mike Stump017460a2009-10-01 22:29:41 +00001637 llvm::BasicBlock *entry = Builder.GetInsertBlock();
1638 llvm::BasicBlock::iterator entry_ptr = Builder.GetInsertPoint();
1639 --entry_ptr;
1640
Eli Friedman2495ab02012-02-25 02:48:22 +00001641 if (IsLambdaConversionToBlock)
1642 EmitLambdaBlockInvokeBody();
Bob Wilsonc845c002014-03-06 20:24:27 +00001643 else {
Serge Pavlov3a561452015-12-06 14:32:39 +00001644 PGO.assignRegionCounters(GlobalDecl(blockDecl), fn);
Justin Bogner66242d62015-04-23 23:06:47 +00001645 incrementProfileCounter(blockDecl->getBody());
Eli Friedman2495ab02012-02-25 02:48:22 +00001646 EmitStmt(blockDecl->getBody());
Bob Wilsonc845c002014-03-06 20:24:27 +00001647 }
Mike Stump017460a2009-10-01 22:29:41 +00001648
Mike Stump7d699112009-10-01 00:27:30 +00001649 // Remember where we were...
1650 llvm::BasicBlock *resume = Builder.GetInsertBlock();
Mike Stump017460a2009-10-01 22:29:41 +00001651
Mike Stump7d699112009-10-01 00:27:30 +00001652 // Go back to the entry.
Mike Stump017460a2009-10-01 22:29:41 +00001653 ++entry_ptr;
1654 Builder.SetInsertPoint(entry, entry_ptr);
1655
John McCall113bee02012-03-10 09:33:50 +00001656 // Emit debug information for all the DeclRefExprs.
John McCall351762c2011-02-07 10:33:21 +00001657 // FIXME: also for 'this'
Mike Stump2e722b92009-09-30 02:43:10 +00001658 if (CGDebugInfo *DI = getDebugInfo()) {
Aaron Ballman9371dd22014-03-14 18:34:04 +00001659 for (const auto &CI : blockDecl->captures()) {
1660 const VarDecl *variable = CI.getVariable();
Eric Christopher7cdf9482011-10-13 21:45:18 +00001661 DI->EmitLocation(Builder, variable->getLocation());
John McCall351762c2011-02-07 10:33:21 +00001662
Benjamin Kramer8c305922016-02-02 11:06:51 +00001663 if (CGM.getCodeGenOpts().getDebugInfo() >=
1664 codegenoptions::LimitedDebugInfo) {
Alexey Samsonov74a38682012-05-04 07:39:27 +00001665 const CGBlockInfo::Capture &capture = blockInfo.getCapture(variable);
1666 if (capture.isConstant()) {
John McCall7f416cc2015-09-08 08:05:57 +00001667 auto addr = LocalDeclMap.find(variable)->second;
Sander de Smalen891af03a2018-02-03 13:55:59 +00001668 (void)DI->EmitDeclareOfAutoVariable(variable, addr.getPointer(),
1669 Builder);
Alexey Samsonov74a38682012-05-04 07:39:27 +00001670 continue;
1671 }
John McCall351762c2011-02-07 10:33:21 +00001672
Duncan P. N. Exon Smith9f5260a2015-11-06 23:00:41 +00001673 DI->EmitDeclareOfBlockDeclRefVariable(
1674 variable, BlockPointerDbgLoc, Builder, blockInfo,
1675 entry_ptr == entry->end() ? nullptr : &*entry_ptr);
Alexey Samsonov74a38682012-05-04 07:39:27 +00001676 }
Mike Stump2e722b92009-09-30 02:43:10 +00001677 }
Manman Renab08a9a2013-01-04 18:51:35 +00001678 // Recover location if it was changed in the above loop.
1679 DI->EmitLocation(Builder,
Adrian Prantl83e30fd2013-04-08 20:52:12 +00001680 cast<CompoundStmt>(blockDecl->getBody())->getRBracLoc());
Mike Stump2e722b92009-09-30 02:43:10 +00001681 }
John McCall351762c2011-02-07 10:33:21 +00001682
Mike Stump7d699112009-10-01 00:27:30 +00001683 // And resume where we left off.
Craig Topper8a13c412014-05-21 05:09:00 +00001684 if (resume == nullptr)
Mike Stump7d699112009-10-01 00:27:30 +00001685 Builder.ClearInsertionPoint();
1686 else
1687 Builder.SetInsertPoint(resume);
Mike Stump2e722b92009-09-30 02:43:10 +00001688
John McCall351762c2011-02-07 10:33:21 +00001689 FinishFunction(cast<CompoundStmt>(blockDecl->getBody())->getRBracLoc());
Anders Carlsson6a60fa22009-02-12 17:55:02 +00001690
John McCall351762c2011-02-07 10:33:21 +00001691 return fn;
Anders Carlsson6a60fa22009-02-12 17:55:02 +00001692}
Mike Stump1db7d042009-02-28 09:07:16 +00001693
Alex Lorenze08e5bc2017-03-06 16:23:04 +00001694static std::pair<BlockCaptureEntityKind, BlockFieldFlags>
1695computeCopyInfoForBlockCapture(const BlockDecl::Capture &CI, QualType T,
1696 const LangOptions &LangOpts) {
1697 if (CI.getCopyExpr()) {
1698 assert(!CI.isByRef());
1699 // don't bother computing flags
1700 return std::make_pair(BlockCaptureEntityKind::CXXRecord, BlockFieldFlags());
1701 }
1702 BlockFieldFlags Flags;
Akira Hatanaka8e57b072018-10-01 21:51:28 +00001703 if (CI.isEscapingByref()) {
Alex Lorenze08e5bc2017-03-06 16:23:04 +00001704 Flags = BLOCK_FIELD_IS_BYREF;
1705 if (T.isObjCGCWeak())
1706 Flags |= BLOCK_FIELD_IS_WEAK;
1707 return std::make_pair(BlockCaptureEntityKind::BlockObject, Flags);
1708 }
Alex Lorenze08e5bc2017-03-06 16:23:04 +00001709
1710 Flags = BLOCK_FIELD_IS_OBJECT;
1711 bool isBlockPointer = T->isBlockPointerType();
1712 if (isBlockPointer)
1713 Flags = BLOCK_FIELD_IS_BLOCK;
1714
Akira Hatanaka7275da02018-02-28 07:15:55 +00001715 switch (T.isNonTrivialToPrimitiveCopy()) {
1716 case QualType::PCK_Struct:
1717 return std::make_pair(BlockCaptureEntityKind::NonTrivialCStruct,
1718 BlockFieldFlags());
Akira Hatanakad791e922018-03-19 17:38:40 +00001719 case QualType::PCK_ARCWeak:
1720 // We need to register __weak direct captures with the runtime.
1721 return std::make_pair(BlockCaptureEntityKind::ARCWeak, Flags);
Akira Hatanaka7275da02018-02-28 07:15:55 +00001722 case QualType::PCK_ARCStrong:
1723 // We need to retain the copied value for __strong direct captures.
1724 // If it's a block pointer, we have to copy the block and assign that to
1725 // the destination pointer, so we might as well use _Block_object_assign.
1726 // Otherwise we can avoid that.
Alex Lorenze08e5bc2017-03-06 16:23:04 +00001727 return std::make_pair(!isBlockPointer ? BlockCaptureEntityKind::ARCStrong
1728 : BlockCaptureEntityKind::BlockObject,
1729 Flags);
Akira Hatanaka7275da02018-02-28 07:15:55 +00001730 case QualType::PCK_Trivial:
1731 case QualType::PCK_VolatileTrivial: {
1732 if (!T->isObjCRetainableType())
1733 // For all other types, the memcpy is fine.
1734 return std::make_pair(BlockCaptureEntityKind::None, BlockFieldFlags());
1735
1736 // Special rules for ARC captures:
1737 Qualifiers QS = T.getQualifiers();
1738
Akira Hatanaka7275da02018-02-28 07:15:55 +00001739 // Non-ARC captures of retainable pointers are strong and
1740 // therefore require a call to _Block_object_assign.
1741 if (!QS.getObjCLifetime() && !LangOpts.ObjCAutoRefCount)
1742 return std::make_pair(BlockCaptureEntityKind::BlockObject, Flags);
1743
1744 // Otherwise the memcpy is fine.
1745 return std::make_pair(BlockCaptureEntityKind::None, BlockFieldFlags());
Alex Lorenze08e5bc2017-03-06 16:23:04 +00001746 }
Akira Hatanaka7275da02018-02-28 07:15:55 +00001747 }
Nico Weberb3897eb2018-02-28 19:28:47 +00001748 llvm_unreachable("after exhaustive PrimitiveCopyKind switch");
Alex Lorenze08e5bc2017-03-06 16:23:04 +00001749}
1750
Akira Hatanaka2ec36f02018-08-17 15:46:07 +00001751static std::pair<BlockCaptureEntityKind, BlockFieldFlags>
1752computeDestroyInfoForBlockCapture(const BlockDecl::Capture &CI, QualType T,
1753 const LangOptions &LangOpts);
1754
Alex Lorenze08e5bc2017-03-06 16:23:04 +00001755/// Find the set of block captures that need to be explicitly copied or destroy.
1756static void findBlockCapturedManagedEntities(
1757 const CGBlockInfo &BlockInfo, const LangOptions &LangOpts,
Akira Hatanaka2ec36f02018-08-17 15:46:07 +00001758 SmallVectorImpl<BlockCaptureManagedEntity> &ManagedCaptures) {
Alex Lorenze08e5bc2017-03-06 16:23:04 +00001759 for (const auto &CI : BlockInfo.getBlockDecl()->captures()) {
1760 const VarDecl *Variable = CI.getVariable();
1761 const CGBlockInfo::Capture &Capture = BlockInfo.getCapture(Variable);
1762 if (Capture.isConstant())
1763 continue;
1764
Akira Hatanaka2a5e4632018-08-22 13:41:19 +00001765 QualType VT = Capture.fieldType();
1766 auto CopyInfo = computeCopyInfoForBlockCapture(CI, VT, LangOpts);
1767 auto DisposeInfo = computeDestroyInfoForBlockCapture(CI, VT, LangOpts);
Akira Hatanaka2ec36f02018-08-17 15:46:07 +00001768 if (CopyInfo.first != BlockCaptureEntityKind::None ||
1769 DisposeInfo.first != BlockCaptureEntityKind::None)
1770 ManagedCaptures.emplace_back(CopyInfo.first, DisposeInfo.first,
1771 CopyInfo.second, DisposeInfo.second, CI,
1772 Capture);
Alex Lorenze08e5bc2017-03-06 16:23:04 +00001773 }
Akira Hatanaka9978da32018-08-10 15:09:24 +00001774
1775 // Sort the captures by offset.
Fangrui Song55fab262018-09-26 22:16:28 +00001776 llvm::sort(ManagedCaptures);
Alex Lorenze08e5bc2017-03-06 16:23:04 +00001777}
1778
Akira Hatanakacb6a9332018-07-26 16:51:21 +00001779namespace {
1780/// Release a __block variable.
1781struct CallBlockRelease final : EHScopeStack::Cleanup {
1782 Address Addr;
1783 BlockFieldFlags FieldFlags;
Akira Hatanaka9978da32018-08-10 15:09:24 +00001784 bool LoadBlockVarAddr, CanThrow;
Akira Hatanakacb6a9332018-07-26 16:51:21 +00001785
Akira Hatanaka9978da32018-08-10 15:09:24 +00001786 CallBlockRelease(Address Addr, BlockFieldFlags Flags, bool LoadValue,
1787 bool CT)
1788 : Addr(Addr), FieldFlags(Flags), LoadBlockVarAddr(LoadValue),
1789 CanThrow(CT) {}
Akira Hatanakacb6a9332018-07-26 16:51:21 +00001790
1791 void Emit(CodeGenFunction &CGF, Flags flags) override {
1792 llvm::Value *BlockVarAddr;
1793 if (LoadBlockVarAddr) {
1794 BlockVarAddr = CGF.Builder.CreateLoad(Addr);
1795 BlockVarAddr = CGF.Builder.CreateBitCast(BlockVarAddr, CGF.VoidPtrTy);
1796 } else {
1797 BlockVarAddr = Addr.getPointer();
1798 }
1799
Akira Hatanaka9978da32018-08-10 15:09:24 +00001800 CGF.BuildBlockRelease(BlockVarAddr, FieldFlags, CanThrow);
Akira Hatanakacb6a9332018-07-26 16:51:21 +00001801 }
1802};
1803} // end anonymous namespace
1804
Akira Hatanaka9978da32018-08-10 15:09:24 +00001805/// Check if \p T is a C++ class that has a destructor that can throw.
1806bool CodeGenFunction::cxxDestructorCanThrow(QualType T) {
1807 if (const auto *RD = T->getAsCXXRecordDecl())
1808 if (const CXXDestructorDecl *DD = RD->getDestructor())
1809 return DD->getType()->getAs<FunctionProtoType>()->canThrow();
1810 return false;
1811}
1812
Akira Hatanaka2ec36f02018-08-17 15:46:07 +00001813// Return a string that has the information about a capture.
1814static std::string getBlockCaptureStr(const BlockCaptureManagedEntity &E,
1815 CaptureStrKind StrKind,
1816 CharUnits BlockAlignment,
1817 CodeGenModule &CGM) {
1818 std::string Str;
Akira Hatanaka9978da32018-08-10 15:09:24 +00001819 ASTContext &Ctx = CGM.getContext();
Akira Hatanaka2ec36f02018-08-17 15:46:07 +00001820 const BlockDecl::Capture &CI = *E.CI;
1821 QualType CaptureTy = CI.getVariable()->getType();
Akira Hatanaka9978da32018-08-10 15:09:24 +00001822
Akira Hatanaka2ec36f02018-08-17 15:46:07 +00001823 BlockCaptureEntityKind Kind;
1824 BlockFieldFlags Flags;
1825
1826 // CaptureStrKind::Merged should be passed only when the operations and the
1827 // flags are the same for copy and dispose.
1828 assert((StrKind != CaptureStrKind::Merged ||
1829 (E.CopyKind == E.DisposeKind && E.CopyFlags == E.DisposeFlags)) &&
1830 "different operations and flags");
1831
1832 if (StrKind == CaptureStrKind::DisposeHelper) {
1833 Kind = E.DisposeKind;
1834 Flags = E.DisposeFlags;
1835 } else {
1836 Kind = E.CopyKind;
1837 Flags = E.CopyFlags;
1838 }
1839
1840 switch (Kind) {
1841 case BlockCaptureEntityKind::CXXRecord: {
1842 Str += "c";
1843 SmallString<256> TyStr;
1844 llvm::raw_svector_ostream Out(TyStr);
Akira Hatanaka32e0a582018-10-20 05:45:01 +00001845 CGM.getCXXABI().getMangleContext().mangleTypeName(CaptureTy, Out);
Akira Hatanaka2ec36f02018-08-17 15:46:07 +00001846 Str += llvm::to_string(TyStr.size()) + TyStr.c_str();
1847 break;
1848 }
1849 case BlockCaptureEntityKind::ARCWeak:
1850 Str += "w";
1851 break;
1852 case BlockCaptureEntityKind::ARCStrong:
1853 Str += "s";
1854 break;
1855 case BlockCaptureEntityKind::BlockObject: {
1856 const VarDecl *Var = CI.getVariable();
1857 unsigned F = Flags.getBitMask();
1858 if (F & BLOCK_FIELD_IS_BYREF) {
1859 Str += "r";
1860 if (F & BLOCK_FIELD_IS_WEAK)
1861 Str += "w";
1862 else {
1863 // If CaptureStrKind::Merged is passed, check both the copy expression
1864 // and the destructor.
1865 if (StrKind != CaptureStrKind::DisposeHelper) {
1866 if (Ctx.getBlockVarCopyInit(Var).canThrow())
1867 Str += "c";
1868 }
1869 if (StrKind != CaptureStrKind::CopyHelper) {
1870 if (CodeGenFunction::cxxDestructorCanThrow(CaptureTy))
1871 Str += "d";
1872 }
1873 }
1874 } else {
1875 assert((F & BLOCK_FIELD_IS_OBJECT) && "unexpected flag value");
1876 if (F == BLOCK_FIELD_IS_BLOCK)
1877 Str += "b";
1878 else
1879 Str += "o";
1880 }
1881 break;
1882 }
1883 case BlockCaptureEntityKind::NonTrivialCStruct: {
1884 bool IsVolatile = CaptureTy.isVolatileQualified();
1885 CharUnits Alignment =
1886 BlockAlignment.alignmentAtOffset(E.Capture->getOffset());
1887
1888 Str += "n";
1889 std::string FuncStr;
1890 if (StrKind == CaptureStrKind::DisposeHelper)
1891 FuncStr = CodeGenFunction::getNonTrivialDestructorStr(
1892 CaptureTy, Alignment, IsVolatile, Ctx);
1893 else
1894 // If CaptureStrKind::Merged is passed, use the copy constructor string.
1895 // It has all the information that the destructor string has.
1896 FuncStr = CodeGenFunction::getNonTrivialCopyConstructorStr(
1897 CaptureTy, Alignment, IsVolatile, Ctx);
1898 // The underscore is necessary here because non-trivial copy constructor
1899 // and destructor strings can start with a number.
1900 Str += llvm::to_string(FuncStr.size()) + "_" + FuncStr;
1901 break;
1902 }
1903 case BlockCaptureEntityKind::None:
1904 break;
1905 }
1906
1907 return Str;
1908}
1909
1910static std::string getCopyDestroyHelperFuncName(
1911 const SmallVectorImpl<BlockCaptureManagedEntity> &Captures,
1912 CharUnits BlockAlignment, CaptureStrKind StrKind, CodeGenModule &CGM) {
1913 assert((StrKind == CaptureStrKind::CopyHelper ||
1914 StrKind == CaptureStrKind::DisposeHelper) &&
1915 "unexpected CaptureStrKind");
1916 std::string Name = StrKind == CaptureStrKind::CopyHelper
1917 ? "__copy_helper_block_"
1918 : "__destroy_helper_block_";
Akira Hatanaka9978da32018-08-10 15:09:24 +00001919 if (CGM.getLangOpts().Exceptions)
1920 Name += "e";
1921 if (CGM.getCodeGenOpts().ObjCAutoRefCountExceptions)
1922 Name += "a";
1923 Name += llvm::to_string(BlockAlignment.getQuantity()) + "_";
1924
1925 for (const BlockCaptureManagedEntity &E : Captures) {
Akira Hatanaka9978da32018-08-10 15:09:24 +00001926 Name += llvm::to_string(E.Capture->getOffset().getQuantity());
Akira Hatanaka2ec36f02018-08-17 15:46:07 +00001927 Name += getBlockCaptureStr(E, StrKind, BlockAlignment, CGM);
Akira Hatanaka9978da32018-08-10 15:09:24 +00001928 }
1929
1930 return Name;
1931}
1932
Akira Hatanakacb6a9332018-07-26 16:51:21 +00001933static void pushCaptureCleanup(BlockCaptureEntityKind CaptureKind,
1934 Address Field, QualType CaptureType,
Akira Hatanaka9978da32018-08-10 15:09:24 +00001935 BlockFieldFlags Flags, bool ForCopyHelper,
1936 VarDecl *Var, CodeGenFunction &CGF) {
1937 bool EHOnly = ForCopyHelper;
1938
Akira Hatanakacb6a9332018-07-26 16:51:21 +00001939 switch (CaptureKind) {
1940 case BlockCaptureEntityKind::CXXRecord:
1941 case BlockCaptureEntityKind::ARCWeak:
1942 case BlockCaptureEntityKind::NonTrivialCStruct:
1943 case BlockCaptureEntityKind::ARCStrong: {
1944 if (CaptureType.isDestructedType() &&
1945 (!EHOnly || CGF.needsEHCleanup(CaptureType.isDestructedType()))) {
1946 CodeGenFunction::Destroyer *Destroyer =
1947 CaptureKind == BlockCaptureEntityKind::ARCStrong
1948 ? CodeGenFunction::destroyARCStrongImprecise
1949 : CGF.getDestroyer(CaptureType.isDestructedType());
1950 CleanupKind Kind =
1951 EHOnly ? EHCleanup
1952 : CGF.getCleanupKind(CaptureType.isDestructedType());
1953 CGF.pushDestroy(Kind, Field, CaptureType, Destroyer, Kind & EHCleanup);
1954 }
1955 break;
1956 }
1957 case BlockCaptureEntityKind::BlockObject: {
1958 if (!EHOnly || CGF.getLangOpts().Exceptions) {
1959 CleanupKind Kind = EHOnly ? EHCleanup : NormalAndEHCleanup;
Akira Hatanaka9978da32018-08-10 15:09:24 +00001960 // Calls to _Block_object_dispose along the EH path in the copy helper
1961 // function don't throw as newly-copied __block variables always have a
1962 // reference count of 2.
1963 bool CanThrow =
1964 !ForCopyHelper && CGF.cxxDestructorCanThrow(CaptureType);
1965 CGF.enterByrefCleanup(Kind, Field, Flags, /*LoadBlockVarAddr*/ true,
1966 CanThrow);
Akira Hatanakacb6a9332018-07-26 16:51:21 +00001967 }
1968 break;
1969 }
1970 case BlockCaptureEntityKind::None:
Akira Hatanaka2ec36f02018-08-17 15:46:07 +00001971 break;
Akira Hatanakacb6a9332018-07-26 16:51:21 +00001972 }
1973}
1974
Akira Hatanaka9978da32018-08-10 15:09:24 +00001975static void setBlockHelperAttributesVisibility(bool CapturesNonExternalType,
1976 llvm::Function *Fn,
1977 const CGFunctionInfo &FI,
1978 CodeGenModule &CGM) {
1979 if (CapturesNonExternalType) {
1980 CGM.SetInternalFunctionAttributes(GlobalDecl(), Fn, FI);
1981 } else {
1982 Fn->setVisibility(llvm::GlobalValue::HiddenVisibility);
1983 Fn->setUnnamedAddr(llvm::GlobalValue::UnnamedAddr::Global);
1984 CGM.SetLLVMFunctionAttributes(nullptr, FI, Fn);
1985 CGM.SetLLVMFunctionAttributesForDefinition(nullptr, Fn);
1986 }
1987}
John McCallf593b102013-01-22 03:56:22 +00001988/// Generate the copy-helper function for a block closure object:
1989/// static void block_copy_helper(block_t *dst, block_t *src);
1990/// The runtime will have previously initialized 'dst' by doing a
1991/// bit-copy of 'src'.
1992///
1993/// Note that this copies an entire block closure object to the heap;
1994/// it should not be confused with a 'byref copy helper', which moves
1995/// the contents of an individual __block variable to the heap.
John McCall351762c2011-02-07 10:33:21 +00001996llvm::Constant *
John McCallad7c5c12011-02-08 08:22:06 +00001997CodeGenFunction::GenerateCopyHelperFunction(const CGBlockInfo &blockInfo) {
Akira Hatanaka9978da32018-08-10 15:09:24 +00001998 SmallVector<BlockCaptureManagedEntity, 4> CopiedCaptures;
Akira Hatanaka2ec36f02018-08-17 15:46:07 +00001999 findBlockCapturedManagedEntities(blockInfo, getLangOpts(), CopiedCaptures);
Akira Hatanaka9978da32018-08-10 15:09:24 +00002000 std::string FuncName =
2001 getCopyDestroyHelperFuncName(CopiedCaptures, blockInfo.BlockAlign,
Akira Hatanaka2ec36f02018-08-17 15:46:07 +00002002 CaptureStrKind::CopyHelper, CGM);
Akira Hatanaka9978da32018-08-10 15:09:24 +00002003
2004 if (llvm::GlobalValue *Func = CGM.getModule().getNamedValue(FuncName))
Akira Hatanaka936240c2018-08-14 00:15:42 +00002005 return llvm::ConstantExpr::getBitCast(Func, VoidPtrTy);
Akira Hatanaka9978da32018-08-10 15:09:24 +00002006
John McCall351762c2011-02-07 10:33:21 +00002007 ASTContext &C = getContext();
2008
2009 FunctionArgList args;
Alexey Bataev56223232017-06-09 13:40:18 +00002010 ImplicitParamDecl DstDecl(getContext(), C.VoidPtrTy,
2011 ImplicitParamDecl::Other);
2012 args.push_back(&DstDecl);
2013 ImplicitParamDecl SrcDecl(getContext(), C.VoidPtrTy,
2014 ImplicitParamDecl::Other);
2015 args.push_back(&SrcDecl);
Mike Stump11289f42009-09-09 15:08:12 +00002016
John McCallc56a8b32016-03-11 04:30:31 +00002017 const CGFunctionInfo &FI =
2018 CGM.getTypes().arrangeBuiltinFunctionDeclaration(C.VoidTy, args);
Mike Stump0c743272009-03-06 01:33:24 +00002019
John McCall351762c2011-02-07 10:33:21 +00002020 // FIXME: it would be nice if these were mergeable with things with
2021 // identical semantics.
John McCalla729c622012-02-17 03:33:10 +00002022 llvm::FunctionType *LTy = CGM.getTypes().GetFunctionType(FI);
Mike Stump0c743272009-03-06 01:33:24 +00002023
2024 llvm::Function *Fn =
Akira Hatanaka9978da32018-08-10 15:09:24 +00002025 llvm::Function::Create(LTy, llvm::GlobalValue::LinkOnceODRLinkage,
2026 FuncName, &CGM.getModule());
Mike Stump0c743272009-03-06 01:33:24 +00002027
2028 IdentifierInfo *II
Akira Hatanaka9978da32018-08-10 15:09:24 +00002029 = &CGM.getContext().Idents.get(FuncName);
Mike Stump0c743272009-03-06 01:33:24 +00002030
John McCall351762c2011-02-07 10:33:21 +00002031 FunctionDecl *FD = FunctionDecl::Create(C,
2032 C.getTranslationUnitDecl(),
Abramo Bagnaradff19302011-03-08 08:55:46 +00002033 SourceLocation(),
Craig Topper8a13c412014-05-21 05:09:00 +00002034 SourceLocation(), II, C.VoidTy,
2035 nullptr, SC_Static,
Douglas Gregorc4df4072010-04-19 22:54:31 +00002036 false,
Eric Christopher56ef3742012-04-12 00:35:04 +00002037 false);
Akira Hatanakaaec6b2c2015-10-08 20:26:34 +00002038
Akira Hatanaka9978da32018-08-10 15:09:24 +00002039 setBlockHelperAttributesVisibility(blockInfo.CapturesNonExternalType, Fn, FI,
2040 CGM);
Adrian Prantl22e66b42014-04-11 01:13:04 +00002041 StartFunction(FD, C.VoidTy, Fn, FI, args);
Stephen Kellyf2ceec42018-08-09 21:08:08 +00002042 ApplyDebugLocation NL{*this, blockInfo.getBlockExpr()->getBeginLoc()};
Chris Lattner2192fe52011-07-18 04:24:23 +00002043 llvm::Type *structPtrTy = blockInfo.StructureType->getPointerTo();
Mike Stumpaeb0ffd2009-03-07 02:35:30 +00002044
Alexey Bataev56223232017-06-09 13:40:18 +00002045 Address src = GetAddrOfLocalVar(&SrcDecl);
John McCall7f416cc2015-09-08 08:05:57 +00002046 src = Address(Builder.CreateLoad(src), blockInfo.BlockAlign);
John McCallad7c5c12011-02-08 08:22:06 +00002047 src = Builder.CreateBitCast(src, structPtrTy, "block.source");
Mike Stumpaeb0ffd2009-03-07 02:35:30 +00002048
Alexey Bataev56223232017-06-09 13:40:18 +00002049 Address dst = GetAddrOfLocalVar(&DstDecl);
John McCall7f416cc2015-09-08 08:05:57 +00002050 dst = Address(Builder.CreateLoad(dst), blockInfo.BlockAlign);
John McCallad7c5c12011-02-08 08:22:06 +00002051 dst = Builder.CreateBitCast(dst, structPtrTy, "block.dest");
Mike Stumpaeb0ffd2009-03-07 02:35:30 +00002052
Alex Lorenze08e5bc2017-03-06 16:23:04 +00002053 for (const auto &CopiedCapture : CopiedCaptures) {
Akira Hatanaka9978da32018-08-10 15:09:24 +00002054 const BlockDecl::Capture &CI = *CopiedCapture.CI;
2055 const CGBlockInfo::Capture &capture = *CopiedCapture.Capture;
Akira Hatanakacb6a9332018-07-26 16:51:21 +00002056 QualType captureType = CI.getVariable()->getType();
Akira Hatanaka2ec36f02018-08-17 15:46:07 +00002057 BlockFieldFlags flags = CopiedCapture.CopyFlags;
John McCall351762c2011-02-07 10:33:21 +00002058
2059 unsigned index = capture.getIndex();
John McCall7f416cc2015-09-08 08:05:57 +00002060 Address srcField = Builder.CreateStructGEP(src, index, capture.getOffset());
2061 Address dstField = Builder.CreateStructGEP(dst, index, capture.getOffset());
John McCall351762c2011-02-07 10:33:21 +00002062
Akira Hatanaka2ec36f02018-08-17 15:46:07 +00002063 switch (CopiedCapture.CopyKind) {
Akira Hatanaka4a6f1902018-08-13 20:59:57 +00002064 case BlockCaptureEntityKind::CXXRecord:
2065 // If there's an explicit copy expression, we do that.
2066 assert(CI.getCopyExpr() && "copy expression for variable is missing");
Alex Lorenze08e5bc2017-03-06 16:23:04 +00002067 EmitSynthesizedCXXCopyCtor(dstField, srcField, CI.getCopyExpr());
Akira Hatanaka4a6f1902018-08-13 20:59:57 +00002068 break;
2069 case BlockCaptureEntityKind::ARCWeak:
John McCall31168b02011-06-15 23:02:42 +00002070 EmitARCCopyWeak(dstField, srcField);
Akira Hatanaka4a6f1902018-08-13 20:59:57 +00002071 break;
2072 case BlockCaptureEntityKind::NonTrivialCStruct: {
2073 // If this is a C struct that requires non-trivial copy construction,
2074 // emit a call to its copy constructor.
Akira Hatanaka7275da02018-02-28 07:15:55 +00002075 QualType varType = CI.getVariable()->getType();
2076 callCStructCopyConstructor(MakeAddrLValue(dstField, varType),
2077 MakeAddrLValue(srcField, varType));
Akira Hatanaka4a6f1902018-08-13 20:59:57 +00002078 break;
2079 }
2080 case BlockCaptureEntityKind::ARCStrong: {
John McCall351762c2011-02-07 10:33:21 +00002081 llvm::Value *srcValue = Builder.CreateLoad(srcField, "blockcopy.src");
Akira Hatanaka4a6f1902018-08-13 20:59:57 +00002082 // At -O0, store null into the destination field (so that the
2083 // storeStrong doesn't over-release) and then call storeStrong.
2084 // This is a workaround to not having an initStrong call.
2085 if (CGM.getCodeGenOpts().OptimizationLevel == 0) {
2086 auto *ty = cast<llvm::PointerType>(srcValue->getType());
2087 llvm::Value *null = llvm::ConstantPointerNull::get(ty);
2088 Builder.CreateStore(null, dstField);
2089 EmitARCStoreStrongCall(dstField, srcValue, true);
John McCalle68b8f42012-10-17 02:28:37 +00002090
Akira Hatanaka4a6f1902018-08-13 20:59:57 +00002091 // With optimization enabled, take advantage of the fact that
2092 // the blocks runtime guarantees a memcpy of the block data, and
2093 // just emit a retain of the src field.
John McCalle68b8f42012-10-17 02:28:37 +00002094 } else {
Akira Hatanaka4a6f1902018-08-13 20:59:57 +00002095 EmitARCRetainNonBlock(srcValue);
John McCall882987f2013-02-28 19:01:20 +00002096
Akira Hatanaka4a6f1902018-08-13 20:59:57 +00002097 // Unless EH cleanup is required, we don't need this anymore, so kill
2098 // it. It's not quite worth the annoyance to avoid creating it in the
2099 // first place.
2100 if (!needsEHCleanup(captureType.isDestructedType()))
2101 cast<llvm::Instruction>(dstField.getPointer())->eraseFromParent();
John McCalle68b8f42012-10-17 02:28:37 +00002102 }
Akira Hatanaka4a6f1902018-08-13 20:59:57 +00002103 break;
2104 }
2105 case BlockCaptureEntityKind::BlockObject: {
2106 llvm::Value *srcValue = Builder.CreateLoad(srcField, "blockcopy.src");
2107 srcValue = Builder.CreateBitCast(srcValue, VoidPtrTy);
2108 llvm::Value *dstAddr =
2109 Builder.CreateBitCast(dstField.getPointer(), VoidPtrTy);
2110 llvm::Value *args[] = {
2111 dstAddr, srcValue, llvm::ConstantInt::get(Int32Ty, flags.getBitMask())
2112 };
2113
2114 if (CI.isByRef() && C.getBlockVarCopyInit(CI.getVariable()).canThrow())
2115 EmitRuntimeCallOrInvoke(CGM.getBlockObjectAssign(), args);
2116 else
2117 EmitNounwindRuntimeCall(CGM.getBlockObjectAssign(), args);
2118 break;
2119 }
2120 case BlockCaptureEntityKind::None:
Akira Hatanaka2ec36f02018-08-17 15:46:07 +00002121 continue;
Mike Stumpaeb0ffd2009-03-07 02:35:30 +00002122 }
Akira Hatanakacb6a9332018-07-26 16:51:21 +00002123
2124 // Ensure that we destroy the copied object if an exception is thrown later
2125 // in the helper function.
Akira Hatanaka2ec36f02018-08-17 15:46:07 +00002126 pushCaptureCleanup(CopiedCapture.CopyKind, dstField, captureType, flags,
Akira Hatanaka9978da32018-08-10 15:09:24 +00002127 /*ForCopyHelper*/ true, CI.getVariable(), *this);
Mike Stumpaeb0ffd2009-03-07 02:35:30 +00002128 }
2129
John McCallad7c5c12011-02-08 08:22:06 +00002130 FinishFunction();
Mike Stump0c743272009-03-06 01:33:24 +00002131
John McCalle3dc1702011-02-15 09:22:45 +00002132 return llvm::ConstantExpr::getBitCast(Fn, VoidPtrTy);
Mike Stump97d01d52009-03-04 03:23:46 +00002133}
2134
Akira Hatanaka7275da02018-02-28 07:15:55 +00002135static BlockFieldFlags
2136getBlockFieldFlagsForObjCObjectPointer(const BlockDecl::Capture &CI,
2137 QualType T) {
2138 BlockFieldFlags Flags = BLOCK_FIELD_IS_OBJECT;
2139 if (T->isBlockPointerType())
2140 Flags = BLOCK_FIELD_IS_BLOCK;
2141 return Flags;
2142}
2143
Alex Lorenze08e5bc2017-03-06 16:23:04 +00002144static std::pair<BlockCaptureEntityKind, BlockFieldFlags>
2145computeDestroyInfoForBlockCapture(const BlockDecl::Capture &CI, QualType T,
2146 const LangOptions &LangOpts) {
Akira Hatanaka8e57b072018-10-01 21:51:28 +00002147 if (CI.isEscapingByref()) {
Akira Hatanaka7275da02018-02-28 07:15:55 +00002148 BlockFieldFlags Flags = BLOCK_FIELD_IS_BYREF;
Alex Lorenze08e5bc2017-03-06 16:23:04 +00002149 if (T.isObjCGCWeak())
2150 Flags |= BLOCK_FIELD_IS_WEAK;
2151 return std::make_pair(BlockCaptureEntityKind::BlockObject, Flags);
2152 }
2153
Akira Hatanaka7275da02018-02-28 07:15:55 +00002154 switch (T.isDestructedType()) {
2155 case QualType::DK_cxx_destructor:
Alex Lorenze08e5bc2017-03-06 16:23:04 +00002156 return std::make_pair(BlockCaptureEntityKind::CXXRecord, BlockFieldFlags());
Akira Hatanaka7275da02018-02-28 07:15:55 +00002157 case QualType::DK_objc_strong_lifetime:
2158 // Use objc_storeStrong for __strong direct captures; the
2159 // dynamic tools really like it when we do this.
2160 return std::make_pair(BlockCaptureEntityKind::ARCStrong,
2161 getBlockFieldFlagsForObjCObjectPointer(CI, T));
2162 case QualType::DK_objc_weak_lifetime:
2163 // Support __weak direct captures.
2164 return std::make_pair(BlockCaptureEntityKind::ARCWeak,
2165 getBlockFieldFlagsForObjCObjectPointer(CI, T));
2166 case QualType::DK_nontrivial_c_struct:
2167 return std::make_pair(BlockCaptureEntityKind::NonTrivialCStruct,
2168 BlockFieldFlags());
2169 case QualType::DK_none: {
2170 // Non-ARC captures are strong, and we need to use _Block_object_dispose.
2171 if (T->isObjCRetainableType() && !T.getQualifiers().hasObjCLifetime() &&
2172 !LangOpts.ObjCAutoRefCount)
2173 return std::make_pair(BlockCaptureEntityKind::BlockObject,
2174 getBlockFieldFlagsForObjCObjectPointer(CI, T));
2175 // Otherwise, we have nothing to do.
2176 return std::make_pair(BlockCaptureEntityKind::None, BlockFieldFlags());
Alex Lorenze08e5bc2017-03-06 16:23:04 +00002177 }
Akira Hatanaka7275da02018-02-28 07:15:55 +00002178 }
Nico Weberb3897eb2018-02-28 19:28:47 +00002179 llvm_unreachable("after exhaustive DestructionKind switch");
Alex Lorenze08e5bc2017-03-06 16:23:04 +00002180}
2181
John McCallf593b102013-01-22 03:56:22 +00002182/// Generate the destroy-helper function for a block closure object:
2183/// static void block_destroy_helper(block_t *theBlock);
2184///
2185/// Note that this destroys a heap-allocated block closure object;
2186/// it should not be confused with a 'byref destroy helper', which
2187/// destroys the heap-allocated contents of an individual __block
2188/// variable.
John McCall351762c2011-02-07 10:33:21 +00002189llvm::Constant *
John McCallad7c5c12011-02-08 08:22:06 +00002190CodeGenFunction::GenerateDestroyHelperFunction(const CGBlockInfo &blockInfo) {
Akira Hatanaka9978da32018-08-10 15:09:24 +00002191 SmallVector<BlockCaptureManagedEntity, 4> DestroyedCaptures;
Akira Hatanaka2ec36f02018-08-17 15:46:07 +00002192 findBlockCapturedManagedEntities(blockInfo, getLangOpts(), DestroyedCaptures);
Akira Hatanaka9978da32018-08-10 15:09:24 +00002193 std::string FuncName =
2194 getCopyDestroyHelperFuncName(DestroyedCaptures, blockInfo.BlockAlign,
Akira Hatanaka2ec36f02018-08-17 15:46:07 +00002195 CaptureStrKind::DisposeHelper, CGM);
Akira Hatanaka9978da32018-08-10 15:09:24 +00002196
2197 if (llvm::GlobalValue *Func = CGM.getModule().getNamedValue(FuncName))
Akira Hatanaka936240c2018-08-14 00:15:42 +00002198 return llvm::ConstantExpr::getBitCast(Func, VoidPtrTy);
Akira Hatanaka9978da32018-08-10 15:09:24 +00002199
John McCall351762c2011-02-07 10:33:21 +00002200 ASTContext &C = getContext();
Mike Stump0c743272009-03-06 01:33:24 +00002201
John McCall351762c2011-02-07 10:33:21 +00002202 FunctionArgList args;
Alexey Bataev56223232017-06-09 13:40:18 +00002203 ImplicitParamDecl SrcDecl(getContext(), C.VoidPtrTy,
2204 ImplicitParamDecl::Other);
2205 args.push_back(&SrcDecl);
Mike Stump11289f42009-09-09 15:08:12 +00002206
John McCallc56a8b32016-03-11 04:30:31 +00002207 const CGFunctionInfo &FI =
2208 CGM.getTypes().arrangeBuiltinFunctionDeclaration(C.VoidTy, args);
Mike Stump0c743272009-03-06 01:33:24 +00002209
Mike Stumpcbc2bca2009-06-05 23:26:36 +00002210 // FIXME: We'd like to put these into a mergable by content, with
2211 // internal linkage.
John McCalla729c622012-02-17 03:33:10 +00002212 llvm::FunctionType *LTy = CGM.getTypes().GetFunctionType(FI);
Mike Stump0c743272009-03-06 01:33:24 +00002213
2214 llvm::Function *Fn =
Akira Hatanaka9978da32018-08-10 15:09:24 +00002215 llvm::Function::Create(LTy, llvm::GlobalValue::LinkOnceODRLinkage,
2216 FuncName, &CGM.getModule());
Mike Stump0c743272009-03-06 01:33:24 +00002217
2218 IdentifierInfo *II
Akira Hatanaka9978da32018-08-10 15:09:24 +00002219 = &CGM.getContext().Idents.get(FuncName);
Mike Stump0c743272009-03-06 01:33:24 +00002220
John McCall351762c2011-02-07 10:33:21 +00002221 FunctionDecl *FD = FunctionDecl::Create(C, C.getTranslationUnitDecl(),
Abramo Bagnaradff19302011-03-08 08:55:46 +00002222 SourceLocation(),
Craig Topper8a13c412014-05-21 05:09:00 +00002223 SourceLocation(), II, C.VoidTy,
2224 nullptr, SC_Static,
Eric Christopher56ef3742012-04-12 00:35:04 +00002225 false, false);
Akira Hatanakaaec6b2c2015-10-08 20:26:34 +00002226
Akira Hatanaka9978da32018-08-10 15:09:24 +00002227 setBlockHelperAttributesVisibility(blockInfo.CapturesNonExternalType, Fn, FI,
2228 CGM);
Adrian Prantl22e66b42014-04-11 01:13:04 +00002229 StartFunction(FD, C.VoidTy, Fn, FI, args);
Akira Hatanaka9978da32018-08-10 15:09:24 +00002230 markAsIgnoreThreadCheckingAtRuntime(Fn);
2231
Stephen Kellyf2ceec42018-08-09 21:08:08 +00002232 ApplyDebugLocation NL{*this, blockInfo.getBlockExpr()->getBeginLoc()};
Mike Stump6f7d9f82009-03-07 02:53:18 +00002233
Chris Lattner2192fe52011-07-18 04:24:23 +00002234 llvm::Type *structPtrTy = blockInfo.StructureType->getPointerTo();
Mike Stump6f7d9f82009-03-07 02:53:18 +00002235
Alexey Bataev56223232017-06-09 13:40:18 +00002236 Address src = GetAddrOfLocalVar(&SrcDecl);
John McCall7f416cc2015-09-08 08:05:57 +00002237 src = Address(Builder.CreateLoad(src), blockInfo.BlockAlign);
John McCallad7c5c12011-02-08 08:22:06 +00002238 src = Builder.CreateBitCast(src, structPtrTy, "block");
Mike Stump6f7d9f82009-03-07 02:53:18 +00002239
John McCallad7c5c12011-02-08 08:22:06 +00002240 CodeGenFunction::RunCleanupsScope cleanups(*this);
John McCall351762c2011-02-07 10:33:21 +00002241
Alex Lorenze08e5bc2017-03-06 16:23:04 +00002242 for (const auto &DestroyedCapture : DestroyedCaptures) {
Akira Hatanaka9978da32018-08-10 15:09:24 +00002243 const BlockDecl::Capture &CI = *DestroyedCapture.CI;
2244 const CGBlockInfo::Capture &capture = *DestroyedCapture.Capture;
Akira Hatanaka2ec36f02018-08-17 15:46:07 +00002245 BlockFieldFlags flags = DestroyedCapture.DisposeFlags;
John McCall351762c2011-02-07 10:33:21 +00002246
John McCall7f416cc2015-09-08 08:05:57 +00002247 Address srcField =
2248 Builder.CreateStructGEP(src, capture.getIndex(), capture.getOffset());
John McCall351762c2011-02-07 10:33:21 +00002249
Akira Hatanaka2ec36f02018-08-17 15:46:07 +00002250 pushCaptureCleanup(DestroyedCapture.DisposeKind, srcField,
Akira Hatanaka9978da32018-08-10 15:09:24 +00002251 CI.getVariable()->getType(), flags,
2252 /*ForCopyHelper*/ false, CI.getVariable(), *this);
Mike Stump6f7d9f82009-03-07 02:53:18 +00002253 }
2254
John McCall351762c2011-02-07 10:33:21 +00002255 cleanups.ForceCleanup();
2256
John McCallad7c5c12011-02-08 08:22:06 +00002257 FinishFunction();
Mike Stump0c743272009-03-06 01:33:24 +00002258
John McCalle3dc1702011-02-15 09:22:45 +00002259 return llvm::ConstantExpr::getBitCast(Fn, VoidPtrTy);
Mike Stump0c743272009-03-06 01:33:24 +00002260}
2261
John McCallf9b056b2011-03-31 08:03:29 +00002262namespace {
2263
2264/// Emits the copy/dispose helper functions for a __block object of id type.
John McCall7f416cc2015-09-08 08:05:57 +00002265class ObjectByrefHelpers final : public BlockByrefHelpers {
John McCallf9b056b2011-03-31 08:03:29 +00002266 BlockFieldFlags Flags;
2267
2268public:
2269 ObjectByrefHelpers(CharUnits alignment, BlockFieldFlags flags)
John McCall7f416cc2015-09-08 08:05:57 +00002270 : BlockByrefHelpers(alignment), Flags(flags) {}
John McCallf9b056b2011-03-31 08:03:29 +00002271
John McCall7f416cc2015-09-08 08:05:57 +00002272 void emitCopy(CodeGenFunction &CGF, Address destField,
2273 Address srcField) override {
John McCallf9b056b2011-03-31 08:03:29 +00002274 destField = CGF.Builder.CreateBitCast(destField, CGF.VoidPtrTy);
2275
2276 srcField = CGF.Builder.CreateBitCast(srcField, CGF.VoidPtrPtrTy);
2277 llvm::Value *srcValue = CGF.Builder.CreateLoad(srcField);
2278
2279 unsigned flags = (Flags | BLOCK_BYREF_CALLER).getBitMask();
2280
2281 llvm::Value *flagsVal = llvm::ConstantInt::get(CGF.Int32Ty, flags);
2282 llvm::Value *fn = CGF.CGM.getBlockObjectAssign();
John McCall882987f2013-02-28 19:01:20 +00002283
John McCall7f416cc2015-09-08 08:05:57 +00002284 llvm::Value *args[] = { destField.getPointer(), srcValue, flagsVal };
John McCall882987f2013-02-28 19:01:20 +00002285 CGF.EmitNounwindRuntimeCall(fn, args);
John McCallf9b056b2011-03-31 08:03:29 +00002286 }
2287
John McCall7f416cc2015-09-08 08:05:57 +00002288 void emitDispose(CodeGenFunction &CGF, Address field) override {
John McCallf9b056b2011-03-31 08:03:29 +00002289 field = CGF.Builder.CreateBitCast(field, CGF.Int8PtrTy->getPointerTo(0));
2290 llvm::Value *value = CGF.Builder.CreateLoad(field);
2291
Akira Hatanaka9978da32018-08-10 15:09:24 +00002292 CGF.BuildBlockRelease(value, Flags | BLOCK_BYREF_CALLER, false);
John McCallf9b056b2011-03-31 08:03:29 +00002293 }
2294
Craig Topper4f12f102014-03-12 06:41:41 +00002295 void profileImpl(llvm::FoldingSetNodeID &id) const override {
John McCallf9b056b2011-03-31 08:03:29 +00002296 id.AddInteger(Flags.getBitMask());
2297 }
2298};
2299
John McCall31168b02011-06-15 23:02:42 +00002300/// Emits the copy/dispose helpers for an ARC __block __weak variable.
John McCall7f416cc2015-09-08 08:05:57 +00002301class ARCWeakByrefHelpers final : public BlockByrefHelpers {
John McCall31168b02011-06-15 23:02:42 +00002302public:
John McCall7f416cc2015-09-08 08:05:57 +00002303 ARCWeakByrefHelpers(CharUnits alignment) : BlockByrefHelpers(alignment) {}
John McCall31168b02011-06-15 23:02:42 +00002304
John McCall7f416cc2015-09-08 08:05:57 +00002305 void emitCopy(CodeGenFunction &CGF, Address destField,
2306 Address srcField) override {
John McCall31168b02011-06-15 23:02:42 +00002307 CGF.EmitARCMoveWeak(destField, srcField);
2308 }
2309
John McCall7f416cc2015-09-08 08:05:57 +00002310 void emitDispose(CodeGenFunction &CGF, Address field) override {
John McCall31168b02011-06-15 23:02:42 +00002311 CGF.EmitARCDestroyWeak(field);
2312 }
2313
Craig Topper4f12f102014-03-12 06:41:41 +00002314 void profileImpl(llvm::FoldingSetNodeID &id) const override {
John McCall31168b02011-06-15 23:02:42 +00002315 // 0 is distinguishable from all pointers and byref flags
2316 id.AddInteger(0);
2317 }
2318};
2319
2320/// Emits the copy/dispose helpers for an ARC __block __strong variable
2321/// that's not of block-pointer type.
John McCall7f416cc2015-09-08 08:05:57 +00002322class ARCStrongByrefHelpers final : public BlockByrefHelpers {
John McCall31168b02011-06-15 23:02:42 +00002323public:
John McCall7f416cc2015-09-08 08:05:57 +00002324 ARCStrongByrefHelpers(CharUnits alignment) : BlockByrefHelpers(alignment) {}
John McCall31168b02011-06-15 23:02:42 +00002325
John McCall7f416cc2015-09-08 08:05:57 +00002326 void emitCopy(CodeGenFunction &CGF, Address destField,
2327 Address srcField) override {
John McCall31168b02011-06-15 23:02:42 +00002328 // Do a "move" by copying the value and then zeroing out the old
2329 // variable.
2330
John McCall7f416cc2015-09-08 08:05:57 +00002331 llvm::Value *value = CGF.Builder.CreateLoad(srcField);
Fangrui Song6907ce22018-07-30 19:24:48 +00002332
John McCall31168b02011-06-15 23:02:42 +00002333 llvm::Value *null =
2334 llvm::ConstantPointerNull::get(cast<llvm::PointerType>(value->getType()));
John McCall3a237aa2011-11-09 03:17:26 +00002335
Fariborz Jahaniana82e9262013-01-04 23:32:24 +00002336 if (CGF.CGM.getCodeGenOpts().OptimizationLevel == 0) {
John McCall7f416cc2015-09-08 08:05:57 +00002337 CGF.Builder.CreateStore(null, destField);
Fariborz Jahaniana82e9262013-01-04 23:32:24 +00002338 CGF.EmitARCStoreStrongCall(destField, value, /*ignored*/ true);
2339 CGF.EmitARCStoreStrongCall(srcField, null, /*ignored*/ true);
2340 return;
2341 }
John McCall7f416cc2015-09-08 08:05:57 +00002342 CGF.Builder.CreateStore(value, destField);
2343 CGF.Builder.CreateStore(null, srcField);
John McCall31168b02011-06-15 23:02:42 +00002344 }
2345
John McCall7f416cc2015-09-08 08:05:57 +00002346 void emitDispose(CodeGenFunction &CGF, Address field) override {
John McCallcdda29c2013-03-13 03:10:54 +00002347 CGF.EmitARCDestroyStrong(field, ARCImpreciseLifetime);
John McCall31168b02011-06-15 23:02:42 +00002348 }
2349
Craig Topper4f12f102014-03-12 06:41:41 +00002350 void profileImpl(llvm::FoldingSetNodeID &id) const override {
John McCall31168b02011-06-15 23:02:42 +00002351 // 1 is distinguishable from all pointers and byref flags
2352 id.AddInteger(1);
2353 }
2354};
2355
John McCall3a237aa2011-11-09 03:17:26 +00002356/// Emits the copy/dispose helpers for an ARC __block __strong
2357/// variable that's of block-pointer type.
John McCall7f416cc2015-09-08 08:05:57 +00002358class ARCStrongBlockByrefHelpers final : public BlockByrefHelpers {
John McCall3a237aa2011-11-09 03:17:26 +00002359public:
John McCall7f416cc2015-09-08 08:05:57 +00002360 ARCStrongBlockByrefHelpers(CharUnits alignment)
2361 : BlockByrefHelpers(alignment) {}
John McCall3a237aa2011-11-09 03:17:26 +00002362
John McCall7f416cc2015-09-08 08:05:57 +00002363 void emitCopy(CodeGenFunction &CGF, Address destField,
2364 Address srcField) override {
John McCall3a237aa2011-11-09 03:17:26 +00002365 // Do the copy with objc_retainBlock; that's all that
2366 // _Block_object_assign would do anyway, and we'd have to pass the
2367 // right arguments to make sure it doesn't get no-op'ed.
John McCall7f416cc2015-09-08 08:05:57 +00002368 llvm::Value *oldValue = CGF.Builder.CreateLoad(srcField);
John McCall3a237aa2011-11-09 03:17:26 +00002369 llvm::Value *copy = CGF.EmitARCRetainBlock(oldValue, /*mandatory*/ true);
John McCall7f416cc2015-09-08 08:05:57 +00002370 CGF.Builder.CreateStore(copy, destField);
John McCall3a237aa2011-11-09 03:17:26 +00002371 }
2372
John McCall7f416cc2015-09-08 08:05:57 +00002373 void emitDispose(CodeGenFunction &CGF, Address field) override {
John McCallcdda29c2013-03-13 03:10:54 +00002374 CGF.EmitARCDestroyStrong(field, ARCImpreciseLifetime);
John McCall3a237aa2011-11-09 03:17:26 +00002375 }
2376
Craig Topper4f12f102014-03-12 06:41:41 +00002377 void profileImpl(llvm::FoldingSetNodeID &id) const override {
John McCall3a237aa2011-11-09 03:17:26 +00002378 // 2 is distinguishable from all pointers and byref flags
2379 id.AddInteger(2);
2380 }
2381};
2382
John McCallf9b056b2011-03-31 08:03:29 +00002383/// Emits the copy/dispose helpers for a __block variable with a
2384/// nontrivial copy constructor or destructor.
John McCall7f416cc2015-09-08 08:05:57 +00002385class CXXByrefHelpers final : public BlockByrefHelpers {
John McCallf9b056b2011-03-31 08:03:29 +00002386 QualType VarType;
2387 const Expr *CopyExpr;
2388
2389public:
2390 CXXByrefHelpers(CharUnits alignment, QualType type,
2391 const Expr *copyExpr)
John McCall7f416cc2015-09-08 08:05:57 +00002392 : BlockByrefHelpers(alignment), VarType(type), CopyExpr(copyExpr) {}
John McCallf9b056b2011-03-31 08:03:29 +00002393
Craig Topper8a13c412014-05-21 05:09:00 +00002394 bool needsCopy() const override { return CopyExpr != nullptr; }
John McCall7f416cc2015-09-08 08:05:57 +00002395 void emitCopy(CodeGenFunction &CGF, Address destField,
2396 Address srcField) override {
John McCallf9b056b2011-03-31 08:03:29 +00002397 if (!CopyExpr) return;
2398 CGF.EmitSynthesizedCXXCopyCtor(destField, srcField, CopyExpr);
2399 }
2400
John McCall7f416cc2015-09-08 08:05:57 +00002401 void emitDispose(CodeGenFunction &CGF, Address field) override {
John McCallf9b056b2011-03-31 08:03:29 +00002402 EHScopeStack::stable_iterator cleanupDepth = CGF.EHStack.stable_begin();
2403 CGF.PushDestructorCleanup(VarType, field);
2404 CGF.PopCleanupBlocks(cleanupDepth);
2405 }
2406
Craig Topper4f12f102014-03-12 06:41:41 +00002407 void profileImpl(llvm::FoldingSetNodeID &id) const override {
John McCallf9b056b2011-03-31 08:03:29 +00002408 id.AddPointer(VarType.getCanonicalType().getAsOpaquePtr());
2409 }
2410};
Akira Hatanaka7275da02018-02-28 07:15:55 +00002411
2412/// Emits the copy/dispose helpers for a __block variable that is a non-trivial
2413/// C struct.
2414class NonTrivialCStructByrefHelpers final : public BlockByrefHelpers {
2415 QualType VarType;
2416
2417public:
2418 NonTrivialCStructByrefHelpers(CharUnits alignment, QualType type)
2419 : BlockByrefHelpers(alignment), VarType(type) {}
2420
2421 void emitCopy(CodeGenFunction &CGF, Address destField,
2422 Address srcField) override {
2423 CGF.callCStructMoveConstructor(CGF.MakeAddrLValue(destField, VarType),
2424 CGF.MakeAddrLValue(srcField, VarType));
2425 }
2426
2427 bool needsDispose() const override {
2428 return VarType.isDestructedType();
2429 }
2430
2431 void emitDispose(CodeGenFunction &CGF, Address field) override {
2432 EHScopeStack::stable_iterator cleanupDepth = CGF.EHStack.stable_begin();
2433 CGF.pushDestroy(VarType.isDestructedType(), field, VarType);
2434 CGF.PopCleanupBlocks(cleanupDepth);
2435 }
2436
2437 void profileImpl(llvm::FoldingSetNodeID &id) const override {
2438 id.AddPointer(VarType.getCanonicalType().getAsOpaquePtr());
2439 }
2440};
John McCallf9b056b2011-03-31 08:03:29 +00002441} // end anonymous namespace
2442
2443static llvm::Constant *
John McCall7f416cc2015-09-08 08:05:57 +00002444generateByrefCopyHelper(CodeGenFunction &CGF, const BlockByrefInfo &byrefInfo,
2445 BlockByrefHelpers &generator) {
John McCallf9b056b2011-03-31 08:03:29 +00002446 ASTContext &Context = CGF.getContext();
2447
2448 QualType R = Context.VoidTy;
Mike Stumpee2a5ee2009-03-06 02:29:21 +00002449
John McCalla738c252011-03-09 04:27:21 +00002450 FunctionArgList args;
Alexey Bataev56223232017-06-09 13:40:18 +00002451 ImplicitParamDecl Dst(CGF.getContext(), Context.VoidPtrTy,
2452 ImplicitParamDecl::Other);
2453 args.push_back(&Dst);
Mike Stumpf89230d2009-03-06 06:12:24 +00002454
Alexey Bataev56223232017-06-09 13:40:18 +00002455 ImplicitParamDecl Src(CGF.getContext(), Context.VoidPtrTy,
2456 ImplicitParamDecl::Other);
2457 args.push_back(&Src);
Mike Stump11289f42009-09-09 15:08:12 +00002458
John McCallc56a8b32016-03-11 04:30:31 +00002459 const CGFunctionInfo &FI =
2460 CGF.CGM.getTypes().arrangeBuiltinFunctionDeclaration(R, args);
Mike Stumpee2a5ee2009-03-06 02:29:21 +00002461
John McCall7f416cc2015-09-08 08:05:57 +00002462 llvm::FunctionType *LTy = CGF.CGM.getTypes().GetFunctionType(FI);
Mike Stumpee2a5ee2009-03-06 02:29:21 +00002463
Mike Stumpcbc2bca2009-06-05 23:26:36 +00002464 // FIXME: We'd like to put these into a mergable by content, with
2465 // internal linkage.
Mike Stumpee2a5ee2009-03-06 02:29:21 +00002466 llvm::Function *Fn =
2467 llvm::Function::Create(LTy, llvm::GlobalValue::InternalLinkage,
John McCallf9b056b2011-03-31 08:03:29 +00002468 "__Block_byref_object_copy_", &CGF.CGM.getModule());
Mike Stumpee2a5ee2009-03-06 02:29:21 +00002469
2470 IdentifierInfo *II
John McCallf9b056b2011-03-31 08:03:29 +00002471 = &Context.Idents.get("__Block_byref_object_copy_");
Mike Stumpee2a5ee2009-03-06 02:29:21 +00002472
John McCallf9b056b2011-03-31 08:03:29 +00002473 FunctionDecl *FD = FunctionDecl::Create(Context,
2474 Context.getTranslationUnitDecl(),
Abramo Bagnaradff19302011-03-08 08:55:46 +00002475 SourceLocation(),
Craig Topper8a13c412014-05-21 05:09:00 +00002476 SourceLocation(), II, R, nullptr,
John McCall8e7d6562010-08-26 03:08:43 +00002477 SC_Static,
Eric Christopher0b1aef22012-04-12 02:16:49 +00002478 false, false);
John McCall31168b02011-06-15 23:02:42 +00002479
Rafael Espindola51ec5a92018-02-28 23:46:35 +00002480 CGF.CGM.SetInternalFunctionAttributes(GlobalDecl(), Fn, FI);
Akira Hatanakaaec6b2c2015-10-08 20:26:34 +00002481
Adrian Prantl22e66b42014-04-11 01:13:04 +00002482 CGF.StartFunction(FD, R, Fn, FI, args);
Mike Stumpf89230d2009-03-06 06:12:24 +00002483
John McCall7f416cc2015-09-08 08:05:57 +00002484 if (generator.needsCopy()) {
2485 llvm::Type *byrefPtrType = byrefInfo.Type->getPointerTo(0);
Mike Stumpf89230d2009-03-06 06:12:24 +00002486
John McCallf9b056b2011-03-31 08:03:29 +00002487 // dst->x
Alexey Bataev56223232017-06-09 13:40:18 +00002488 Address destField = CGF.GetAddrOfLocalVar(&Dst);
John McCall7f416cc2015-09-08 08:05:57 +00002489 destField = Address(CGF.Builder.CreateLoad(destField),
2490 byrefInfo.ByrefAlignment);
John McCallf9b056b2011-03-31 08:03:29 +00002491 destField = CGF.Builder.CreateBitCast(destField, byrefPtrType);
John McCall7f416cc2015-09-08 08:05:57 +00002492 destField = CGF.emitBlockByrefAddress(destField, byrefInfo, false,
2493 "dest-object");
Mike Stumpee2a5ee2009-03-06 02:29:21 +00002494
John McCallf9b056b2011-03-31 08:03:29 +00002495 // src->x
Alexey Bataev56223232017-06-09 13:40:18 +00002496 Address srcField = CGF.GetAddrOfLocalVar(&Src);
John McCall7f416cc2015-09-08 08:05:57 +00002497 srcField = Address(CGF.Builder.CreateLoad(srcField),
2498 byrefInfo.ByrefAlignment);
John McCallf9b056b2011-03-31 08:03:29 +00002499 srcField = CGF.Builder.CreateBitCast(srcField, byrefPtrType);
John McCall7f416cc2015-09-08 08:05:57 +00002500 srcField = CGF.emitBlockByrefAddress(srcField, byrefInfo, false,
2501 "src-object");
John McCallf9b056b2011-03-31 08:03:29 +00002502
John McCall7f416cc2015-09-08 08:05:57 +00002503 generator.emitCopy(CGF, destField, srcField);
Fangrui Song6907ce22018-07-30 19:24:48 +00002504 }
John McCallf9b056b2011-03-31 08:03:29 +00002505
2506 CGF.FinishFunction();
2507
2508 return llvm::ConstantExpr::getBitCast(Fn, CGF.Int8PtrTy);
Mike Stumpee2a5ee2009-03-06 02:29:21 +00002509}
2510
John McCallf9b056b2011-03-31 08:03:29 +00002511/// Build the copy helper for a __block variable.
2512static llvm::Constant *buildByrefCopyHelper(CodeGenModule &CGM,
John McCall7f416cc2015-09-08 08:05:57 +00002513 const BlockByrefInfo &byrefInfo,
2514 BlockByrefHelpers &generator) {
John McCallf9b056b2011-03-31 08:03:29 +00002515 CodeGenFunction CGF(CGM);
John McCall7f416cc2015-09-08 08:05:57 +00002516 return generateByrefCopyHelper(CGF, byrefInfo, generator);
John McCallf9b056b2011-03-31 08:03:29 +00002517}
2518
2519/// Generate code for a __block variable's dispose helper.
2520static llvm::Constant *
2521generateByrefDisposeHelper(CodeGenFunction &CGF,
John McCall7f416cc2015-09-08 08:05:57 +00002522 const BlockByrefInfo &byrefInfo,
2523 BlockByrefHelpers &generator) {
John McCallf9b056b2011-03-31 08:03:29 +00002524 ASTContext &Context = CGF.getContext();
2525 QualType R = Context.VoidTy;
Mike Stumpee2a5ee2009-03-06 02:29:21 +00002526
John McCalla738c252011-03-09 04:27:21 +00002527 FunctionArgList args;
Alexey Bataev56223232017-06-09 13:40:18 +00002528 ImplicitParamDecl Src(CGF.getContext(), Context.VoidPtrTy,
2529 ImplicitParamDecl::Other);
2530 args.push_back(&Src);
Mike Stump11289f42009-09-09 15:08:12 +00002531
John McCallc56a8b32016-03-11 04:30:31 +00002532 const CGFunctionInfo &FI =
2533 CGF.CGM.getTypes().arrangeBuiltinFunctionDeclaration(R, args);
Mike Stumpee2a5ee2009-03-06 02:29:21 +00002534
John McCall7f416cc2015-09-08 08:05:57 +00002535 llvm::FunctionType *LTy = CGF.CGM.getTypes().GetFunctionType(FI);
Mike Stumpee2a5ee2009-03-06 02:29:21 +00002536
Mike Stumpcbc2bca2009-06-05 23:26:36 +00002537 // FIXME: We'd like to put these into a mergable by content, with
2538 // internal linkage.
Mike Stumpee2a5ee2009-03-06 02:29:21 +00002539 llvm::Function *Fn =
2540 llvm::Function::Create(LTy, llvm::GlobalValue::InternalLinkage,
Fariborz Jahanian50198092010-12-02 17:02:11 +00002541 "__Block_byref_object_dispose_",
John McCallf9b056b2011-03-31 08:03:29 +00002542 &CGF.CGM.getModule());
Mike Stumpee2a5ee2009-03-06 02:29:21 +00002543
2544 IdentifierInfo *II
John McCallf9b056b2011-03-31 08:03:29 +00002545 = &Context.Idents.get("__Block_byref_object_dispose_");
Mike Stumpee2a5ee2009-03-06 02:29:21 +00002546
John McCallf9b056b2011-03-31 08:03:29 +00002547 FunctionDecl *FD = FunctionDecl::Create(Context,
2548 Context.getTranslationUnitDecl(),
Abramo Bagnaradff19302011-03-08 08:55:46 +00002549 SourceLocation(),
Craig Topper8a13c412014-05-21 05:09:00 +00002550 SourceLocation(), II, R, nullptr,
John McCall8e7d6562010-08-26 03:08:43 +00002551 SC_Static,
Eric Christopher0b1aef22012-04-12 02:16:49 +00002552 false, false);
Akira Hatanakaaec6b2c2015-10-08 20:26:34 +00002553
Rafael Espindola51ec5a92018-02-28 23:46:35 +00002554 CGF.CGM.SetInternalFunctionAttributes(GlobalDecl(), Fn, FI);
Akira Hatanakaaec6b2c2015-10-08 20:26:34 +00002555
Adrian Prantl22e66b42014-04-11 01:13:04 +00002556 CGF.StartFunction(FD, R, Fn, FI, args);
Mike Stumpfbe25dd2009-03-06 04:53:30 +00002557
John McCall7f416cc2015-09-08 08:05:57 +00002558 if (generator.needsDispose()) {
Alexey Bataev56223232017-06-09 13:40:18 +00002559 Address addr = CGF.GetAddrOfLocalVar(&Src);
John McCall7f416cc2015-09-08 08:05:57 +00002560 addr = Address(CGF.Builder.CreateLoad(addr), byrefInfo.ByrefAlignment);
2561 auto byrefPtrType = byrefInfo.Type->getPointerTo(0);
2562 addr = CGF.Builder.CreateBitCast(addr, byrefPtrType);
2563 addr = CGF.emitBlockByrefAddress(addr, byrefInfo, false, "object");
John McCallad7c5c12011-02-08 08:22:06 +00002564
John McCall7f416cc2015-09-08 08:05:57 +00002565 generator.emitDispose(CGF, addr);
Fariborz Jahanian50198092010-12-02 17:02:11 +00002566 }
Mike Stumpee2a5ee2009-03-06 02:29:21 +00002567
John McCallf9b056b2011-03-31 08:03:29 +00002568 CGF.FinishFunction();
John McCallad7c5c12011-02-08 08:22:06 +00002569
John McCallf9b056b2011-03-31 08:03:29 +00002570 return llvm::ConstantExpr::getBitCast(Fn, CGF.Int8PtrTy);
Mike Stumpee2a5ee2009-03-06 02:29:21 +00002571}
2572
John McCallf9b056b2011-03-31 08:03:29 +00002573/// Build the dispose helper for a __block variable.
2574static llvm::Constant *buildByrefDisposeHelper(CodeGenModule &CGM,
John McCall7f416cc2015-09-08 08:05:57 +00002575 const BlockByrefInfo &byrefInfo,
2576 BlockByrefHelpers &generator) {
John McCallf9b056b2011-03-31 08:03:29 +00002577 CodeGenFunction CGF(CGM);
John McCall7f416cc2015-09-08 08:05:57 +00002578 return generateByrefDisposeHelper(CGF, byrefInfo, generator);
Mike Stumpee2a5ee2009-03-06 02:29:21 +00002579}
2580
John McCallf593b102013-01-22 03:56:22 +00002581/// Lazily build the copy and dispose helpers for a __block variable
2582/// with the given information.
David Blaikie92551612015-08-13 23:53:09 +00002583template <class T>
John McCall7f416cc2015-09-08 08:05:57 +00002584static T *buildByrefHelpers(CodeGenModule &CGM, const BlockByrefInfo &byrefInfo,
2585 T &&generator) {
John McCallf9b056b2011-03-31 08:03:29 +00002586 llvm::FoldingSetNodeID id;
John McCall7f416cc2015-09-08 08:05:57 +00002587 generator.Profile(id);
John McCallf9b056b2011-03-31 08:03:29 +00002588
2589 void *insertPos;
John McCall7f416cc2015-09-08 08:05:57 +00002590 BlockByrefHelpers *node
John McCallf9b056b2011-03-31 08:03:29 +00002591 = CGM.ByrefHelpersCache.FindNodeOrInsertPos(id, insertPos);
2592 if (node) return static_cast<T*>(node);
2593
John McCall7f416cc2015-09-08 08:05:57 +00002594 generator.CopyHelper = buildByrefCopyHelper(CGM, byrefInfo, generator);
2595 generator.DisposeHelper = buildByrefDisposeHelper(CGM, byrefInfo, generator);
John McCallf9b056b2011-03-31 08:03:29 +00002596
Malcolm Parsonsf92d44c2016-12-06 14:49:18 +00002597 T *copy = new (CGM.getContext()) T(std::forward<T>(generator));
John McCallf9b056b2011-03-31 08:03:29 +00002598 CGM.ByrefHelpersCache.InsertNode(copy, insertPos);
2599 return copy;
2600}
2601
John McCallf593b102013-01-22 03:56:22 +00002602/// Build the copy and dispose helpers for the given __block variable
2603/// emission. Places the helpers in the global cache. Returns null
2604/// if no helpers are required.
John McCall7f416cc2015-09-08 08:05:57 +00002605BlockByrefHelpers *
Chris Lattner2192fe52011-07-18 04:24:23 +00002606CodeGenFunction::buildByrefHelpers(llvm::StructType &byrefType,
John McCallf9b056b2011-03-31 08:03:29 +00002607 const AutoVarEmission &emission) {
2608 const VarDecl &var = *emission.Variable;
Akira Hatanaka8e57b072018-10-01 21:51:28 +00002609 assert(var.isEscapingByref() &&
2610 "only escaping __block variables need byref helpers");
2611
John McCallf9b056b2011-03-31 08:03:29 +00002612 QualType type = var.getType();
2613
John McCall7f416cc2015-09-08 08:05:57 +00002614 auto &byrefInfo = getBlockByrefInfo(&var);
2615
2616 // The alignment we care about for the purposes of uniquing byref
2617 // helpers is the alignment of the actual byref value field.
2618 CharUnits valueAlignment =
2619 byrefInfo.ByrefAlignment.alignmentAtOffset(byrefInfo.FieldOffset);
John McCallf593b102013-01-22 03:56:22 +00002620
John McCallf9b056b2011-03-31 08:03:29 +00002621 if (const CXXRecordDecl *record = type->getAsCXXRecordDecl()) {
Akira Hatanaka9978da32018-08-10 15:09:24 +00002622 const Expr *copyExpr =
2623 CGM.getContext().getBlockVarCopyInit(&var).getCopyExpr();
Craig Topper8a13c412014-05-21 05:09:00 +00002624 if (!copyExpr && record->hasTrivialDestructor()) return nullptr;
John McCallf9b056b2011-03-31 08:03:29 +00002625
David Blaikie92551612015-08-13 23:53:09 +00002626 return ::buildByrefHelpers(
John McCall7f416cc2015-09-08 08:05:57 +00002627 CGM, byrefInfo, CXXByrefHelpers(valueAlignment, type, copyExpr));
John McCallf9b056b2011-03-31 08:03:29 +00002628 }
2629
Akira Hatanaka7275da02018-02-28 07:15:55 +00002630 // If type is a non-trivial C struct type that is non-trivial to
2631 // destructly move or destroy, build the copy and dispose helpers.
2632 if (type.isNonTrivialToPrimitiveDestructiveMove() == QualType::PCK_Struct ||
2633 type.isDestructedType() == QualType::DK_nontrivial_c_struct)
2634 return ::buildByrefHelpers(
2635 CGM, byrefInfo, NonTrivialCStructByrefHelpers(valueAlignment, type));
2636
John McCall31168b02011-06-15 23:02:42 +00002637 // Otherwise, if we don't have a retainable type, there's nothing to do.
2638 // that the runtime does extra copies.
Craig Topper8a13c412014-05-21 05:09:00 +00002639 if (!type->isObjCRetainableType()) return nullptr;
John McCall31168b02011-06-15 23:02:42 +00002640
2641 Qualifiers qs = type.getQualifiers();
2642
2643 // If we have lifetime, that dominates.
2644 if (Qualifiers::ObjCLifetime lifetime = qs.getObjCLifetime()) {
John McCall31168b02011-06-15 23:02:42 +00002645 switch (lifetime) {
2646 case Qualifiers::OCL_None: llvm_unreachable("impossible");
2647
2648 // These are just bits as far as the runtime is concerned.
2649 case Qualifiers::OCL_ExplicitNone:
2650 case Qualifiers::OCL_Autoreleasing:
Craig Topper8a13c412014-05-21 05:09:00 +00002651 return nullptr;
John McCall31168b02011-06-15 23:02:42 +00002652
2653 // Tell the runtime that this is ARC __weak, called by the
2654 // byref routines.
David Blaikie92551612015-08-13 23:53:09 +00002655 case Qualifiers::OCL_Weak:
John McCall7f416cc2015-09-08 08:05:57 +00002656 return ::buildByrefHelpers(CGM, byrefInfo,
2657 ARCWeakByrefHelpers(valueAlignment));
John McCall31168b02011-06-15 23:02:42 +00002658
2659 // ARC __strong __block variables need to be retained.
2660 case Qualifiers::OCL_Strong:
John McCall3a237aa2011-11-09 03:17:26 +00002661 // Block pointers need to be copied, and there's no direct
2662 // transfer possible.
John McCall31168b02011-06-15 23:02:42 +00002663 if (type->isBlockPointerType()) {
John McCall7f416cc2015-09-08 08:05:57 +00002664 return ::buildByrefHelpers(CGM, byrefInfo,
2665 ARCStrongBlockByrefHelpers(valueAlignment));
John McCall31168b02011-06-15 23:02:42 +00002666
2667 // Otherwise, we transfer ownership of the retain from the stack
2668 // to the heap.
2669 } else {
John McCall7f416cc2015-09-08 08:05:57 +00002670 return ::buildByrefHelpers(CGM, byrefInfo,
2671 ARCStrongByrefHelpers(valueAlignment));
John McCall31168b02011-06-15 23:02:42 +00002672 }
2673 }
2674 llvm_unreachable("fell out of lifetime switch!");
2675 }
2676
John McCallf9b056b2011-03-31 08:03:29 +00002677 BlockFieldFlags flags;
2678 if (type->isBlockPointerType()) {
2679 flags |= BLOCK_FIELD_IS_BLOCK;
Fangrui Song6907ce22018-07-30 19:24:48 +00002680 } else if (CGM.getContext().isObjCNSObjectType(type) ||
John McCallf9b056b2011-03-31 08:03:29 +00002681 type->isObjCObjectPointerType()) {
2682 flags |= BLOCK_FIELD_IS_OBJECT;
2683 } else {
Craig Topper8a13c412014-05-21 05:09:00 +00002684 return nullptr;
John McCallf9b056b2011-03-31 08:03:29 +00002685 }
2686
2687 if (type.isObjCGCWeak())
2688 flags |= BLOCK_FIELD_IS_WEAK;
2689
John McCall7f416cc2015-09-08 08:05:57 +00002690 return ::buildByrefHelpers(CGM, byrefInfo,
2691 ObjectByrefHelpers(valueAlignment, flags));
Mike Stumpee2a5ee2009-03-06 02:29:21 +00002692}
2693
John McCall7f416cc2015-09-08 08:05:57 +00002694Address CodeGenFunction::emitBlockByrefAddress(Address baseAddr,
2695 const VarDecl *var,
2696 bool followForward) {
2697 auto &info = getBlockByrefInfo(var);
2698 return emitBlockByrefAddress(baseAddr, info, followForward, var->getName());
John McCall73064872011-03-31 01:59:53 +00002699}
2700
John McCall7f416cc2015-09-08 08:05:57 +00002701Address CodeGenFunction::emitBlockByrefAddress(Address baseAddr,
2702 const BlockByrefInfo &info,
2703 bool followForward,
2704 const llvm::Twine &name) {
2705 // Chase the forwarding address if requested.
2706 if (followForward) {
2707 Address forwardingAddr =
2708 Builder.CreateStructGEP(baseAddr, 1, getPointerSize(), "forwarding");
2709 baseAddr = Address(Builder.CreateLoad(forwardingAddr), info.ByrefAlignment);
2710 }
2711
2712 return Builder.CreateStructGEP(baseAddr, info.FieldIndex,
2713 info.FieldOffset, name);
John McCall73064872011-03-31 01:59:53 +00002714}
2715
John McCall7f416cc2015-09-08 08:05:57 +00002716/// BuildByrefInfo - This routine changes a __block variable declared as T x
John McCall73064872011-03-31 01:59:53 +00002717/// into:
2718///
2719/// struct {
2720/// void *__isa;
2721/// void *__forwarding;
2722/// int32_t __flags;
2723/// int32_t __size;
2724/// void *__copy_helper; // only if needed
2725/// void *__destroy_helper; // only if needed
Fariborz Jahaniana9d44642012-11-14 17:15:51 +00002726/// void *__byref_variable_layout;// only if needed
John McCall73064872011-03-31 01:59:53 +00002727/// char padding[X]; // only if needed
2728/// T x;
2729/// } x
2730///
John McCall7f416cc2015-09-08 08:05:57 +00002731const BlockByrefInfo &CodeGenFunction::getBlockByrefInfo(const VarDecl *D) {
2732 auto it = BlockByrefInfos.find(D);
2733 if (it != BlockByrefInfos.end())
2734 return it->second;
John McCall73064872011-03-31 01:59:53 +00002735
John McCall7f416cc2015-09-08 08:05:57 +00002736 llvm::StructType *byrefType =
Chris Lattner5ec04a52011-08-12 17:43:31 +00002737 llvm::StructType::create(getLLVMContext(),
2738 "struct.__block_byref_" + D->getNameAsString());
Fangrui Song6907ce22018-07-30 19:24:48 +00002739
John McCall7f416cc2015-09-08 08:05:57 +00002740 QualType Ty = D->getType();
2741
2742 CharUnits size;
2743 SmallVector<llvm::Type *, 8> types;
Fangrui Song6907ce22018-07-30 19:24:48 +00002744
John McCall73064872011-03-31 01:59:53 +00002745 // void *__isa;
John McCall9dc0db22011-05-15 01:53:33 +00002746 types.push_back(Int8PtrTy);
John McCall7f416cc2015-09-08 08:05:57 +00002747 size += getPointerSize();
Fangrui Song6907ce22018-07-30 19:24:48 +00002748
John McCall73064872011-03-31 01:59:53 +00002749 // void *__forwarding;
John McCall7f416cc2015-09-08 08:05:57 +00002750 types.push_back(llvm::PointerType::getUnqual(byrefType));
2751 size += getPointerSize();
Fangrui Song6907ce22018-07-30 19:24:48 +00002752
John McCall73064872011-03-31 01:59:53 +00002753 // int32_t __flags;
John McCall9dc0db22011-05-15 01:53:33 +00002754 types.push_back(Int32Ty);
John McCall7f416cc2015-09-08 08:05:57 +00002755 size += CharUnits::fromQuantity(4);
Fangrui Song6907ce22018-07-30 19:24:48 +00002756
John McCall73064872011-03-31 01:59:53 +00002757 // int32_t __size;
John McCall9dc0db22011-05-15 01:53:33 +00002758 types.push_back(Int32Ty);
John McCall7f416cc2015-09-08 08:05:57 +00002759 size += CharUnits::fromQuantity(4);
2760
Fariborz Jahanian998f0a32012-11-28 23:12:17 +00002761 // Note that this must match *exactly* the logic in buildByrefHelpers.
John McCall7f416cc2015-09-08 08:05:57 +00002762 bool hasCopyAndDispose = getContext().BlockRequiresCopying(Ty, D);
2763 if (hasCopyAndDispose) {
John McCall73064872011-03-31 01:59:53 +00002764 /// void *__copy_helper;
John McCall9dc0db22011-05-15 01:53:33 +00002765 types.push_back(Int8PtrTy);
John McCall7f416cc2015-09-08 08:05:57 +00002766 size += getPointerSize();
Fangrui Song6907ce22018-07-30 19:24:48 +00002767
John McCall73064872011-03-31 01:59:53 +00002768 /// void *__destroy_helper;
John McCall9dc0db22011-05-15 01:53:33 +00002769 types.push_back(Int8PtrTy);
John McCall7f416cc2015-09-08 08:05:57 +00002770 size += getPointerSize();
John McCall73064872011-03-31 01:59:53 +00002771 }
John McCall7f416cc2015-09-08 08:05:57 +00002772
Fariborz Jahaniana9d44642012-11-14 17:15:51 +00002773 bool HasByrefExtendedLayout = false;
2774 Qualifiers::ObjCLifetime Lifetime;
2775 if (getContext().getByrefLifetime(Ty, Lifetime, HasByrefExtendedLayout) &&
John McCall7f416cc2015-09-08 08:05:57 +00002776 HasByrefExtendedLayout) {
Fariborz Jahaniana9d44642012-11-14 17:15:51 +00002777 /// void *__byref_variable_layout;
2778 types.push_back(Int8PtrTy);
John McCall7f416cc2015-09-08 08:05:57 +00002779 size += CharUnits::fromQuantity(PointerSizeInBytes);
John McCall73064872011-03-31 01:59:53 +00002780 }
2781
2782 // T x;
John McCall7f416cc2015-09-08 08:05:57 +00002783 llvm::Type *varTy = ConvertTypeForMem(Ty);
2784
2785 bool packed = false;
2786 CharUnits varAlign = getContext().getDeclAlign(D);
Rui Ueyama83aa9792016-01-14 21:00:27 +00002787 CharUnits varOffset = size.alignTo(varAlign);
John McCall7f416cc2015-09-08 08:05:57 +00002788
2789 // We may have to insert padding.
2790 if (varOffset != size) {
2791 llvm::Type *paddingTy =
2792 llvm::ArrayType::get(Int8Ty, (varOffset - size).getQuantity());
2793
2794 types.push_back(paddingTy);
2795 size = varOffset;
2796
2797 // Conversely, we might have to prevent LLVM from inserting padding.
2798 } else if (CGM.getDataLayout().getABITypeAlignment(varTy)
2799 > varAlign.getQuantity()) {
2800 packed = true;
2801 }
2802 types.push_back(varTy);
2803
2804 byrefType->setBody(types, packed);
2805
2806 BlockByrefInfo info;
2807 info.Type = byrefType;
2808 info.FieldIndex = types.size() - 1;
2809 info.FieldOffset = varOffset;
2810 info.ByrefAlignment = std::max(varAlign, getPointerAlign());
2811
2812 auto pair = BlockByrefInfos.insert({D, info});
2813 assert(pair.second && "info was inserted recursively?");
2814 return pair.first->second;
John McCall73064872011-03-31 01:59:53 +00002815}
2816
2817/// Initialize the structural components of a __block variable, i.e.
2818/// everything but the actual object.
2819void CodeGenFunction::emitByrefStructureInit(const AutoVarEmission &emission) {
John McCallf9b056b2011-03-31 08:03:29 +00002820 // Find the address of the local.
John McCall7f416cc2015-09-08 08:05:57 +00002821 Address addr = emission.Addr;
John McCall73064872011-03-31 01:59:53 +00002822
John McCallf9b056b2011-03-31 08:03:29 +00002823 // That's an alloca of the byref structure type.
Chris Lattner2192fe52011-07-18 04:24:23 +00002824 llvm::StructType *byrefType = cast<llvm::StructType>(
John McCall7f416cc2015-09-08 08:05:57 +00002825 cast<llvm::PointerType>(addr.getPointer()->getType())->getElementType());
2826
2827 unsigned nextHeaderIndex = 0;
2828 CharUnits nextHeaderOffset;
2829 auto storeHeaderField = [&](llvm::Value *value, CharUnits fieldSize,
2830 const Twine &name) {
2831 auto fieldAddr = Builder.CreateStructGEP(addr, nextHeaderIndex,
2832 nextHeaderOffset, name);
2833 Builder.CreateStore(value, fieldAddr);
2834
2835 nextHeaderIndex++;
2836 nextHeaderOffset += fieldSize;
2837 };
John McCallf9b056b2011-03-31 08:03:29 +00002838
2839 // Build the byref helpers if necessary. This is null if we don't need any.
John McCall7f416cc2015-09-08 08:05:57 +00002840 BlockByrefHelpers *helpers = buildByrefHelpers(*byrefType, emission);
John McCall73064872011-03-31 01:59:53 +00002841
2842 const VarDecl &D = *emission.Variable;
2843 QualType type = D.getType();
2844
Fariborz Jahaniana9d44642012-11-14 17:15:51 +00002845 bool HasByrefExtendedLayout;
2846 Qualifiers::ObjCLifetime ByrefLifetime;
2847 bool ByRefHasLifetime =
2848 getContext().getByrefLifetime(type, ByrefLifetime, HasByrefExtendedLayout);
John McCall7f416cc2015-09-08 08:05:57 +00002849
John McCallf9b056b2011-03-31 08:03:29 +00002850 llvm::Value *V;
John McCall73064872011-03-31 01:59:53 +00002851
2852 // Initialize the 'isa', which is just 0 or 1.
2853 int isa = 0;
John McCallf9b056b2011-03-31 08:03:29 +00002854 if (type.isObjCGCWeak())
John McCall73064872011-03-31 01:59:53 +00002855 isa = 1;
2856 V = Builder.CreateIntToPtr(Builder.getInt32(isa), Int8PtrTy, "isa");
John McCall7f416cc2015-09-08 08:05:57 +00002857 storeHeaderField(V, getPointerSize(), "byref.isa");
John McCall73064872011-03-31 01:59:53 +00002858
2859 // Store the address of the variable into its own forwarding pointer.
John McCall7f416cc2015-09-08 08:05:57 +00002860 storeHeaderField(addr.getPointer(), getPointerSize(), "byref.forwarding");
John McCall73064872011-03-31 01:59:53 +00002861
2862 // Blocks ABI:
2863 // c) the flags field is set to either 0 if no helper functions are
Fariborz Jahaniana9d44642012-11-14 17:15:51 +00002864 // needed or BLOCK_BYREF_HAS_COPY_DISPOSE if they are,
John McCall73064872011-03-31 01:59:53 +00002865 BlockFlags flags;
Fariborz Jahaniana9d44642012-11-14 17:15:51 +00002866 if (helpers) flags |= BLOCK_BYREF_HAS_COPY_DISPOSE;
2867 if (ByRefHasLifetime) {
2868 if (HasByrefExtendedLayout) flags |= BLOCK_BYREF_LAYOUT_EXTENDED;
2869 else switch (ByrefLifetime) {
2870 case Qualifiers::OCL_Strong:
2871 flags |= BLOCK_BYREF_LAYOUT_STRONG;
2872 break;
2873 case Qualifiers::OCL_Weak:
2874 flags |= BLOCK_BYREF_LAYOUT_WEAK;
2875 break;
2876 case Qualifiers::OCL_ExplicitNone:
2877 flags |= BLOCK_BYREF_LAYOUT_UNRETAINED;
2878 break;
2879 case Qualifiers::OCL_None:
2880 if (!type->isObjCObjectPointerType() && !type->isBlockPointerType())
2881 flags |= BLOCK_BYREF_LAYOUT_NON_OBJECT;
2882 break;
2883 default:
2884 break;
2885 }
2886 if (CGM.getLangOpts().ObjCGCBitmapPrint) {
2887 printf("\n Inline flag for BYREF variable layout (%d):", flags.getBitMask());
2888 if (flags & BLOCK_BYREF_HAS_COPY_DISPOSE)
2889 printf(" BLOCK_BYREF_HAS_COPY_DISPOSE");
2890 if (flags & BLOCK_BYREF_LAYOUT_MASK) {
2891 BlockFlags ThisFlag(flags.getBitMask() & BLOCK_BYREF_LAYOUT_MASK);
2892 if (ThisFlag == BLOCK_BYREF_LAYOUT_EXTENDED)
2893 printf(" BLOCK_BYREF_LAYOUT_EXTENDED");
2894 if (ThisFlag == BLOCK_BYREF_LAYOUT_STRONG)
2895 printf(" BLOCK_BYREF_LAYOUT_STRONG");
2896 if (ThisFlag == BLOCK_BYREF_LAYOUT_WEAK)
2897 printf(" BLOCK_BYREF_LAYOUT_WEAK");
2898 if (ThisFlag == BLOCK_BYREF_LAYOUT_UNRETAINED)
2899 printf(" BLOCK_BYREF_LAYOUT_UNRETAINED");
2900 if (ThisFlag == BLOCK_BYREF_LAYOUT_NON_OBJECT)
2901 printf(" BLOCK_BYREF_LAYOUT_NON_OBJECT");
2902 }
2903 printf("\n");
2904 }
2905 }
John McCall7f416cc2015-09-08 08:05:57 +00002906 storeHeaderField(llvm::ConstantInt::get(IntTy, flags.getBitMask()),
2907 getIntSize(), "byref.flags");
John McCall73064872011-03-31 01:59:53 +00002908
John McCallf9b056b2011-03-31 08:03:29 +00002909 CharUnits byrefSize = CGM.GetTargetTypeStoreSize(byrefType);
2910 V = llvm::ConstantInt::get(IntTy, byrefSize.getQuantity());
John McCall7f416cc2015-09-08 08:05:57 +00002911 storeHeaderField(V, getIntSize(), "byref.size");
John McCall73064872011-03-31 01:59:53 +00002912
John McCallf9b056b2011-03-31 08:03:29 +00002913 if (helpers) {
John McCall7f416cc2015-09-08 08:05:57 +00002914 storeHeaderField(helpers->CopyHelper, getPointerSize(),
2915 "byref.copyHelper");
2916 storeHeaderField(helpers->DisposeHelper, getPointerSize(),
2917 "byref.disposeHelper");
John McCall73064872011-03-31 01:59:53 +00002918 }
John McCall7f416cc2015-09-08 08:05:57 +00002919
Fariborz Jahaniana9d44642012-11-14 17:15:51 +00002920 if (ByRefHasLifetime && HasByrefExtendedLayout) {
John McCall7f416cc2015-09-08 08:05:57 +00002921 auto layoutInfo = CGM.getObjCRuntime().BuildByrefLayout(CGM, type);
2922 storeHeaderField(layoutInfo, getPointerSize(), "byref.layout");
Fariborz Jahaniana9d44642012-11-14 17:15:51 +00002923 }
John McCall73064872011-03-31 01:59:53 +00002924}
2925
Akira Hatanaka9978da32018-08-10 15:09:24 +00002926void CodeGenFunction::BuildBlockRelease(llvm::Value *V, BlockFieldFlags flags,
2927 bool CanThrow) {
Daniel Dunbar900546d2010-07-16 00:00:15 +00002928 llvm::Value *F = CGM.getBlockObjectDispose();
John McCall882987f2013-02-28 19:01:20 +00002929 llvm::Value *args[] = {
2930 Builder.CreateBitCast(V, Int8PtrTy),
2931 llvm::ConstantInt::get(Int32Ty, flags.getBitMask())
2932 };
Akira Hatanaka9978da32018-08-10 15:09:24 +00002933
2934 if (CanThrow)
2935 EmitRuntimeCallOrInvoke(F, args);
2936 else
2937 EmitNounwindRuntimeCall(F, args);
Mike Stump626aecc2009-03-05 01:23:13 +00002938}
John McCall73064872011-03-31 01:59:53 +00002939
Akira Hatanakacb6a9332018-07-26 16:51:21 +00002940void CodeGenFunction::enterByrefCleanup(CleanupKind Kind, Address Addr,
2941 BlockFieldFlags Flags,
Akira Hatanaka9978da32018-08-10 15:09:24 +00002942 bool LoadBlockVarAddr, bool CanThrow) {
2943 EHStack.pushCleanup<CallBlockRelease>(Kind, Addr, Flags, LoadBlockVarAddr,
2944 CanThrow);
John McCall73064872011-03-31 01:59:53 +00002945}
John McCall7959fee2011-09-09 20:41:01 +00002946
2947/// Adjust the declaration of something from the blocks API.
2948static void configureBlocksRuntimeObject(CodeGenModule &CGM,
2949 llvm::Constant *C) {
Rafael Espindola2ae250c2014-05-09 00:08:36 +00002950 auto *GV = cast<llvm::GlobalValue>(C->stripPointerCasts());
Saleem Abdulrasool442b88b2016-05-28 19:41:35 +00002951
2952 if (CGM.getTarget().getTriple().isOSBinFormatCOFF()) {
2953 IdentifierInfo &II = CGM.getContext().Idents.get(C->getName());
2954 TranslationUnitDecl *TUDecl = CGM.getContext().getTranslationUnitDecl();
2955 DeclContext *DC = TranslationUnitDecl::castToDeclContext(TUDecl);
2956
Saleem Abdulrasool7bae9ad2016-06-03 23:26:30 +00002957 assert((isa<llvm::Function>(C->stripPointerCasts()) ||
2958 isa<llvm::GlobalVariable>(C->stripPointerCasts())) &&
2959 "expected Function or GlobalVariable");
Saleem Abdulrasool442b88b2016-05-28 19:41:35 +00002960
2961 const NamedDecl *ND = nullptr;
2962 for (const auto &Result : DC->lookup(&II))
2963 if ((ND = dyn_cast<FunctionDecl>(Result)) ||
2964 (ND = dyn_cast<VarDecl>(Result)))
2965 break;
2966
2967 // TODO: support static blocks runtime
2968 if (GV->isDeclaration() && (!ND || !ND->hasAttr<DLLExportAttr>())) {
2969 GV->setDLLStorageClass(llvm::GlobalValue::DLLImportStorageClass);
2970 GV->setLinkage(llvm::GlobalValue::ExternalLinkage);
2971 } else {
2972 GV->setDLLStorageClass(llvm::GlobalValue::DLLExportStorageClass);
2973 GV->setLinkage(llvm::GlobalValue::ExternalLinkage);
2974 }
2975 }
2976
Rafael Espindola3c8a39c2018-03-14 18:19:26 +00002977 if (CGM.getLangOpts().BlocksRuntimeOptional && GV->isDeclaration() &&
2978 GV->hasExternalLinkage())
John McCall7959fee2011-09-09 20:41:01 +00002979 GV->setLinkage(llvm::GlobalValue::ExternalWeakLinkage);
Rafael Espindola3c8a39c2018-03-14 18:19:26 +00002980
2981 CGM.setDSOLocal(GV);
John McCall7959fee2011-09-09 20:41:01 +00002982}
2983
2984llvm::Constant *CodeGenModule::getBlockObjectDispose() {
2985 if (BlockObjectDispose)
2986 return BlockObjectDispose;
2987
2988 llvm::Type *args[] = { Int8PtrTy, Int32Ty };
2989 llvm::FunctionType *fty
2990 = llvm::FunctionType::get(VoidTy, args, false);
2991 BlockObjectDispose = CreateRuntimeFunction(fty, "_Block_object_dispose");
2992 configureBlocksRuntimeObject(*this, BlockObjectDispose);
2993 return BlockObjectDispose;
2994}
2995
2996llvm::Constant *CodeGenModule::getBlockObjectAssign() {
2997 if (BlockObjectAssign)
2998 return BlockObjectAssign;
2999
3000 llvm::Type *args[] = { Int8PtrTy, Int8PtrTy, Int32Ty };
3001 llvm::FunctionType *fty
3002 = llvm::FunctionType::get(VoidTy, args, false);
3003 BlockObjectAssign = CreateRuntimeFunction(fty, "_Block_object_assign");
3004 configureBlocksRuntimeObject(*this, BlockObjectAssign);
3005 return BlockObjectAssign;
3006}
3007
3008llvm::Constant *CodeGenModule::getNSConcreteGlobalBlock() {
3009 if (NSConcreteGlobalBlock)
3010 return NSConcreteGlobalBlock;
3011
3012 NSConcreteGlobalBlock = GetOrCreateLLVMGlobal("_NSConcreteGlobalBlock",
Craig Topper8a13c412014-05-21 05:09:00 +00003013 Int8PtrTy->getPointerTo(),
3014 nullptr);
John McCall7959fee2011-09-09 20:41:01 +00003015 configureBlocksRuntimeObject(*this, NSConcreteGlobalBlock);
3016 return NSConcreteGlobalBlock;
3017}
3018
3019llvm::Constant *CodeGenModule::getNSConcreteStackBlock() {
3020 if (NSConcreteStackBlock)
3021 return NSConcreteStackBlock;
3022
3023 NSConcreteStackBlock = GetOrCreateLLVMGlobal("_NSConcreteStackBlock",
Craig Topper8a13c412014-05-21 05:09:00 +00003024 Int8PtrTy->getPointerTo(),
3025 nullptr);
John McCall7959fee2011-09-09 20:41:01 +00003026 configureBlocksRuntimeObject(*this, NSConcreteStackBlock);
Saleem Abdulrasool442b88b2016-05-28 19:41:35 +00003027 return NSConcreteStackBlock;
John McCall7959fee2011-09-09 20:41:01 +00003028}