获取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的调用。