IOS切面编程Hook
最近项目需要统计用户操作App的一些行为。我分析了一下，这里可以使用hook操作，把所有的事件都hook到一个方法中，然后在方法中进行统一进行处理。这样对原代码的入侵是最小的。
具体做法，我这里不细讲（网上有很多介绍这方面的方案）。只分析下其中遇到的问题。
一、简单介绍下Hook。
我以Hook的UIViewController-viewDidLoad为例：
目的：UIViewController的实例方法viewDidLoad与另外的一个方法比如ZX_viewDidLoad进行交换。之后，我在ZX_viewDidLoad中就可以监听到原来的viewDidLoad操作。
1、我们得为UIViewController添加ZX_viewDidLoad方法。这里我们可以通过类别的方式实现。
2、在类别中的load进行方法交换。load方法，只在加载（编译）此类时候调用一次，所以非常适合在这里进行操作。
#import <objc/runtime.h>
@implementation UIViewController (Analysis)
+(void)load
{
Method originalMethod = class_getInstanceMethod([self class], @selector(viewWillAppear:));
Method swizzingMethod = class_getInstanceMethod([self class], @selector(ZX_viewWillAppear:));
method_exchangeImplementations(originalMethod, swizzingMethod);
}
-(void)ZX_viewDidLoad
{
[self ZX_viewDidLoad];//这里调用的是原来的实现，所以不会导致死循环
//Action_identifier=BCUserSettingVC_ViewDidLoad
}
@end
3、上面已经基本实现了Hook。但需要注意的是：我们只是Hook了UIViewController的viewDidLoad方法。并不是Hook子类的viewDidLoad方法。但UIViewController都是被继承使用的。所以子类VC必须调用[super viewDidLoad]才能触发。比如：
//子类BCUserSettingVC
- (void)viewDidLoad{
NSLog(@"BCUserSettingVC--->super before");
[super viewDidLoad];
NSLog(@"BCUserSettingVC--->super after");
}
结果如下：
BCUserSettingVC--->super before
Action_identifier=BCUserSettingVC_ViewDidLoad
BCUserSettingVC--->super after
Action_identifier=BCUserSettingVC_ViewDidLoad是我在Hook方法中的输出。可见，调用[super viewDidLoad]才生效。
二、遇到的问题分析：
1）、对于Hook tableView的点击事件，网上基本都是这样实现的：
@implementation UITableView (Analysis)
+(void)load
{
static dispatch_once_t onceToken;
dispatch_once(&onceToken, ^{
SEL originalAppearSelector = @selector(setDelegate:);
SEL swizzingAppearSelector = @selector(ZX_setDelegate:);
[MethodSwizzingTool swizzingForClass:[self class] originalSel:originalAppearSelector swizzingSel:swizzingAppearSelector];
});
}
-(void)ZX_setDelegate:(id<UITableViewDelegate>)delegate
{
[self ZX_setDelegate:delegate];
SEL sel = @selector(tableView:didSelectRowAtIndexPath:);
SEL sel_t = @selector(ZX_tableView:didSelectRowAtIndexPath:);
//如果没实现tableView:didSelectRowAtIndexPath:就不需要hook
if (![delegate respondsToSelector:sel]){
return;
}
BOOL addsuccess = class_addMethod([delegate class],
sel_t,
method_getImplementation(class_getInstanceMethod([self class], sel_t)),
nil);
//如果添加成功了就直接交换实现， 如果没有添加成功，说明之前已经添加过并交换过实现了
if (addsuccess) {
Method selMethod = class_getInstanceMethod([delegate class], sel);
Method sel_Method = class_getInstanceMethod([delegate class], sel_t);
method_exchangeImplementations(selMethod, sel_Method);
}
}
// 由于我们交换了方法， 所以在tableview的 didselected 被调用的时候， 实质调用的是以下方法：
-(void)ZX_tableView:(UITableView *)tableView didSelectRowAtIndexPath:(NSIndexPath *)indexPath
{
[self ZX_tableView:tableView didSelectRowAtIndexPath:indexPath];
NSString * identifier = [NSString stringWithFormat:@"%@/%@/%ld", [self class],[tableView class], tableView.tag];
NSLog(@"tableView_identifier=%@",identifier);
}
先Hook setDelegate方法。再去Hook代理方法tableView:didSelectRowAtIndexPath。
这里有个很有意思的点，就是往代理类动态添加方法ZX_tableView:didSelectRowAtIndexPath。这个方法的实现是在UITableView中。并且这个Hook操作，在程序运行生命周期可能多次调用。不像上面Hook UIViewController一样，只调用一次。
因为tableView:didSelectRowAtIndexPath和ZX_tableView:didSelectRowAtIndexPath都是代理类的方法，所以怎么Hook也只是影响当前的代理类。所以一般情况下是可行的。
2）、但如果是以下的情况，就会出问题了：
1、UITableView的代理类为A，分别有子类B和C。因为有B和C，那么类A中self.tableView.delegate=self就会调用两次。
2、生成B的时候，调用类A中self.tableView.delegate=self后：
SEL sel = @selector(tableView:didSelectRowAtIndexPath:);
SEL sel_t = @selector(ZX_tableView:didSelectRowAtIndexPath:);
我们定义
tableView:didSelectRowAtIndexPath的SEL为->sel
ZX_tableView:didSelectRowAtIndexPath:的SEL为->sel_t
交换实现后为：
sel-->ZX_tableView:didSelectRowAtIndexPath:
sel_t-->tableView:didSelectRowAtIndexPath:
这时候是正确的。
class_addMethod这个运行时的添加方法，只对当前类实例有效。B生成了ZX_tableView:didSelectRowAtIndexPath后与父类A的方法tableView:didSelectRowAtIndexPath进行了交互。所以类A的原sel指向了ZX_tableView:didSelectRowAtIndexPath:
3、生成C的时候，又会调用类A中self.tableView.delegate=sel。此时，又得交换A类中的方法。
C生成方法ZX_tableView:didSelectRowAtIndexPath。但对于类A是公用的，所以
类A：sel-->ZX_tableView:didSelectRowAtIndexPath:
类C：sel_t-->ZX_tableView:didSelectRowAtIndexPath:
sel与sel_t指向了同一个实现，进行了交换，还是指向同一个实现。那么就会导致死循环。
4、解决办法：
使用Aspects替换自己写的交互逻辑：
-(void)ZX_setDelegate:(id<UITableViewDelegate>)delegate{
[self ZX_setDelegate:delegate];
NSObject *obg = (NSObject *)delegate;
if(![obg isKindOfClass:[NSObject class]]){
return;
}
SEL sel = @selector(tableView:didSelectRowAtIndexPath:);
[obg aspect_hookSelector:sel withOptions:AspectPositionAfter usingBlock:^(id<AspectInfo> aspectInfo){
NSArray *arr = aspectInfo.arguments;
// NSLog(@"UITableViewDelegate-aspect_hookSelector");
if(arr.count>1){
[self ZX_tableView:arr[0] didSelectRowAtIndexPath:arr[1]];
}
} error:nil];
}
-(void)ZX_tableView:(UITableView *)tableView didSelectRowAtIndexPath:(NSIndexPath *)indexPath
{
NSString *pathStr = [MethodSwizzingTool gainIdentifier:tableView];
NSString * identifier = [NSString stringWithFormat:@"%@/(%zi_%zi)", pathStr,indexPath.section,indexPath.row];
[ZXFireBaseManage ZX_TableViewReport:tableView didSelectRowAtIndexPath:indexPath identifier:identifier];
}
因为Aspects会生成一个新的类，然后对此类方法进行操作。所以就不会影响到公共的父类了。想引用好hook，得好好思考下，因为比较绕，稍不留神可能就铸成大错。
