前言：在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下生效，避免给开发带来困难，如果想要排查问题可以把数组信息上传服务器，帮助定位问题。
