经过代码注入（上）的努力，现在终于能让其他的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
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 <objc/runtime.h>
@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其实就是把指向方法实现的地址交换了，也就是上方举例目录中的页码。具体如下图所示：
4、使用 Method Swizzle 破坏 WeChat 注册
使用代码注入（上）的方法注入我们的WeChatHook.framework,然后做方法交换。要达到破坏微信注册的效果，需要以下2点：
- 需要获取到点击注册按钮的方法实现
- 需要获取到点击注册按钮方法实现所在的类
当我们运行起来WeChat的时候，我们可以使用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中所有的头文件。
将Header文件夹拖入Sublime Text软件中打开，使用Commend + Shift + F全局搜索一下WCAccountLoginControlLogic类，找到了WCAccountLoginControlLogic类之后寻找onFirstViewRegister方法。
寻找之后，发现onFirstViewRegister并没有参数，找到了我们需要的东西后，开始实现我们的需求。
//
// HookObject.m
// WeChatHook
//
// Created by ABC on 2019/10/23.
// Copyright © 2019 ABC. All rights reserved.
//
#import "HookObject.h"
#import <objc/runtime.h>
@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
发现成功了拦截了注册，输出了我们的代码，
5、使用 Method Swizzle 登录时获取输入的密码后继续登录
明确了我们的目的，思路如下：
- 查找点击登录按钮后调用的方法
- 查找登录按钮的所在的类
- 查找输入框的类
-
Method Swizzle方法交换
继续使用ViewDebug查找登录方法：onNext，所在类为WCAccountMainLoginViewController，然后头文件核实一下。
接着找到了输入框的类为WCUITextField，然而并没有看到在哪个View上，只能打开header寻找线索了。
我们继续在header中寻找，Command + Shift + F 查看 WCAccountMainLoginViewController 类，发现了WCAccountTextFieldItem *_textFieldUserPwdItem代码，感觉应该是密码框的View了。
查看 WCAccountTextFieldItem 类却没有发现，寻找其父类找到了WCUITextField。
最后查看WCUITextField是继承UIKit基类UITextField。
到这里我们找到了我们所有需要东西，接下来我们不要急于动手实现，先验证一下我们查找的过程是否正确，正确了再着手实现也不急。
控制台lldb中：
(lldb) po 0x10f015c00
<WCAccountMainLoginViewController: 0x10f015c00>
(lldb) po [(WCAccountMainLoginViewController *) 0x10f015c00 valueForKey:@"_textFieldUserPwdItem"];
<WCAccountTextFieldItem: 0x280f8ed60>
(lldb) po [(WCAccountTextFieldItem *) 0x280f8ed60 valueForKey:@"m_textField"];
<WCUITextField: 0x10f042400; baseClass = UITextField; frame = (20 0; 345 44); text = '123456'; opaque = NO; autoresize = W+H; tintColor = UIExtendedSRGBColorSpace 0.00784314 0.733333 0 1; gestureRecognizers = <NSArray: 0x282418c60>; layer = <CALayer: 0x282b38560>>
(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 <objc/runtime.h>
#import <UIKit/UIKit.h>
@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方法（因为方法交换了）。结果如下：
注意：调用之后崩溃了。为什么会崩溃呢？
回顾一下上方代码，我们通过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 <objc/runtime.h>
#import <objc/message.h>
#import <UIKit/UIKit.h>
@implementation LoginHook
void 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 <objc/runtime.h>
#import <objc/message.h>
#import <UIKit/UIKit.h>
@implementation LoginHook
IMP (*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 <objc/runtime.h>
#import <objc/message.h>
#import <UIKit/UIKit.h>
@implementation LoginHook
IMP (*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
以上就是代码注入的所有内容了。
