iOS App 無代碼入侵的方法hook

繼續Objective-C runtime的研究

最近公司項目在做用戶行為分析

於是App端在某些頁面切換,交互操作的時候需要給統計系統發送一條消息

在幾十個Controller 的項目里,一個一個地加代碼那完全是不可能的,維護起來也是吃力

但這里需要處理的是 Controller, 可以有以下方式實現上述需求

1. 利用Objective-C 中的對象繼承

　　繼承 在面向對象開發中是非常常用的,像我們現在做的項目工程中都會有一個BaseViewController,

所有新建的ViewController都繼承BaseViewController,通過往BaseViewController中添加一些公共方法\屬性 可以被他們的子類所調用

這是統一我們工程中所有視圖控制器樣式的一個主要途徑

 

2.利用Category 和Runtime實行方法hook 

　　hook方案有一個好處,就是可以避免代碼入侵,做到更加廣泛的通用性.通過swizzling我們可以將原method與自己加入的method相結合,

即不需要在原有工程中加入代碼,又能做到全局覆蓋

 

兩種方案對比:

　　通過繼承父類來實現 相對於hook來說 是較為准確的,因為需要被統計的頁面都是繼承於這個父類的控制器,而其他的如UINavigationController,系統自帶的UIAlertController等則不會誤入統計數據當中

　　上面提到 hook方案是通過hook UIViewController viewdidload/viewdidappear等方法,而這些方法實際上 每個Controller 都會調用,那么就會出現不該出現的Controller 也出現在這里(如上面說到的UINavigationController和UIAlertController).但hook方案一個比較好的特點是無代碼入侵,在不修改項目代碼的前提下完成工作.

考慮到 行為分析統計系統 有可能被公司其他項目中所應用,這里采用hook方案.那么當中必然會出現 不該統計的卻被統計 的情況,后面再作分析.

 

既然用到hook方案,又要用runtime 的swizzling

首先 新建一個UIViewController 的category

實現swizzling代碼

+ (void)load{
    [super load];
    
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        // 假如要打開controller的統計 ,則把下面這行代碼打開
        __gbh_tracer_swizzleMethod([self class], @selector(viewDidAppear:), @selector(__gbh_tracer_viewDidAppear:));
    });
}

嗯,看到這里大家會發現 這里調用的是一個C的方法,然而這個C方法是怎么實現的呢?看下面

 

void __gbh_tracer_swizzleMethod(Class class, SEL originalSelector, SEL swizzledSelector){
    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);
    }
}

這是一個標准的swizzling寫法,當然了 github上面也有關於swizzling的開源庫,用起來也順手 這里就不多說

 

看回第一塊代碼,紅色的viewDidAppear是即將被我hook的方法,__gbh_tracer_viewDidAppear 則是我需要實現的方法

- (void)__gbh_tracer_viewDidAppear:(BOOL)animated{
    [self __gbh_tracer_viewDidAppear:animated];  //由於方法已經被交換,這里調用的實際上是viewDidAppear:方法
    
　　 //設置不允許發送數據的Controller
    NSArray *filter = @[@"UINavigationController",@"UITabBarController"];
    NSString *className = NSStringFromClass(self.class);
    if ([filter containsObject:className]) return ; //如果該Controller在不允許發送log的列表里,則不能繼續往下走
   
    if ([self.title isKindOfClass:[NSString class]] && self.title.length > 0){ //有標題的才符合我的要求
        // 這里發送log
    }

}

嗯,剛剛說到有部分Controller我是不發數據的,這里有兩重判斷,一個是加入到黑名單,另一個是 判斷Controller的title屬性是否為空

以上判斷基本能滿足我這個行為分析統計系統的需求,若還需要什么判斷還可以繼續加

以此 我只需要往工程里面添加這個Category,這個viewDidAppear就會被hook出來,可以為所欲為..

 

另外 需求中還提到 需要在應用啟動的時候發送一次init消息

hook?可以,但我更傾向與利用category+NSNotification,因為系統中已經有 UIApplicationDidFinishLaunchingNotification

這種通知,直接用就可以

@implementation UIApplication (GBHTracer)
+ (void)load{
    [super load];
    
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{ //只執行一次就可以了
        [[NSNotificationCenter defaultCenter]addObserver:self selector:@selector(__gbh_tracer_applicationDidFinishLaunching:) name:UIApplicationDidFinishLaunchingNotification object:nil];
    });
}

+ (void)__gbh_tracer_applicationDidFinishLaunching:(NSNotification *)noti{
    //應用啟動時為所欲為!
}

@end

嗯..我們的行為分析統計系統就在原工程不Import一個頭文件 不調用 任何一個方法就可以達到統計效果.

但是像什么操作響應的時候的統計,還是需要各位看官在響應中調用相應的方法