ios runtime IMP指针 消息转发机制Demo

本文代码是根据消息转发机制来写的, 有不妥之处, 请大神指正

githubDemo地址欢迎大家下载

1. UIViewController (ViewDidLoadName)文件 UIViewController的category

在实现viewDidLoad系统方法的前提下 添加自定义的方法

2. Person类有一个run的方法(没有实现),这里展示了OC中的消息转发机制, 使其不崩溃并实现方法,或者转到Car的run方法来实现

直接上代码(注释很全, 简单易懂)

ViewController.m文件

<span style="font-size:18px;">//  ViewController.m
//  MethodSwizzlingIMPdemo
//
//  Created by 帝炎魔 on 16/5/12.
//  Copyright © 2016年 帝炎魔. All rights reserved.
//

#import "ViewController.h"
#import "UIViewController+ViewDidLoadName.h"
#import "Person.h"

@interface ViewController ()

@end

@implementation ViewController

- (void)viewDidLoad {
    [super viewDidLoad];
    
    
    // 创建Person对象 执行run方法, 但是Person类中并没有实现run方法
    // 我们利用消息转发机制防止其崩溃,或者用其他方法来代替[per run]的方法
    Person *per = [[Person alloc] init];
    [per run];
    
    // Do any additional setup after loading the view, typically from a nib.
}



- (void)didReceiveMemoryWarning {
    [super didReceiveMemoryWarning];
    // Dispose of any resources that can be recreated.
}

@end</span>

UIViewController (ViewDidLoadName).h文件

<span style="font-size:18px;">//  UIViewController+ViewDidLoadName.h
//  MethodSwizzlingIMPdemo
//
//  Created by 帝炎魔 on 16/5/12.
//  Copyright © 2016年 帝炎魔. All rights reserved.
//

#import <UIKit/UIKit.h>

@interface UIViewController (ViewDidLoadName)

@end</span>

UIViewController (ViewDidLoadName).m文件

<span style="font-size:18px;">//  UIViewController+ViewDidLoadName.m
//  MethodSwizzlingIMPdemo
//
//  Created by 帝炎魔 on 16/5/12.
//  Copyright © 2016年 帝炎魔. All rights reserved.
//

#import "UIViewController+ViewDidLoadName.h"
#import <objc/runtime.h>

// 有返回值的IMP
typedef id (* _IMP) (id, SEL, ...);
// 没有返回值的IMP(定义为VIMP)
typedef void (* _VIMP) (id, SEL, ...);

@implementation UIViewController (ViewDidLoadName)


+(void)load
{
    // 保证交换方法只执行一次
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        
        // 获取原始方法
        Method viewDidLoad = class_getInstanceMethod(self, @selector(viewDidLoad));
        // 获取原始方法的实现指针(IMP)
        _VIMP viewDidLoad_IMP = (_VIMP)method_getImplementation(viewDidLoad);
        
        // 重新设置方法的实现
        method_setImplementation(viewDidLoad, imp_implementationWithBlock(^(id target, SEL action) {
            // 调用系统的原生方法
            viewDidLoad_IMP(target, @selector(viewDidLoad));
            // 新增的功能代码
            NSLog(@"%@ did load", target);
        }));
        
    });
}

//+ (void)load
//{
//    // 保证交换方法只执行一次
//    static dispatch_once_t onceToken;
//    dispatch_once(&onceToken, ^{
//        // 获取到这个类的viewDidLoad方法, 它的类型是一个objc_method结构体的指针
//        Method viewDidLoad = class_getInstanceMethod(self, @selector(viewDidLoad));
//        
//        // 获取到自己刚刚创建的一个方法
//        Method myViewDidLoad = class_getInstanceMethod(self, @selector(myViewDidLoad));
//        
//        // 交换两个方法的实现
//        method_exchangeImplementations(viewDidLoad, myViewDidLoad);
//        
//      
//        
//    });
//}
//
//- (void)myViewDidLoad
//{
//    // 调用系统的方法
//    [self myViewDidLoad];
//    NSLog(@"%@ did load", self);
//}

@end</span>

Person.h 文件

<span style="font-size:18px;">//  Person.h
//  MethodSwizzlingIMPdemo
//
//  Created by 帝炎魔 on 16/5/12.
//  Copyright © 2016年 帝炎魔. All rights reserved.
//

#import <Foundation/Foundation.h>

@interface Person : NSObject

