Fariborz Jahanian | 5c8e13f | 2010-11-15 17:06:17 +0000 | [diff] [blame] | 1 | // RUN: %clang_cc1 -fblocks -triple x86_64-apple-darwin9 %s -emit-llvm -o - | FileCheck %s -check-prefix=X64 |
| 2 | // RUN: %clang_cc1 -fblocks -triple i686-apple-darwin9 %s -emit-llvm -o - | FileCheck %s -check-prefix=X32 |
Stephen Hines | 651f13c | 2014-04-23 16:59:28 -0700 | [diff] [blame] | 3 | // RUN: %clang_cc1 -fblocks -triple arm64-apple-darwin %s -emit-llvm -o - | FileCheck %s -check-prefix=ARM64 |
Fariborz Jahanian | 5c8e13f | 2010-11-15 17:06:17 +0000 | [diff] [blame] | 4 | |
Chris Lattner | 9cbe4f0 | 2011-07-09 17:41:47 +0000 | [diff] [blame] | 5 | // X64: internal constant {{.*}} { i8** @_NSConcreteGlobalBlock, i32 1879048192 |
Stephen Hines | 651f13c | 2014-04-23 16:59:28 -0700 | [diff] [blame] | 6 | // X64: store i32 1610612736, i32* %want |
Fariborz Jahanian | 5c8e13f | 2010-11-15 17:06:17 +0000 | [diff] [blame] | 7 | |
| 8 | // X32: @_NSConcreteGlobalBlock, i32 1879048192, i32 0, |
| 9 | // X32: store i32 1610612736, i32* %want |
| 10 | |
| 11 | // rdar://7677537 |
Stephen Hines | 651f13c | 2014-04-23 16:59:28 -0700 | [diff] [blame] | 12 | |
| 13 | // ARM64: @_NSConcreteGlobalBlock, i32 1342177280, i32 0, |
| 14 | // ARM64: store i32 1610612736, i32* %want |
| 15 | |
| 16 | // rdar://9757126 |
| 17 | |
Fariborz Jahanian | 5c8e13f | 2010-11-15 17:06:17 +0000 | [diff] [blame] | 18 | int printf(const char *, ...); |
| 19 | void *malloc(__SIZE_TYPE__ size); |
| 20 | |
| 21 | typedef struct bigbig { |
| 22 | int array[512]; |
| 23 | char more[32]; |
| 24 | } BigStruct_t; |
| 25 | |
| 26 | BigStruct_t (^global)(void) = ^{ return *(BigStruct_t *)malloc(sizeof(struct bigbig)); }; |
| 27 | |
| 28 | const char * getBlockSignature(void *); |
| 29 | |
| 30 | BigStruct_t foo(int param) { |
| 31 | BigStruct_t x; |
| 32 | BigStruct_t (^f)(int) = ^(int param) { |
| 33 | BigStruct_t *result = malloc(sizeof(BigStruct_t)); |
| 34 | result->array[23] = param; |
| 35 | return *result; |
| 36 | }; |
| 37 | getBlockSignature(f); |
| 38 | return x; |
| 39 | } |
| 40 | |
| 41 | enum { |
| 42 | BLOCK_HAS_COPY_DISPOSE = (1 << 25), |
| 43 | BLOCK_HAS_CXX_OBJ = (1 << 26), |
| 44 | BLOCK_IS_GLOBAL = (1 << 28), |
| 45 | BLOCK_USE_STRET = (1 << 29), |
| 46 | BLOCK_HAS_OBJC_TYPE = (1 << 30) |
| 47 | }; |
| 48 | |
| 49 | struct block_descriptor_big { |
| 50 | unsigned long int reserved; |
| 51 | unsigned long int size; |
| 52 | void (*copy)(void *dst, void *src); // conditional on BLOCK_HAS_COPY_DISPOSE |
| 53 | void (*dispose)(void *); // conditional on BLOCK_HAS_COPY_DISPOSE |
| 54 | const char *signature; // conditional on BLOCK_HAS_OBJC |
| 55 | const char *layout; // conditional on BLOCK_HAS_OBJC |
| 56 | }; |
| 57 | struct block_descriptor_small { |
| 58 | unsigned long int reserved; |
| 59 | unsigned long int size; |
| 60 | const char *signature; // conditional on BLOCK_HAS_OBJC |
| 61 | const char *layout; // conditional on BLOCK_HAS_OBJC |
| 62 | }; |
| 63 | |
| 64 | struct block_layout_abi { // can't change |
| 65 | void *isa; |
| 66 | int flags; |
| 67 | int reserved; |
| 68 | void (*invoke)(void *, ...); |
| 69 | struct block_descriptor_big *descriptor; |
| 70 | }; |
| 71 | |
| 72 | const char *getBlockSignature(void *block) { |
| 73 | struct block_layout_abi *layout = (struct block_layout_abi *)block; |
| 74 | if ((layout->flags & BLOCK_HAS_OBJC_TYPE) != BLOCK_HAS_OBJC_TYPE) return 0; |
| 75 | if (layout->flags & BLOCK_HAS_COPY_DISPOSE) |
| 76 | return layout->descriptor->signature; |
| 77 | else |
| 78 | return ((struct block_descriptor_small *)layout->descriptor)->signature; |
| 79 | } |
| 80 | |
| 81 | int usesStruct(void *block) { |
| 82 | struct block_layout_abi *layout = (struct block_layout_abi *)block; |
| 83 | int want = BLOCK_HAS_OBJC_TYPE | BLOCK_USE_STRET; |
| 84 | return (layout->flags & want) == want; |
| 85 | } |
| 86 | |
| 87 | |
| 88 | int main(int argc, char *argv[]) { |
| 89 | printf("desired global flags: %d\n", BLOCK_USE_STRET | BLOCK_IS_GLOBAL | BLOCK_HAS_OBJC_TYPE); |
| 90 | printf("desired stack flags: %d\n", BLOCK_USE_STRET | BLOCK_HAS_OBJC_TYPE); |
| 91 | |
| 92 | printf("should be non-zero: %d\n", usesStruct(global)); |
| 93 | BigStruct_t x; |
| 94 | BigStruct_t (^local)(int) = ^(int param) { |
| 95 | BigStruct_t *result = (BigStruct_t *)malloc(sizeof(BigStruct_t)); |
| 96 | result->array[23] = argc; |
| 97 | return *result; |
| 98 | }; |
| 99 | printf("should be non-zero: %d\n", usesStruct(global)); |
| 100 | printf("should be non-zero: %d\n", usesStruct(local)); |
| 101 | printf("should be zero: %d\n", usesStruct(^void(int x){ })); |
| 102 | return 0; |
| 103 | } |
| 104 | |
| 105 | /* |
| 106 | desired global flags: 1879048192 |
| 107 | desired stack flags: 1610612736 |
| 108 | should be non-zero: 1 |
| 109 | should be non-zero: 1 |
| 110 | should be non-zero: 1 |
| 111 | should be zero: 0 |
| 112 | |
| 113 | */ |