iOS利用runtime技术实现容器的越界和nil保护

姓名的连嘉玮 学号 16040120089

转自：http://www.jianshu.com/p/e124031eb997?utm_campaign=hugo&utm_medium=reader_share&utm_content=note&utm_source=qq

有删节

【嵌牛导读】：iOS利用runtime技术实现容器的越界和nil保护

【嵌牛鼻子】：类介绍，mrc

【嵌牛提问】：为什么开发过程中总遇到错误？

【嵌牛正文】：

原因

在实际的项目中，经常会遇到这样的错误 ：

1.png

2.png

3.png

他们分别是因为添加nil到NSDictionary，添加nil到NSArray，数组越界。为了防止这些因为代码疏忽而造成的崩溃，所以写了WQContainerSafe这个类。

类介绍

这文件一定要在工程中弄成mrc的。不然会有这样一个bug 在弹出键盘后home出程序就会crash，并且最后报[UIKeyboardLayoutStar release]这样一个错误。

重要的事情说三遍：

这文件一定要在工程中弄成mrc的。

这文件一定要在工程中弄成mrc的。

这文件一定要在工程中弄成mrc的。

WQContainerSafe这个类只有WQContainerSafe.m这一个文件。它并不会要求在其他代码的地方引用，调用它，只需要把该文件加到工程里面即可。原因就是：

+(void) load {

  //code

}

load函数是在app装载的时候调用。（NSObject的load和initialize方法可以google一下这两个的区别）

一个类的+load方法在其父类的+load方法后调用

一个Category的+load方法在被其扩展的类的自有+load方法后调用

所以在WQContainerSafe.m文件里面我们实现了NSArray和NSMutableDictionary的load方法。然后利用runtime中method_exchangeImplementations方法来替换原有的objectAtIndex、insertObject:atIndex:、setObject:forKey:方法。在新的方法里面写上越界保护，nil保护。

我们拿NSMutableDictionary为例：

//在load方法中调用方法交换的函数。这样代码就为非侵入式的。默默的就自己贡献了力量。:-)

+(void)load{

      [self swizze];

}

//交换方法，把[setObject:forKey:]的调用指向我们自己实现的函数[setSaveObject:forKey:]

+(void)swizze

{

    Method old = class_getInstanceMethod(NSClassFromString(@"__NSDictionaryM"), @selector(setObject:forKey:));

    Method new = class_getInstanceMethod(self, @selector(setSaveObject:forKey:));

    if (!old || !new) {

        return;

    }

    method_exchangeImplementations(old, new);

}

//实现安全的setObject:forKey:函数。添加了nil保护。这里之所以是[setSaveObject:forKey:]调用自己。

//是因为方法交换了以后，这里实际上是调用原生的[setObject:forKey:]

-(void)setSaveObject:(id)value forKey:(id)key

{

    if (value!=nil) {

        [self setSaveObject:value forKey:key];

    }

    else{

        NSAssert(NO, @"WQContainerSafe: setObject:nil");

    }

}

遇到的坑

用方法交换做NSArray的越界保护，要交换3个类的objectAtIndex:方法： __NSArrayM (可变数组) __NSArrayI (一般数组) __NSArray0 (一般的空数组) ,而且不能用NSArrayI的objectiveAtIndex方法替换NSArrayM的。

NSArray的addObject方法最终会走到insertObject:atIndex:这个方法，所以只用替换后者就可以了。

TODO

现在只支持了NSArray和NSDictionary，以后会添加其他容器的支持

Demo地址：

https://github.com/decemberLi/WQContainerSafe