iOS安全防护之二：数组越界

前言：在release发布之后，会有一些偶发crash是数组越界造成的。其本质原因大多是代码逻辑的错误，如果我们暂时没办法定位问题，也要尽量减少crash的发生，避免影响用户体验。

---

可以利用分类地添加方法给数组添加一个我们自己实现地方法，在取出数值之前做判断，但是这样地方法非常不灵活。对于系统调用也无能为力。这时候我们就可以利用runtime的Method-Swizzling技术去替换系统的实现，达到防护目的。

先手动制造几个数据越界，看一下结果

NSArray *array = @[@1];
[array objectAtIndex:4];
array[4];
错误提示：[__NSSingleObjectArrayI objectAtIndex:]: index 4 beyond bounds [0 .. 0]


NSArray *array = @[@1,@2];
[array objectAtIndex:4];
错误提示：[__NSArrayI objectAtIndex:]: index 4 beyond bounds [0 .. 1]'
array[4];
错误提示：[__NSArrayI objectAtIndexedSubscript:]: index 4 beyond bounds [0 .. 1]'

NSArray *arrayM = @[@1].mutableCopy;
[arrayM objectAtIndex:4];
错误提示：[__NSArrayM objectAtIndex:]: index 4 beyond bounds [0 .. 0]
    
arrayM[4];
错误提示：[__NSArrayM objectAtIndexedSubscript:]: index 4 beyond bounds [0 .. 0]
    

由于NSArray和NSMutableArray类簇的特殊性，其真身并是__NSSingleObjectArrayI（不可变数组只有一个对象）， __NSArrayI（不可变数组多个对象），__NSArrayM（可变数组）。不同的调用方法锁最终地调用方法也不同，有objectAtIndex：和objectAtIndexedSubscript：

所以需要替换这个几种方法实现即可，还是在分类里面实现即可，直接上代码：

#import "NSArray+Extension.h"
#import <objc/runtime.h>

@implementation NSArray (Extension)

+ (void)load
{
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{

        Class clsI = NSClassFromString(@"__NSArrayI");
        
        Method method1 = class_getInstanceMethod(clsI, @selector(objectAtIndexedSubscript:));
        Method method2 = class_getInstanceMethod(clsI, @selector(yye_objectAtIndexedSubscript:));
        method_exchangeImplementations(method1, method2);
        
        Method method3 = class_getInstanceMethod(clsI, @selector(objectAtIndex:));
        Method method4 = class_getInstanceMethod(clsI, @selector(yye_objectAtIndex:));
        method_exchangeImplementations(method3, method4);
        
        Class clsSingleI = NSClassFromString(@"__NSSingleObjectArrayI");
        Method method5 = class_getInstanceMethod(clsSingleI, @selector(objectAtIndex:));
        Method method6 = class_getInstanceMethod(clsSingleI, @selector(yyeSingle_objectAtIndex:));
        method_exchangeImplementations(method5, method6);
    });
}

- (id)yye_objectAtIndexedSubscript:(NSUInteger)index{
    if(index>=self.count) return self.lastObject;
    
    return [self yye_objectAtIndexedSubscript:index];
}

- (id)yye_objectAtIndex:(NSUInteger)index{
    if(index>=self.count) return self.lastObject;
    
    return [self yye_objectAtIndex:index];
}

- (id)yyeSingle_objectAtIndex:(NSUInteger)index{
    if(index>=self.count){
        return self.lastObject;
    }
    
    return [self yyeSingle_objectAtIndex:index];
}
@end

可变数组地代码我就不贴出来了，只要把__NSArrayI更换为__NSArrayM就好。同时测试通过放在release下生效，避免给开发带来困难，如果想要排查问题可以把数组信息上传服务器，帮助定位问题。