From d9ddb65b8c27cff9929ca86834add18dcb4cd5c2 Mon Sep 17 00:00:00 2001
From: wheremyfoodat <44909372+wheremyfoodat@users.noreply.github.com>
Date: Sun, 24 Sep 2023 18:22:03 +0300
Subject: [PATCH] [NDM] Add exclusive state

---
 include/services/ndm.hpp  |  5 +++++
 src/core/services/ndm.cpp | 25 +++++++++++++++++++++++--
 2 files changed, 28 insertions(+), 2 deletions(-)

diff --git a/include/services/ndm.hpp b/include/services/ndm.hpp
index f8a964c5..6d4e5ad8 100644
--- a/include/services/ndm.hpp
+++ b/include/services/ndm.hpp
@@ -6,6 +6,8 @@
 #include "result/result.hpp"
 
 class NDMService {
+	enum class ExclusiveState : u32 { None = 0, Infrastructure = 1, LocalComms = 2, StreetPass = 3, StreetPassData = 4 };
+
 	Handle handle = KernelHandles::NDM;
 	Memory& mem;
 	MAKE_LOG_FUNCTION(log, ndmLogger)
@@ -15,11 +17,14 @@ class NDMService {
 	void enterExclusiveState(u32 messagePointer);
 	void exitExclusiveState(u32 messagePointer);
 	void overrideDefaultDaemons(u32 messagePointer);
+	void queryExclusiveState(u32 messagePointer);
 	void resumeDaemons(u32 messagePointer);
 	void resumeScheduler(u32 messagePointer);
 	void suspendDaemons(u32 messagePointer);
 	void suspendScheduler(u32 messagePointer);
 
+	ExclusiveState exclusiveState = ExclusiveState::None;
+
 public:
 	NDMService(Memory& mem) : mem(mem) {}
 	void reset();
diff --git a/src/core/services/ndm.cpp b/src/core/services/ndm.cpp
index 2c68e81f..65de1399 100644
--- a/src/core/services/ndm.cpp
+++ b/src/core/services/ndm.cpp
@@ -4,7 +4,8 @@
 namespace NDMCommands {
 	enum : u32 {
 		EnterExclusiveState = 0x00010042,
-		ExitExclusiveState = 0x00020002,	
+		ExitExclusiveState = 0x00020002,
+		QueryExclusiveMode = 0x00030000,
 		OverrideDefaultDaemons = 0x00140040,
 		SuspendDaemons = 0x00060040,
 		ResumeDaemons = 0x00070040,
@@ -14,7 +15,7 @@ namespace NDMCommands {
 	};
 }
 
-void NDMService::reset() {}
+void NDMService::reset() { exclusiveState = ExclusiveState::None; }
 
 void NDMService::handleSyncRequest(u32 messagePointer) {
 	const u32 command = mem.read32(messagePointer);
@@ -23,6 +24,7 @@ void NDMService::handleSyncRequest(u32 messagePointer) {
 		case NDMCommands::ExitExclusiveState: exitExclusiveState(messagePointer); break;
 		case NDMCommands::ClearHalfAwakeMacFilter: clearHalfAwakeMacFilter(messagePointer); break;
 		case NDMCommands::OverrideDefaultDaemons: overrideDefaultDaemons(messagePointer); break;
+		case NDMCommands::QueryExclusiveMode: queryExclusiveState(messagePointer); break;
 		case NDMCommands::ResumeDaemons: resumeDaemons(messagePointer); break;
 		case NDMCommands::ResumeScheduler: resumeScheduler(messagePointer); break;
 		case NDMCommands::SuspendDaemons: suspendDaemons(messagePointer); break;
@@ -33,16 +35,35 @@ void NDMService::handleSyncRequest(u32 messagePointer) {
 
 void NDMService::enterExclusiveState(u32 messagePointer) {
 	log("NDM::EnterExclusiveState (stubbed)\n");
+	const u32 state = mem.read32(messagePointer + 4);
+
+	// Check that the exclusive state config is valid
+	if (state > 4) {
+		Helpers::warn("NDM::EnterExclusiveState: Invalid state %d", state);
+	} else {
+		exclusiveState = static_cast<ExclusiveState>(state);
+	}
+
 	mem.write32(messagePointer, IPC::responseHeader(0x1, 1, 0));
 	mem.write32(messagePointer + 4, Result::Success);
 }
 
 void NDMService::exitExclusiveState(u32 messagePointer) {
 	log("NDM::ExitExclusiveState (stubbed)\n");
+	exclusiveState = ExclusiveState::None;
+
 	mem.write32(messagePointer, IPC::responseHeader(0x2, 1, 0));
 	mem.write32(messagePointer + 4, Result::Success);
 }
 
+void NDMService::queryExclusiveState(u32 messagePointer) {
+	log("NDM::QueryExclusiveState\n");
+
+	mem.write32(messagePointer, IPC::responseHeader(0x3, 1, 0));
+	mem.write32(messagePointer + 4, Result::Success);
+	mem.write32(messagePointer + 8, static_cast<u32>(exclusiveState));
+}
+
 void NDMService::overrideDefaultDaemons(u32 messagePointer) {
 	log("NDM::OverrideDefaultDaemons (stubbed)\n");
 	mem.write32(messagePointer, IPC::responseHeader(0x14, 1, 0));