有時(shí)候項(xiàng)目中總是出現(xiàn)一些無法預(yù)知的情況，導(dǎo)致數(shù)組越界是程序crash，如果這種意外情況無法避免，那么只能從側(cè)面采取保護(hù)措施。我先從網(wǎng)上找答案，我想其他人也肯定遇到過相同的情況，如果有好的解決方案，直接采用就可以了。但是實(shí)際上，網(wǎng)上搜索的結(jié)果令人有些失望。下面還是記錄一下我自己的解決方案，以及和網(wǎng)上解決方案的差異。
crash的具體幾種情況
- 取值：index超出array的索引范圍
- 添加：插入的object為nil或者Null
- 插入：index大于count、插入的object為nil或者Null
- 刪除：index超出array的索引范圍
- 替換：index超出array的索引范圍、替換的object為nil或者Null
解決思路
任何代碼都需要圍繞"高內(nèi)聚，低耦合"的思想來實(shí)現(xiàn)，尤其是這種工具類的代碼，更是應(yīng)該對(duì)原代碼入侵越少越好。一個(gè)很容易想到的方法，就是采用runtime, 把a(bǔ)rray中的以上幾種情況的方法替換成自己的方法，然后再執(zhí)行方法的時(shí)候加以判斷。而我在網(wǎng)上搜到的結(jié)果全是以這種方案解決的，不排除有更好的方法我沒找到。附上一個(gè)我找到的代碼比較詳細(xì)的demo。我試了一下，效果是可以達(dá)到，不過我還是毫不猶豫的拒絕這種方式。直接替換了系統(tǒng)的方法必然會(huì)導(dǎo)致更多無法預(yù)知的問題。這些問題，我在后面會(huì)講幾個(gè)我遇到的。而我準(zhǔn)備這樣解決：
-
這是系統(tǒng)原本的調(diào)用方式
這是系統(tǒng)原本的調(diào)用方式 -
這是改變之后的調(diào)用方式
這是改變之后的調(diào)用方式
我是先勾住array自帶的方法，進(jìn)行判斷，如果沒有越界等幾種情況，再繼續(xù)執(zhí)行它自身的方法，相當(dāng)于在執(zhí)行方法前多了一步判斷，而網(wǎng)上是直接把方法替換成自己的方法了，這里還是有本質(zhì)的區(qū)別。
具體實(shí)現(xiàn)原理
這里舉例說明 NSArray 的 addObject: 方法，其他也類似。
先定義一個(gè)靜態(tài)變量
static IMP array_old_func_imap_object = NULL;
這個(gè)變量用來記錄array自帶方法的指針地址
獲取方法，然后記錄方法的指針地址
Method old_func_imap_object = class_getInstanceMethod(NSClassFromString(@"__NSArrayI"), @selector(objectAtIndex:));
array_old_func_imap_object = method_getImplementation(old_func_imap_object);
改變原方法的指針地址，并指向自定義方法
method_setImplementation(old_func_imap_object, [self methodForSelector:@selector(fm_objectAtIndex:)]);
自定義方法的實(shí)現(xiàn)
- (id)fm_objectAtIndex:(NSUInteger)index {
if (index < [(NSArray*)self count]) {
return ((id(*)(id, SEL, NSUInteger))array_old_func_imap_object)(self, @selector(objectAtIndex:), index);
}
NSLog(@"NArray objectAtIndex 失敗--%@", [NSThread callStackSymbols]);
return nil;
}
最后一步
到這里已經(jīng)差不多完成了，就剩最后一個(gè)問題了，就是怎么運(yùn)用到項(xiàng)目中，讓這個(gè)工具類繼承自NSObject，把這個(gè)工具類寫成一個(gè)單例，然后在load方法中調(diào)用單例。load 方法會(huì)在本類第一次使用的時(shí)候調(diào)用一次，所以，把這個(gè)工具類拖到項(xiàng)目中，不用寫其他代碼，就實(shí)現(xiàn)了以上的功能。
+ (void)load {
[FMDetecter sharedInstance];
}
static dispatch_once_t onceToken;
static FMDetecter *sharedInstance;
+ (instancetype)sharedInstance {
dispatch_once(&onceToken, ^{
sharedInstance = [[FMDetecter alloc] init];
});
return sharedInstance;
}
這里有完整的代碼，有興趣可查看demo
實(shí)際出現(xiàn)的問題
我用這兩種方式都試了試，新建一個(gè)空項(xiàng)目，然后把上面幾個(gè)方法都試一遍，似乎都沒問題，然后我把他們公司的項(xiàng)目中，程序有時(shí)候卡死，還會(huì)crash，還是沒法用，兩種方式都有問題，找了找原因，發(fā)現(xiàn)NSArray和NSMutableArray的那幾個(gè)方法，系統(tǒng)自己會(huì)調(diào)用很多很多次，極大的影響了性能，還有網(wǎng)友遇到了其他的問題：替換了objectAtIndex方法有輸入的地方出來了軟鍵盤按手機(jī)Home鍵就Crash了。簡直無解，最后，還是決定寫個(gè)分類，雖然low一點(diǎn)，畢竟還是能解決我的問題，并且不會(huì)帶來新的問題。
這是給NSArray添加的方法
#import "NSArray+beyond.h"
@implementation NSArray (beyond)
-(id)objectAtIndexCheck:(NSUInteger)index
{
if (index < self.count) {
return [self objectAtIndex:index];
}
return nil;
}
@end
這是給NSMutableArray添加的方法
#import "NSMutableArray+beyond.h"
@implementation NSMutableArray (beyond)
-(id)objectAtIndexCheck:(NSUInteger)index
{
if (index < self.count) {
return [self objectAtIndex:index];
}
NSLog(@"%@", [NSThread callStackSymbols]);
return nil;
}
- (void)addObjectCheck:(id)anObject
{
if (anObject != nil && [anObject isKindOfClass:[NSNull class]] == NO) {
[self addObject:anObject];
} else {
NSLog(@"%@", [NSThread callStackSymbols]);
}
}
- (void)insertObjectCheck:(id)anObject atIndex:(NSUInteger)index
{
if (index <= self.count && anObject != nil && [anObject isKindOfClass:[NSNull class]] == NO) {
[self insertObject:anObject atIndex:index];
} else {
NSLog(@"%@", [NSThread callStackSymbols]);
}
}
- (void)removeObjectAtIndexCheck:(NSUInteger)index
{
if (index < self.count) {
[self removeObjectAtIndex:index];
} else {
NSLog(@"%@", [NSThread callStackSymbols]);
}
}
- (void)replaceObjectAtIndexCheck:(NSUInteger)index withObject:(id)anObject
{
if (index < self.count && anObject != nil && [anObject isKindOfClass:[NSNull class]] == NO) {
[self replaceObjectAtIndex:index withObject:anObject];
} else {
NSLog(@"%@", [NSThread callStackSymbols]);
}
}
@end
