From 46bed8d09f32360377350d71816a76abdfbf2c7c Mon Sep 17 00:00:00 2001
From: BobTheBob <32057864+BobTheBob9@users.noreply.github.com>
Date: Wed, 24 Aug 2022 00:32:31 +0100
Subject: Adjust folder structure (#242)

* Adjust folder structure

* change launcher directory name
---
 NorthstarDLL/miscserverscript.cpp | 76 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 76 insertions(+)
 create mode 100644 NorthstarDLL/miscserverscript.cpp

(limited to 'NorthstarDLL/miscserverscript.cpp')

diff --git a/NorthstarDLL/miscserverscript.cpp b/NorthstarDLL/miscserverscript.cpp
new file mode 100644
index 00000000..3b06ca10
--- /dev/null
+++ b/NorthstarDLL/miscserverscript.cpp
@@ -0,0 +1,76 @@
+#include "pch.h"
+#include "miscserverscript.h"
+#include "squirrel.h"
+#include "masterserver.h"
+#include "serverauthentication.h"
+#include "gameutils.h"
+#include "dedicated.h"
+
+// annoying helper function because i can't figure out getting players or entities from sqvm rn
+// wish i didn't have to do it like this, but here we are
+void* GetPlayerByIndex(int playerIndex)
+{
+	const int PLAYER_ARRAY_OFFSET = 0x12A53F90;
+	const int PLAYER_SIZE = 0x2D728;
+
+	void* playerArrayBase = (char*)GetModuleHandleA("engine.dll") + PLAYER_ARRAY_OFFSET;
+	void* player = (char*)playerArrayBase + (playerIndex * PLAYER_SIZE);
+
+	return player;
+}
+
+// void function NSEarlyWritePlayerIndexPersistenceForLeave( int playerIndex )
+SQRESULT SQ_EarlyWritePlayerIndexPersistenceForLeave(void* sqvm)
+{
+	int playerIndex = ServerSq_getinteger(sqvm, 1);
+	void* player = GetPlayerByIndex(playerIndex);
+
+	if (!g_ServerAuthenticationManager->m_additionalPlayerData.count(player))
+	{
+		ServerSq_pusherror(sqvm, fmt::format("Invalid playerindex {}", playerIndex).c_str());
+		return SQRESULT_ERROR;
+	}
+
+	g_ServerAuthenticationManager->m_additionalPlayerData[player].needPersistenceWriteOnLeave = false;
+	g_ServerAuthenticationManager->WritePersistentData(player);
+	return SQRESULT_NULL;
+}
+
+// bool function NSIsWritingPlayerPersistence()
+SQRESULT SQ_IsWritingPlayerPersistence(void* sqvm)
+{
+	ServerSq_pushbool(sqvm, g_MasterServerManager->m_savingPersistentData);
+	return SQRESULT_NOTNULL;
+}
+
+// bool function NSIsPlayerIndexLocalPlayer( int playerIndex )
+SQRESULT SQ_IsPlayerIndexLocalPlayer(void* sqvm)
+{
+	int playerIndex = ServerSq_getinteger(sqvm, 1);
+	void* player = GetPlayerByIndex(playerIndex);
+	if (!g_ServerAuthenticationManager->m_additionalPlayerData.count(player))
+	{
+		ServerSq_pusherror(sqvm, fmt::format("Invalid playerindex {}", playerIndex).c_str());
+		return SQRESULT_ERROR;
+	}
+
+	ServerSq_pushbool(sqvm, !strcmp(g_LocalPlayerUserID, (char*)player + 0xF500));
+	return SQRESULT_NOTNULL;
+}
+
+// bool function NSIsDedicated()
+
+SQRESULT SQ_IsDedicated(void* sqvm)
+{
+	ServerSq_pushbool(sqvm, IsDedicated());
+	return SQRESULT_NOTNULL;
+}
+
+void InitialiseMiscServerScriptCommand(HMODULE baseAddress)
+{
+	g_ServerSquirrelManager->AddFuncRegistration(
+		"void", "NSEarlyWritePlayerIndexPersistenceForLeave", "int playerIndex", "", SQ_EarlyWritePlayerIndexPersistenceForLeave);
+	g_ServerSquirrelManager->AddFuncRegistration("bool", "NSIsWritingPlayerPersistence", "", "", SQ_IsWritingPlayerPersistence);
+	g_ServerSquirrelManager->AddFuncRegistration("bool", "NSIsPlayerIndexLocalPlayer", "int playerIndex", "", SQ_IsPlayerIndexLocalPlayer);
+	g_ServerSquirrelManager->AddFuncRegistration("bool", "NSIsDedicated", "", "", SQ_IsDedicated);
+}
\ No newline at end of file
-- 
cgit v1.2.3