OC的hook_交换方法

作者: melody5 | 来源:发表于2018-05-28 18:00 被阅读16次

    首先来了解几个概念

    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) }
    

    完美交换!!!

    相关文章

      网友评论

        本文标题:OC的hook_交换方法

        本文链接:https://www.haomeiwen.com/subject/czpgjftx.html