首先来了解几个概念
- SEL 方法编号
- IMP 方法实现(本质上是函数指针)
在OC中调用方法其实都是消息转发的过程,给某个对象,发送方法的编号消息。
通过SEL 可以找到对应的IMP(方法实现)
SEL和IMP是一一对应的关系,就像一本书的目录(SEL)和页面(IMP)
- 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的对应关系,来实现方法实现的交换,就像这样:
那怎么交换呢?那就用到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启动的过程:
- 项目装在手机上,都是二进制文件(match_O),在硬盘上面。
- 点进启动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) }
完美交换!!!