锐客网

  1. 首页 > it技术 > >

代码注入（下）

经过代码注入（上）的努力，现在终于能让其他的App执行我们的代码了，之前的代码仅仅是为了测试代码的注入是否成功，接下来就可以试试代码注入的真正用途了。
要让代码注入有实际的意义，我们就需要OC中runtime的Method Swizzle了。
1、什么是Method Swizzle ？ 【代码注入（下）】Method Swizzle中文翻译就是方法交换或者方法重排。
我们知道OC是动态语言，其核心的runtime是在运行时才确定属性的类、方法的实现地址，所以我们根据runtime的这种特性，动态改变SEL(方法编号)和IMP(方法实现)的映射关系，来达到改变OC方法调用流程的目的。这便是我们的Method Swizzle。
2、Method Swizzle的小Demo 举个简单的例子：我们平常使用NSURL的使用会用到类似于这样的网址www.baidu.com/中文，那我们就需要进行UTF-8的转码。
平常使用的方法

NSString *string = @"www.baidu.com/中文"; string = [string stringByAddingPercentEncodingWithAllowedCharacters:[NSCharacterSet URLQueryAllowedCharacterSet]]; NSURL *url = [NSURL URLWithString:string]; NSLog(@"%@",url); 打印结果： 2019-10-26 17:56:21.414290+0800 MethodSwizzleDemo[39740:299124] www.baidu.com/%E4%B8%AD%E6%96%87

然后我们用Method Swizzle实现：
1、首先新建一个NSURL的分类NSURL+UTF8

代码注入（下）
文章图片
image.png

代码注入（下）
文章图片
image.png
2、 + (vod)load; 方法中实现方法交换
// //NSURL+UTF8.m //MethodSwizzleDemo // //Created by ABC on 2019/10/26. //Copyright ? 2019 ABC. All rights reserved. //#import "NSURL+UTF8.h" #import @implementation NSURL (UTF8)+ (void)load { Method URLWithString_Method = class_getClassMethod(self, @selector(URLWithString:)); Method my_URLWithString_Method = class_getClassMethod(self, @selector(my_URLWithString:)); method_exchangeImplementations(URLWithString_Method, my_URLWithString_Method); }+ (nullable instancetype)my_URLWithString:(NSString *)URLString { NSURL *url = [self my_URLWithString:URLString]; if (url == nil) { URLString = [URLString stringByAddingPercentEncodingWithAllowedCharacters:[NSCharacterSet URLQueryAllowedCharacterSet]]; url = [self my_URLWithString:URLString]; } return url; }@end

那我们现在的实现就是
NSString *string = @"www.baidu.com/中文"; NSURL *url = [NSURL URLWithString:string]; NSLog(@"%@",url); 打印结果： 2019-10-26 18:02:54.426813+0800 MethodSwizzleDemo[40109:313573] www.baidu.com/%E4%B8%AD%E6%96%87

这就是一个简单的方法交换。
3、Method Swizzle 的原理 上述代码中使用了method_exchangeImplementations方法,方法需要两个Method参数,生成Method需要SEL。
该方法交换了Method m1和Method m2的函数指针地址，即IMP。
OBJC_EXPORT Method _Nullable class_getClassMethod(Class _Nullable cls, SEL _Nonnull name)OBJC_EXPORT void method_exchangeImplementations(Method _Nonnull m1, Method _Nonnull m2)

那什么是SEL和IMP? 举个简单的例子：
我们的书都会有一个目录，目录中左边是标题，右边页码。我们看到标题，需要查看后方的页码才能到知道这个标题下的内容在哪里。所以：
  • SEL就是和我们的标题是一样的，告诉我们这个方法叫什么，也就是所谓的方法编号。
  • IMP就是标题对应的页码，告诉我们这个方法在哪儿放着，也就是所谓的函数指针地址。
我们交换了Method m1和Method m2其实就是把指向方法实现的地址交换了，也就是上方举例目录中的页码。具体如下图所示：
代码注入（下）
文章图片
image.png 4、使用 Method Swizzle 破坏 WeChat 注册 使用代码注入（上）的方法注入我们的WeChatHook.framework,然后做方法交换。要达到破坏微信注册的效果，需要以下2点：
  • 需要获取到点击注册按钮的方法实现
  • 需要获取到点击注册按钮方法实现所在的类
