iOS利用runtime追踪对象的每一个方法

iOS利用runtime追踪对象的每一个方法

我们会用到runtime替换方法来监听某个方法的调用。例如，项目中每个Controller都直接继承了UIViewController，但是现在想监听每个Controller的viewDidAppear 和 viewDidDisappear，用法如下:

void qhd_exchangeInstanceMethod(Class class, SEL originalSelector, SEL newSelector) {
Method originalMethod = class_getInstanceMethod(class, originalSelector);
Method newMethod = class_getInstanceMethod(class, newSelector);
if(class_addMethod(class, originalSelector, method_getImplementation(newMethod), method_getTypeEncoding(newMethod))) {
class_replaceMethod(class, newSelector, method_getImplementation(originalMethod), method_getTypeEncoding(originalMethod));
} else {
method_exchangeImplementations(originalMethod, newMethod);
}
}

@implementation UIViewController (Test)

+ (void)load {
qhd_exchangeInstanceMethod([self class], @selector(viewDidAppear:), @selector(qhd_viewDidAppear:));
qhd_exchangeInstanceMethod([self class], @selector(viewDidDisappear:), @selector(qhd_viewDidDisappear:));
}

- (void)qhd_viewDidAppear:(BOOL)animated {
//[MobClick beginLogPageView:self.title];
[self qhd_viewDidAppear:animated];
}

- (void)qhd_viewDidDisappear:(BOOL)animated {
//[MobClick endLogPageView:self.title];
[self qhd_viewDidDisappear:animated];
}

@end

最近产生了一个的想法：替换一个类的所有方法，每一个方法都打印一个log，看看调用顺序是怎样的，例如我想知道UIViewController在运行时到底都调用了哪些方法，包括私有方法。

思路是这样的:

1.通过class_copyMethodList得出一个类的所有方法。

2.通过method_getTypeEncoding和method_copyReturnType得出方法的参数类型和返回值。

3.创建出SEL和IMP，通过class_addMethod动态添加新方法。

4.通过交换的思想，在新方法里通过NSInvocation来调用原方法。

难点在于，新方法里面怎么把方法的“实现”(即IMP)绑定上，并且在“实现”里调用原方法。在runtime的头文件中Method的结构：

typedef struct objc_method *Method;

struct objc_method {
SEL method_name
char *method_types
IMP method_imp
}

可以看到Method包含了是三个元素：一个SEL，一个char *，一个IMP。

SEL是方法名，char *是方法的类型，IMP就是实现的地址。

具体代码查看github：https://github.com/qhd/ANYMethodLog.git