rumtime Method Swizzling Hook的三种方式

runtime 大家都很熟悉，iOS的动态特性是靠runtime来维护的。那么通过runtime我们可以做什么呢？

AOP编程

AOP即面向切片编程，很著名的Aspects 就是基于runtime特性

防止Hook

一般动态注入的方式有framework 和 Dylib， 那么如何动态注入呢？
有一些强调安全的app，会做Hook防范，这里也是基于runtime的特性来的。
当然还有著名的Facebook 的fishhook 是基于iOS编译过程的来做的系统C函数的钩子，这个在执行时间上就要比runtime 更早的防范了。

---

本篇文章主要是说一下日常开发中hook的几种方式。了解了大概的注入方式，也即可根据不同情况进行对应的防护

首先我们先来说一下分类的方式

一般系统自带类

平时有些需求需要我们对已经写好的代码进行适当的修改，来完善业务需求如：解析url 但是我们不能每个地方都去更改一遍，这时候就适合做分类
demo中做了NSURL+correctUrl.h分类来解析中文。代码如下：

+(void)load{
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        
        Method URLWithStr = class_getClassMethod(self, @selector(URLWithString:));
        
        Method HKURL = class_getClassMethod(self, @selector(HKURLWithStr:));
        
        //交换
        method_exchangeImplementations(URLWithStr, HKURL);
    });
}
+(instancetype)HKURLWithStr:(NSString *)str{
    //调用系统原来的方法
    NSURL * url = [NSURL HKURLWithStr:str];
    if (url == nil) {
        str = [str stringByAddingPercentEncodingWithAllowedCharacters:[NSCharacterSet URLQueryAllowedCharacterSet]];
    }
    url = [NSURL HKURLWithStr:str];
    
    return url;
}

因为是系统类分类所以这样既可完成方法的交换，最后记得在新方法里面一定要执行一下系统方法，否则会破坏系统方法功能。

类簇

对于NSArray、NSMutableArray、NSDictionary、NSMutableDictionary等类进行swizzleMethod，不能直接使用类似NSClassFromString(@"NSMutableArray") 因为这些类簇类，其实是一种抽象工厂的设计模式。抽象工厂内部有很多其它继承自当前类的子类，抽象工厂类会根据不同情况，创建不同的抽象对象来进行使用。
class 的获取一般使用 [self class] 或者 类似objc_getClass("__NSArrayM")取到真正的类。如下代码：

+(void)load{
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        id obj = [[self alloc]init];
        [obj swizzleMethod:@selector(addObject:) withMethod:@selector(LFAddObject:)];
    });
    
}
-(void)LFAddObject:(id)obj{
    if (obj) {
        [self LFAddObject:obj];
    }else{
        NSLog(@"添加元素为nil");
    }
    
}
- (void)swizzleMethod:(SEL)origSelector withMethod:(SEL)newSelector
{
   //这里打印cls 可以看见是__NSArrayM  对于NSArray、NSMutableArray、NSDictionary、NSMutableDictionary等类进行swizzleMethod，不能直接使用类似NSClassFromString(@"NSMutableArray") 因为这些类簇类，其实是一种抽象工厂的设计模式。抽象工厂内部有很多其它继承自当前类的子类，抽象工厂类会根据不同情况，创建不同的抽象对象来进行使用
    //class 的获取一般使用 [self class] 或者 类似objc_getClass("__NSArrayM")取到真正的类
    Class cls = [self class];
//    Class cls = objc_getClass("__NSArrayM");
    Method originalMethod = class_getInstanceMethod(cls, origSelector);
    Method swizzledMethod = class_getInstanceMethod(cls, newSelector);
    
    BOOL didAddMethod = class_addMethod(cls,
                                        origSelector,
                                        method_getImplementation(swizzledMethod),
                                        method_getTypeEncoding(swizzledMethod));
    if (didAddMethod) {
        class_replaceMethod(cls,
                            newSelector,
                            method_getImplementation(originalMethod),
                            method_getTypeEncoding(originalMethod));
    } else {
        method_exchangeImplementations(originalMethod, swizzledMethod);
    }
}

上述swizzleMethod方法是目前方法交换比较普遍的写法，手懒的同学可以直接拷贝去用哈

自定义

replace方式

有时候我们要单独封装一个injectionCode ，这个时候就需要我们注意：上面说过，在新方法里面一定要执行以下原方法，否则会破坏原方法的功能
但是在封装的类中我们是没有原函数的，所以要先保存一下原函数。

IMP (*old_tz)(id self,SEL _cmd);
Method ViewController_tz = class_getInstanceMethod(objc_getClass("ViewController"), @selector(tz));
    old_tz = method_getImplementation(ViewController_tz);

将原方法保存在old_tz中，此时只需要将新函数替换掉原函数即可

 class_replaceMethod(objc_getClass("ViewController"), @selector(tz), my_tz, "v@:");

class_replaceMethod 会在没有函数的时候自己创建一个新的函数
执行的时候如下

void my_tz(id self,SEL _cmd){
    NSLog(@"截获了跳转事件");
    old_tz(self,_cmd);
}

除了class_replaceMethod 还有一种方式即get/set方式

get/set方式

直接上代码，这个也比较简单。

+(void)getAndSetTohook{
    Method ViewController_tz = class_getInstanceMethod(objc_getClass("ViewController"), @selector(tz));
    old_tz = method_getImplementation(ViewController_tz);
    method_setImplementation(ViewController_tz, (IMP)my_tz);
}

将原函数的指针，直接指向新函数

以上即为runtime Method Swizzling 的几种常见方式，做防护的时候可从这几个函数进行考虑。
完整demo