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 4 // X64: internal constant {{.*}} { i8** @_NSConcreteGlobalBlock, i32 1879048192 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 */ 107
