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