Added preframe hook refresh to fix disconnect crash

Sixthworks · Jun 23, 2024 · 5ba60da · 5ba60da
1 parent 62a66ce
commit 5ba60da
Show file tree

Hide file tree

Showing 12 changed files with 79 additions and 3 deletions.
diff --git a/.vs/spankerfield/v17/.suo b/.vs/spankerfield/v17/.suo
diff --git a/.vs/spankerfield/v17/Browse.VC.db b/.vs/spankerfield/v17/Browse.VC.db
diff --git a/spankerfield/Features/Refresh Hook/refresh_hook.cpp b/spankerfield/Features/Refresh Hook/refresh_hook.cpp
@@ -0,0 +1,40 @@
+#include "refresh_hook.h"
+#include "../../Hooks/hooks.h"
+#include "../../Utilities/vtablehook.h"
+
+using namespace big;
+namespace plugins
+{
+	static bool refresh = false;
+	void refresh_preframe_hook()
+	{
+		const auto border_input_node = BorderInputNode::GetInstance();
+		if (!border_input_node) return;
+
+		const auto game_context = ClientGameContext::GetInstance();
+		if (!game_context) return;
+
+		const auto player_manager = game_context->m_pPlayerManager;
+		if (!player_manager) return;
+
+		// Rehooking PreFrameUpdate vtable to prevent crashes when leaving the server (credits Menool)
+		if (!IsValidPtr(player_manager->m_pLocalPlayer))
+		{
+			if (!refresh)
+			{
+				refresh = true;
+
+				hook_vtable_func(reinterpret_cast<PDWORD64*>(border_input_node->m_Vtable), reinterpret_cast<PBYTE>(PreFrame::oPreFrameUpdate), 3);
+			}
+
+			return;
+		}
+		else if (refresh)
+		{
+			refresh = false;
+
+			PreFrame::oPreFrameUpdate = reinterpret_cast<PreFrame::PreFrameUpdate_t>(hook_vtable_func(reinterpret_cast<PDWORD64*>(border_input_node->m_Vtable), reinterpret_cast<PBYTE>(&PreFrame::hkPreFrame), 3));
+			LOG(INFO) << xorstr_("Refreshed PreFrame.");
+		}
+	}
+}
diff --git a/spankerfield/Features/Refresh Hook/refresh_hook.h b/spankerfield/Features/Refresh Hook/refresh_hook.h
@@ -0,0 +1,7 @@
+#pragma once
+#include "../../SDK/sdk.h"
+
+namespace plugins
+{
+	void refresh_preframe_hook();
+}
diff --git a/spankerfield/Features/main.cpp b/spankerfield/Features/main.cpp
@@ -6,6 +6,9 @@ namespace big
 {
 	void features::draw()
 	{
+		// This has to be done before the visuals run
+		plugins::refresh_preframe_hook();
+
 		plugins::draw_screenshots();
 		plugins::draw_crosshair();
 		plugins::draw_spectators();

diff --git a/spankerfield/Features/main.h b/spankerfield/Features/main.h
@@ -17,6 +17,7 @@
 #include "Anti AFK/anti_afk.h"
 #include "Name Spoofer/name_spoofer.h"
 #include "Kill Sound/kill_sound.h"
+#include "Refresh Hook/refresh_hook.h"
 
 namespace big
 {

diff --git a/spankerfield/Hooks/hooks.cpp b/spankerfield/Hooks/hooks.cpp
@@ -2,16 +2,14 @@
 #include "../MinHook.h"
 #include "../Utilities/vtablehook.h"
 #include "../Utilities/other.h"
+#include "../Utilities/thread_pool.h"
 #include "../Rendering/renderer.h"
 #include "../Rendering/gui.h"
 #include "../Features/main.h"
 #include "../settings.h"
 #include "../global.h"
 
-
-#ifdef _WIN64
 #define GWL_WNDPROC GWLP_WNDPROC
-#endif
 
 namespace big
 {

diff --git a/spankerfield/Hooks/hooks.h b/spankerfield/Hooks/hooks.h
@@ -3,13 +3,24 @@
 
 namespace big
 {
+	// For the refresh hook plugin
+	namespace PreFrame
+	{
+		using PreFrameUpdate_t = void(*)(float dt);
+		extern PreFrameUpdate_t oPreFrameUpdate;
+		void hkPreFrame(float delta_time);
+	}
+
+	// Hooking
 	class hooking
 	{
 	public:
 		void initialize();
 		void uninitialize();
 		void enable();
 		void disable();
+
+		bool refresh_preframe;
 	};
 
 	inline hooking* g_hooking{};

diff --git a/spankerfield/Utilities/xorstr.h b/spankerfield/Utilities/xorstr.h
@@ -30,8 +30,13 @@
 #include <utility>
 #include <type_traits>
 
+#ifdef BUILD_FOR_UC
+#define xorstr(str) (str)
+#define xorstr_(str) (str)
+#else
 #define xorstr(str) ::jm::xor_string([]() { return str; }, std::integral_constant<std::size_t, sizeof(str) / sizeof(*str)>{}, std::make_index_sequence<::jm::detail::_buffer_size<sizeof(str)>()>{})
 #define xorstr_(str) xorstr(str).crypt_get()
+#endif
 
 #ifdef _MSC_VER
 #define XORSTR_FORCEINLINE __forceinline

diff --git a/spankerfield/common.h b/spankerfield/common.h
@@ -2,6 +2,9 @@
 
 #pragma comment(lib, "Winmm.lib")
 
+// Made this for myself, disables string encryption so I can upload the .dll on UnknownCheats
+// #define BUILD_FOR_UC
+
 #include <SDKDDKVer.h>
 #include <Windows.h>
 #include <TlHelp32.h>

diff --git a/spankerfield/spankerfield.vcxproj b/spankerfield/spankerfield.vcxproj
@@ -161,6 +161,7 @@
     <ClCompile Include="Features\Name Spoofer\name_spoofer.cpp" />
     <ClCompile Include="Features\Overheat\overheat.cpp" />
     <ClCompile Include="Features\Radar\radar.cpp" />
+    <ClCompile Include="Features\Refresh Hook\refresh_hook.cpp" />
     <ClCompile Include="Features\Risky\risky.cpp" />
     <ClCompile Include="Features\Screenshots\screenshots.cpp" />
     <ClCompile Include="Features\Spectator List\spectators.cpp" />
@@ -213,6 +214,7 @@
     <ClInclude Include="Features\Name Spoofer\name_spoofer.h" />
     <ClInclude Include="Features\Overheat\overheat.h" />
     <ClInclude Include="Features\Radar\radar.h" />
+    <ClInclude Include="Features\Refresh Hook\refresh_hook.h" />
     <ClInclude Include="Features\Risky\risky.h" />
     <ClInclude Include="Features\Screenshots\screenshots.h" />
     <ClInclude Include="logger.h" />

diff --git a/spankerfield/spankerfield.vcxproj.filters b/spankerfield/spankerfield.vcxproj.filters
@@ -183,6 +183,9 @@
     <ClCompile Include="Features\Kill Sound\kill_sound.cpp">
       <Filter>Исходные файлы</Filter>
     </ClCompile>
+    <ClCompile Include="Features\Refresh Hook\refresh_hook.cpp">
+      <Filter>Исходные файлы</Filter>
+    </ClCompile>
   </ItemGroup>
   <ItemGroup>
     <ClInclude Include="Features\ESP\esp.h">
@@ -422,6 +425,9 @@
     <ClInclude Include="Features\Kill Sound\kill_sound.h">
       <Filter>Файлы заголовков</Filter>
     </ClInclude>
+    <ClInclude Include="Features\Refresh Hook\refresh_hook.h">
+      <Filter>Файлы заголовков</Filter>
+    </ClInclude>
   </ItemGroup>
   <ItemGroup>
     <None Include="packages.config" />