forked from OSchip/llvm-project
				
			
		
			
				
	
	
		
			114 lines
		
	
	
		
			3.6 KiB
		
	
	
	
		
			C
		
	
	
	
			
		
		
	
	
			114 lines
		
	
	
		
			3.6 KiB
		
	
	
	
		
			C
		
	
	
	
| // RUN: %clang_cc1 -fblocks -triple x86_64-apple-darwin9 %s -emit-llvm -o - | FileCheck %s -check-prefix=X64
 | |
| // RUN: %clang_cc1 -fblocks -triple i686-apple-darwin9 %s -emit-llvm -o - | FileCheck %s -check-prefix=X32
 | |
| // RUN: %clang_cc1 -fblocks -triple arm64-apple-darwin %s -emit-llvm -o - | FileCheck %s -check-prefix=ARM64
 | |
| 
 | |
| // X64:   internal constant {{.*}} { i8** @_NSConcreteGlobalBlock, i32 1879048192
 | |
| // X64:   store i32 1610612736, i32* %want
 | |
| 
 | |
| // X32:   @_NSConcreteGlobalBlock, i32 1879048192, i32 0,
 | |
| // X32:   store i32 1610612736, i32* %want
 | |
| 
 | |
| // rdar://7677537
 | |
| 
 | |
| // ARM64: @_NSConcreteGlobalBlock, i32 1342177280, i32 0,
 | |
| // ARM64: store i32 1610612736, i32* %want
 | |
| 
 | |
| // rdar://9757126
 | |
| 
 | |
| int printf(const char *, ...);
 | |
| void *malloc(__SIZE_TYPE__ size);
 | |
| 
 | |
| typedef struct bigbig {
 | |
|    int array[512];
 | |
|    char more[32];
 | |
| } BigStruct_t;
 | |
| 
 | |
| BigStruct_t (^global)(void) = ^{ return *(BigStruct_t *)malloc(sizeof(struct bigbig)); };
 | |
| 
 | |
| const char * getBlockSignature(void *);
 | |
|  
 | |
| BigStruct_t foo(int param) {
 | |
|    BigStruct_t x;
 | |
|    BigStruct_t (^f)(int) = ^(int param) {
 | |
|      BigStruct_t *result = malloc(sizeof(BigStruct_t));
 | |
|      result->array[23] = param;
 | |
|      return *result;
 | |
|    };
 | |
|    getBlockSignature(f);
 | |
|    return x;
 | |
| }
 | |
| 
 | |
| enum {
 | |
|     BLOCK_HAS_COPY_DISPOSE =  (1 << 25),
 | |
|     BLOCK_HAS_CXX_OBJ =       (1 << 26),
 | |
|     BLOCK_IS_GLOBAL =         (1 << 28),
 | |
|     BLOCK_USE_STRET =    (1 << 29),
 | |
|     BLOCK_HAS_OBJC_TYPE  =    (1 << 30)
 | |
| };
 | |
| 
 | |
| struct block_descriptor_big {
 | |
|     unsigned long int reserved;
 | |
|     unsigned long int size;
 | |
|     void (*copy)(void *dst, void *src); // conditional on BLOCK_HAS_COPY_DISPOSE
 | |
|     void (*dispose)(void *);            // conditional on BLOCK_HAS_COPY_DISPOSE
 | |
|     const char *signature;                  // conditional on BLOCK_HAS_OBJC
 | |
|     const char *layout;                 // conditional on BLOCK_HAS_OBJC
 | |
| };
 | |
| struct block_descriptor_small {
 | |
|     unsigned long int reserved;
 | |
|     unsigned long int size;
 | |
|     const char *signature;              // conditional on BLOCK_HAS_OBJC
 | |
|     const char *layout;                 // conditional on BLOCK_HAS_OBJC
 | |
| };
 | |
| 
 | |
| struct block_layout_abi { // can't change
 | |
|   void *isa;
 | |
|   int flags;
 | |
|   int reserved; 
 | |
|   void (*invoke)(void *, ...);
 | |
|   struct block_descriptor_big *descriptor;
 | |
| };
 | |
| 
 | |
| const char *getBlockSignature(void *block) {
 | |
|    struct block_layout_abi *layout = (struct block_layout_abi *)block;
 | |
|    if ((layout->flags & BLOCK_HAS_OBJC_TYPE) != BLOCK_HAS_OBJC_TYPE) return 0;
 | |
|    if (layout->flags & BLOCK_HAS_COPY_DISPOSE) 
 | |
|       return layout->descriptor->signature;
 | |
|    else
 | |
|       return ((struct block_descriptor_small *)layout->descriptor)->signature;
 | |
| }
 | |
| 
 | |
| int usesStruct(void *block) {
 | |
|    struct block_layout_abi *layout = (struct block_layout_abi *)block;
 | |
|    int want = BLOCK_HAS_OBJC_TYPE | BLOCK_USE_STRET;
 | |
|    return (layout->flags & want) == want;
 | |
| }
 | |
|     
 | |
|    
 | |
| int main(int argc, char *argv[]) {
 | |
|    printf("desired global flags: %d\n", BLOCK_USE_STRET | BLOCK_IS_GLOBAL  | BLOCK_HAS_OBJC_TYPE);
 | |
|    printf("desired stack flags: %d\n",  BLOCK_USE_STRET | BLOCK_HAS_OBJC_TYPE);
 | |
|    
 | |
|    printf("should be non-zero: %d\n", usesStruct(global));
 | |
|    BigStruct_t x;
 | |
|    BigStruct_t (^local)(int) = ^(int param) {
 | |
|      BigStruct_t *result = (BigStruct_t *)malloc(sizeof(BigStruct_t));
 | |
|      result->array[23] = argc;
 | |
|      return *result;
 | |
|    };
 | |
|    printf("should be non-zero: %d\n", usesStruct(global));
 | |
|    printf("should be non-zero: %d\n", usesStruct(local));
 | |
|    printf("should be zero: %d\n", usesStruct(^void(int x){ }));
 | |
|    return 0;
 | |
| }
 | |
| 
 | |
| /*
 | |
| desired global flags: 1879048192
 | |
| desired stack flags: 1610612736
 | |
| should be non-zero: 1
 | |
| should be non-zero: 1
 | |
| should be non-zero: 1
 | |
| should be zero: 0
 | |
| 
 | |
| */
 |