当我们运行起来WeChat的时候，我们可以使用ViewDebug来获取注册按钮的方法实现以及方法实现所在的类。
代码注入（下）
文章图片
ViewDebug 我们知道了方法是onFirstViewRegister,方法所在的类是WCAccountLoginControlLogic，但是这里并不知道onFirstViewRegister方法有没有参数,所以我们还需要借助一个工具class-dump。
class-dump：它可以将MachO里面类的描述拷贝出来，其实就是把所有类的头文件导出一份。
class-dump网上有可以去下载一份，代码注入（上）中说过了，把class-dump放在/usr/local/bin/目录下就可以任何地方调用了。
将WeChat.ipa解压，将其中的MachO拷贝出来，打开终端，输入：class-dump -H WeChat -o ./header/ 会提取出WeChat中所有的头文件。

代码注入（下）
文章图片
WeChat 头文件提取
将Header文件夹拖入Sublime Text软件中打开，使用Commend + Shift + F全局搜索一下WCAccountLoginControlLogic类，找到了WCAccountLoginControlLogic类之后寻找onFirstViewRegister方法。
代码注入（下）
文章图片
image.png 寻找之后，发现onFirstViewRegister并没有参数，找到了我们需要的东西后，开始实现我们的需求。
// //HookObject.m //WeChatHook // //Created by ABC on 2019/10/23. //Copyright ? 2019 ABC. All rights reserved. //#import "HookObject.h" #import @implementation HookObject+ (void)load { Method method_WeChatMethod = class_getInstanceMethod(objc_getClass("WCAccountLoginControlLogic"), @selector(onFirstViewRegister)); Method method_MyMethod = class_getInstanceMethod(self, @selector(my_onFirstViewRegister)); method_exchangeImplementations(method_WeChatMethod, method_MyMethod); }- (void)my_onFirstViewRegister { NSLog(@"你点击注册，但是很遗憾的是被我拦截了"); }@end

发现成功了拦截了注册，输出了我们的代码，
代码注入（下）
文章图片
Log输出 5、使用 Method Swizzle 登录时获取输入的密码后继续登录 明确了我们的目的，思路如下：
  • 查找点击登录按钮后调用的方法
  • 查找登录按钮的所在的类
  • 查找输入框的类
  • Method Swizzle方法交换
继续使用ViewDebug查找登录方法：onNext，所在类为WCAccountMainLoginViewController，然后头文件核实一下。

代码注入（下）
文章图片
image.png
代码注入（下）
文章图片
接着找到了输入框的类为WCUITextField，然而并没有看到在哪个View上，只能打开header寻找线索了。
代码注入（下）
文章图片
输入框 我们继续在header中寻找，Command + Shift + F 查看 WCAccountMainLoginViewController 类，发现了WCAccountTextFieldItem *_textFieldUserPwdItem代码，感觉应该是密码框的View了。
代码注入（下）
文章图片
image.png 查看 WCAccountTextFieldItem 类却没有发现，寻找其父类找到了WCUITextField。
代码注入（下）
文章图片
WCBaseTextFieldItem 最后查看WCUITextField是继承UIKit基类UITextField。
代码注入（下）
文章图片
WCUITextField 到这里我们找到了我们所有需要东西，接下来我们不要急于动手实现，先验证一下我们查找的过程是否正确，正确了再着手实现也不急。
控制台lldb中：
(lldb) po 0x10f015c00 (lldb) po [(WCAccountMainLoginViewController *) 0x10f015c00valueForKey:@"_textFieldUserPwdItem"]; (lldb) po[(WCAccountTextFieldItem *) 0x280f8ed60valueForKey:@"m_textField"]; ; layer = >(lldb) po [(UITextField *) 0x10f042400 text] 123456(lldb)

经过我们的测试我们正确的打印出了我们密码123456。
现在开始代码实现：
  • 先输出我的密码
  • 然后再继续请求WeChat服务器
下方代码输出了我的密码
// //LoginHook.m //WeChatHook // //Created by ABC on 2019/10/26. //Copyright ? 2019 ABC. All rights reserved. //#import "LoginHook.h" #import #import @implementation LoginHook+ (void)load { Method method_WeChatMethod = class_getInstanceMethod(objc_getClass("WCAccountMainLoginViewController"), @selector(onNext)); Method method_MyMethod = class_getInstanceMethod(self, @selector(my_onNext)); method_exchangeImplementations(method_WeChatMethod, method_MyMethod); }- (void)my_onNext { UITextField *textField = [[self valueForKey:@"_textFieldUserPwdItem"] valueForKey:@"m_textField"]; NSLog(@"我输入的密码是 : %@",textField.text); }@end

