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:] 
}