快學會這個技能-.NET API攔截技法

大家好,我是沙漠盡頭的狼。

本文先拋出以下問題,請在文中尋找答案,可在評論區回答:

  1. 什么是API攔截?
  2. 一個方法被很多地方調用,怎么在不修改這個方法源碼情況下,記錄這個方法調用的前后時間?
  3. 同2,不修改源碼的情況下,怎么對方法的參數進行校正(篡改)?
  4. 同3,不修改源碼的情況下,怎么對方法的返回值進行偽造?
    ...

1. 前言

前言翻譯自一個國外的文章,他寫的更容易讓人理解 - Hacking .NET – rewriting code you don’t control

您是否曾經遇到過不屬于您但想要更改其行為的類庫方法?通常,該方法是非公開的,并且沒有很好的方法來覆蓋其行為。你可以看到它是如何工作的(因為你很棒,并且使用像Resharper、dnSpy之類反編譯工具,對吧?),你只是無法改變它。你真的需要改變它,因為XXX原因。

有幾個選項可供您使用:

  1. 通過反編譯或下載源代碼(如果首先可用)獲取源代碼。這通常很冒險,因為它經常伴隨著復雜的構建過程,許多依賴項,現在你負責維護庫的整個分支,即使你只想做一個很小的改變。

  2. 使用 ILDasm 反編譯應用,直接修補 IL 代碼,然后使用 ILAs 將其組裝回來。在許多方面,這更好,因為您可以創建一個戰略性的手術切口,而不是全面的“從頭開始”的方法。缺點是您必須完全在IL中實現您的方法,這是一個不平凡的冒險。

如果您正在處理已簽名的庫,上述兩種方法也不起作用。

現在讓我們看一下另一種解決方法-內存修補。這與游戲作弊引擎幾十年來使用的技術相同,這些引擎附加到正在運行的進程,查找內存位置并改變其行為。聽起來很復雜? 實際上,在 .NET 中做到這一點比聽起來容易得多。我們將使用一個名為Harmony的庫,該庫在NuGet上可通過“Lib.Harmony”包獲得。這是一個用于 .NET 的內存修補引擎,主要針對使用 Unity 構建的游戲,當然不止Unity

站長將在本文向您展示如何更改您認為不可能的事情 - 從攔截(Hook)自己的庫開始,到攔截(Hook) WPF庫和.NET基礎庫結束。

2. 攔截(Hook)自己的庫

2.1. 準備工作

  1. 創建一個控制臺程序 HelloHook,添加類 Student
namespace HelloHook;

public class Student
{
    public string GetDetails(string name)
    {
        return $"大家好,我是Dotnet9網站站長:{name}";
    }
}

Student類中定義了一個GetDetails方法,返回格式化的個人介紹信息,這個方法后面攔截試驗使用。

  1. Program.cs中添加Student調用:
using HelloHook;

var student = new Student();
Console.WriteLine(student.GetDetails("沙漠盡頭的狼"));

運行程序輸出如下:

大家好,我是Dotnet9網站站長:沙漠盡頭的狼

基本工作準備完成,這就是一個簡單的控制臺程序,后文的內容就根據這兩個工程展開細說。

2.2. 攔截GetDetails方法

  1. 引入攔截包-Lib.Harmony

我們使用Lib.Harmony包,API的攔截就靠它了,在HelloHook工程中添加如下Nuget包:

<PackageReference Include="Lib.Harmony" Version="2.2.2" />
  1. 攔截處理

添加攔截類HookStudent

using HarmonyLib;

namespace HelloHook;

[HarmonyPatch(typeof(Student))]
[HarmonyPatch(nameof(Student.GetDetails))]
public class HookStudent
{
    public static bool Prefix()
    {
        Console.WriteLine($"Prefix");
        return true;
    }

    public static void Postfix()
    {
        Console.WriteLine($"Postfix");
    }

    public static void Finalizer()
    {
        Console.WriteLine($"Finalizer");
    }
}

看代碼中的注釋,HookStudent類上添加了兩個HarmonyPatch特性:

  • 第一個是關聯被攔截的類Student類型;
  • 第二個是關聯被攔截的類方法GetDetails

即當程序中調用Student類的GetDetails方法時,HookStudent內定義的方法就會分別執行,三個方法執行順序是Prefix->Postfix->Finalizer,當然約定的方法不止這三個,其實我們常用的應該是PrefixPostfix,約定方法的意義見后方說明,沒說就看Harmony wiki...

