这篇博文是我的另一篇 中的一部分,考虑到这部分内容相对独立,单独成篇以便查询。 在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
的代码如下:
//代码来自Aspectstatic 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;}复制代码
你可以通过 来进行一步了解_objc_msgForward
,_objc_msgForward_stret
,method swizzling
如何配合使用完成消息转发和对消息的统一处理。