从NullSafe源码分析理解消息转发

前言

有时候无法避免地,服务器会传回一些脏数据,但这个锅我们自己得接好。

场景

比如我们接收的是一个NSMutableDictionary ,传回来的却是一个NSNull,在调用方法的时候必然会奔溃,因为NSMutableDictionary 的这个方法我NSNull 类里没有实现。可以模拟情况如下:

    NSMutableDictionary * nullDic = [NSNull null];
    [nullDic objectForKey:@"123"];

可以简单看一下nilNSNull 的区别

null nil.png

当我们给一个NSNull对象发送消息的话,可能会崩溃(NSNull是有内存的),而发送给nil的话,是不会崩溃的。
我们也可以看一下objc_msgSend的源代码,如果传递给 objc_msgSendself 参数是 nil ,该函数不会执行有意义的操作,直接返回。不会走到消息转发。

所以解决问题的一个思路就是把 NSNull 替换成 nil

怎么替换?如果是我来写的话。

我可能也会写一个NULL(safeCategory) 直接在消息转发的第三步只会返回一个自定义的方法签名,然后再把 target = nil

- (void)forwardInvocation:(NSInvocation *)anInvocation {
    anInvocation.target = nil;
    [anInvocation invoke];
}

- (NSMethodSignature *)methodSignatureForSelector:(SEL)aSelector {
    NSMethodSignature *Signature = [super methodSignatureForSelector:aSelector];
    if (!signature) {
      signature = [NSMethodSignature signatureWithObjCTypes:"^v^c"];
}

      return signature;
}

NullSafe的实现

但作者在他的issue里写了这么一段话,原本应该返回的方法签名返回值和实际的方法签名需要保持一致

What about the return type? If the expected return type is a struct (> 8 bytes), and you return an integer, couldn't that corrupt the stack?

issue

这里我有些存疑,我既然生成了新的方法签名我怎么还要和之前的返回类型保持一致。我觉得可能是作者想错了。
但我想可能还有些其他原因,如果像我这么写,performselector一个未知的方法都不会走到崩溃
所以作者的写法会更复杂一些

作者把发送给NSNull的而NSNull又无法处理的消息经过如下几步处理:
1 创建缓存,缓存项目中类的所有类名
2 遍历缓存,寻找是否已经有可以执行此方法的类
3 如果有的话,返回这个NSMethodSignature
4 如果没有的话,返回nil奔溃
5 如果有的话,[invocation invokeWithTarget:nil];将消息转发给nil

我解释一下。1的代码就如下cacheSignatures函数,没什么好说的,实现中规中矩(一般几w个类还会有性能瓶颈)。将所有的类加到了一个缓存字典中,除了performselector的方法都应该能找到对应归属的类

static void cacheSignatures()
{
    
}
// check implementation cache first
            NSString *selectorString = NSStringFromSelector(selector);
            signature = signatureCache[selectorString];
            if (!signature)
            {
                for (Class someClass in classList)
                {
                    // 判断 这个基于NSObject类的子类是否能够响应传入的方法
                    if ([someClass instancesRespondToSelector:selector])
                    {
                        // someClass类能够响应selector方法
                        // 返回NSMethodSignature对象,这个对象包含被标示的实例方法的描述。
                        signature = [someClass instanceMethodSignatureForSelector:selector];
                        break;
                    }
                }
                
                // cache for next time
                signatureCache[selectorString] = signature ?: [NSNull null];
            }
            else if ([signature isKindOfClass:[NSNull class]])
            {
                // 永远不会走到这里
                signature = nil;
            }

第二步第三步如上,之所以我要找到对应的类就是为了 signature = [someClass instanceMethodSignatureForSelector:selector]通过类去获得正确的方法签名。

会出现第四步的是情况很少见,一般在所有类能找到对应的方法签名,否则编译都无法通过,除了performselector的方法。
这里的Crash作者没有去做避免,也不应该在这里去做避免。
这个时候就走向Crash了, 所以即使缓存了signatureNSNull
下次也不会走到这个判断 ([signature isKindOfClass:[NSNull class]]),(存的是内存,我需要重新启动APP

signature返回nil为什么奔溃?

这里就需要讲一下消息转发了,消息进入消息转发后有三步
1 + (BOOL)resolveInstanceMethod:(SEL)sel 或者 Class
2 - (id)forwardingTargetForSelector:(SEL)aSelector
3 - (NSMethodSignature *)methodSignatureForSelector:(SEL)aSelector
- (void)forwardInvocation:(NSInvocation *)anInvocation

刚接触消息转发的时候我困惑为什么消息转发要设计成3步,明明任何一步都能处理这个本来无法处理的消息
不知道读者会不会和我有一样的困惑
其实这里每一步做的事情都是不一样的
1 给原本的实例或者类一次增加方法的机会
2 将原本的方法子交给其他的类实现
3 第三步能做的事情就更多了,NSMethodSignaturetaget 我都可以自定义,很多黑魔法的实现都依赖于第三步,比如JSPatch

在第三步中的2个方法,如果第一步的NSMethodSignature返回了nil,那么forwardInvocation就不会再走了,直接返回无法处理该方法。

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

推荐阅读更多精彩内容