iOS hook webviewdelegate && 继承关系hook

主要有两个注意事项
1.判断方法是否被交换过,已经交换过就不再重复交换

     if ([delegate respondsToSelector:@selector(secre_webview_webView:decidePolicyForNavigationAction:decisionHandler:)]) {
            TLogD(@"WKNavigationDelegate class %@ has hooked yet.", [NSString stringWithUTF8String:object_getClassName(delegateClass)]);
            return;
        }

2.判断交换的对象为方法实现的类,不要在继承子类中进行方法交换。
原因:
2.1
父类superClass 中实现了methodA
子类selfClass 中没有实现methodA
2.2
如果selfClass中进入hook,那么实际交换的是selfClass中的新方法与superClass中的methodA; 这时selfClass已经hook,但是在superClass中找不过交换的新方法,会再次hook,最终进入死循环crash
2.3
所以,我们需要在selfClass中获取到实现methodA的superClass,直接交换superClass中的methodA,所有的子类方法都能调用到交换后的方法。

        
         //直接交换实现方法的类,避免继承之间循环交换

         //遍历父类找到实现方法的父类
        Class c = [self enumerateClasses:delegateClass];
        if (!c) {
           //如果没有父类方法实现需要交换的方法,就默认交换自己类的方法
            c = delegateClass;  
        }

        [SECREMethodSwizzling swizzleMethod:@selector(webView:decidePolicyForNavigationAction:decisionHandler:) ofClass:c withMethod:@selector(secre_webview_webView:decidePolicyForNavigationAction:decisionHandler:) defaultMethod:@selector(secre_webview_default_webView:decidePolicyForNavigationAction:decisionHandler:) fromClass:WKWebView.class];

以webview为例代码实现

1.方法交换

+ (void)load {
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        // swizzling method
        [SECREMethodSwizzling swizzleMethod:@selector(setNavigationDelegate:) withMethod:@selector(secre_webview_setNavigationDelegate:) ofClass:WKWebView.class];
        TLogD(@"has exchanged setNavigationDelegate: method.");
    });
}

2.交换方法自己与需要添加处理的方法

- (void)secre_webview_setNavigationDelegate:(id<WKNavigationDelegate>)delegate {
    if (delegate) {
        //    NSLog(@"Now is using %s as -[WKWebView setNavigationDelegate:]", __PRETTY_FUNCTION__);
        [self secre_webview_swizzleMethodOfWKNavigationDelegate:delegate];
        [self secre_webview_setNavigationDelegate:delegate];
    }
}
- (void)secre_webview_swizzleMethodOfWKNavigationDelegate:(id<WKNavigationDelegate>)delegate {
    Class delegateClass = delegate.class;
    @synchronized (sc_hookLock) {
        
         //直接交换实现方法的类,避免继承之间循环交换
        Class c = [self enumerateClasses:delegateClass];
        if (!c) {
            c = delegateClass;
        }
        
        if ([delegate respondsToSelector:@selector(secre_webview_webView:decidePolicyForNavigationAction:decisionHandler:)]) {
            TLogD(@"WKNavigationDelegate class %@ has hooked yet.", [NSString stringWithUTF8String:object_getClassName(delegateClass)]);
            return;
        }
        Class wkWebViewClass = WKWebView.class;
        // exchange implementation
        [SECREMethodSwizzling swizzleMethod:@selector(webView:decidePolicyForNavigationAction:decisionHandler:) ofClass:c withMethod:@selector(secre_webview_webView:decidePolicyForNavigationAction:decisionHandler:) defaultMethod:@selector(secre_webview_default_webView:decidePolicyForNavigationAction:decisionHandler:) fromClass:wkWebViewClass];
        // 将hook过的类添加进来
        // [hookedClassNames addObject:NSStringFromClass(delegateClass)];
        TLogD(@"WKNavigationDelegate class %@ has hooked.", [NSString stringWithUTF8String:object_getClassName(delegateClass)]);
    }
}

#pragma mark - helper
//判断页面是否实现了某个sel
- (Class)enumerateClasses:(Class)delegateClass {
    SEL sel = @selector(webView:decidePolicyForNavigationAction:decisionHandler:);
    if ([self isContainSel:sel inClass:[delegateClass class]]) {
        return delegateClass;
    }
    
    Class c = [delegateClass class];
    while (c) {
        c = class_getSuperclass(c);
        if ([self isContainSel:sel inClass:[c class]]) {
            return [c class];
        }
    }
    return nil;
}
- (BOOL)isContainSel:(SEL)sel inClass:(Class)class {
    unsigned int count;
    
    Method *methodList = class_copyMethodList(class,&count);
    for (int i = 0; i < count; i++) {
        Method method = methodList[i];
        NSString *tempMethodString = [NSString stringWithUTF8String:sel_getName(method_getName(method))];
        if ([tempMethodString isEqualToString:NSStringFromSelector(sel)]) {
            return YES;
        }
    }
    return NO;
}

3.默认方法与自己需要添加的实现

#pragma mark - default Methods
- (void)secre_webview_default_webView:(WKWebView *)webView decidePolicyForNavigationAction:(WKNavigationAction *)navigationAction decisionHandler:(void (^)(WKNavigationActionPolicy))decisionHandler {
    decisionHandler(WKNavigationActionPolicyAllow);
}

#pragma mark - swizzling Methods
- (void)secre_webview_webView:(WKWebView *)webView decidePolicyForNavigationAction:(WKNavigationAction *)navigationAction decisionHandler:(void (^)(WKNavigationActionPolicy))decisionHandler {
    TLogD(@"%s: will decide whether can visit url: %@, webview url: %@", __PRETTY_FUNCTION__, navigationAction.request.URL, webView.URL);
    
    NSLog(@"URL :%@", navigationAction.request.URL);
    // do something 
    [self secre_webview_webView:webView decidePolicyForNavigationAction:navigationAction decisionHandler:decisionHandler];
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 158,736评论 4 362
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,167评论 1 291
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 108,442评论 0 243
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 43,902评论 0 204
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,302评论 3 287
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,573评论 1 216
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,847评论 2 312
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,562评论 0 197
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,260评论 1 241
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,531评论 2 245
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,021评论 1 258
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,367评论 2 253
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,016评论 3 235
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,068评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,827评论 0 194
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,610评论 2 274
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,514评论 2 269

推荐阅读更多精彩内容

  • WebView·开车指南 2016-08-31BugDev 北京市东城区首席Bug布道师开山之作,一整月交通事故血...
    53c021c38a1d阅读 798评论 0 1
  • View 自定义View中在onDraw()方法中可以设置padding吗?答案是不能,设置padding后,Vi...
    ElvenShi阅读 1,730评论 0 0
  • 一、WebView 谷歌提供的系统组件,用来加载和展现html网页,其采用webkit内核驱动,来实现网页浏览功能...
    闲庭阅读 6,717评论 2 12
  • 一、深复制和浅复制的区别? 1、浅复制:只是复制了指向对象的指针,即两个指针指向同一块内存单元!而不复制指向对象的...
    iOS_Alex阅读 1,314评论 1 27
  • 当你达到一定境界的时候,你所看到的和你想的都是高于别人的,名家大师亦是如此,市井平民亦是如此,特别是咱们的国画精神...
    竹本良子阅读 402评论 0 6