2.3. 注冊攔截

Program.cs進行修改,添加Harmony對整個程序集的攔截:

using HarmonyLib;
using HelloHook;
using System.Reflection;

var student = new Student();
Console.WriteLine(student.GetDetails("沙漠盡頭的狼"));  

var harmony = new Harmony("https://dotnet9.com");
harmony.PatchAll(Assembly.GetExecutingAssembly());

Console.WriteLine(student.GetDetails("沙漠盡頭的狼"));

Console.ReadLine();

運行程序輸出如下:

大家好,我是Dotnet9網站站長:沙漠盡頭的狼
Prefix
Postfix
Finalizer
大家好,我是Dotnet9網站站長:沙漠盡頭的狼

上面代碼就完成了一個自定義類的攔截處理,使用PatchAll能自動發現補丁類HookStudent,從而自動攔截Student類的GetDetails方法調用,發現第二次調用student.GetDetails("沙漠盡頭的狼")時,Harmony的三個生命周期方法都被調用了。

我們可以在攔截類的約定方法(PrefixPostfix等)里做一些日志記錄(Console.WriteLine\ILogger.LogInfo等),類似于B/S中的AOP攔截,操作日志在這里記錄正合適。

這就完了?說啥呢,這才開始。

2.4. 說好的參數篡改呢?還有API結果偽造呢?

修改Program.cs,多打印幾行數據方便區分:

using HarmonyLib;
using HelloHook;
using System.Reflection;

var student = new Student();
Console.WriteLine(student.GetDetails("沙漠盡頭的狼"));

var harmony = new Harmony("https://dotnet9.com");
harmony.PatchAll(Assembly.GetExecutingAssembly());

Console.WriteLine(student.GetDetails("沙漠之狐"));
Console.WriteLine(student.GetDetails("Dotnet"));

Console.ReadLine();

在注冊Harmony之前,打印一次,注冊后打印兩次,注意看參數的不同。

修改HookStudent,我們只使用Prefix方法,其他Postfix等方法類似,可看Harmony wiki了解更多的使用方法,修改如下:

using HarmonyLib;

namespace HelloHook;

[HarmonyPatch(typeof(Student))]
[HarmonyPatch(nameof(Student.GetDetails))]
public class HookStudent
{
    public static bool Prefix(ref string name, ref string __result)
    {
        if ("沙漠之狐".Equals(name))
        {
            __result = $"這是我的曾用網名";
            return false;
        }

        if (!"沙漠盡頭的狼".Equals(name))
        {
            name = "非站長名";
        }

        return true;
    }
}

先運行看輸出:

大家好,我是Dotnet9網站站長:沙漠盡頭的狼
這是我的曾用網名
大家好,我是Dotnet9網站站長:非站長名
  • 第1行“大家好,我是Dotnet9網站站長:沙漠盡頭的狼”,這是沒有注冊攔截之前的正常格式化輸出;
  • 第2行“這是我的曾用網名”,這里實現的是結果的偽造;
  • 第3行“大家好,我是Dotnet9網站站長:非站長名”,這里實現的是參數的篡改。

結果偽造

注意看Prefix方法傳入的參數ref string __result:其中ref表示引用傳遞,允許對結果進行修改;string與原方法的返回值類型必須一致;__result為返回值的約定命名,前面是兩個"_",即命名必須為__result

if ("沙漠之狐".Equals(name))
{
    __result = $"這是我的曾用網名";
    return false;
}

注意返回值是false,表示不調用原生方法,這里就將被攔截的方法返回值偽造成功了。

參數篡改

看傳入的參數ref string nameref表示參數是引用傳遞,允許對參數進行修改;string name必須與原方法參數定義一樣。

if (!"沙漠盡頭的狼".Equals(name))
{
    name = "非站長名";
}

Prefix方法默認返回的true,表示需要調用原生方法,這里會將篡改的參數傳入原生方法,原生方法執行結果會將篡改的參數組合返回為“大家好,我是Dotnet9網站站長:非站長名”。

注意:

原生參數name和返回值__result是可選的,如果不進行篡改,去掉ref也是可以的。

上面的示例源碼點這

3. 攔截(Hook)WPF的API

