經(jīng)過代碼注入（上）的努力呆瞻，現(xiàn)在終于能讓其他的App執(zhí)行我們的代碼了，之前的代碼僅僅是為了測試代碼的注入是否成功径玖，接下來就可以試試代碼注入的真正用途了痴脾。
要讓代碼注入有實際的意義，我們就需要OC中runtime的Method Swizzle了梳星。
1赞赖、什么是Method Swizzle ？
Method Swizzle中文翻譯就是方法交換或者方法重排冤灾。
我們知道OC是動態(tài)語言前域，其核心的runtime是在運行時才確定屬性的類、方法的實現(xiàn)地址瞳购，所以我們根據(jù)runtime的這種特性话侄，動態(tài)改變SEL(方法編號)和IMP(方法實現(xiàn))的映射關系亏推，來達到改變OC方法調用流程的目的学赛。這便是我們的Method Swizzle。
2吞杭、Method Swizzle的小Demo
舉個簡單的例子：我們平常使用NSURL的使用會用到類似于這樣的網(wǎng)址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實現(xiàn)：
1芽狗、首先新建一個NSURL的分類NSURL+UTF8
2绢掰、
+ (vod)load;方法中實現(xiàn)方法交換
//
// 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
那我們現(xiàn)在的實現(xiàn)就是
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參數(shù),生成Method需要SEL滴劲。
該方法交換了Method m1和Method m2的函數(shù)指針地址，即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就是標題對應的頁碼，告訴我們這個方法在哪兒放著渔期，也就是所謂的函數(shù)指針地址运吓。
我們交換了Method m1和Method m2其實就是把指向方法實現(xiàn)的地址交換了，也就是上方舉例目錄中的頁碼疯趟。具體如下圖所示：
4羽德、使用 Method Swizzle 破壞 WeChat 注冊
使用代碼注入（上）的方法注入我們的WeChatHook.framework,然后做方法交換。要達到破壞微信注冊的效果迅办，需要以下2點：
- 需要獲取到點擊注冊按鈕的方法實現(xiàn)
- 需要獲取到點擊注冊按鈕方法實現(xiàn)所在的類
當我們運行起來WeChat的時候宅静，我們可以使用ViewDebug來獲取注冊按鈕的方法實現(xiàn)以及方法實現(xiàn)所在的類。
我們知道了方法是onFirstViewRegister,方法所在的類是WCAccountLoginControlLogic站欺，但是這里并不知道onFirstViewRegister方法有沒有參數(shù),所以我們還需要借助一個工具class-dump姨夹。
class-dump：它可以將MachO里面類的描述拷貝出來，其實就是把所有類的頭文件導出一份矾策。
class-dump網(wǎng)上有可以去下載一份磷账，代碼注入（上）中說過了，把class-dump放在/usr/local/bin/目錄下就可以任何地方調用了贾虽。
將WeChat.ipa解壓逃糟，將其中的MachO拷貝出來，打開終端蓬豁，輸入：class-dump -H WeChat -o ./header/ 會提取出WeChat中所有的頭文件绰咽。
將Header文件夾拖入Sublime Text軟件中打開地粪，使用Commend + Shift + F全局搜索一下WCAccountLoginControlLogic類取募，找到了WCAccountLoginControlLogic類之后尋找onFirstViewRegister方法。
尋找之后蟆技，發(fā)現(xiàn)onFirstViewRegister并沒有參數(shù)玩敏，找到了我們需要的東西后，開始實現(xiàn)我們的需求质礼。
//
// 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
發(fā)現(xiàn)成功了攔截了注冊，輸出了我們的代碼眶蕉，
5砰粹、使用 Method Swizzle 登錄時獲取輸入的密碼后繼續(xù)登錄
明確了我們的目的，思路如下：
- 查找點擊登錄按鈕后調用的方法
- 查找登錄按鈕的所在的類
- 查找輸入框的類
-
Method Swizzle方法交換
繼續(xù)使用ViewDebug查找登錄方法：onNext妻坝，所在類為WCAccountMainLoginViewController伸眶，然后頭文件核實一下惊窖。
接著找到了輸入框的類為WCUITextField，然而并沒有看到在哪個View上厘贼，只能打開header尋找線索了界酒。
我們繼續(xù)在header中尋找，Command + Shift + F 查看 WCAccountMainLoginViewController 類嘴秸，發(fā)現(xiàn)了WCAccountTextFieldItem *_textFieldUserPwdItem代碼毁欣，感覺應該是密碼框的View了。
查看 WCAccountTextFieldItem 類卻沒有發(fā)現(xiàn)岳掐，尋找其父類找到了WCUITextField凭疮。
最后查看WCUITextField是繼承UIKit基類UITextField。
到這里我們找到了我們所有需要東西串述，接下來我們不要急于動手實現(xiàn)执解，先驗證一下我們查找的過程是否正確，正確了再著手實現(xiàn)也不急纲酗。
控制臺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)
經(jīng)過我們的測試我們正確的打印出了我們密碼123456衰腌。
現(xiàn)在開始代碼實現(xiàn)：
- 先輸出我的密碼
- 然后再繼續(xù)請求
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
接下來我需要繼續(xù)請求WeChat服務器了，通過Method Swizzle交換了方法觅赊，現(xiàn)在我需要再次調用my_onNext即調用了原來的onNext方法（因為方法交換了）右蕊。結果如下：
注意：調用之后崩潰了。為什么會崩潰呢吮螺？
回顧一下上方代碼饶囚，我們通過runtime將- (void)onNext;和- (void)my_onNext;方法交換了，但是交換的時候-(void)my_onNext卻是在我們的LoginHook類里面, WCAccountMainLoginViewController 并沒有這個方法鸠补，所以崩潰了萝风。
明白了上述問題，其實我們只需要向WCAccountMainLoginViewController動態(tài)生成一個方法然后再做方法交換就沒有問題了莫鸭。接下來闹丐，修改上述代碼：
//
// 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 發(fā)消息 等同于 [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 動態(tài)添加 my_onNext 方法
class_addMethod(objc_getClass("WCAccountMainLoginViewController"), @selector(my_onNext), (IMP)my_onNext, "v@:");
//獲取我動態(tài)添加的方法
Method method_MyMethod = class_getInstanceMethod(objc_getClass("WCAccountMainLoginViewController"), @selector(my_onNext));
//交換方法
method_exchangeImplementations(method_WeChatMethod, method_MyMethod);
}
@end
這樣就實現(xiàn)上述需求，先獲取到輸入的密碼被因，再請求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
以上就是代碼注入的所有內容了衫仑。
