前言
最近在開發(fā)的時候遇到了一個Swizzling的問題,特別在此記錄,希望有相同遭遇的朋友能參考犁功。
問題
問題概況:被Swizzling的方法在調(diào)用原有實現(xiàn)的時候提示“Selector無法被響應(yīng)”案糙。
問題過程:項目中A框架使用Aspects(一個幫助開發(fā)者實現(xiàn)快速Swizzling的框架)來替換UIViewController
的- (void)viewDidLoad
方法,這個步驟是先被執(zhí)行的稚叹。之后的某個情況下,項目中B框架又使用自己寫的Swizzling方法替換了UIViewController
的- (void)viewDidLoad
方法。這時我發(fā)現(xiàn)蚀同,當代碼運行到B框架的替換方法迟蜜,并且要執(zhí)行原有的方法時髓霞,會報Selector無法被響應(yīng)的錯。很奇怪徐鹤,因為從B框架中Swizzling的實現(xiàn)上來看寥院,會保存之前的IMP只磷,所以應(yīng)該能找到才對。
原因
于是我去看了Aspects的源碼轧叽,發(fā)現(xiàn)它的實現(xiàn)是這樣的(假如現(xiàn)在要Swizzling classA類的methodA方法):
- 在methodA被Swizzling的時候先看classA的
-forwardInvocation:
方法是否被替換甥角,如果不是,就使用自己的-AspectsForwardInvocation:
方法來替換classA的-forwardInvocation:
垢村。
- 將methodA的IMP指向
_objc_msgForward
,這是一個全局 IMP,OC 調(diào)用方法不存在時都會轉(zhuǎn)發(fā)到這個 IMP 上胚宦,這樣做了之后井联,當methodA被調(diào)用的時候鹤盒,就會先進入-AspectsForwardInvocation:
方法。 - 接下來我們看
-AspectsForwardInvocation:
的實現(xiàn)挣棕,這個方法會將NSInvocation
的Selector拿出來词身,再拼上一個前綴(aspects_),然后檢查這個方法是否是已經(jīng)被Aspects替換過,如果是墓塌,就查詢相關(guān)的實現(xiàn)并執(zhí)行,如果不是,就執(zhí)行原有的-forwardInvocation:
方法來進行轉(zhuǎn)發(fā)涡相。
這下問題就很清楚了育特,當Aspects的Swizzling方法先被執(zhí)行的時候,原方法Selector對應(yīng)的IMP已經(jīng)指向_objc_msgForward
,所以當另一個框架再進行Swizzling的時候乾戏,存起來的原有實現(xiàn)就是這個“錯的”_objc_msgForward
就漾。那么當執(zhí)行完自己加的代碼后,想要再通過objc_msgSend
執(zhí)行原有實現(xiàn),就是會找不到象踊,因為原有實現(xiàn)已經(jīng)被替換為_objc_msgForward
,而真的IMP由于被Aspects先Swizzling掉了魂务,所以找不到相艇!
具體Swizzling結(jié)果如下圖:
具體執(zhí)行邏輯如下圖:
解決辦法
依照上面的說法阴颖,當項目里面有類似Aspects這種Swizzling方式帅矗,而且它先于其他Swizzling方式執(zhí)行,且Swizzling了相同的代碼紊馏,那么這個問題就會發(fā)生了原叮,目前知道的另一個這么做的庫是JSPatch(具體可以參考它Wiki中方法替換章節(jié)的第5小節(jié))赡若。使用Aspects庫的人數(shù)我無法估計躺苦,但JSPatch的應(yīng)該不少吧。所以給出合理的解決辦法還是很有必要的她按。
方案1:如果在你的Swizzling方法內(nèi)部需要調(diào)用原有方法匕累,那么在執(zhí)行原有方法的IMP之前先判斷一下,如果為_objc_msgForward
(或_objc_msgForward_stret
),那么就使用原有Selector拼成一個NSInvocation
,再使用objc_msgSend
執(zhí)行斗搞。使用原有Selector的原因是:在進入已被替換的-forwardInvocation:
時膝擂,只有原有Selector可以幫助找到真實的實現(xiàn)全闷。使用NSInvocation
進行消息轉(zhuǎn)發(fā)的原因是:這樣可以走到-forwardInvocation:
方法勘纯。這種方法的缺點是需要在每個地方都做處理,而且之前的判斷也比較特殊。
方案2:參考Aspects或JSPatch相關(guān)代碼媒惕,將-forwardInvocation:
也進行Swizzling月弛,在自己的-forwardInvocation:
方法中進行同樣的操作贞绵,就是判斷傳入的NSInvocation
的Selector,如果是自己可以識別的Selector,那么就將Selector變?yōu)樵蠸elector在執(zhí)行蚪缀,如果不識別,就直接轉(zhuǎn)發(fā)。(這也是為什么我們項目中Aspects和JSPatch不沖突的原因,因為都將被Swizzling的方法指向了_objc_msgForward
(或_objc_msgForward_stret
)抒线,然后再監(jiān)控-forwardInvocation:
方法眨猎,使得方法最終通過原有Selector經(jīng)消息轉(zhuǎn)發(fā)流程得到正確的實現(xiàn))匿情。
最后
歡迎討論,歡迎指出問題据德。