iOS 解决addObject`传入一个`nil`导致崩溃问题

本文链接：https://blog.csdn.net/huanglinxiao/article/details/90043385

本文介绍了一个封装类XTSafeCollection，用于防止在使用NSArray、NSMutableArray、NSDictionary和NSMutableDictionary时因传入nil或越界index而导致的崩溃。通过重写关键方法，确保在不安全操作发生时进行错误记录，提高应用稳定性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

我们经常在使用`NSArray``NSMutableArray``NSDictionary``NSMutableDictionary`的时候。我们可能会调用`addObject`传入一个`nil`, 也有可能是会`objectAtIndex`传入一个越界的index。尤其是在数据基本依赖于服务端返回的的情况，这种crash大幅增加。最近项目上经常出现`NSDictionary`的`setObject:forKey:`的`nil object`的崩溃。

下面是封装的一个类，直接在项目中创建这个类，并在setting设置中设置该文件 -fno-objc-arc 即可

//
//  XTSafeCollection.h
//  XTSafeCollection
//

#import <Foundation/Foundation.h>

@interface XTSafeCollection : NSObject

+ (void)setLogEnabled:(BOOL)enabled;

@end


//  XTSafeCollection.m
//  XTSafeCollection


#import "XTSafeCollection.h"

#import <objc/runtime.h>

#if __has_feature(objc_arc)

#error "Should disable arc (-fno-objc-arc)"

#endif

static BOOL logEnabled = NO;

#define XTSCLOG(...) safeCollectionLog(__VA_ARGS__)

void safeCollectionLog(NSString *fmt, ...) NS_FORMAT_FUNCTION(1, 2);

void safeCollectionLog(NSString *fmt, ...)

{

    if (!logEnabled)
    {

        return;

    }

    va_list ap;

    va_start(ap, fmt);

    NSString *content = [[NSString alloc] initWithFormat:fmt arguments:ap];

//    NSLog(@"%@", content);

    va_end(ap);

    

    NSLog(@" ============= call stack ========== \n%@", [NSThread callStackSymbols]);

}

#pragma mark - NSArray

@interface NSArray (XTSafe)

@end

@implementation NSArray (XTSafe)

+ (Method)methodOfSelector:(SEL)selector

{

    return class_getInstanceMethod(NSClassFromString(@"__NSArrayI"),selector);

}

- (id)xt_objectAtIndexI:(NSUInteger)index

{

    if (index >= self.count)

    {

        XTSCLOG(@"[%@ %@] index {%lu} beyond bounds [0...%lu]",

                NSStringFromClass([self class]),

                NSStringFromSelector(_cmd),

                (unsigned long)index,

                MAX((unsigned long)self.count - 1, 0));

        return nil;

    }

    

    return [self xt_objectAtIndexI:index];

}

+ (id)xt_arrayWithObjects:(const id [])objects count:(NSUInteger)cnt
{

    id validObjects[cnt];

    NSUInteger count = 0;

    for (NSUInteger i = 0; i < cnt; i++)

    {

        if (objects[i])

        {

            validObjects[count] = objects[i];

            count++;

        }

        else

        {

            XTSCLOG(@"[%@ %@] NIL object at index {%lu}",

                    NSStringFromClass([self class]),

                    NSStringFromSelector(_cmd),

                    (unsigned long)i);

            

        }

    }

    return [self xt_arrayWithObjects:validObjects count:count];

}

@end


#pragma mark - NSMutableArray

@interface NSMutableArray (XTSafe)

@end

@implementation NSMutableArray (XTSafe)

+ (Method)methodOfSelector:(SEL)selector

{

    return class_getInstanceMethod(NSClassFromString(@"__NSArrayM"),selector);

}

- (id)xt_objectAtIndexM:(NSUInteger)index

{

    if (index >= self.count)

    {

        XTSCLOG(@"[%@ %@] index {%lu} beyond bounds [0...%lu]",

                NSStringFromClass([self class]),

                NSStringFromSelector(_cmd),

                (unsigned long)index,

                MAX((unsigned long)self.count - 1, 0));

        return nil;

    }

    

    return [self xt_objectAtIndexM:index];

}

- (void)xt_addObject:(id)anObject

{

    if (!anObject)

    {

        XTSCLOG(@"[%@ %@], NIL object.", NSStringFromClass([self class]), NSStringFromSelector(_cmd));

        return;

    }

    [self xt_addObject:anObject];

}

- (void)xt_replaceObjectAtIndex:(NSUInteger)index withObject:(id)anObject

{

    if (index >= self.count)

    {

        XTSCLOG(@"[%@ %@] index {%lu} beyond bounds [0...%lu].",

                NSStringFromClass([self class]),

                NSStringFromSelector(_cmd),

                (unsigned long)index,

                MAX((unsigned long)self.count - 1, 0));

        return;

    }

    

    if (!anObject)

    {

        XTSCLOG(@"[%@ %@] NIL object.", NSStringFromClass([self class]), NSStringFromSelector(_cmd));

        return;

    }

    

    [self xt_replaceObjectAtIndex:index withObject:anObject];

}

- (void)xt_insertObject:(id)anObject atIndex:(NSUInteger)index

{

    if (index > self.count)

    {

        XTSCLOG(@"[%@ %@] index {%lu} beyond bounds [0...%lu].",

                NSStringFromClass([self class]),

                NSStringFromSelector(_cmd),

                (unsigned long)index,

                MAX((unsigned long)self.count - 1, 0));

        return;

    }

    

    if (!anObject)

    {

        XTSCLOG(@"[%@ %@] NIL object.", NSStringFromClass([self class]), NSStringFromSelector(_cmd));

        return;

    }

    

    [self xt_insertObject:anObject atIndex:index];

}

@end

#pragma mark - NSDictionary

@interface NSDictionary (XTSafe)

@end

@implementation NSDictionary (XTSafe)

+ (instancetype)xt_dictionaryWithObjects:(const id [])objects forKeys:(const id<NSCopying> [])keys count:(NSUInteger)cnt

{

    id validObjects[cnt];

    id<NSCopying> validKeys[cnt];

    NSUInteger count = 0;

    for (NSUInteger i = 0; i < cnt; i++)

    {

        if (objects[i] && keys[i])

        {

            validObjects[count] = objects[i];

            validKeys[count] = keys[i];

            count ++;

        }

        else

        {

            XTSCLOG(@"[%@ %@] NIL object or key at index{%lu}.",

                    NSStringFromClass(self),

                    NSStringFromSelector(_cmd),

                    (unsigned long)i);

        }

    }

    

    return [self xt_dictionaryWithObjects:validObjects forKeys:validKeys count:count];

}

@end

#pragma mark - NSMutableDictionary

@interface NSMutableDictionary (XTSafe)

@end

@implementation NSMutableDictionary (XTSafe)

+ (Method)methodOfSelector:(SEL)selector

{

    return class_getInstanceMethod(NSClassFromString(@"__NSDictionaryM"),selector);

}

- (void)xt_setObject:(id)anObject forKey:(id<NSCopying>)aKey

{

    if (!aKey)

    {

        XTSCLOG(@"[%@ %@] NIL key.", NSStringFromClass([self class]), NSStringFromSelector(_cmd));

        return;

    }

    if (!anObject)

    {

        XTSCLOG(@"[%@ %@] NIL object.", NSStringFromClass([self class]), NSStringFromSelector(_cmd));

        return;

    }

    

    [self xt_setObject:anObject forKey:aKey];

}



@end

#pragma mark - Mama

@implementation XTSafeCollection

+ (void)load

{

    static dispatch_once_t onceToken;

    dispatch_once(&onceToken, ^{

        // NSArray

        [self exchangeOriginalMethod:[NSArray methodOfSelector:@selector(objectAtIndex:)] withNewMethod:[NSArray methodOfSelector:@selector(xt_objectAtIndexI:)]];

        [self exchangeOriginalMethod:class_getClassMethod([NSArray class], @selector(arrayWithObjects:count:))

                       withNewMethod:class_getClassMethod([NSArray class], @selector(xt_arrayWithObjects:count:))];

        // NSMutableArray

        [self exchangeOriginalMethod:[NSMutableArray methodOfSelector:@selector(objectAtIndex:)] withNewMethod:[NSMutableArray methodOfSelector:@selector(xt_objectAtIndexM:)]];

        [self exchangeOriginalMethod:[NSMutableArray methodOfSelector:@selector(replaceObjectAtIndex:withObject:)] withNewMethod:[NSMutableArray methodOfSelector:@selector(xt_replaceObjectAtIndex:withObject:)]];

        [self exchangeOriginalMethod:[NSMutableArray methodOfSelector:@selector(addObject:)] withNewMethod:[NSMutableArray methodOfSelector:@selector(xt_addObject:)]];

        [self exchangeOriginalMethod:[NSMutableArray methodOfSelector:@selector(insertObject:atIndex:)] withNewMethod:[NSMutableArray methodOfSelector:@selector(xt_insertObject:atIndex:)]];

        

        // NSDictionary

        [self exchangeOriginalMethod:class_getClassMethod([NSDictionary class], @selector(dictionaryWithObjects:forKeys:count:))

                       withNewMethod:class_getClassMethod([NSDictionary class], @selector(xt_dictionaryWithObjects:forKeys:count:))];

        // NSMutableDictionary

        [self exchangeOriginalMethod:[NSMutableDictionary methodOfSelector:@selector(setObject:forKey:)] withNewMethod:[NSMutableDictionary methodOfSelector:@selector(xt_setObject:forKey:)]];

    });

}

+ (void)exchangeOriginalMethod:(Method)originalMethod withNewMethod:(Method)newMethod

{

    method_exchangeImplementations(originalMethod, newMethod);

}

+ (void)setLogEnabled:(BOOL)enabled

{

    logEnabled = enabled;

}

@end