NSInvocation调用block

起

一日在开发之中，遇到这样一个问题，在某些场合，需要用NSInvocation来调用Block，而Block签名并不是固定，即，Block参数类型个数可以不同。

问题

回忆NSInvocation 一般用法

自然想到了NSInvocation，譬如如下代码：

NSString* string = @"Hello";NSString* anotherString = [string stringByAppendingString:@" World!"];

写成Invocataion大致是这样的：

NSString* string = @"Hello";NSString* anotherString;NSString* stringToAppend = @" World!";NSInvocation* inv = [NSInvocation invocationWithMethodSignature:[NSString instanceMethodSignatureForSelector:@selector(stringByAppendingString:)]];inv.target = string;[inv setArgument:&stringToAppend atIndex:2];[inv invoke];[inv getReturnValue:&anotherString];

具体就不详细介绍了，文档里讲得很详细。请移步Apple Doc

MethodSignature

一个问题是如何Block获得MethodSignature。Block没有selector，但发现NSMethodSignature有这样一个方法-[NSMethodSignature signatureWithObjCTypes:]，那问题转化成如何从Block获得编码的Signature。

一搜索。发现Clang官方文档和stackoverflow都有说这个问题。(Clang官方文档真是个宝库啊)。

按Clang的文档，Block定义如下：

struct Block_literal_1 {    void *isa; // initialized to &_NSConcreteStackBlock or &_NSConcreteGlobalBlock    int flags;    int reserved;    void (*invoke)(void *, ...);    struct Block_descriptor_1 {    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)    void* rest[1];    } *descriptor;    // imported variables};

中间注释部分是对做了些小改造，因为对于可以copy的Block，上述两个函数指针才存在。（另外，发现其实Block还能通过block->invoke(...)来调用，先按下不表）。

static const char *__BlockSignature__(id blockObj){    struct Block_literal_1 *block = (__bridge void *)blockObj;    struct Block_descriptor_1 *descriptor = block->descriptor;    int copyDisposeFlag = 1 << 25;    int signatureFlag = 1 << 30;    assert(block->flags & signatureFlag);    int offset = 0;    if(block->flags & copyDisposeFlag)        offset += 2;    return (const char*)(descriptor->rest[offset]);}NSInvocation * invocation = [NSInvocation invocationWithMethodSignature:[NSMethodSignature signatureWithObjCTypes:__BlockSignature__(block)]];

最重要的一个问题解决了。之后就是对invocation调用setArgument，进行参数传递。（先简化一下问题，参数都是NSObject，放在NSArray里,关于参数获取后面还有坑，以后再写）

    for(NSUInteger i = 0; i < args.count ; ++i){        id object = args[i];        [invocation setArgument:&object atIndex:i + 2];    }    [invocation invokeWithTarget:block];

调用，Crash！越界了

Reason: -[NSInvocation setArgument:atIndex:]: index (5) out of bounds [-1, 4]

文档是这样描述setArgument的

Indices 0 and 1 indicate the hidden arguments self and _cmd, respectively; these values can be retrieved directly with the target and selector methods. Use indices 2 and greater for the arguments normally passed in a message.

其实上述代码问题很多，刚才有点撞大运编程

1. selector(即_cmd)哪里去了？并没有传递给NSInvocation

2. 为什么越界，按照文档说法应该从2开始。

3. 为什么从-1开始

文档说的言之凿凿，第一个参数传self，第二个是selector(即_cmd)，但block调用并没有selector，参数个数其实可以从MethodSignature获取：invocation.methodSignature.numberOfArguments

所以这就是会越界的原因，正确的做法是从1开始：

[invocation setArgument:&object atIndex:i + 1]

一调试，果然。

另外从-1开始原因是-1的位置是存储return result，当然这个结论我查了文档并没有找到，也是试出来的。囧。

源码及其他

源码我放在了github，戳这里

用法也很简单：

NSInvocation* inv = [NSInvocation invocationWithBlock:block];

后续会增加一些接口如：
+ (instancetype) invocationWithBlockAndArguments:(id) block ,...;

更新

恩 已经增加了。

增加的接口用法：
对于

void (^myBlock)(id, NSArray*,double, int**) = ^(id obj1, NSArray* array, double dNum,int** pi) {  NSLog(@"%@",@"Hey!");};

int* i = NULL;NSInvocation* inv = [NSInvocation invocationWithBlockAndArguments:myBlock,[NSObject new],@[@1,@2,@3],1.23,&i];

参数支持id,所有简单值类型,IMP,SEL,Class,Block,指针, 但Struct,Union,C-style Array 不支持，比较预想的tricky，研究中。

原作写于segmentfault 链接