在日常項目中肯污,常見的crash包括:給NSNull發(fā)送消息翘单,數(shù)組越界吨枉,字典傳空值等。我們可以對這些crash簡單的處理哄芜,來優(yōu)化項目貌亭,減少安全隱患。
NSNull
NSNull的crash常見于后臺返回數(shù)據(jù)中可能會有null字段认臊。絕大多數(shù)情況下給一個NSNull對象發(fā)送消息的話圃庭,會產(chǎn)生crash(null是有內(nèi)存的)。但是給nil發(fā)送消息失晴,就可以規(guī)避這種crash剧腻。這就是NullSafe的處理原理。
NullSafe的處理步驟:
- 創(chuàng)建一個方法緩存涂屁,這個緩存會緩存項目中類的所有類名书在。
- 遍歷緩存,尋找是否已經(jīng)有可以執(zhí)行此方法的類拆又。
- 如果有的話儒旬,返回這個NSMethodSignature。
- 如果沒有的話遏乔,返回nil,接下來會走forwardInvocation:方法义矛。
- [invocation invokeWithTarget:nil]將消息轉(zhuǎn)發(fā)給nil。
在OC中盟萨,系統(tǒng)如果對某個實(shí)例發(fā)送消息之后凉翻,它(及其父類)無法處理(比如,沒有這個方法等)捻激,系統(tǒng)就會發(fā)送methodSignatureForSelector消息制轰,如果這個方法返回非空,那么就去執(zhí)行返回的方法胞谭,如果為nil,則發(fā)送forwardInvocation消息垃杖。重寫這兩個方法將沒能力處理消息的方法簽名轉(zhuǎn)發(fā)給nil對象則不會產(chǎn)生崩潰
其實(shí)我們可以在解析json時對NSNull進(jìn)行處理,或者直接要后臺不返回null(比如丈屹,將空對象過濾掉)调俘。
NSObject
對于數(shù)組越界,字典傳空值的crash的處理方式是一樣的旺垒,通過Runtime的Method Swizzle,將原生的方法hook掉彩库。
1.抽出公共的SwizzleMethod方法放在NSObject分類中
#import "NSObject+SwizzleMethod.h"
#import <objc/runtime.h>
#define NSOBJECT_SWIZZLEMETHOD_ENABLED 1
@implementation NSObject (SwizzleMethod)
//runtime交換方法
- (void)swizzleMethod:(SEL)origSelector withMethod:(SEL)newSelector
{
if(!NSOBJECT_SWIZZLEMETHOD_ENABLED) return;
Class class = [self class];
//獲取方法
Method originalMethod = class_getInstanceMethod(class, origSelector);
Method swizzledMethod = class_getInstanceMethod(class, newSelector);
//添加origSelector方法,并將origSelector的實(shí)現(xiàn)指向swizzledMethod先蒋,以達(dá)到交換方法實(shí)現(xiàn)的目的骇钦。
//如果didAddMethod返回YES,說明origSelectorz在Class中不存在竞漾,是新方法眯搭,并將origSelector的實(shí)現(xiàn)指向swizzledMethod
//返回NO窥翩,說明Class中已經(jīng)存在origSelector方法
BOOL didAddMethod = class_addMethod(class,
origSelector,
method_getImplementation(swizzledMethod),
method_getTypeEncoding(swizzledMethod));
if (didAddMethod) {
//利用class_replaceMethod將newSelector的實(shí)現(xiàn)指向originalMethod(替換newSelector的實(shí)現(xiàn))。
class_replaceMethod(class,
newSelector,
method_getImplementation(originalMethod),
method_getTypeEncoding(originalMethod));
} else {
//利用method_exchangeImplementations交換方法的實(shí)現(xiàn)
method_exchangeImplementations(originalMethod, swizzledMethod);
}
}
@end
開發(fā)階段可以將NSOBJECT_SWIZZLEMETHOD_ENABLED設(shè)置為0鳞仙,因為我們需要準(zhǔn)確的定位問題寇蚊。發(fā)布階段設(shè)置為1,用來規(guī)避不可預(yù)見的crash繁扎。
2.NSMutableDictionary字典傳空的crash
#import "NSMutableDictionary+NullSafe.h"
#import "NSObject+SwizzleMethod.m"
@implementation NSMutableDictionary (NullSafe)
+ (void)load {
static dispatch_once_t onceToken;
dispatch_once(&onceToken, ^{
id obj = [[self alloc] init];
[obj swizzleMethod:@selector(setObject:forKey:) withMethod:@selector(safe_setObject:forKey:)];
});
}
- (void)safe_setObject:(id)value forKey:(NSString *)key {
if (value) {
[self safe_setObject:value forKey:key];
}else {
NSLog(@"***[NSMutableDictionary setObject: forKey:], Object cannot be nil");
}
}
@end
3.NSMutableArray的removeObjectAtIndex:越界crash
#import "NSMutableArray+BeyondSafe.h"
#import "NSObject+SwizzleMethod.m"
@implementation NSMutableArray (BeyondSafe)
+ (void)load {
static dispatch_once_t onceToken;
dispatch_once(&onceToken, ^{
id obj = [[self alloc] init];
[obj swizzleMethod:@selector(removeObjectAtIndex:) withMethod:@selector(safe_removeObjectAtIndex:)];
});
}
- (void)safe_removeObjectAtIndex:(NSUInteger)index {
if (index >= self.count) {
NSLog((@"***[NSArrayM removeObjectAtIndex:], index %lu beyond bounds count %lu"),(unsigned long)index,(unsigned long)self.count);
return;
}
[self safe_removeObjectAtIndex:index];
}
@end
4.NSArray的objectAtIndex:越界的crash
#import "NSArray+BeyondSafe.h"
#import "NSObject+SwizzleMethod.m"
@implementation NSArray (BeyondSafe)
+ (void)load {
static dispatch_once_t onceToken;
dispatch_once(&onceToken, ^{
[objc_getClass("__NSArrayI") swizzleMethod:@selector(objectAtIndex:) withMethod:@selector(safeI_objectAtIndex:)];
[objc_getClass("__NSArrayM") swizzleMethod:@selector(objectAtIndex:) withMethod:@selector(safeM_objectAtIndex:)];
});
}
//雖然兩方法的實(shí)現(xiàn)是一樣的幔荒,但是不能指向同一方法,會報錯
- (id)safeI_objectAtIndex:(NSUInteger)index
{
if (index >= self.count) {
NSLog((@"***[NSArrayI objectAtIndex:], index %lu beyond bounds count %lu"),(unsigned long)index,(unsigned long)self.count);
return nil;
}
return [self safeI_objectAtIndex:index];
}
- (id)safeM_objectAtIndex:(NSUInteger)index
{
if (index >= self.count) {
NSLog(@"***[NSArrayM objectAtIndex:], index %lu beyond bounds count %lu",(unsigned long)index,(unsigned long)self.count);
return nil;
}
return [self safeM_objectAtIndex:index];
}
@end
常見的解決形式:
1梳玫、通過category給類添加方法用來替換掉原本存在潛在崩潰的方法爹梁。
2、利用runtime方法交換技術(shù)提澎,將系統(tǒng)方法替換成我們給類添加的新方法姚垃。
3、利用異常的捕獲來防止程序的崩潰盼忌,并且進(jìn)行相應(yīng)的處理积糯。
以上就是項目中常見crash的簡單處理。更多的項目優(yōu)化可以學(xué)習(xí)iOS 如何優(yōu)化項目谦纱。感謝大神的分享看成,我只是一名知識搬運(yùn)工。