110 lines · c
1// RUN: %clang_cc1 -fblocks -triple x86_64-apple-darwin9 %s -emit-llvm -o - | FileCheck %s -check-prefix=X642// RUN: %clang_cc1 -fblocks -triple i686-apple-darwin9 %s -emit-llvm -o - | FileCheck %s -check-prefix=X323// RUN: %clang_cc1 -fblocks -triple arm64-apple-darwin %s -emit-llvm -o - | FileCheck %s -check-prefix=ARM644 5// X64: internal constant {{.*}} { ptr @_NSConcreteGlobalBlock, i32 18790481926// X64: store i32 1610612736, ptr %want7 8// X32: @_NSConcreteGlobalBlock, i32 1879048192, i32 0,9// X32: store i32 1610612736, ptr %want10 11// ARM64: @_NSConcreteGlobalBlock, i32 1342177280, i32 0,12// ARM64: store i32 1610612736, ptr %want13 14int printf(const char *, ...);15void *malloc(__SIZE_TYPE__ size);16 17typedef struct bigbig {18 int array[512];19 char more[32];20} BigStruct_t;21 22BigStruct_t (^global)(void) = ^{ return *(BigStruct_t *)malloc(sizeof(struct bigbig)); };23 24const char * getBlockSignature(void *);25 26BigStruct_t foo(int param) {27 BigStruct_t x;28 BigStruct_t (^f)(int) = ^(int param) {29 BigStruct_t *result = malloc(sizeof(BigStruct_t));30 result->array[23] = param;31 return *result;32 };33 getBlockSignature(f);34 return x;35}36 37enum {38 BLOCK_HAS_COPY_DISPOSE = (1 << 25),39 BLOCK_HAS_CXX_OBJ = (1 << 26),40 BLOCK_IS_GLOBAL = (1 << 28),41 BLOCK_USE_STRET = (1 << 29),42 BLOCK_HAS_OBJC_TYPE = (1 << 30)43};44 45struct block_descriptor_big {46 unsigned long int reserved;47 unsigned long int size;48 void (*copy)(void *dst, void *src); // conditional on BLOCK_HAS_COPY_DISPOSE49 void (*dispose)(void *); // conditional on BLOCK_HAS_COPY_DISPOSE50 const char *signature; // conditional on BLOCK_HAS_OBJC51 const char *layout; // conditional on BLOCK_HAS_OBJC52};53struct block_descriptor_small {54 unsigned long int reserved;55 unsigned long int size;56 const char *signature; // conditional on BLOCK_HAS_OBJC57 const char *layout; // conditional on BLOCK_HAS_OBJC58};59 60struct block_layout_abi { // can't change61 void *isa;62 int flags;63 int reserved; 64 void (*invoke)(void *, ...);65 struct block_descriptor_big *descriptor;66};67 68const char *getBlockSignature(void *block) {69 struct block_layout_abi *layout = (struct block_layout_abi *)block;70 if ((layout->flags & BLOCK_HAS_OBJC_TYPE) != BLOCK_HAS_OBJC_TYPE) return 0;71 if (layout->flags & BLOCK_HAS_COPY_DISPOSE) 72 return layout->descriptor->signature;73 else74 return ((struct block_descriptor_small *)layout->descriptor)->signature;75}76 77int usesStruct(void *block) {78 struct block_layout_abi *layout = (struct block_layout_abi *)block;79 int want = BLOCK_HAS_OBJC_TYPE | BLOCK_USE_STRET;80 return (layout->flags & want) == want;81}82 83 84int main(int argc, char *argv[]) {85 printf("desired global flags: %d\n", BLOCK_USE_STRET | BLOCK_IS_GLOBAL | BLOCK_HAS_OBJC_TYPE);86 printf("desired stack flags: %d\n", BLOCK_USE_STRET | BLOCK_HAS_OBJC_TYPE);87 88 printf("should be non-zero: %d\n", usesStruct(global));89 BigStruct_t x;90 BigStruct_t (^local)(int) = ^(int param) {91 BigStruct_t *result = (BigStruct_t *)malloc(sizeof(BigStruct_t));92 result->array[23] = argc;93 return *result;94 };95 printf("should be non-zero: %d\n", usesStruct(global));96 printf("should be non-zero: %d\n", usesStruct(local));97 printf("should be zero: %d\n", usesStruct(^void(int x){ }));98 return 0;99}100 101/*102desired global flags: 1879048192103desired stack flags: 1610612736104should be non-zero: 1105should be non-zero: 1106should be non-zero: 1107should be zero: 0108 109*/110