iOS 消息转发流程

这篇博文是我的另一篇 Aspects源码剖析中的一部分，考虑到这部分内容相对独立，单独成篇以便查询。 在Objective-C中调用一个方法，其实是向一个对象发送消息。如果这个消息没有对应的实现时就会进行消息转发。转发流程图如下：

下面用代码演示一遍

• resolveInstanceMethod

当根据selector没有找到对应的method时，首先会调用这个方法，在该方法中你可以为一个类添加一个方法。并返回yes。下面的代码只是声明了runTo方法，没有实现。

//Car.h
@interface Car : NSObject
- (void)runTo:(NSString *)place;
@end

//Car.m
@implementation Car
+ (BOOL)resolveInstanceMethod:(SEL)sel{
    if (sel == @selector(runTo:)) {
        class_addMethod(self, sel, (IMP)dynamicMethodIMPRunTo, "v@:@");
        return YES;
    }
    return [super resolveInstanceMethod:sel];
}
//动态添加的@selector(runTo:) 对应的实现
static void dynamicMethodIMPRunTo(id self, SEL _cmd,id place){
    NSLog(@"dynamicMethodIMPRunTo %@",place);
}
@end
复制代码

• forwardingTargetForSelector

如果resolveInstanceMethod没有实现，返回No，或者没有动态添加方法的话，就会执行forwardingTargetForSelector。 在这里你可以返回一个能够执行这个selector的对象otherTarget，接下来消息会重新发送到这个otherTarget。

//Person.h
@interface Person : NSObject
- (void)runTo:(NSString *)place;
@end

//Person.m
@implementation Person
- (void)runTo:(NSString *)place;{
    NSLog(@"person runTo %@",place);
}
@end

//Car.h
@interface Car : NSObject
- (void)runTo:(NSString *)place;
@end

//Car.m
@implementation Car
- (id)forwardingTargetForSelector:(SEL)aSelector{
    //将消息转发给Person的实例
    if (aSelector == @selector(runTo:)){
        return [[Person alloc]init];
    }
    return [super forwardingTargetForSelector:aSelector];
}
@end
复制代码

• forwardInvocation

如果上面两种情况没有执行，就会执行通过forwardInvocation进行消息转发。

@implementation Car
- (NSMethodSignature*)methodSignatureForSelector:(SEL)aSelector{
    //判断selector是否为需要转发的，如果是则手动生成方法签名并返回。
    if (aSelector == @selector(runTo:)){
        return [NSMethodSignature signatureWithObjCTypes:"v@:@"];
    }
    return [super forwardingTargetForSelector:aSelector];
}

- (void)forwardInvocation:(NSInvocation *)anInvocation{
    //判断待处理的anInvocation是否为我们要处理的
    if (anInvocation.selector == @selector(runTo:)){
    		
    }else{
    }
}
@end
复制代码

在NSInvocation对象中保存着我们调用一个method的所有信息。可以看下其属性和方法：

• methodSignature 含有返回值类型，参数个数及每个参数的类型 等信息。
• - (void)getArgument:(void *)argumentLocation atIndex:(NSInteger)idx;获取调用method时传的参数
• - (void)setArgument:(void *)argumentLocation atIndex:(NSInteger)idx; 设置第index参数。
• - (void)invoke; 开始执行
• - (void)getReturnValue:(void *)retLoc; 获取返回值

下面的代码演示如何获取调用method时所传的各参数值

- (void)forwardInvocation:(NSInvocation *)anInvocation{
    if (anInvocation.selector == @selector(runTo:)){
        void *argBuf = NULL;
        NSUInteger numberOfArguments = anInvocation.methodSignature.numberOfArguments;
        for (NSUInteger idx = 2; idx < numberOfArguments; idx++) {
            const char *type = [anInvocation.methodSignature getArgumentTypeAtIndex:idx];
            NSUInteger argSize;
            NSGetSizeAndAlignment(type, &argSize, NULL);
            if (!(argBuf = reallocf(argBuf, argSize))) {
                NSLog(@"Failed to allocate memory for block invocation.");
                return ;
            }
            
            [anInvocation getArgument:argBuf atIndex:idx];
            //现在argBuf 中保存着第index 参数的值。 你可以使用这些值进行其他处理，例如为block中各参数赋值，并调用。
        }
    }else{
        
    }
}
复制代码