我們創建一個簡單的WPF程序HookWpf,攔截MessageBox.Show方法:

public static MessageBoxResult Show(string messageBoxText, string caption)

首先在App中使用自動攔截注冊:

public partial class App : Application
{
    protected override void OnStartup(StartupEventArgs e)
    {

        base.OnStartup(e);

        var harmony = new Harmony("https://dotnet9.com");
        harmony.PatchAll(Assembly.GetExecutingAssembly());
    }
}

定義攔截類HookMessageBox

using HarmonyLib;
using System.Windows;

namespace HookWpf;

[HarmonyPatch(typeof(MessageBox))]
[HarmonyPatch(nameof(MessageBox.Show))]
[HarmonyPatch(new [] { typeof(string), typeof(string) })]
public class HookMessageBox
{
    public static bool Prefix(ref string messageBoxText, string caption)
    {
        if (messageBoxText.Contains("垃圾"))
        {
            messageBoxText = "這是一個不錯的網站喲";
        }

        return true;
    }
}

HookMessageBox上關聯攔截的MessageBox.Show重載方法,在Prefix里對提示框內容進行合法性驗證,不合法進行修正。

最后就是在窗體MainWindow.xaml里添加兩個彈出提示框的按鈕:

<Window x:Class="HookWpf.MainWindow"
        xmlns="http://schemas.microsoft.com/winfx/2006/xaml/presentation"
        xmlns:d="http://schemas.microsoft.com/expression/blend/2008"
        xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
        xmlns:mc="http://schemas.openxmlformats.org/markup-compatibility/2006"
        mc:Ignorable="d"
        Title="MainWindow" Height="450" Width="800">
    <StackPanel>
        <Button Content="彈出默認提示框" Width="120" Height="30" Click="ShowDialog_OnClick"></Button>
        <Button Content="這是一個垃圾網站" Width="120" Height="30" Click="ShowBadMessageDialog_OnClick"></Button>
    </StackPanel>
</Window>

后臺處理按鈕點擊事件,彈出提示框:

using System.Windows;

namespace HookWpf;

public partial class MainWindow : Window
{
    public MainWindow()
    {
        InitializeComponent();
    }

    private void ShowDialog_OnClick(object sender, RoutedEventArgs e)
    {
        MessageBox.Show("https://dotnet9.com 是一個熱衷于技術分享的程序員網站", "Dotnet9");
    }

    private void ShowBadMessageDialog_OnClick(object sender, RoutedEventArgs e)
    {
        MessageBox.Show("這是一個垃圾網站", "https://dotnet9.com");
    }
}

運行結果如下:

image

上面效果即完成了提示框內容的驗證,如果內容含有“垃圾”關鍵字,就換成好聽的話(這是一個不錯的網站喲)。

本示例源碼在這

4. 攔截(Hook).NET默認的API

創建控制臺程序HookDotnetAPI,引入Lib.Harmonynuget包,Program.cs修改如下:

using HarmonyLib;

var dotnet9Domain = "https://dotnet9.com";
Console.WriteLine($"9的位置:{dotnet9Domain.IndexOf('9',0)}");

var harmony = new Harmony("com.dotnet9");
harmony.PatchAll();

Console.WriteLine($"9的位置:{dotnet9Domain.IndexOf('9', 0)}");

[HarmonyPatch(typeof(String))]
[HarmonyPatch(nameof(string.IndexOf))]
[HarmonyPatch(new Type[] { typeof(char), typeof(int) })]
public static class HookClass
{
    public static bool Prefix(ref int __result)
    {
        __result = 100;
        return false;
    }
}

使用方法和前面類似,string.IndexOf方法被攔截后,總是返回100,不論查找的字符位置在哪,當然這個測試代碼沒有任何意義,這里只是演示而已,運行結果如下:

9的位置:14
9的位置:100

5. 總結及分享

5.1. 總結

Harmony的原理是利用反射獲取對應類中的方法,然后加上特性標簽進行邏輯控制,達到不破壞原有代碼進行更新的效果。

Harmony用于在運行時修補替換和裝飾 .NET/.NET Core 方法的庫。 但是該技術可以與任何.NET版本一起使用。它對同一方法的多次更改是累積而不是覆蓋。

