前言

上一篇簡單介紹了一下通過運行時添加方法。這篇文章呢，側重于方法的hook。

Method Swizzling 原理

在Objective-C中調用一個方法，其實是向一個對象發送消息，查找消息的唯一依據是selector的名字。利用Objective-C的動態特性，可以實現在運行時偷換selector對應的方法實現。每個類都有一個方法列表，存放著selector的名字和方法實現的映射關系。IMP有點類似函數指針，指向具體的Method實現。

1.png

利用運行時的一些方法，我們可以做到對系統原有方法的hook,也叫method swizzling。

2.png

實例舉例

最近在項目中接入了友盟的數據統計，其中有一個功能，可以統計app各個頁面的打開次數，時間等。

實現頁面的統計需要在每個View中配對調用如下方法：
- (void)viewWillAppear:(BOOL)animated { 
[super viewWillAppear:animated]; 
[MobClick beginLogPageView:@"PageOne"];//("PageOne"為頁面名稱，可自定義) 
}
- (void)viewWillDisappear:(BOOL)animated { 
[super viewWillDisappear:animated]; 
[MobClick endLogPageView:@"PageOne"];
 }

就像文檔描述的，我們需要在生命周期函數中添加語句，但由于目前項目已經基本穩定，因此一個頁面一個頁面的改，難免費時費力。于是考慮AOP，直接hook掉這些函數。

//demo
#import <objc/runtime.h>
 
@implementation UIViewController (Tracking)
 
+ (void)load {
        static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        Class class = [self class];         
        // When swizzling a class method, use the following:
       // Class class = object_getClass((id)self);
 
        SEL originalSelector = @selector(viewWillAppear:);
                    SEL swizzledSelector = @selector(xxx_viewWillAppear:);
 
        Method originalMethod = class_getInstanceMethod(class, originalSelector);
                    Method swizzledMethod = class_getInstanceMethod(class, swizzledSelector);
 
        BOOL didAddMethod =
                        class_addMethod(class,
                originalSelector,
                method_getImplementation(swizzledMethod),
                method_getTypeEncoding(swizzledMethod));
 
        if (didAddMethod) {
                        class_replaceMethod(class,
                swizzledSelector,
                method_getImplementation(originalMethod),
                method_getTypeEncoding(originalMethod));
        } else {
            method_exchangeImplementations(originalMethod, swizzledMethod);
        }
    });
}
 
#pragma mark - Method Swizzling
 
- (void)xxx_viewWillAppear:(BOOL)animated {
        [self xxx_viewWillAppear:animated];
    NSLog(@"viewWillAppear: %@", self);
}
 
@end

選擇器、方法與實現

在Objective-C中，選擇器(selector)、方法(method)和實現(implementation)是運行時中一個特殊點，雖然在一般情況下，這些術語更多的是用在消息發送的過程描述中。

以下是Objective-C Runtime Reference中的對這幾個術語一些描述：

Selector(typedef struct objc_selector *SEL)：用于在運行時中表示一個方法的名稱。一個方法選擇器是一個C字符串，它是在Objective-C運行時被注冊的。選擇器由編譯器生成，并且在類被加載時由運行時自動做映射操作。
Method(typedef struct objc_method Method)：在類定義中表示方法的類型
Implementation(typedef id (IMP)(id, SEL, …))：這是一個指針類型，指向方法實現函數的開始位置。這個函數使用為當前CPU架構實現的標準C調用規范。第一個參數是指向對象自身的指針(self)，第二個參數是方法選擇器。然后是方法的實際參數。
理解這幾個術語之間的關系最好的方式是：一個類維護一個運行時可接收的消息分發表；分發表中的每個入口是一個方法(Method)，其中key是一個特定名稱，即選擇器(SEL)，其對應一個實現(IMP)，即指向底層C函數的指針。

為了swizzle一個方法，我們可以在分發表中將一個方法的現有的選擇器映射到不同的實現，而將該選擇器對應的原始實現關聯到一個新的選擇器中。

參考

Objective-C Runtime 運行時之四：Method Swizzling
iOS黑魔法－Method Swizzling