代码注入（下）
文章图片
密码输出 接下来我需要继续请求WeChat服务器了，通过Method Swizzle交换了方法，现在我需要再次调用my_onNext即调用了原来的onNext方法（因为方法交换了）。结果如下：
代码注入（下）
文章图片
image.png 注意：调用之后崩溃了。为什么会崩溃呢？
回顾一下上方代码，我们通过runtime将- (void)onNext; 和- (void)my_onNext; 方法交换了，但是交换的时候-(void)my_onNext却是在我们的LoginHook类里面, WCAccountMainLoginViewController 并没有这个方法，所以崩溃了。
明白了上述问题，其实我们只需要向WCAccountMainLoginViewController动态生成一个方法然后再做方法交换就没有问题了。接下来，修改上述代码：
// //LoginHook.m //WeChatHook // //Created by ABC on 2019/10/26. //Copyright ? 2019 ABC. All rights reserved. //#import "LoginHook.h" #import #import #import @implementation LoginHookvoid my_onNext (id self, SEL _cmd) { //获取密码 UITextField *textField = [[self valueForKey:@"_textFieldUserPwdItem"] valueForKey:@"m_textField"]; NSLog(@"我输入的密码是 : %@",textField.text); //给 my_onNext 发消息 等同于 [self my_onNext]; ((void(*)(id,SEL))objc_msgSend)(self,@selector(my_onNext)); }+ (void)load {//首先获取微信的方法 Method method_WeChatMethod = class_getInstanceMethod(objc_getClass("WCAccountMainLoginViewController"), @selector(onNext)); //向 WCAccountMainLoginViewController 动态添加 my_onNext 方法 class_addMethod(objc_getClass("WCAccountMainLoginViewController"), @selector(my_onNext), (IMP)my_onNext, "v@:"); //获取我动态添加的方法 Method method_MyMethod = class_getInstanceMethod(objc_getClass("WCAccountMainLoginViewController"), @selector(my_onNext)); //交换方法 method_exchangeImplementations(method_WeChatMethod, method_MyMethod); }@end

这样就实现上述需求，先获取到输入的密码，再请求WeChat服务器。
其他方法解决崩溃：
方法一：方法替换
// //LoginHook.m //WeChatHook // //Created by ABC on 2019/10/26. //Copyright ? 2019 ABC. All rights reserved. //#import "LoginHook.h" #import #import #import @implementation LoginHookIMP (*old_onNext) (id self, SEL _cmd); void my_onNext (id self, SEL _cmd) { //获取密码 UITextField *textField = [[self valueForKey:@"_textFieldUserPwdItem"] valueForKey:@"m_textField"]; NSLog(@"我输入的密码是 : %@",textField.text); //等同于 [self my_onNext]; old_onNext(self,_cmd); }+ (void)load { //首先获取微信的方法 Method method_WeChatMethod = class_getInstanceMethod(objc_getClass("WCAccountMainLoginViewController"), @selector(onNext)); //获取IMP old_onNext = method_getImplementation(method_WeChatMethod); //方法替换 class_replaceMethod(objc_getClass("WCAccountMainLoginViewController"), @selector(onNext), my_onNext, "v@:"); }@end

方法二：方法覆盖
// //LoginHook.m //WeChatHook // //Created by ABC on 2019/10/26. //Copyright ? 2019 ABC. All rights reserved. //#import "LoginHook.h" #import #import #import @implementation LoginHookIMP (*old_onNext) (id self, SEL _cmd); void my_onNext (id self, SEL _cmd) { //获取密码 UITextField *textField = [[self valueForKey:@"_textFieldUserPwdItem"] valueForKey:@"m_textField"]; NSLog(@"我输入的密码是 : %@",textField.text); //等同于 [self my_onNext]; old_onNext(self,_cmd); }+ (void)load { //首先获取微信的方法 Method method_WeChatMethod = class_getInstanceMethod(objc_getClass("WCAccountMainLoginViewController"), @selector(onNext)); //获取IMP old_onNext = method_getImplementation(method_WeChatMethod); //setIMP method_setImplementation(method_WeChatMethod, (IMP)my_onNext); }@end

以上就是代码注入的所有内容了。

    推荐阅读


    上一篇：秦岭太白山温饱思人生

    下一篇：记录下最近觉得美好的事情吧