iOS利用runtime技术实现容器的越界和nil保护
原因
我们经常会因为添加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
,以后会添加其他容器的支持
Hi! I am a robot. I just upvoted you! I found similar content that readers might be interested in:
http://www.itwendao.com/article/detail/24311.html