在OC中：
API： class_addMethod往一个Class里添加method
API： class_getInstanceMethod或class_getClassMethod可以判断某个SEL是否存在于Class
API： method_exchangeImplementations 交换方法。
最近工作上做了一件事，简单点说就是需要把一些特定Class里的方法func，替换成Hook_func，当Hook_func执行完之后，再执行func。于是很简单地想到了往Class添加一个Hook_func，然后再交换func与Hook_func，就能到达目的。
但是，在实现后，却出现了死循环，纠其原因，因为部分Class间存在着继承关系，没有正确地将Method添加到正确的Class中导致。
当时为了解决这个问题，重新去理清楚了一下Class中 SEL 与 Method的关系。SEL是一个选择器，相当于指向一个Method的指针，将SEL指向不同的Method，它就会有不一样的特性，method_exchangeImplementations也就是交换SEL指向的Method值来实现方法交换的。
在调用class_getInstanceMethod时，是会检查superClass的。
如下图，当ClassB继承ClassA时, ClassB中没有SEL func，而ClassA中有SEL func。
这时调用class_getInstanceMethod(ClassB, @selector(func))，是能拿到SEL func的Method的，然后再将ClassB中添加SEL Hook_func后，变成下图。
对ClassB调用method_exchangeImplementations后，得到下图。
这里可以看到，其实ClassA中的SEL：func已经是指向Method:Hook_func。在对ClassB的实例调用SEL:func时，能达到之前预订的效果，即先执行Hook_func后再执行func。
但在这个时候，如果再对ClassA做类似ClassB的处理，将得到下图：
这时SEL:func和SEL:Hook_func都指向了Method:Hook_func，于是便出现了死循环的问题。
最终的解决方法从上图已经可以很明显的看出来了，即在对ClassB做处理的时候，添加的SEL:Hook_func不应该添加到ClassB上，而应该添加到ClassA上，如下图，则问题得已解决。
执行下面的这个函数，找到正确的Class，然后再往Class里添加方法和交换方法，问题就解决了：
Class GetSelectorInsClass(Class hclass, SEL sel, Method sel_method) {
Class super_class = [hclass superclass]; if (!super_class) { return hclass;
}
Method method = class_getInstanceMethod(super_class, sel); if (method != sel_method) { return hclass;
} return GetSelectorInsClass(super_class, sel, sel_method);
}
延伸阅读
- Objective-c runtime方法替换引发的死循环 2017-05-11 16:46:38.99
- Objective-C 实用关键字详解1「面试、工作」看我就 ?? 了 ^_^. 2017-05-06 15:58:17.663
- Swift、Objective-C 单例模式 (Singleton) 2017-05-04 17:11:27.643
- Objective-C Runtime使用之全局字体替换为第三方字体(iOS) 2016-11-22 11:07:42.61
- Objective-C编程 - 关于Block的几点 2016-11-19 09:23:35.623
- Objective-C中block的底层原理 2016-11-18 11:13:38.793
- Objective-C 装饰模式--简单介绍和使用 2016-11-07 16:02:38.13
Objective-c runtime方法替换引发的死循环