再次分析可能產生的場景需要攔截,加深您對本文的記憶:

  1. .NET的一些方法,我們直接在代碼層面可能無法直接修改;
  2. 第三庫未提供源碼,但我們想改它的部分方法;
  3. 第三庫提供了源碼,雖然可以修改它的源碼,但萬一第三庫后面迭代升級,我們又不得不更新時,那自己做的修改跟著升級可能麻煩了;

攔截注意:如您所見,這提供了大量新的可能性。請記住,權力越大,責任越大。由于您以原始開發人員不打算的方式覆蓋行為,因此無法保證您的補丁代碼在他們發布新版本的代碼時會起作用。即上面第3點,不排除第三庫升級API結構也變了,我們也要跟著修改攔截邏輯哦

從《Harmony wiki patching》中翻譯出以下使用注意事項:

  1. 最新2.0版支持 .NET Core ;
  2. Harmony支持手動(Patch,參考Harmony wiki上的使用)和自動(PatchAll,本文演示使用的這種方式,Lib.Harmony使用的是C#的特性機制);
  3. 它為每個原始方法創建DynamicMethod方法,并向其織入代碼,該代碼在開始(Prefix)和結束時(Postfix)調用自定義方法。它還允許您編寫過濾器(Transpiler)來處理原始的IL代碼,從而可以對原始方法進行更詳細的操作;
  4. Getter/Setter、虛/ 非虛 方法、 靜態 方法;
  5. 補丁方法必須是靜態方法;
  6. Prefix需要返回void或者bool類型(void即不攔截);
  7. Postfix需要返回void類型,或者返回的類型要與第一個參數一致(直通模式);
  8. 如果原方法不是靜態方法,則可以使用名為__instance(兩個下劃線)的參數來訪問對象實例;
  9. 可以使用名為__result(兩個下劃線)的參數來訪問方法的返回值,如果是Prefix,則得到返回值的默認值;
  10. 可以使用名為__state(兩個下劃線)的參數在Prefix補丁中存儲任意類型的值,然后在Postfix中使用它,你有責任在Prefix中初始化它的值;
  11. 可以使用與原方法中同名的參數來訪問對應的參數,如果你要寫入非引用類型,記得使用ref關鍵字;
  12. 補丁使用的參數必須嚴格對應類型(或者使用object類型)和名字;
  13. 我們的補丁只需要定義我們需要用到的參數,不用把所有參數都寫上;
  14. 要允許補丁重用,可以使用名為__originalMethod(兩個下劃線)的參數注入原始方法。

最后忘了補一條,.NET 7中使用Harmony還有點點問題,站長在測試WPF API和.NET基礎庫攔截Demo時一直不生效,折騰了2、3個晚上,以為是自己的使用問題,最后看到Harmony issue .NET 7 Runtime Skipping Patches #504,將程序降級為.NET 6即可。

5.2 分享

讀者朋友們,相信不少人使用過Harmony或者其他的 .NET Hook庫,可在評論中留言分享,可提出自己的疑問,或自己的使用心得:

  1. 我使用過這個庫進行API的Hook,它是XXX;
  2. 我自己實現過類似功能,分享文章鏈接是XXX;
  3. 想問,我能攔截這個API嗎?場景是XXXX

[我的分享] + [你的分享] ∈ [.NET圈子的一份小力量]

6. 參考

寫作本文時以下文章都做了參考,建議大家都看看,特別是 Harmony wiki中寫了Harmony的詳細使用方法:

?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 230,527評論 6 544
  • 序言:濱河連續發生了三起死亡事件,死亡現場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發現死者居然都...
    沈念sama閱讀 99,687評論 3 429
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 178,640評論 0 383
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 63,957評論 1 318
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 72,682評論 6 413
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發上,一...
    開封第一講書人閱讀 56,011評論 1 329
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 44,009評論 3 449
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 43,183評論 0 290
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當地人在樹林里發現了一具尸體,經...
    沈念sama閱讀 49,714評論 1 336
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 41,435評論 3 359
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發現自己被綠了。 大學時的朋友給我發了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 43,665評論 1 374
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 39,148評論 5 365
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發生泄漏。R本人自食惡果不足惜,卻給世界環境...
    茶點故事閱讀 44,838評論 3 350
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 35,251評論 0 28
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 36,588評論 1 295
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 52,379評論 3 400
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 48,627評論 2 380

推薦閱讀更多精彩內容