##通过手动触发消息转发（method已经实现） 前面所描述的消息转发都是在selector没有对应实现时自动进行的，我们称之为自动消息转发。现在有个需求：即使Car类实现了 runTo:，执行[objOfCar runTo:@"shangHai"]； 时也进行消息转发（手动触发），如何实现？ 实现方法如下：利用 method swizzling 将selector的实现改变为_objc_msgForward或者_objc_msgForward_stret。在调selector时就会进行消息转发。 看下面的代码:

//对 runTo: 进行消息转发
@implementation Car

//进行 method swizzling。此时调用runTo：就会进行消息转发
+ (void)load{
    SEL selector = @selector(runTo:);
    Method targetMethod = class_getInstanceMethod(self.class, @selector(selector));
    const char *typeEncoding = method_getTypeEncoding(targetMethod);
    IMP targetMethodIMP = _objc_msgForward;
    class_replaceMethod(self.class, selector, targetMethodIMP, typeEncoding);
}

- (void)runTo:(NSString *)place{
    NSLog(@"car runTo %@",place);
}

//消息转发，调用这个方法。anInvocation中保存着调用方法时传递的参数信息
- (void)forwardInvocation:(NSInvocation *)anInvocation{
    if (anInvocation.selector == @selector(runTo:)){
    
    }else{
        
    }
}
复制代码

上面提到了_objc_msgForward或者_objc_msgForward_stret， 该如何选择？首先两者都是进行消息转发的，大概是这样：如果转发的消息的返回值是struct类型，就使用_objc_msgForward_stret，否则使用_objc_msgForward。参考资料。简单引用JSPatch作者的解释

大多数CPU在执行C函数时会把前几个参数放进寄存器里，对 obj_msgSend 来说前两个参数固定是 self / _cmd，它们会放在寄存器上，在最后执行完后返回值也会保存在寄存器上，取这个寄存器的值就是返回值。普通的返回值(int/pointer)很小，放在寄存器上没问题，但有些 struct 是很大的，寄存器放不下，所以要用另一种方式，在一开始申请一段内存，把指针保存在寄存器上，返回值往这个指针指向的内存写数据，所以寄存器要腾出一个位置放这个指针，self / _cmd 在寄存器的位置就变了。objc_msgSend 不知道 self / _cmd 的位置变了，所以要用另一个方法 objc_msgSend_stret 代替。原理大概就是这样。在 NSMethodSignature 的 debugDescription 上打出了是否 special struct，只能通过这字符串判断。所以最终的处理是，在非 arm64 下，是 special struct 就走 _objc_msgForward_stret，否则走 _objc_msgForward。

根据selector返回值类型获取_objc_msgForward或者_objc_msgForward_stret 的代码如下：

//代码来自Aspect
static IMP aspect_getMsgForwardIMP(NSObject *self, SEL selector) {
    IMP msgForwardIMP = _objc_msgForward;
#if !defined(__arm64__)
    Method method = class_getInstanceMethod(self.class, selector);
    const char *encoding = method_getTypeEncoding(method);
    BOOL methodReturnsStructValue = encoding[0] == _C_STRUCT_B;
    if (methodReturnsStructValue) {
        @try {
            NSUInteger valueSize = 0;
            NSGetSizeAndAlignment(encoding, &valueSize, NULL);

            if (valueSize == 1 || valueSize == 2 || valueSize == 4 || valueSize == 8) {
                methodReturnsStructValue = NO;
            }
        } @catch (__unused NSException *e) {}
    }
    if (methodReturnsStructValue) {
        msgForwardIMP = (IMP)_objc_msgForward_stret;
    }
#endif
    return msgForwardIMP;
}
复制代码

你可以通过iOS Aspects源码剖析 来进行一步了解_objc_msgForward，_objc_msgForward_stret，method swizzling如何配合使用完成消息转发和对消息的统一处理。