From c80eb26135cb01b0711651bc1be321cf0f8cb955 Mon Sep 17 00:00:00 2001
From: Michael Scire <SciresM@gmail.com>
Date: Wed, 27 Feb 2019 04:27:49 -0800
Subject: [PATCH] dmnt-cheat: continue fleshing out cheat manager code

---
 .../dmnt/source/dmnt_cheat_manager.cpp        | 82 ++++++++++++++++---
 .../dmnt/source/dmnt_cheat_manager.hpp        |  8 +-
 2 files changed, 79 insertions(+), 11 deletions(-)

diff --git a/stratosphere/dmnt/source/dmnt_cheat_manager.cpp b/stratosphere/dmnt/source/dmnt_cheat_manager.cpp
index 0c849f49d..6d29e3d7d 100644
--- a/stratosphere/dmnt/source/dmnt_cheat_manager.cpp
+++ b/stratosphere/dmnt/source/dmnt_cheat_manager.cpp
@@ -21,10 +21,57 @@
 static HosMutex g_cheat_lock;
 static HosThread g_detect_thread, g_vm_thread;
 
-static bool g_has_cheat_process;
 static CheatProcessMetadata g_cheat_process_metadata = {0};
 static Handle g_cheat_process_debug_hnd = 0;
 
+void DmntCheatManager::CloseActiveCheatProcess() {
+    if (g_cheat_process_debug_hnd != 0) {
+        svcCloseHandle(g_cheat_process_debug_hnd);
+        g_cheat_process_debug_hnd = 0;
+        g_cheat_process_metadata = (CheatProcessMetadata){0};
+    }
+}
+
+bool DmntCheatManager::HasActiveCheatProcess() {
+    u64 tmp;
+    bool has_cheat_process = g_cheat_process_debug_hnd != 0;
+    
+    if (has_cheat_process) {
+        has_cheat_process &= R_SUCCEEDED(svcGetProcessId(&tmp, g_cheat_process_debug_hnd));
+    }
+    
+    if (has_cheat_process) {
+        has_cheat_process &= R_SUCCEEDED(pmdmntGetApplicationPid(&tmp));
+    }
+    
+    if (has_cheat_process) {
+        has_cheat_process &= tmp == g_cheat_process_metadata.process_id;
+    }
+    
+    if (!has_cheat_process) {
+        CloseActiveCheatProcess();
+    }
+    
+    return has_cheat_process;
+}
+
+void DmntCheatManager::ContinueCheatProcess() {
+    if (HasActiveCheatProcess()) {
+        /* Loop getting debug events. */
+        u8 tmp;
+        while (R_SUCCEEDED(svcGetDebugEvent(&tmp, g_cheat_process_debug_hnd))) {
+            /* ... */
+        }
+        
+        /* Continue the process. */
+        if (kernelAbove300()) {
+            svcContinueDebugEvent(g_cheat_process_debug_hnd, 5, nullptr, 0);
+        } else {
+            svcLegacyContinueDebugEvent(g_cheat_process_debug_hnd, 5, 0);
+        }
+    }
+}
+
 Handle DmntCheatManager::PrepareDebugNextApplication() {
     Result rc;
     Handle event_h;
@@ -53,11 +100,7 @@ void DmntCheatManager::OnNewApplicationLaunch() {
     Result rc;
     
     /* Close the current application, if it's open. */
-    if (g_cheat_process_debug_hnd != 0) {
-        svcCloseHandle(g_cheat_process_debug_hnd);
-        g_cheat_process_debug_hnd = 0;
-        g_cheat_process_metadata = (CheatProcessMetadata){0};
-    }
+    CloseActiveCheatProcess();
     
     /* Get the new application's process ID. */
     if (R_FAILED((rc = pmdmntGetApplicationPid(&g_cheat_process_metadata.process_id)))) {
@@ -88,7 +131,7 @@ void DmntCheatManager::OnNewApplicationLaunch() {
     {
         LoaderModuleInfo proc_modules[2];
         u32 num_modules;
-        if (R_FAILED((rc = ldrDmntGetModuleInfos(g_cheat_process_metadata.process_id, &proc_modules, 2, &num_modules)))) {
+        if (R_FAILED((rc = ldrDmntGetModuleInfos(g_cheat_process_metadata.process_id, proc_modules, 2, &num_modules)))) {
             fatalSimple(rc);
         }
         
@@ -102,15 +145,18 @@ void DmntCheatManager::OnNewApplicationLaunch() {
         
         g_cheat_process_metadata.main_nso_extents.base = proc_modules[1].base_address;
         g_cheat_process_metadata.main_nso_extents.size = proc_modules[1].size;
-        g_cheat_process_metadata.main_nso_build_id = proc_modules[1].build_id;
+        memcpy(g_cheat_process_metadata.main_nso_build_id, proc_modules[1].build_id, sizeof(g_cheat_process_metadata.main_nso_build_id));
     }
     
+    /* TODO: Read cheats off the SD. */
+    
     /* Open a debug handle. */
     if (R_FAILED((rc = svcDebugActiveProcess(&g_cheat_process_debug_hnd, g_cheat_process_metadata.process_id)))) {
         fatalSimple(rc);
     }
     
-    /* TODO: Continue debug events, etc. */
+    /* Continue debug events, etc. */
+    ContinueCheatProcess();
 }
 
 void DmntCheatManager::DetectThread(void *arg) {
@@ -131,11 +177,27 @@ void DmntCheatManager::DetectThread(void *arg) {
  
 void DmntCheatManager::VmThread(void *arg) {
     while (true) {
-        /* TODO */
+        /* Execute Cheat VM. */
+        {
+            /* Acquire lock. */
+            std::scoped_lock<HosMutex> lk(g_cheat_lock);
+            
+            if (HasActiveCheatProcess()) {
+                ContinueCheatProcess();
+                
+                /* TODO: Execute VM. */
+            }
+        }
         svcSleepThread(0x5000000ul);
     }
 }
 
+bool DmntCheatManager::GetHasActiveCheatProcess() {
+    std::scoped_lock<HosMutex> lk(g_cheat_lock);
+    
+    return HasActiveCheatProcess();
+}
+
 void DmntCheatManager::InitializeCheatManager() {
     /* Spawn application detection thread, spawn cheat vm thread. */
     if (R_FAILED(g_detect_thread.Initialize(&DmntCheatManager::DetectThread, nullptr, 0x4000, 28))) {
diff --git a/stratosphere/dmnt/source/dmnt_cheat_manager.hpp b/stratosphere/dmnt/source/dmnt_cheat_manager.hpp
index a0ad6d3c7..afbd63715 100644
--- a/stratosphere/dmnt/source/dmnt_cheat_manager.hpp
+++ b/stratosphere/dmnt/source/dmnt_cheat_manager.hpp
@@ -26,6 +26,12 @@ class DmntCheatManager {
         static void OnNewApplicationLaunch();
         static void DetectThread(void *arg);
         static void VmThread(void *arg);
-    public:
+        
+        static bool HasActiveCheatProcess();
+        static void CloseActiveCheatProcess();
+        static void ContinueCheatProcess();
+    public:;
+        static bool GetHasActiveCheatProcess();
+        
         static void InitializeCheatManager();
 };