有關Swizzling的一個問題

前言

最近在開發(fā)的時候遇到了一個Swizzling的問題,特別在此記錄,希望有相同遭遇的朋友能參考。

問題

問題概況:被Swizzling的方法在調用原有實現的時候提示“Selector無法被響應”。
問題過程:項目中A框架使用Aspects(一個幫助開發(fā)者實現快速Swizzling的框架)來替換UIViewController- (void)viewDidLoad方法,這個步驟是先被執(zhí)行的。之后的某個情況下,項目中B框架又使用自己寫的Swizzling方法替換了UIViewController- (void)viewDidLoad方法。這時我發(fā)現,當代碼運行到B框架的替換方法,并且要執(zhí)行原有的方法時,會報Selector無法被響應的錯。很奇怪,因為從B框架中Swizzling的實現上來看,會保存之前的IMP,所以應該能找到才對。

原因

于是我去看了Aspects的源碼,發(fā)現它的實現是這樣的(假如現在要Swizzling classA類的methodA方法):

  1. 在methodA被Swizzling的時候先看classA的-forwardInvocation:方法是否被替換,如果不是,就使用自己的-AspectsForwardInvocation:方法來替換classA的-forwardInvocation:。
  1. 將methodA的IMP指向_objc_msgForward,這是一個全局 IMP,OC 調用方法不存在時都會轉發(fā)到這個 IMP 上,這樣做了之后,當methodA被調用的時候,就會先進入-AspectsForwardInvocation:方法。
  2. 接下來我們看-AspectsForwardInvocation:的實現,這個方法會將NSInvocation的Selector拿出來,再拼上一個前綴(aspects_),然后檢查這個方法是否是已經被Aspects替換過,如果是,就查詢相關的實現并執(zhí)行,如果不是,就執(zhí)行原有的-forwardInvocation:方法來進行轉發(fā)。

這下問題就很清楚了,當Aspects的Swizzling方法先被執(zhí)行的時候,原方法Selector對應的IMP已經指向_objc_msgForward,所以當另一個框架再進行Swizzling的時候,存起來的原有實現就是這個“錯的”_objc_msgForward。那么當執(zhí)行完自己加的代碼后,想要再通過objc_msgSend執(zhí)行原有實現,就是會找不到,因為原有實現已經被替換為_objc_msgForward,而真的IMP由于被Aspects先Swizzling掉了,所以找不到!

具體Swizzling結果如下圖:

Aspects處理結果(先)
另一處理結果(后)

具體執(zhí)行邏輯如下圖:

執(zhí)行流程

解決辦法

依照上面的說法,當項目里面有類似Aspects這種Swizzling方式,而且它先于其他Swizzling方式執(zhí)行,且Swizzling了相同的代碼,那么這個問題就會發(fā)生了,目前知道的另一個這么做的庫是JSPatch(具體可以參考它Wiki中方法替換章節(jié)的第5小節(jié))。使用Aspects庫的人數我無法估計,但JSPatch的應該不少吧。所以給出合理的解決辦法還是很有必要的。

方案1:如果在你的Swizzling方法內部需要調用原有方法,那么在執(zhí)行原有方法的IMP之前先判斷一下,如果為_objc_msgForward(或_objc_msgForward_stret),那么就使用原有Selector拼成一個NSInvocation,再使用objc_msgSend執(zhí)行。使用原有Selector的原因是:在進入已被替換的-forwardInvocation:時,只有原有Selector可以幫助找到真實的實現。使用NSInvocation進行消息轉發(fā)的原因是:這樣可以走到-forwardInvocation:方法。這種方法的缺點是需要在每個地方都做處理,而且之前的判斷也比較特殊。

方案2:參考Aspects或JSPatch相關代碼,將-forwardInvocation:也進行Swizzling,在自己的-forwardInvocation:方法中進行同樣的操作,就是判斷傳入的NSInvocation的Selector,如果是自己可以識別的Selector,那么就將Selector變?yōu)樵蠸elector在執(zhí)行,如果不識別,就直接轉發(fā)。(這也是為什么我們項目中Aspects和JSPatch不沖突的原因,因為都將被Swizzling的方法指向了_objc_msgForward(或_objc_msgForward_stret),然后再監(jiān)控-forwardInvocation:方法,使得方法最終通過原有Selector經消息轉發(fā)流程得到正確的實現)。

最后

歡迎討論,歡迎指出問題。

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
【社區(qū)內容提示】社區(qū)部分內容疑似由AI輔助生成,瀏覽時請結合常識與多方信息審慎甄別。
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發(fā)布,文章內容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

相關閱讀更多精彩內容

友情鏈接更多精彩內容