最近在网易云捕上看到一些数组越界导致的崩溃日志，所以决定数组的越界做一些处理。
崩溃报错信息
在项目的开发中，笔者一般遇到的问题就是，数组越界：
- -[__NSArrayM objectAtIndex:]: index 0 beyond bounds for empty array；
- -[__NSArrayM objectAtIndexedSubscript:]: index 0 beyond bounds for empty array；
很明显，这两个函数是在数组取值的时候发生的越界情况，在网上搜索了很多大神的文章，也自己总结了一下，下面罗列出两种处理方法：
一、为NSArray、NSMutableArray添加分类并添加方法
首先，我们为NSarray创建分类并添加方法,在.h文件中：
@interface NSArray (ErrorHandle)
/**
为数组分类添加的方法 可以在应用中直接调用 可以防止数组越界导致的crash
@param index 传入的取值下标
@return id类型的数据
*/
- (id)objectAtIndexVerify:(NSUInteger)index;
- (id)objectAtIndexedSubscriptVerify:(NSUInteger)idx;
@end
在.m文件中，我们可以将这两个方法实现出来:
@implementation NSArray (ErrorHandle)
/**
* 防止数组越界
*/
- (id)objectAtIndexVerify:(NSUInteger)index{
if (index < self.count) {
return [self objectAtIndex:index];
}else{
return nil;
}
}
/**
* 防止数组越界
*/
- (id)objectAtIndexedSubscriptVerify:(NSUInteger)idx{
if (idx < self.count) {
return [self objectAtIndexedSubscript:idx];
}else{
return nil;
}
}
类似的，我们可以为NSMutableArray创建分类,(在可变数组中，我们插入nil对象也会产生crash，所以我们要对可变数组做特殊处理)
#import
@interface NSMutableArray (ErrorHandle)
/**
数组中插入数据
@param object 数据
@param index 下标
*/
- (void)insertObjectVerify:(id)object atIndex:(NSInteger)index;
/**
数组中添加数据
@param object 数据
*/
- (void)addObjectVerify:(id)object;
@end
在可变数组的.m文件中
@implementation NSMutableArray (ErrorHandle)
/**
* 数组中插入数据
*/
- (void)insertObjectVerify:(id)object atIndex:(NSInteger)index{
if (index < self.count && object) {
[self insertObject:object atIndex:index];
}
}
/**
* 数组中添加数据
*/
- (void)addObjectVerify:(id)object{
if (object) {
[self addObject:object];
}
}
特别说明：以上方法在项目的实际运用中，要想防止数组越界，就需要调用我们自己添加的方法了，而不要调用系统的了。
二、用runtime处理数组越界
不到万不得已，笔者一般是不想用runtime的。不过runtime确确实实也能解决数组越界的问题，在我们数组越界处理的第一种方法中，我们可以看见，我们无法使用索引来从数组中取值了（即类似：cell.textLabel.text = self.dataSource[indexPath.row];这样的取值方式）。那如果我们想要这种取值方式的话，就需要用runtime来实现了。
数组索引调用的方法
通过报错的函数，我们可以发现，数组索引调用的是objectAtIndexedSubscript:这个函数。找到了报错的函数，我们就可以通过runtime来实现函数的交换。首先，我们为NSObject写一个分类，方便我们调用交换系统和自定义的方法：
#import
@interface NSObject (SwizzleMethod)
/**
* 对系统方法进行替换(交换实例方法)
*
* @param systemSelector 被替换的方法
* @param swizzledSelector 实际使用的方法
* @param error 替换过程中出现的错误消息
*
* @return 是否替换成功
*/
+ (BOOL)SystemSelector:(SEL)systemSelector swizzledSelector:(SEL)swizzledSelector error:(NSError *)error;
@end
在.m文件中，我们需要将方法实现出来：
#import "NSObject+SwizzleMethod.h"
#import
@implementation NSObject (SwizzleMethod)
/**
* 对系统方法进行替换
*
* @param systemSelector 被替换的方法
* @param swizzledSelector 实际使用的方法
* @param error 替换过程中出现的错误消息
*
* @return 是否替换成功
*/
+ (BOOL)SystemSelector:(SEL)systemSelector swizzledSelector:(SEL)swizzledSelector error:(NSError *)error{
Method systemMethod = class_getInstanceMethod(self, systemSelector);
if (!systemMethod) {
return NO;
}
Method swizzledMethod = class_getInstanceMethod(self, swizzledSelector);
if (!swizzledMethod) {
return NO;
}
if (class_addMethod([self class], systemSelector, method_getImplementation(swizzledMethod), method_getTypeEncoding(swizzledMethod))) {
class_replaceMethod([self class], swizzledSelector, method_getImplementation(systemMethod), method_getTypeEncoding(systemMethod));
}else{
method_exchangeImplementations(systemMethod, swizzledMethod);
}
return YES;
}
@end
在方法交换和替换的过程中，如果被替换的方法或者我们将要使用的方法没有的话，直接ruturn，不进行方法互换，经过双重检验才进行方法的互换。
我们以NSMutableArray为例子，同样的NSMutableArray添加分类，在.h文件中只需要写下如下代码：
+(void)load{
[super load];
//无论怎样 都要保证方法只交换一次
static dispatch_once_t onceToken;
dispatch_once(&onceToken, ^{
//交换NSMutableArray中的方法
[objc_getClass("__NSArrayM") SystemSelector:@selector(objectAtIndex:) swizzledSelector:@selector(jz_objectAtIndex:) error:nil];
//交换NSMutableArray中的方法
[objc_getClass("__NSArrayM") SystemSelector:@selector(objectAtIndexedSubscript:) swizzledSelector:@selector(jz_objectAtIndexedSubscript:) error:nil];
});
}
- (id)jz_objectAtIndex:(NSUInteger)index{
if (index < self.count) {
return [self jz_objectAtIndex:index];
}else{
NSLog(@" 你的NSMutableArray数组已经越界 帮你处理好了%ld %ld %@", index, self.count, [self class]);
return nil;
}
}
- (id)jz_objectAtIndexedSubscript:(NSUInteger)index{
if (index < self.count) {
return [self jz_objectAtIndexedSubscript:index];
}else{
NSLog(@" 你的NSMutableArray数组已经越界 帮你处理好了%ld %ld %@", index, self.count, [self class]);
return nil;
}
}
同样的，我们也可以在NSArray的分类中添加如下代码：
+(void)load{
[super load];
//无论怎样 都要保证方法只交换一次
static dispatch_once_t onceToken;
dispatch_once(&onceToken, ^{
//交换NSArray中的objectAtIndex方法
[objc_getClass("__NSArrayI") SystemSelector:@selector(objectAtIndex:) swizzledSelector:@selector(sxy_objectAtIndex:) error:nil];
//交换NSArray中的objectAtIndexedSubscript方法
[objc_getClass("__NSArrayI") SystemSelector:@selector(objectAtIndexedSubscript:) swizzledSelector:@selector(sxy_objectAtIndexedSubscript:) error:nil];
});
}
- (id)sxy_objectAtIndexedSubscript:(NSUInteger)idx{
if (idx < self.count) {
return [self sxy_objectAtIndexedSubscript:idx];
}else{
NSLog(@" 你的 NSArray数组已经越界了 但是已经帮你处理好了 %ld %ld", idx, self.count);
return nil;
}
}
- (id)sxy_objectAtIndex:(NSUInteger)index{
if (index < self.count) {
return [self sxy_objectAtIndex:index];
}else{
NSLog(@" 你的 NSArray数组已经越界了 但是已经帮你处理好了 %ld %ld", index, self.count);
return nil;
}
}
关于上面的Demo，笔者已经上传git，需要的小伙伴去下载吧！数组越界Demo