fix Test CallLoop bugs when builded with win32

DarthTon · Jul 17, 2023 · 5ede6ce · 5ede6ce
1 parent 61fe525
commit 5ede6ce
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/BlackBone/Process/RPC/RemoteExec.cpp b/src/BlackBone/Process/RPC/RemoteExec.cpp
@@ -553,6 +553,11 @@ NTSTATUS RemoteExec::PrepareCallAssembly(
     }
 
     a.GenPrologue();
+    if (_process.core().isWow64())
+    {
+        a->pusha();
+        a->pushf();
+    }
     a.GenCall( pfn, args, cc );
 
     // Retrieve result from XMM0 or ST0
@@ -571,6 +576,11 @@ NTSTATUS RemoteExec::PrepareCallAssembly(
     }
 
     AddReturnWithEvent( a, mt_default, retType );
+    if (_process.core().isWow64())
+    {
+        a->popf();
+        a->popa();
+    }
     a.GenEpilogue();
 
     return STATUS_SUCCESS;