简述
最近在做一个简单的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:]
}
网友评论