Hook之 Method Swizzling

Method Swizzling利用了Objective-C的runtime特性,使得我们能动态替换方法的实现,实现hook。

runtime

从敲下NSlog(@"hello world!")开始,就不断看到runtime这个词。对于之前没接触过动态语言的学习者来说,runtime听起来陌生又晦涩。

Objective-C不同于C语言,它是一门动态语言,这就意味着它将C语言等静态语言在编译和链接时所做的一些工作留到了运行时再处理。比如,在C语言中,编译时就可以确定真正调用哪些函数;而Objective-C并不能在编译时确定,只有在运行时才会由名称找到对应方法。

因为这种特性,Objective-C需要一种机制在运行时来处理编译后的代码,这种机制就是runtime。Runtime从实质来讲,就是一套用C语言和汇编写成的底层API,用于处理编译后的代码。

method

methodStruct
method_name 表示的是方法的名称,用于唯一标识某个方法;
method_types 表示的是方法的返回值和参数类型;
method_imp 是一个函数指针,指向方法的实现。
方法的名称 name 和方法的实现 imp 是一一对应的。而且可以发现方法的名称和参数类型是分离的。
在我们调用一个方法时

[someObject messageName:paremeter]

在runtime底层会转换成

Objc_msgSend(someObject, @selector(messageName), paremeter)

调用方法其实是向一个对象发送消息,而查找方法实现的唯一依据是selector后接的方法名称。所以,利用这一特性,我们就可以实现在运行时更改selector对应的方法实现,也就实现了HOOK。

method swizzling

原理

methodIMP
changeIMP

图源:念茜

每个类都维护着一个方法列表,里面存放着selector的名字与方法实现间的映射关系。Method Swizzling就是改换了这种对应关系。

可以调用三种方法来实现method swizzling:
method_exchangeImplementations 交换2个方法中的IMP(IMPlication);
class_replaceMethod 修改类;
利用 method_setImplementation直接设置某个方法的IMP。
这些方法的声明都写在runtime.h中。

1.method_exchangeImplementations

1

method_exchangeImplementations通过交换两个方法的IMP实现hook的,通过
method_exchangeImplementations(A_Method, B_Method);
可以实现如下图的交换。
1

2.method_setImplementation

1

method_setImplementation直接设置更改某个方法的IMP。
如下图,假设有方法A,我们想要把它的实现替换为方法B的实现,但又不想改变方法B的实现,就可以使用method_setImplementation,直接把A的IMP设置为B的IMP。

1

3.class_replaceMethod

1

class_replaceMethod其实相当于class_addMethodmethod_setImplementation的一个结合。

当调用
class_replaceMethod(cls, name, imp, types)
时,将把属于cls类的name方法实现指向imp
如果cls类原本没有name方法,就相当于调用class_addMethodcls类增添了一个方法及其实现,这个方法的返回类型由types给定。
如果cls类原本就有name方法,就相当于调用method_setImplementation更改了name方法的实现为imp,返回类型由替换的实现imp决定,给定types将被忽略。

参考

Objective-C的hook方案(一): Method Swizzling
Hook 原理之 Method Swizzling