优雅解决 iOS 8 UIScrollView delegate

背景

从近期 Crash 上报记录来看，有相当一部分 EXC_BAD_ACCESS 和 UIScrollView 有关，而且都是在 iOS 8 上才会出现，通过排查发现是因为在 iOS 8 的 UIScrollView 的 delegate 是 assign 属性，在某些场景下 scrollView 的 delegate 被释放而 scrollView 继续访问其 delegate 则会出现 EXC_BAD_ACCESS 的问题。

解决的方法也很简单，在 viewController 的 dealloc 方法将对应的 scrollView 方法置空即可

- (void)dealloc {
    self.scrollView.delegate = nil;
}

但是由于项目历史原因，要逐一排查并修改成本比较大，利用 hook 来解决这个问题的想法油然而生，基本思路是当一个对象成为 UIScrollView 的 delegate 时，去监听这个 delegate 对象的释放，一旦 delegate 对象被释放，则手动把 scrollView 的 delegate 设置为 nil （有点像我们手动实现 weak）

DelegateCleaner

由于 hook 对象的 -(void)dealloc 方法比较复杂，并且有一定的危险性，这里采用了一种宿主的方法，利用 objc_setAssociatedObject 给 delegate 添加一个 DelegateCleaner 对象，这个对象有以下功能：
1、记录将宿主对象作为 delegate 的所有 scrollView
2、当宿主对象被释放的时候，对记录的全部 scrollView 调用 scrollView.delegate = nil

因为 AssociatedObject 机制，当 delegate dealloc 的时候，DelegateCleaner 对象也会释放，调用 -(void)dealloc 方法，所以我们可以实现以上功能。

ReleaseDelegateCleaner 用到了 NSPointerArray 来记录 scrollView（功能等同于 weak 数组） ， 这样一来当 scrollView 被释放掉，我们就无需要去对其 delegate 置 nil。
另外在对 scrollView 的 delegate 置 nil 的时候，是利用设置 Ivar 的直接方式，避免和其他第三方库 hook setDelegate 的方法造成冲突。

ReleaseDelegateCleaner 的具体实现如下：

@interface ReleaseDelegateCleaner : NSObject
@property (nonatomic, strong) NSPointerArray *scrollViews;
@end

@implementation ReleaseDelegateCleaner

- (void)dealloc {
    [self cleanScrollViewsDelegate];
}

- (void)cleanScrollViewsDelegate {
    [self.scrollViews.allObjects enumerateObjectsUsingBlock:^(UIScrollView *scrollView, NSUInteger idx, BOOL * _Nonnull stop) {
        object_setIvarValue(scrollView, "_delegate", nil);
    }];
}

- (void)recordDelegatedScrollView:(UIScrollView *)scrollView {
    NSUInteger index = [self.scrollViews.allObjects indexOfObject:scrollView];
    if (index == NSNotFound) {
        [self.scrollViews addPointer:(__bridge void *)(scrollView)];
    }
}

- (void)removeDelegatedScrollView:(UIScrollView *)scrollView {
    NSUInteger index = [self.scrollViews.allObjects indexOfObject:scrollView];
    if (index != NSNotFound) {
        [self.scrollViews removePointerAtIndex:index];
    }
}

- (void)setScrollViews:(NSMutableSet *)scrollViews {
    objc_setAssociatedObject(self, @selector(scrollViews), scrollViews, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}

- (NSPointerArray *)scrollViews {
    NSPointerArray *scrollViews = objc_getAssociatedObject(self, _cmd);
    if (!scrollViews) {
        scrollViews = [NSPointerArray weakObjectsPointerArray];
        [self setScrollViews:scrollViews];
    }
    return scrollViews;
}

为 delegate 关联 DelegateCleaner 对象

我们要找一个时机对 scrollview 的 delegate 进行 AssociatedObject 关联，这个很容易想到就是 setDelegate: 方法，另外当 delegate 置 nil 的时候，需要判断当前的 delegate 是否记录了 scrollView，这里一样直接取 Ivar 从而避免触发 scrollView 的 getter 方法。

@implementation UIScrollView (iOS8Safe)

+ (void)load {
    if (IOS_VERSION < 9.0) {
        SwizzleMethod([UIScrollView class], @selector(setDelegate:), @selector(safe_setDelegate:));
    }
}

- (void)safe_setDelegate:(id<UIScrollViewDelegate>)delegate {
    
    id _delegate = object_getIvarValue(self, "_delegate");
    
    if (_delegate != delegate) {
        [[(NSObject *)_delegate iOS8DelegateCleaner] removeDelegatedScrollView:self];
        [[(NSObject *)delegate iOS8DelegateCleaner] recordDelegatedScrollView:self];
    }
    
    [self safe_setDelegate:delegate];
}

@end

除此之外，利用这一方法也可以对 UICollectionView 和 UITableView 的 setDataSource 做防护处理，完整的源码详见：

GitHub ： UIScrollView-iOS8Safe