如何為一個實(shí)例動態(tài)替換方法

這個 Tip 來源于一道面試題,感覺很是考察知識變通的能力,對 KVO 深入了解的同學(xué),應(yīng)該很容易就可以答出來。這里拋磚引玉,簡單聊聊這個 Tip

首先簡單總結(jié)下 KVO 的大概原理

  • 當(dāng)你觀察一個對象時(shí),會動態(tài)的創(chuàng)建該對象類的子類,這個子類重寫了被觀察屬性的 setter 方法,同時(shí)將該對象的 isa 指針指向了新創(chuàng)建的子類。在 Objective-C 中對象是通過 isa 指針來查找對應(yīng)類中的方法列表的,所以這里可以把該對象看為新子類的實(shí)例對象。重寫的 setter 方法會在調(diào)用原 setter 方法之后,通知觀察者對象屬性值的更改。

好的,下面進(jìn)入正題,聊聊如何為一個實(shí)例動態(tài)替換方法。

首先創(chuàng)建一個初始化工程,直接對 ViewController 進(jìn)行實(shí)戰(zhàn)即可,在 ViewController 中加入一個 eat 方法,如下

- (void)viewDidLoad {
    [super viewDidLoad];

    self.view.backgroundColor = [UIColor redColor];
}

- (void)eat {
    NSLog(@"original eat");
}

然后寫一個 NSObject 的 Category 負(fù)責(zé)進(jìn)行方法交換,將原對象的 isa 指針指向該對象類的子類(LDSubclass),并在子類中重寫 eat 方法

@implementation NSObject (Hook)

+ (void)hookWithInstance:(id)instance method:(SEL)selector {
 
    Method originMethod = class_getInstanceMethod([instance class], selector);
    if (!originMethod) {
        // exception ..
    }
    
    Class newClass = [LDSubclass class];
    
    // 修改 isa 指針的指向
    object_setClass(instance, newClass);

}

@end

子類的代碼很簡單,就是重寫 eat 方法,如果有需要,可以調(diào)用原方法的實(shí)現(xiàn)

@implementation LDSubclass

- (void)eat {
    NSLog(@"newSubClass eat");
    
    struct objc_super superClazz = {
        .receiver = self,
        .super_class = class_getSuperclass(object_getClass(self))
    };
    
    // 調(diào)用原方法
    void (*objc_msgSendSuperCasted)(void *, SEL) = (void *)objc_msgSendSuper;
    
    objc_msgSendSuperCasted(&superClazz, _cmd);
}
@end

最后在 ViewControlller 中進(jìn)行測試即可,此時(shí)的 ViewController 代碼如下

@implementation ViewController

- (void)viewDidLoad {
    [super viewDidLoad];

    self.view.backgroundColor = [UIColor redColor];
    
    ViewController * vc = [[ViewController alloc] init];

    [vc eat];
    
    NSLog(@"-----------");
    
    ViewController * hookedInstance= [[ViewController alloc] init];
    
    [ViewController hookWithInstance:hookedInstance method:@selector(eat)];
    
    [hookedInstance eat];
}

- (void)eat {
    NSLog(@"original eat");
}
@end

來看看打印的結(jié)果,第一個沒有 Hook 的實(shí)例,正常執(zhí)行;第二個Hook 后的實(shí)例,先執(zhí)行重寫的方法,后執(zhí)行原方法。

2017-03-20 14:30:21.244 JYHookInstanceMethod[91153:3422584] original eat
2017-03-20 14:30:21.244 JYHookInstanceMethod[91153:3422584] -----------
2017-03-20 14:30:21.245 JYHookInstanceMethod[91153:3422584] newSubClass eat
2017-03-20 14:30:21.245 JYHookInstanceMethod[91153:3422584] original eat

原理

與 KVO 的 isa-swizzling 思路相同,對想要 Hook 實(shí)例的類創(chuàng)建一個子類,并在子類中重寫想要 Hook 的方法,將該實(shí)例的 isa 指針指向子類,這樣進(jìn)行方法查找時(shí),便會在子類方法列表進(jìn)行查找,如果想要執(zhí)行更多操作,可以在替換后的新方法中加入自己的邏輯。

這里只是一個超級簡單的 Demo,很多邊界情況沒有考慮,后期可以自己完善,Demo 可以參考JYHookInstanceMethod

Reference

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 228,546評論 6 533
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機(jī),發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 98,570評論 3 418
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 176,505評論 0 376
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經(jīng)常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 63,017評論 1 313
  • 正文 為了忘掉前任,我火速辦了婚禮,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 71,786評論 6 410
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 55,219評論 1 324
  • 那天,我揣著相機(jī)與錄音,去河邊找鬼。 笑死,一個胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,287評論 3 441
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 42,438評論 0 288
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 48,971評論 1 335
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 40,796評論 3 354
  • 正文 我和宋清朗相戀三年,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點(diǎn)故事閱讀 42,995評論 1 369
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,540評論 5 359
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 44,230評論 3 347
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,662評論 0 26
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 35,918評論 1 286
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 51,697評論 3 392
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 47,991評論 2 374

推薦閱讀更多精彩內(nèi)容