runtime - 子类动态实现父类的方法

简述

最近在做一个简单的UI自动化框架,只针对iOS平台的,基本思路:Hook(可利用SWIZZLE技术实现)目标UIViewController的viewDidAppear:方法(目的就是要让控件都显示完毕),在执行完viewDidAppear:的方法之后,再执行UI自动化脚本命令;
是在一个UIViewController加载并显示完成之后(ViewDidAppear:),里面开始执行自动化脚本利用KVC获取里面的控件;
然后向控件发送控件事件(UIControlEvents);这样可以保证0侵入代码;
然而使用过程中发现一个问题:如果一个UIViewController子类没有重写ViewDidAppear:方法,那第一步的Hook操作直接把基类(UIViewController)的viewDidAppear:方法给hook了,导致的结果就是所有的UIViewController子类的该方法都被hook了,结果就是其他VC子类找不到调用原本的方法而崩溃;

解决方法

所以想了一个解决方法,如果一个VC子类重写了viewDidAppear:,则大胆的swizzle进行hook就行啦;
如果一个一个VC子类未重写viewDidAppear:,那我们就用runtime给他添加上,难点就是[super viewDidAppear:animated]这句话怎么实现,翻了好久资料,原来有个objc_msgSendSuper方法,啊呀,还是太年轻啊,只知道有这个东西,也从来没用过;具体代码如下:

- (void)swizzle {
    Class clz = [NSClassFromString(@"XXX_VC") class];
    Method method = class_getInstanceMethod([self class], @selector(fakeViewDidAppear:));
    IMP imp = class_getMethodImplementation([self class], @selector(fakeViewDidAppear:));
    if(
       class_addMethod(clz, @selector(viewDidAppear:), imp, method_getTypeEncoding(method))) {
        NSLog(@"子类VC未实现viewDidAppear,这下加上了");
        class_replaceMethod(clz, @selector(viewDidAppear:), imp, method_getTypeEncoding(method));
    }else {
        NSLog(@"子类VC已经实现viewDidAppear:");
        //exchange IMP,请自己查阅资料,不再赘述
    }
}
- (void)fakeViewDidAppear:(BOOL)animated {
// Do some hook things
    struct objc_super sup = {
        .receiver = self,
        .super_class = class_getSuperclass([self class]) //
    };
    void(*func)(struct objc_super*,SEL) = (void*)&objc_msgSendSuper;
    func(&sup, _cmd); // 等同于[super viewDidAppear:] 
}
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • /* UIViewController is a generic controller base class th...
    DanDanC阅读 1,844评论 0 2
  • 这篇文章完全是基于南峰子老师博客的转载 这篇文章完全是基于南峰子老师博客的转载 这篇文章完全是基于南峰子老师博客的...
    西木阅读 30,589评论 33 466
  • 《假如爱有天意》又名《不可不信缘》,讲述了母女两代人爱的故事。 母亲叫珠喜——国会议员的孙女,一次偶然的机会认识了...
    Ashley麦阅读 897评论 0 0
  • 春节假期最重要的三项收获。 一、原计划 其实最开始没有给自己定太多计划,就想着趁着春节好好休息一下,所以也就制定了...
    有钱YQ阅读 363评论 1 0
  • 我上班时间是从早上七点半到十二点,再从下午三点到六点。 由于晚上不用上班,所以经常有大把的时间可以挥霍。我一直想利...
    梦想中大奖阅读 625评论 0 2