From f37b47f2d08c75266e46c9fff54d52b032bef5e3 Mon Sep 17 00:00:00 2001
From: wheremyfoodat <44909372+wheremyfoodat@users.noreply.github.com>
Date: Sun, 10 Sep 2023 16:57:55 +0300
Subject: [PATCH] More NDM stubbing

---
 include/services/ndm.hpp  |  2 ++
 src/core/services/ndm.cpp | 16 ++++++++++++++++
 2 files changed, 18 insertions(+)

diff --git a/include/services/ndm.hpp b/include/services/ndm.hpp
index 4d21ea3e..f8a964c5 100644
--- a/include/services/ndm.hpp
+++ b/include/services/ndm.hpp
@@ -12,6 +12,8 @@ class NDMService {
 
 	// Service commands
 	void clearHalfAwakeMacFilter(u32 messagePointer);
+	void enterExclusiveState(u32 messagePointer);
+	void exitExclusiveState(u32 messagePointer);
 	void overrideDefaultDaemons(u32 messagePointer);
 	void resumeDaemons(u32 messagePointer);
 	void resumeScheduler(u32 messagePointer);
diff --git a/src/core/services/ndm.cpp b/src/core/services/ndm.cpp
index ecda3df9..2c68e81f 100644
--- a/src/core/services/ndm.cpp
+++ b/src/core/services/ndm.cpp
@@ -3,6 +3,8 @@
 
 namespace NDMCommands {
 	enum : u32 {
+		EnterExclusiveState = 0x00010042,
+		ExitExclusiveState = 0x00020002,	
 		OverrideDefaultDaemons = 0x00140040,
 		SuspendDaemons = 0x00060040,
 		ResumeDaemons = 0x00070040,
@@ -17,6 +19,8 @@ void NDMService::reset() {}
 void NDMService::handleSyncRequest(u32 messagePointer) {
 	const u32 command = mem.read32(messagePointer);
 	switch (command) {
+		case NDMCommands::EnterExclusiveState: enterExclusiveState(messagePointer); break;
+		case NDMCommands::ExitExclusiveState: exitExclusiveState(messagePointer); break;
 		case NDMCommands::ClearHalfAwakeMacFilter: clearHalfAwakeMacFilter(messagePointer); break;
 		case NDMCommands::OverrideDefaultDaemons: overrideDefaultDaemons(messagePointer); break;
 		case NDMCommands::ResumeDaemons: resumeDaemons(messagePointer); break;
@@ -27,6 +31,18 @@ void NDMService::handleSyncRequest(u32 messagePointer) {
 	}
 }
 
+void NDMService::enterExclusiveState(u32 messagePointer) {
+	log("NDM::EnterExclusiveState (stubbed)\n");
+	mem.write32(messagePointer, IPC::responseHeader(0x1, 1, 0));
+	mem.write32(messagePointer + 4, Result::Success);
+}
+
+void NDMService::exitExclusiveState(u32 messagePointer) {
+	log("NDM::ExitExclusiveState (stubbed)\n");
+	mem.write32(messagePointer, IPC::responseHeader(0x2, 1, 0));
+	mem.write32(messagePointer + 4, Result::Success);
+}
+
 void NDMService::overrideDefaultDaemons(u32 messagePointer) {
 	log("NDM::OverrideDefaultDaemons (stubbed)\n");
 	mem.write32(messagePointer, IPC::responseHeader(0x14, 1, 0));