关于 UITableView 的 delegate 被 hook 时的问题

开发中必不可少的一个环节就是产品经理要求添加的埋点, 从是否需要开发者添加代码的角度, 分为无侵入埋点和事件埋点. 其中 无侵入埋点 SDK 的实现原理就是在 hook 了所有UI 控件的事件.
页面展示的 UI 控件, 最常见的就是 UITableView, 那么如何 hook 它的点击事件呢?

UITableView 的点击事件 tableView:didSelectRowAtIndexPath: 由其 delegate 实现. 最直接的想法, 就是在设置 UITableView 的 delegate 时, 交换 delegate 中实现的方法:

@implementation UITableView (switchDelegate)

+ (void)load {
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        [self methodSwizzlingWithOriginalSelector:@selector(setDelegate:) bySwizzledSelector:@selector(ZD_setDelegate:)];
    });
}

-(void)ZD_setDelegate:(id)delegate
{    
    SEL sel = @selector(tableView:didSelectRowAtIndexPath:);
    SEL sel_t = @selector(ZD_tableView:didSelectRowAtIndexPath:);

    //如果 delegate 没实现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);
    }
    [self ZD_setDelegate:delegate];
}

- (void)ZD_tableView:(UITableView *)tableView didSelectRowAtIndexPath:(NSIndexPath *)indexPath {
    NSLog(@"%s", __func__);
    [self ZD_tableView:tableView didSelectRowAtIndexPath:indexPath];
}

//NSObject 分类 Swizzling 中封装了方法交换方式
@implementation NSObject (Swizzling)
+ (void)methodSwizzlingWithOriginalSelector:(SEL)originalSelector bySwizzledSelector:(SEL)swizzledSelector{
    Class class = [self class];
    Method originalMethod = class_getInstanceMethod(class, originalSelector);
    Method swizzledMethod = class_getInstanceMethod(class, swizzledSelector);
    BOOL didAddMethod = class_addMethod(class,originalSelector,
                                        method_getImplementation(swizzledMethod),
                                        method_getTypeEncoding(swizzledMethod));
    if (didAddMethod) {
        class_replaceMethod(class,swizzledSelector,
                            method_getImplementation(originalMethod),
                            method_getTypeEncoding(originalMethod));
    } else {
        method_exchangeImplementations(originalMethod, swizzledMethod);
    }
}

+ (void)methodSwizzlingWithClassSelector:(SEL)originalClassSelector
                         byClassSelector:(SEL)swizzledClassSelector{
    
    Class class = object_getClass([self class]);
    Method originalMethod = class_getInstanceMethod(class, originalClassSelector);
    Method swizzledMethod = class_getInstanceMethod(class, swizzledClassSelector);
    BOOL didAddMethod = class_addMethod(class,originalClassSelector,
                                        method_getImplementation(swizzledMethod),
                                        method_getTypeEncoding(swizzledMethod));
    if (didAddMethod) {
        class_replaceMethod(class,swizzledClassSelector,
                            method_getImplementation(originalMethod),
                            method_getTypeEncoding(originalMethod));
    } else {
        method_exchangeImplementations(originalMethod, swizzledMethod);
    }
    
}
@end

使用以上方式, 能实现 hook 单独 UITableView 的 Delegate. 单独的意思是 该UITableView 没有子类 . 比如新建 MyTableView, 在其初始化方法中, 设置其本身为 delegate:

@interface MyTableView ()

@end

@implementation MyTableView

- (instancetype)initWithFrame:(CGRect)frame style:(UITableViewStyle)style {
    self = [super initWithFrame:frame style:style];
    if (self) {
        self.dataSource = self;
        self.delegate = self;
    }
    return self;
}

- (void)tableView:(UITableView *)tableView didSelectRowAtIndexPath:(NSIndexPath *)indexPath {
    NSLog(@"didSelectRowAtIndexPath");
}
@end

如果有两个子类 OneTableView 和 TwoTableView 分别继承了 MyTableView, 在创建 OneTableView 和 TwoTableView 使用时, 会在 tableView:didSelectRowAtIndexPath: 中陷入死循环. 原因如下:

image.png

在 OneTableView 设置 delegate 后, 其父类 MyTableView 中的 tableView:didSelectRowAtIndexPath: 方法实现, 已经指向了 UITableView+switchDelegate 中的 ZD_tableView:didSelectRowAtIndexPath:. 当在 TwoTableView 中再次设置 delegate 时, 将父类 MyTableView 中的 ZD_tableView:didSelectRowAtIndexPath: 再次指向 UITableView+switchDelegate 中的函数本身, 在调用时就发生了循环调用.

解决方式, 下回分解.

喜欢和关注都是对我的鼓励和支持~