自己动手写一个野指针拦截工具的基础原理

刷微博看到大佬分析评价了一个库对野指针拦截处理,通读之后若有所思,随即点了收藏。

大概过了两周,重新翻看这条分析,发现自己并没有完整理解消化了这个野指针拦截原理,之前收藏的时候评论了原博,希望博主贴个源码地址学习一下~然而大佬并没有回复,于是自己试着用截图里的keyword去搜了下,最终还是找到了--> JJException

这个库的野指针拦截处理的原理其实就是模仿Xcode的僵尸对象捕获的功能(即Zombie)

Zombie.png

套用大佬的话:hook了dealloc,然后对该对象的内存做个废弃的标记但是不释放,之后替换该对象所属的类,对替换的类添加一个通用的方法识别,只要有调用任何一个方法就属于野指针,日志上报即可。

那么,我们从代码层面看一下是具体是怎么实现的:

@implementation NSObject (ZombieHook)
+ (void)jj_swizzleZombie{
    [self jj_swizzleInstanceMethod:@selector(dealloc) withSwizzleMethod:@selector(hookDealloc)];
}

- (void)hookDealloc{
    Class currentClass = self.class;
    //Check black list(如果该类在黑名单中,则还是走原有dealloc方法)
    if (![[[JJExceptionProxy shareExceptionProxy] blackClassesSet] containsObject:currentClass]) {
        [self hookDealloc];
        return;
    }
    //Check the array max size
    if ([JJExceptionProxy shareExceptionProxy].currentClassSize > MAX_ARRAY_SIZE) {
        id object = [[JJExceptionProxy shareExceptionProxy] objectFromCurrentClassesSet];
        [[JJExceptionProxy shareExceptionProxy] removeCurrentZombieClass:object_getClass(object)];
        object?free(object):nil;
    }
    objc_destructInstance(self);//注1:销毁该对象的引用关系
    object_setClass(self, [JJZombieSub class]);//注2:替换该对象所属的类
    [[JJExceptionProxy shareExceptionProxy] addCurrentZombieClass:currentClass];
}
@end

首先用Category的方式Hook NSObject的dealloc方法,野指针发生的前提是:这个对象的内存已经被系统回收了,对象调用方法时指向的是一个已经被回收了的,无效的内存地址。对象被回收必走dealloc方法,因此Hook这个方法是一个很好的切入点。

接着通过runtime替换该对象所属的类,注意这是一个通用的类(JJZombieSub)。所有的野指针方法的Class都会被替换成JJZombieSub。

替换该对象所属的类干啥?接下来我们把目光移到JJZombieSub这个替代类。

@interface JJZombieSub : NSObject
@end

@implementation JJZombieSub
- (id)forwardingTargetForSelector:(SEL)selector{
    NSMethodSignature* sign = [self methodSignatureForSelector:selector];
    if (!sign) {//注3:获取所调方法的签名,如果签名不存在则说明该Target下没有这个selector。
        id stub = [[ZombieSelectorHandle new] autorelease];
        [stub setFromObject:self];//注4:生成一个后续处理的对象,并绑定原类的信息,用于上传日志时标记是哪个类出现了野指针。
        class_addMethod([stub class], selector, (IMP)unrecognizedSelectorZombie, "v@:");
        return stub;//注5:将所调方法添加到后续处理类上ZombieSelectorHandle,并替换该方法的IMP(即最终执行unrecognizedSelectorZombie方法)
    }
    return [super forwardingTargetForSelector:selector];
}
@end

这个通用替代类里重写了forwardingTargetForSelector:方法,这个方法属于消息转发机制里的第二步,重写这个方法可以动态的去替换所调方法的Target。

这里重写这个方法就是为了将野指针的方法重定向到后续处理对象上执行。

消息转发三部曲
@interface ZombieSelectorHandle : NSObject
@property(nonatomic,readwrite,assign)id fromObject;
@end

@implementation ZombieSelectorHandle
void unrecognizedSelectorZombie(ZombieSelectorHandle* self, SEL _cmd){
}
@end

这个ZombieSelectorHandle就是最终要处理野指针、上传日志的类。unrecognizedSelectorZombie方法有两个参数:后续处理对象和野指针方法的SEL,后续处理对象的fromObject属性即是野指针方法所属的类。

拿到了发生野指针的类名和方法名,在unrecognizedSelectorZombie方法中去上传日志,以便于后续的排查。

代码层面的流程就出来了:

-->Hook NSObject的dealloc方法
-->将野指针发生的类替换为JJZombieSub通用类
-->消息转发时重定向野指针方法的类为ZombieSelectorHandle处理类、同时替换野指针发生的方法为通用的unrecognizedSelectorZombie方法。
--> ZombieSelectorHandle对象的unrecognizedSelectorZombie进行日志上传。
为什么作者要转换两次类JJZombieSub、ZombieSelectorHandle而不是直接全部在JJZombieSub中处理?

我猜测应该是基于设计层面去考虑的,JJZombieSub只负责拦截、ZombieSelectorHandle负责后续处理,互不干扰各司其职。

一点思考:

在看到大佬的那条微博前,自己用过Zombie,但是基本没去考虑过系统的Zombie是怎么去处理的,对原理的好奇心还远远不够,好奇心是进步的源动力啊,fighting~

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 194,242评论 5 459
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 81,769评论 2 371
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 141,484评论 0 319
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 52,133评论 1 263
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 61,007评论 4 355
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 46,080评论 1 272
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 36,496评论 3 381
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 35,190评论 0 253
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 39,464评论 1 290
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 34,549评论 2 309
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 36,330评论 1 326
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,205评论 3 312
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 37,567评论 3 298
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 28,889评论 0 17
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,160评论 1 250
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 41,475评论 2 341
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 40,650评论 2 335

推荐阅读更多精彩内容