From 2d7e01f9386fad169baf9508a6ebe40de1eb1edd Mon Sep 17 00:00:00 2001
From: SciresM <Sciresm@gmail.com>
Date: Thu, 1 Nov 2018 06:54:59 +0900
Subject: [PATCH] Fix crash when taking over some games on 5.0.0+ (#10)

* HBL: Fix crash when taking over some games on 5.0.0+ where the title supports automatic video-recording via the nacp.
---
 source/main.c | 35 +++++++++++++++++++++++++++++++++--
 1 file changed, 33 insertions(+), 2 deletions(-)

diff --git a/source/main.c b/source/main.c
index 36b4db6..2cf4ecf 100644
--- a/source/main.c
+++ b/source/main.c
@@ -14,6 +14,9 @@ static u64  g_nroSize = 0;
 static NroHeader g_nroHeader;
 static bool g_isApplication = 0;
 
+static NsApplicationControlData g_applicationControlData;
+static bool g_isAutomaticGameplayRecording = 0;
+
 static u8 g_savedTls[0x100];
 
 // Used by trampoline.s
@@ -74,6 +77,10 @@ void setupHbHeap(void)
     if (size==0)
         size = 0x2000000*16;
 
+    if (size > 0x6000000 && g_isAutomaticGameplayRecording) {
+        size -= 0x6000000;
+    }
+
     rc = svcSetHeapSize(&addr, size);
 
     if (R_FAILED(rc) || addr==NULL)
@@ -136,6 +143,29 @@ void getIsApplication(void) {
     if (R_SUCCEEDED(rc) && cur_pid == app_pid) g_isApplication = 1;
 }
 
+//Gets the control.nacp for the current title id, and then sets g_isAutomaticGameplayRecording if less memory should be allocated.
+void getIsAutomaticGameplayRecording(void) {
+    if (kernelAbove500() && g_isApplication) {
+        Result rc=0;
+        u64 cur_tid=0;
+
+        rc = svcGetInfo(&cur_tid, 18, CUR_PROCESS_HANDLE, 0);
+        if (R_FAILED(rc)) return;
+
+        g_isAutomaticGameplayRecording = 0;
+
+        rc = nsInitialize();
+
+        if (R_SUCCEEDED(rc)) {
+            size_t dummy;
+            rc = nsGetApplicationControlData(0x1, cur_tid, &g_applicationControlData, sizeof(g_applicationControlData), &dummy);
+            nsExit();
+        }
+
+        if (R_SUCCEEDED(rc) && (((g_applicationControlData.nacp.x3034_unk >> 8) & 0xFF) == 2)) g_isAutomaticGameplayRecording = 1;
+    }
+}
+
 void getOwnProcessHandle(void)
 {
     static Thread t;
@@ -170,7 +200,7 @@ void getOwnProcessHandle(void)
     raw->x = raw->y = 0;
 
     rc = serviceIpcDispatch(&srv);
-    
+
     threadWaitForExit(&t);
     threadClose(&t);
 
@@ -375,8 +405,9 @@ int main(int argc, char **argv)
 {
     memcpy(g_savedTls, (u8*)armGetTls() + 0x100, 0x100);
 
-    setupHbHeap();
     getIsApplication();
+    getIsAutomaticGameplayRecording();
+    setupHbHeap();
     getOwnProcessHandle();
     loadNro();