- (void)run;

@end</span>

Person.m文件

<span style="font-size:18px;">//  Person.m
//  MethodSwizzlingIMPdemo
//
//  Created by 帝炎魔 on 16/5/12.
//  Copyright © 2016年 帝炎魔. All rights reserved.
//

#import "Person.h"
#import <objc/runtime.h>
#import "Car.h"

@implementation Person

/**
 *  首先，该方法在调用时，系统会查看这个对象能否接收这个消息（查看这个类有没有这个方法，或者有没有实现这个方法。），如果不能并且只在不能的情况下，就会调用下面这几个方法，给你“补救”的机会，你可以先理解为几套防止程序crash的备选方案，我们就是利用这几个方案进行消息转发，注意一点，前一套方案实现后一套方法就不会执行。如果这几套方案你都没有做处理，那么程序就会报错crash。
    
    方案一：
 
    + (BOOL)resolveInstanceMethod:(SEL)sel
    + (BOOL)resolveClassMethod:(SEL)sel
 
    方案二：
 
    - (id)forwardingTargetForSelector:(SEL)aSelector
 
    方案三：
 
    - (NSMethodSignature *)methodSignatureForSelector:(SEL)aSelector;
    - (void)forwardInvocation:(NSInvocation *)anInvocation;
 
*/

void run (id self, SEL _cmd)
{
    // 程序会走我们C语言的部分
    NSLog(@"%@ %s", self, sel_getName(_cmd));
}


/**
 *   方案一
 *
 *   为Person类动态增加了run方法的实现
    由于没有实现run对应的方法, 那么系统会调用resolveInstanceMethod让你去做一些其他操作
 */

+ (BOOL)resolveInstanceMethod:(SEL)sel
{
//    if(sel == @selector(run)) {
//        class_addMethod([self class], sel, (IMP)run, "v@:");
//        return YES;
//    }
    return [super respondsToSelector:sel];
}

/** 方案二
 *  现在不对方案一做任何的处理, 直接调用父类的方法
    系统会走到forwardingTargetForSelector方法
 */

//- (id)forwardingTargetForSelector:(SEL)aSelector
//{
//    return [[Car alloc] init];
//}


/**
 *   不实现forwardingTargetForSelector,
     系统就会调用方案三的两个方法
    methodSignatureForSelector 和 forwardInvocation
 */

/**
 *  方案三
    开头我们要找的错误unrecognized selector sent to instance原因，原来就是因为methodSignatureForSelector这个方法中，由于没有找到run对应的实现方法，所以返回了一个空的方法签名，最终导致程序报错崩溃。
 
     所以我们需要做的是自己新建方法签名，再在forwardInvocation中用你要转发的那个对象调用这个对应的签名，这样也实现了消息转发。
  */

/**
 *  methodSignatureForSelector
 *  用来生成方法签名, 这个签名就是给forwardInvocation中参数NSInvocation调用的
 *
 */
- (NSMethodSignature *)methodSignatureForSelector:(SEL)aSelector
{
    NSString *sel = NSStringFromSelector(aSelector);
    // 判断你要转发的SEL
    if ([sel isEqualToString:@"run"]) {
        // 为你的转发方法手动生成签名
        return [NSMethodSignature signatureWithObjCTypes:"v@:"];
        
    }
    return [super methodSignatureForSelector:aSelector];
}
/**
 *  关于生成签名类型"v@:"解释一下, 每个方法会默认隐藏两个参数, self, _cmd
    self 代表方法调用者, _cmd 代表这个方法SEL, 签名类型就是用来描述这个方法的返回值, 参数的, 
    v代表返回值为void, @表示self, :表示_cmd
 */
-(void)forwardInvocation:(NSInvocation *)anInvocation
{
    SEL selector = [anInvocation selector];
    // 新建需要转发消息的对象
    Car *car = [[Car alloc] init];
    if ([car respondsToSelector:selector]) {
        // 唤醒这个方法
        [anInvocation invokeWithTarget:car];
    }
}

@end</span>

Car .m文件

<span style="font-size:18px;">//  Car.m
//  MethodSwizzlingIMPdemo
//
//  Created by 帝炎魔 on 16/5/12.
//  Copyright © 2016年 帝炎魔. All rights reserved.
//

#import "Car.h"

@implementation Car


- (void)run
{
     NSLog(@"%@ %s", self, sel_getName(_cmd));
}
@end</span>