修改Hook测试代码

Author: Co1Swet

bin/MulNX/CS2Injector/CS2Injector.exe

@@ -7,4 +7,7 @@ class ITimeAdapter {
 
     virtual float GetTime() = 0;
     virtual bool SetTime(float time) = 0;
+
+    virtual void OnSetOn(float time) {};
+    virtual void OnSetOff() {};
 };

bin/MulNX/CS2InternalHelper/CS2InternalHelper.dll

@@ -6,7 +6,6 @@
 void TimeLiner::Menu() {
     MulNX::UI::RAIIWindow w("时间轴");
 
-    // 没有绑定适配器时，仅绘制静态时间轴
     if (!pTimeAdapter) {
         ImGui::Text("未连接时间适配器");
         return;

bin/MulNX/CS2OBTool/CS2OBTool.dll

@@ -1,11 +1,29 @@
 #include <MulNXUtils/WinExt/WinExt.hpp>
 int Test(int rcx, int rdx, int r8, int r9, int stack4, int stack5) {
-    auto str1 = std::to_string(stack4);
-    auto str2 = std::to_string(stack5);
-    MessageBoxA(NULL, str1.c_str(), str2.c_str(), MB_OK);
+    auto str = std::format("{} {} {}", rcx, stack4, stack5);
+    MessageBoxA(NULL, str.c_str(), str.c_str(), MB_OK);
     return 0;
 }
+class Hooker {
+    std::unique_ptr<MulNX::Hook> hook = nullptr;
+    MulNX::Hook::Then OnTest(MulNX::Hook* hk, RegContext* ctx) {
+        if (!this->enable)return MulNX::Hook::Then::Continue;
+        ctx->rcx = std::bit_cast<uint32_t>(-1);
+        *hk->GetStackParam<int>(ctx, 4) = 42;
+        hk->CallMaybeOrigin(2, ctx); // 栈上两个参数
+        return MulNX::Hook::Then::Return;
+    }
+public:
+    void DoHook() {
+        this->hook = MulNX::Hook::Create((uint8_t*)&Test, [this](MulNX::Hook* hk, RegContext* ctx) {
+            return this->OnTest(hk, ctx);
+            }).value();
+        this->hook->Attach();
+    }
+    bool enable = true;
+};
 int main() {
+    Hooker hooker{};
     {
         auto hkMessageBoxW = MulNX::Hook::Create((uint8_t*)&MessageBoxW, [](MulNX::Hook* hk, RegContext* ctx) {
             if (wcscmp((wchar_t*)(ctx->rdx), L"no show"))return MulNX::Hook::Then::Return;
@@ -25,10 +43,8 @@ int main() {
         MessageBoxW(NULL, L"show hacked", L"example", MB_OK);
     }
     Test(0, 1, 2, 3, 4, 5);
-    auto hkTest = MulNX::Hook::Create((uint8_t*)&Test, [](MulNX::Hook* hk, RegContext* ctx) {
-        hk->CallMaybeOrigin(2, ctx); // 栈上两个参数
-        return MulNX::Hook::Then::Return;
-        }).value();
-    hkTest->Attach();
+    hooker.DoHook();
+    Test(0, 1, 2, 3, 4, 5);
+    hooker.enable = false;
     Test(0, 1, 2, 3, 4, 5);
 }