有時候項目中總是出現(xiàn)一些無法預(yù)知的情況,導(dǎo)致數(shù)組越界是程序crash,如果這種意外情況無法避免,那么只能從側(cè)面采取保護措施。我先從網(wǎng)上找答案,我想其他人也肯定遇到過相同的情況,如果有好的解決方案,直接采用就可以了。但是實際上,網(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)聚,低耦合"的思想來實現(xiàn),尤其是這種工具類的代碼,更是應(yīng)該對原代碼入侵越少越好。一個很容易想到的方法,就是采用runtime, 把array中的以上幾種情況的方法替換成自己的方法,然后再執(zhí)行方法的時候加以判斷。而我在網(wǎng)上搜到的結(jié)果全是以這種方案解決的,不排除有更好的方法我沒找到。附上一個我找到的代碼比較詳細的demo。我試了一下,效果是可以達到,不過我還是毫不猶豫的拒絕這種方式。直接替換了系統(tǒng)的方法必然會導(dǎo)致更多無法預(yù)知的問題。這些問題,我在后面會講幾個我遇到的。而我準備這樣解決:
-
這是系統(tǒng)原本的調(diào)用方式
這是系統(tǒng)原本的調(diào)用方式 -
這是改變之后的調(diào)用方式
這是改變之后的調(diào)用方式
我是先勾住array自帶的方法,進行判斷,如果沒有越界等幾種情況,再繼續(xù)執(zhí)行它自身的方法,相當于在執(zhí)行方法前多了一步判斷,而網(wǎng)上是直接把方法替換成自己的方法了,這里還是有本質(zhì)的區(qū)別。
具體實現(xiàn)原理
這里舉例說明 NSArray 的 addObject: 方法,其他也類似。
先定義一個靜態(tài)變量
static IMP array_old_func_imap_object = NULL;
這個變量用來記錄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:)]);
自定義方法的實現(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)差不多完成了,就剩最后一個問題了,就是怎么運用到項目中,讓這個工具類繼承自NSObject,把這個工具類寫成一個單例,然后在load方法中調(diào)用單例。load 方法會在本類第一次使用的時候調(diào)用一次,所以,把這個工具類拖到項目中,不用寫其他代碼,就實現(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
實際出現(xiàn)的問題
我用這兩種方式都試了試,新建一個空項目,然后把上面幾個方法都試一遍,似乎都沒問題,然后我把他們公司的項目中,程序有時候卡死,還會crash,還是沒法用,兩種方式都有問題,找了找原因,發(fā)現(xiàn)NSArray和NSMutableArray的那幾個方法,系統(tǒng)自己會調(diào)用很多很多次,極大的影響了性能,還有網(wǎng)友遇到了其他的問題:替換了objectAtIndex方法有輸入的地方出來了軟鍵盤按手機Home鍵就Crash了。簡直無解,最后,還是決定寫個分類,雖然low一點,畢竟還是能解決我的問題,并且不會帶來新的問題。
這是給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