test dll changes

dMasterServer/CMakeLists.txt

@@ -1,11 +1,23 @@
 set(DMASTERSERVER_SOURCES
 	"InstanceManager.cpp"
 	"PersistentIDManager.cpp"
-	"Start.cpp"
+	"StartLocal.cpp"
 )
 
+if(WIN32 AND LOCAL_SERVER)
+	set(DMASTERSERVER_SOURCES ${DMASTERSERVER_SOURCES} "StartLocal.cpp") 
+else()
+	set(DMASTERSERVER_SOURCES ${DMASTERSERVER_SOURCES} "Start.cpp") 
+endif()
+
 add_library(dMasterServer ${DMASTERSERVER_SOURCES})
-add_executable(MasterServer "MasterServer.cpp")
+
+if (WIN32 AND LOCAL_SERVER)
+	add_library(MasterServer SHARED "MasterServer.cpp")
+else()
+	add_executable(MasterServer "MasterServer.cpp")
+endif()
+
 target_compile_definitions(MasterServer PRIVATE PROJECT_VERSION="\"${PROJECT_VERSION}\"")
 target_include_directories(dMasterServer PUBLIC "."
 	"${PROJECT_SOURCE_DIR}/dZoneManager" # InstanceManager.h uses dZMCommon.h

dMasterServer/MasterServer.cpp

@@ -62,7 +62,7 @@ std::map<uint32_t, std::string> activeSessions;
 SystemAddress authServerMasterPeerSysAddr;
 SystemAddress chatServerMasterPeerSysAddr;
 
-int main(int argc, char** argv) {
+int start(int argc, char** argv) {
 	constexpr uint32_t masterFramerate = mediumFramerate;
 	constexpr uint32_t masterFrameDelta = mediumFrameDelta;
 	Diagnostics::SetProcessName("Master");
@@ -429,6 +429,44 @@ int main(int argc, char** argv) {
 	return ShutdownSequence(EXIT_SUCCESS);
 }
 
+#ifdef LOCAL_SERVER
+BOOL WINAPI DllMain(
+	HINSTANCE hinstDLL,  // handle to DLL module
+	DWORD fdwReason,     // reason for calling function
+	LPVOID lpvReserved)  // reserved
+{
+	// Perform actions based on the reason for calling.
+	switch (fdwReason) {
+	case DLL_PROCESS_ATTACH:
+		AllocConsole();
+		start(0, nullptr);
+		break;
+
+	case DLL_THREAD_ATTACH:
+		// Do thread-specific initialization.
+		break;
+
+	case DLL_THREAD_DETACH:
+		// Do thread-specific cleanup.
+		break;
+
+	case DLL_PROCESS_DETACH:
+
+		if (lpvReserved != nullptr) {
+			break; // do not do cleanup if process termination scenario
+		}
+
+		// Perform any necessary cleanup.
+		break;
+	}
+	return TRUE;  // Successful DLL_PROCESS_ATTACH.
+}
+#else
+int main(int argc, char** argv) {
+	return start(argc, argv);
+}
+#endif
+
 void HandlePacket(Packet* packet) {
 	if (packet->length < 1) return;
 	if (packet->data[0] == ID_DISCONNECTION_NOTIFICATION) {

dMasterServer/StartLocal.cpp

@@ -0,0 +1,121 @@
+#include <windows.h>
+#include <iostream>
+#include <string>
+#include "Start.h"
+#include "Logger.h"
+#include "dConfig.h"
+#include "Game.h"
+#include "BinaryPathFinder.h"
+
+// Function to create a process and execute a DLL
+bool RunDLLAsApplication(const char* dllPath) {
+	STARTUPINFOA si = { sizeof(STARTUPINFOA) };
+	PROCESS_INFORMATION pi = { 0 };
+
+	// Path to a dummy executable
+	const char* dummyProcessPath = "C:\\Windows\\System32\\notepad.exe";
+
+	// Create a suspended dummy process
+	if (!CreateProcessA(
+		dummyProcessPath,
+		NULL,
+		NULL,
+		NULL,
+		FALSE,
+		CREATE_SUSPENDED,
+		NULL,
+		NULL,
+		&si,
+		&pi)) {
+		std::cerr << "Failed to create dummy process. Error: " << GetLastError() << std::endl;
+		return false;
+	}
+
+	// Get the address of LoadLibraryA in the current process
+	HMODULE kernel32 = GetModuleHandleA("kernel32.dll");
+	if (!kernel32) {
+		std::cerr << "Failed to get handle for kernel32.dll. Error: " << GetLastError() << std::endl;
+		TerminateProcess(pi.hProcess, 1);
+		return false;
+	}
+
+	void* loadLibraryAddr = (void*)GetProcAddress(kernel32, "LoadLibraryA");
+	if (!loadLibraryAddr) {
+		std::cerr << "Failed to get address of LoadLibraryA. Error: " << GetLastError() << std::endl;
+		TerminateProcess(pi.hProcess, 1);
+		return false;
+	}
+
+	// Allocate memory in the target process for the DLL path
+	void* remoteMemory = VirtualAllocEx(pi.hProcess, NULL, strlen(dllPath) + 1, MEM_COMMIT | MEM_RESERVE, PAGE_READWRITE);
+	if (!remoteMemory) {
+		std::cerr << "Failed to allocate memory in target process. Error: " << GetLastError() << std::endl;
+		TerminateProcess(pi.hProcess, 1);
+		return false;
+	}
+
+	// Write the DLL path into the allocated memory
+	if (!WriteProcessMemory(pi.hProcess, remoteMemory, dllPath, strlen(dllPath) + 1, NULL)) {
+		std::cerr << "Failed to write DLL path to target process. Error: " << GetLastError() << std::endl;
+		VirtualFreeEx(pi.hProcess, remoteMemory, 0, MEM_RELEASE);
+		TerminateProcess(pi.hProcess, 1);
+		return false;
+	}
+
+	// Create a remote thread in the target process to load the DLL
+	HANDLE remoteThread = CreateRemoteThread(pi.hProcess, NULL, 0, (LPTHREAD_START_ROUTINE)loadLibraryAddr, remoteMemory, 0, NULL);
+	if (!remoteThread) {
+		std::cerr << "Failed to create remote thread. Error: " << GetLastError() << std::endl;
+		VirtualFreeEx(pi.hProcess, remoteMemory, 0, MEM_RELEASE);
+		TerminateProcess(pi.hProcess, 1);
+		return false;
+	}
+
+	// Wait for the remote thread to complete
+	WaitForSingleObject(remoteThread, INFINITE);
+	CloseHandle(remoteThread);
+
+	// Free the allocated memory
+	VirtualFreeEx(pi.hProcess, remoteMemory, 0, MEM_RELEASE);
+
+	// Resume the main thread of the process
+	ResumeThread(pi.hThread);
+
+	// Clean up process and thread handles
+	CloseHandle(pi.hThread);
+	CloseHandle(pi.hProcess);
+
+	return true;
+}
+
+void StartChatServer() {
+	if (Game::ShouldShutdown()) {
+		LOG("Currently shutting down.  Chat will not be restarted.");
+		return;
+	}
+
+	RunDLLAsApplication("ChatServer.dll");
+}
+
+void StartAuthServer() {
+	if (Game::ShouldShutdown()) {
+		LOG("Currently shutting down.  Auth will not be restarted.");
+		return;
+	}
+
+	RunDLLAsApplication("AuthServer.dll");
+}
+
+void StartWorldServer(LWOMAPID mapID, uint16_t port, LWOINSTANCEID lastInstanceID, int maxPlayers, LWOCLONEID cloneID) {
+	RunDLLAsApplication("WorldServer.dll");
+
+	//cmd.append(std::to_string(mapID));
+	//cmd.append(" -port ");
+	//cmd.append(std::to_string(port));
+	//cmd.append(" -instance ");
+	//cmd.append(std::to_string(lastInstanceID));
+	//cmd.append(" -maxclients ");
+	//cmd.append(std::to_string(maxPlayers));
+	//cmd.append(" -clone ");
+	//cmd.append(std::to_string(cloneID));
+}