日常知识点收集

OC的hook_交换方法

2018-05-28  本文已影响16人  melody5

首先来了解几个概念

  1. SEL 方法编号
  2. IMP 方法实现(本质上是函数指针)

在OC中调用方法其实都是消息转发的过程,给某个对象,发送方法的编号消息。
通过SEL 可以找到对应的IMP(方法实现)
SEL和IMP是一一对应的关系,就像一本书的目录(SEL)和页面(IMP)

  1. HOOK:钩子!! -- 修改原来方法的调用顺序

编程思想:面向切面编程,核心技术就是用的 HOOK思想

举例

接下来看看怎么用它,比如我们创建一个URL的时候,如果URLStr里边有中文的话,那么request会返回一个nil,并且我们不会受到任何报错,就像这样:

    NSURL *url = [NSURL URLWithString:@"https://www.sina.com空了"];
    NSURLRequest *request = [NSURLRequest requestWithURL:url];
    NSLog(@"%@",request);
<NSURLRequest: 0x100aba8e0> { URL: (null) }

对于这种情况,我们来建一个NSURL的分类,在里边处理一下nill的情况,当然这个例子可以直接重写URLWithString方法来处理,但是我们既然是说hook,还是要用hook的方法来处理。

交换两个方法的对应关系

首先创建+(instancetype)hook_URLWithString:(NSString *)URLString;方法,在这里处理nil,但是如果项目大,这样做的话,需要修改的地方也太多了,如果调用URLWithString方法的时候系统就默认先来调用我们自定义的hook_URLWithString,这样就只需要在分类里实现hook_URLWithString就可以了,所以我们要交换他们的SEL和IMP的对应关系,来实现方法实现的交换,就像这样:


image.png

那怎么交换呢?那就用到runtime的了,其中有一个方法

method_exchangeImplementations(<#Method  _Nonnull m1#>, <#Method  _Nonnull m2#>)

分别传入两个方法

    // class_getClassMethod     获取类方法
    // class_getInstanceMethod  获取对象方法
    Method URLWithString = class_getClassMethod(self, @selector(URLWithString:));
    Method hook_URLWithString = class_getClassMethod(self, @selector(hook_URLWithString:));
    // 交换方法
    method_exchangeImplementations(URLWithString, hook_URLWithString);

调用时机

我们知道了怎么交换方法,但是在哪里调用呢?首先得来了解下,app启动的过程:

  1. 项目装在手机上,都是二进制文件(match_O),在硬盘上面。
  2. 点进启动app的时候,会将二进制文件加载到内存中,等待CPU调用(装载过程)。

装载的时候就会调用+(void)load,之后才会进入main函数,所以load是在项目启动的最初就调用了,在这里做交换再合适不过了

#import "NSURL+hook.h"
#import <objc/runtime.h>

@implementation NSURL (hook)

+(void)load {
    // 下勾子
    /*
     SEL -- IMP(才是指针)
     */
    
    // 防止被别人手动调用,再次调换回来,所以要保证他只执行一次
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        
        // class_getClassMethod     获取类方法
        // class_getInstanceMethod  获取对象方法
        Method URLWithString = class_getClassMethod(self, @selector(URLWithString:));
        Method hook_URLWithString = class_getClassMethod(self, @selector(hook_URLWithString:));
        // 交换方法
        method_exchangeImplementations(URLWithString, hook_URLWithString);
        
    });
    
}

+(instancetype)hook_URLWithString:(NSString *)URLString {
    
    // 两个方法实现已经调换了,所以这里要调用hook_URLWithString,如果调用URLWithString的会造成递归
    NSURL *url = [NSURL hook_URLWithString:URLString];
    if (!url) {
        NSLog(@"url为nil");
    }
    
    return url;
}

@end
2018-05-28 17:52:42.601303+0800 hook_方法交换[331:70399] url为nil
2018-05-28 17:52:42.601609+0800 hook_方法交换[331:70399] <NSURLRequest: 0x10de9e0e0> { URL: (null) }

完美交换!!!

上一篇 下一篇

猜你喜欢

热点阅读