前言
上一篇簡單介紹了一下通過運行時添加方法延欠。這篇文章呢，側(cè)重于方法的hook。
Method Swizzling 原理
在Objective-C中調(diào)用一個方法撮执，其實是向一個對象發(fā)送消息薄翅，查找消息的唯一依據(jù)是selector的名字沙兰。利用Objective-C的動態(tài)特性氓奈，可以實現(xiàn)在運行時偷換selector對應(yīng)的方法實現(xiàn)。每個類都有一個方法列表鼎天，存放著selector的名字和方法實現(xiàn)的映射關(guān)系舀奶。IMP有點類似函數(shù)指針，指向具體的Method實現(xiàn)斋射。
利用運行時的一些方法育勺，我們可以做到對系統(tǒng)原有方法的hook,也叫method swizzling。
實例舉例
最近在項目中接入了友盟的數(shù)據(jù)統(tǒng)計罗岖，其中有一個功能涧至，可以統(tǒng)計app各個頁面的打開次數(shù)，時間等呀闻。
實現(xiàn)頁面的統(tǒng)計需要在每個View中配對調(diào)用如下方法：
- (void)viewWillAppear:(BOOL)animated {
[super viewWillAppear:animated];
[MobClick beginLogPageView:@"PageOne"];//("PageOne"為頁面名稱化借，可自定義)
}
- (void)viewWillDisappear:(BOOL)animated {
[super viewWillDisappear:animated];
[MobClick endLogPageView:@"PageOne"];
}
就像文檔描述的，我們需要在生命周期函數(shù)中添加語句捡多，但由于目前項目已經(jīng)基本穩(wěn)定蓖康，因此一個頁面一個頁面的改，難免費時費力垒手。于是考慮AOP蒜焊，直接hook掉這些函數(shù)。
//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
選擇器科贬、方法與實現(xiàn)
在Objective-C中泳梆，選擇器(selector)、方法(method)和實現(xiàn)(implementation)是運行時中一個特殊點榜掌，雖然在一般情況下优妙，這些術(shù)語更多的是用在消息發(fā)送的過程描述中。
以下是Objective-C Runtime Reference中的對這幾個術(shù)語一些描述：
Selector(typedef struct objc_selector *SEL)：用于在運行時中表示一個方法的名稱憎账。一個方法選擇器是一個C字符串套硼，它是在Objective-C運行時被注冊的。選擇器由編譯器生成胞皱，并且在類被加載時由運行時自動做映射操作邪意。
Method(typedef struct objc_method Method)：在類定義中表示方法的類型
Implementation(typedef id (IMP)(id, SEL, …))：這是一個指針類型，指向方法實現(xiàn)函數(shù)的開始位置反砌。這個函數(shù)使用為當前CPU架構(gòu)實現(xiàn)的標準C調(diào)用規(guī)范雾鬼。第一個參數(shù)是指向?qū)ο笞陨淼闹羔?self)，第二個參數(shù)是方法選擇器宴树。然后是方法的實際參數(shù)策菜。
理解這幾個術(shù)語之間的關(guān)系最好的方式是：一個類維護一個運行時可接收的消息分發(fā)表；分發(fā)表中的每個入口是一個方法(Method)，其中key是一個特定名稱做入，即選擇器(SEL)冒晰，其對應(yīng)一個實現(xiàn)(IMP)同衣，即指向底層C函數(shù)的指針竟块。
為了swizzle一個方法，我們可以在分發(fā)表中將一個方法的現(xiàn)有的選擇器映射到不同的實現(xiàn)耐齐，而將該選擇器對應(yīng)的原始實現(xiàn)關(guān)聯(lián)到一個新的選擇器中浪秘。
參考
Objective-C Runtime 運行時之四：Method Swizzling
iOS黑魔法－Method Swizzling
