From 0160d05e8ad20ef9666e98108d5c95fb9d74ceb1 Mon Sep 17 00:00:00 2001
From: Ryan Fleury <ryan.j.fleury@gmail.com>
Date: Wed, 22 May 2024 11:39:23 -0700
Subject: [PATCH] further tweaks to mule_peb_trample

---
 build.bat                          |  2 ++
 src/mule/mule_peb_trample.c        |  7 +------
 src/mule/mule_peb_trample_reload.c | 10 ++++++++++
 3 files changed, 13 insertions(+), 6 deletions(-)
 create mode 100644 src/mule/mule_peb_trample_reload.c

diff --git a/build.bat b/build.bat
index eb6653d8..e23cdf5f 100644
--- a/build.bat
+++ b/build.bat
@@ -111,6 +111,8 @@ if "%mule_module%"=="1"                %compile%             ..\src\mule\mule_mo
 if "%mule_hotload%"=="1"               %compile% ..\src\mule\mule_hotload_main.c %compile_link% %out%mule_hotload.exe & %compile% ..\src\mule\mule_hotload_module_main.c %compile_link% %link_dll% %out%mule_hotload_module.dll || exit /b 1
 if "%mule_peb_trample%"=="1" (
   if exist mule_peb_trample.exe move mule_peb_trample.exe mule_peb_trample_old_%random%.exe
+  if exist mule_peb_trample_new.pdb move mule_peb_trample_new.pdb mule_peb_trample_old_%random%.pdb
+  if exist mule_peb_trample_new.rdi move mule_peb_trample_new.rdi mule_peb_trample_old_%random%.rdi
   %compile% ..\src\mule\mule_peb_trample.c %compile_link% %out%mule_peb_trample_new.exe || exit /b 1
   move mule_peb_trample_new.exe mule_peb_trample.exe
 )
diff --git a/src/mule/mule_peb_trample.c b/src/mule/mule_peb_trample.c
index f44bcea1..a0aaa2c9 100644
--- a/src/mule/mule_peb_trample.c
+++ b/src/mule/mule_peb_trample.c
@@ -1,5 +1,6 @@
 #include <windows.h>
 #include <winternl.h>
+#include "mule_peb_trample_reload.c"
 
 static void
 HideModuleFromWindowsReload(HMODULE ModuleToFlush)
@@ -32,12 +33,6 @@ HideModuleFromWindowsReload(HMODULE ModuleToFlush)
   }
 }
 
-__declspec(dllexport) int
-loop_iteration(int it)
-{
-  return it*it;
-}
-
 int main(int argument_count, char **arguments)
 {
   char *exe_name = arguments[0];
diff --git a/src/mule/mule_peb_trample_reload.c b/src/mule/mule_peb_trample_reload.c
new file mode 100644
index 00000000..8e90550f
--- /dev/null
+++ b/src/mule/mule_peb_trample_reload.c
@@ -0,0 +1,10 @@
+__declspec(dllexport) int
+loop_iteration(int it)
+{
+  int sum = 0;
+  for(int i = 0; i < 1000; i += 1)
+  {
+    sum += it*i;
+  }
+  return sum;
+}