iOS數(shù)組防止越界crash

有時候項目中總是出現(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)用方式
    這是系統(tǒng)原本的調(diào)用方式
  • 這是改變之后的調(diào)用方式


    這是改變之后的調(diào)用方式
    這是改變之后的調(diào)用方式

我是先勾住array自帶的方法,進行判斷,如果沒有越界等幾種情況,再繼續(xù)執(zhí)行它自身的方法,相當于在執(zhí)行方法前多了一步判斷,而網(wǎng)上是直接把方法替換成自己的方法了,這里還是有本質(zhì)的區(qū)別。

具體實現(xiàn)原理

這里舉例說明 NSArrayaddObject: 方法,其他也類似。

先定義一個靜態(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
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容