Prevent the app from sleeping when the lid is closed

windows-server-2003 · Nov 28, 2021 · 4022e99 · 4022e99
1 parent e1f08cc
commit 4022e99
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 0 deletions.
diff --git a/include/network/network_io.hpp b/include/network/network_io.hpp
@@ -59,6 +59,9 @@ NetworkResult Access_http_post(NetworkSessionList &session_list, const std::stri
 
 std::string url_get_host_name(const std::string &url);
 
+void lock_network_state();
+void unlock_network_state();
+
 #define HTTP_STATUS_CODE_OK 200
 #define HTTP_STATUS_CODE_NO_CONTENT 204
 #define HTTP_STATUS_CODE_PARTIAL_CONTENT 206

diff --git a/source/network/network_io.cpp b/source/network/network_io.cpp
@@ -641,5 +641,24 @@ std::string NetworkResult::get_header(std::string key) {
 	}
 }
 
+static bool exclusive_state_entered = false;
+void lock_network_state() {
+	int res = 0;
+	if (!exclusive_state_entered) {
+		Util_log_save("init", "ndmuInit()...", ndmuInit());
+		res = NDMU_EnterExclusiveState(NDM_EXCLUSIVE_STATE_INFRASTRUCTURE);
+		if (R_SUCCEEDED(res)) res = NDMU_LockState(); // prevents ndm from switching to StreetPass when the lid is closed
+		exclusive_state_entered = R_SUCCEEDED(res);
+	}
+}
+void unlock_network_state() {
+	int res = 0;
+	if (exclusive_state_entered) {
+		res = NDMU_UnlockState();
+		if (R_SUCCEEDED(res)) res = NDMU_LeaveExclusiveState();
+		ndmuExit();
+		exclusive_state_entered = R_FAILED(res);
+	}
+}
 
 
diff --git a/source/scene_switcher.cpp b/source/scene_switcher.cpp
@@ -58,6 +58,9 @@ void Menu_init(void)
 	Util_log_save(DEF_MENU_INIT_STR, "amInit()...", amInit());
 	Util_log_save(DEF_MENU_INIT_STR, "ndspInit()...", (sound_init_result = ndspInit()));//0xd880A7FA
 	Util_log_save(DEF_MENU_INIT_STR, "APT_SetAppCpuTimeLimit()...", APT_SetAppCpuTimeLimit(30));
+	lock_network_state();
+
+	aptSetSleepAllowed(false);
 
 	APT_CheckNew3DS(&is_new_3ds);
 
@@ -142,6 +145,7 @@ void Menu_exit(void)
 	thumbnail_downloader_thread_exit_request();
 	async_task_thread_exit_request();
 	misc_tasks_thread_exit_request();
+	unlock_network_state();
 	Util_log_save(DEF_MENU_EXIT_STR, "threadJoin()...", threadJoin(menu_worker_thread, time_out));
 	Util_log_save(DEF_MENU_EXIT_STR, "threadJoin()...", threadJoin(menu_check_connectivity_thread, time_out));
 	// Util_log_save(DEF_MENU_EXIT_STR, "threadJoin()...", threadJoin(menu_send_app_info_thread, time_out));