獲取block的方法簽名，從而獲取block的參數和返回值類型

首先我們要了解block的真實結構如下：參考：這里

struct ZBBlockLiteral {
    void *isa; // initialized to &_NSConcreteStackBlock or &_NSConcreteGlobalBlock
    int flags;
    int reserved;
    void (*invoke)(void *, ...);
    struct block_descriptor {
        unsigned long int reserved;    // NULL
        unsigned long int size;         // sizeof(struct Block_literal_1)
        // optional helper functions
        void (*copy_helper)(void *dst, void *src);     // IFF (1<<25)
        void (*dispose_helper)(void *src);             // IFF (1<<25)
        // required ABI.2010.3.16
        const char *signature;                         // IFF (1<<30)
    } *descriptor;
    // imported variables
};

// flags enum
enum {
    ZBBlockDescriptionFlagsHasCopyDispose = (1 << 25),
    ZBBlockDescriptionFlagsHasCtor = (1 << 26), // helpers have C++ code
    ZBBlockDescriptionFlagsIsGlobal = (1 << 28),
    ZBBlockDescriptionFlagsHasStret = (1 << 29), // IFF BLOCK_HAS_SIGNATURE
    ZBBlockDescriptionFlagsHasSignature = (1 << 30)
};
typedef int ZBBlockDescriptionFlags;

block其實就是個struct類型，其中的descriptor中的signature正是我們想要的block的方法簽名。那么我們只需要獲取到signature，然后轉換成NSMethodSignature，即可輕易的獲取到我們想要的block的參數和返回值類型，並且通過NSInvocation執行block。

以下就是如何獲取signature的方法：

+ (NSMethodSignature *)getSignatureWithBlock:(id)block{
    struct ZBBlockLiteral *blockRef = (__bridge struct ZBBlockLiteral *)block;
    ZBBlockDescriptionFlags _flags = blockRef->flags;
    if (_flags & ZBBlockDescriptionFlagsHasSignature) {
        void *signatureLocation = blockRef->descriptor;
        signatureLocation += sizeof(unsigned long int); 
        signatureLocation += sizeof(unsigned long int);
        
        if (_flags & ZBBlockDescriptionFlagsHasCopyDispose) {
            signatureLocation += sizeof(void(*)(void *dst, void *src));
            signatureLocation += sizeof(void (*)(void *src));
        }
        
        const char *signature = (*(const char **)signatureLocation);
        return [NSMethodSignature signatureWithObjCTypes:signature];
    }
    return nil;
}

其中 signatureLocation += sizeof(unsigned long int);的加號，就是c/c++中的指針移動，通過指針移動，我們就可以找到signature的指針。然后通過[NSMethodSignature signatureWithObjCTypes:signature]構建方法簽名的objc對象。

下面是測試代碼：

BOOL(^testBlock)(NSString *, NSString *) = ^(NSString *str1, NSString *str2) {
        NSLog(@"-----:%@-------:%@", str1, str2);
        return  [str1 isEqualToString:str2];
    };
    
    NSMethodSignature *sign = [MyBlockDesc getSignatureWithBlock:testBlock];
    
    if (sign) {
        
        NSLog(@"----參數個數：%@", @(sign.numberOfArguments));
        NSLog(@"----返回值類型：%@", [NSString stringWithUTF8String:sign.methodReturnType]);
        for (int i=0; i<sign.numberOfArguments; i++) {
            NSLog(@"----第%@個參數：%@", @(i), [NSString stringWithUTF8String:[sign getArgumentTypeAtIndex:i]]);
        }
        NSString *str1 = @"111";
        NSString *str2 = @"111";
        NSInvocation *inv = [NSInvocation invocationWithMethodSignature:sign];
        [inv setArgument:&str1 atIndex:1];
        [inv setArgument:&str2 atIndex:2];
        inv.target = testBlock;
        [inv invoke];
        BOOL res;
        [inv getReturnValue:&res];
        
        NSLog(@"----返回結果：%@", @(res));
        
    }else {
        NSLog(@"------參數簽名為空");
    }

打印日志如下：

2020-01-09 09:32:04.600723+0800 Test[2503:27368] ----參數個數：3
2020-01-09 09:32:04.600908+0800 Test[2503:27368] ----返回值類型：B
2020-01-09 09:32:04.601046+0800 Test[2503:27368] ----第0個參數：@?
2020-01-09 09:32:04.601166+0800 Test[2503:27368] ----第1個參數：@"NSString"
2020-01-09 09:32:04.601277+0800 Test[2503:27368] ----第2個參數：@"NSString"
2020-01-09 09:32:04.601380+0800 Test[2503:27368] -----:111-------:111
2020-01-09 09:32:04.601493+0800 Test[2503:27368] ----返回結果：1

其中第0個參數就是block對象自身。因此我們既可以通過如上手段，來擴展實現一個線程安全的block調用，結合可變參數的使用，就可以實現任意參數的block的調用。