iOS防止数组越界

可以直接hook数组的objectIndex方法进行判断,但是一定要记住,NSArray真正的类型是__NSArrayI

所以我们hook的时候直接用NSArray可能会取不到方法,所以我们要用objc_getClass("__NSArrayI")
来操作。

代码如下:

+ (void)load {

static dispatch_once_t onceToken;

dispatch_once(&onceToken, ^{

Method ori_Method1 =  class_getInstanceMethod(objc_getClass("__NSArrayI"), @selector(objectAtIndex:));

Method my_Method1 = class_getInstanceMethod(objc_getClass("__NSArrayI"), @selector(hook_objectAtIndex:));

method_exchangeImplementations(ori_Method1, my_Method1);

});

}

- (id)hook_objectAtIndex:(NSUInteger)index {

if (!self.count ||self.count-1 < index) {

@try {

return [self hook_objectAtIndex:index];

}

@catch (NSException *exception) {

NSLog(@"---------- %s Crash Because Method %s  ----------\n", class_getName(self.class), __func__);

NSLog(@"%@", [exception callStackSymbols]);

return nil;

}

@finally {}

} else {

return [self hook_objectAtIndex:index];

}

}

这时候以为万事大吉了,结果发现,其实NSArray的类簇中还有__NSArrayM是可变数组__NSArray0是一个空数组的时候会直接闪退,所以我们还要重新hook __NSArrayM __NSArray0的objectAtIndex.


你可能感兴趣的:(iOS防止数组越界)