2 files changed, 118 insertions, 4 deletions

diff --git a/win/inspircd_win32wrapper.cpp b/win/inspircd_win32wrapper.cpp
index ad45bb0ae..ee1d54c25 100644
--- a/win/inspircd_win32wrapper.cpp
+++ b/win/inspircd_win32wrapper.cpp
@@ -22,7 +22,9 @@ using namespace std;
 #define INADDR_NONE 0xffffffff
 #endif
 
-HANDLE hIPCPipe;
+/* This MUST remain static and delcared outside the class, so that WriteProcessMemory can reference it properly */
+static DWORD owner_processid = 0;
+
 
 int inet_aton(const char *cp, struct in_addr *addr)
 {
@@ -511,3 +513,104 @@ void ClearConsole()
 	}
 	return;
 }
+
+DWORD WindowsForkStart(InspIRCd * Instance)
+{
+        /* Windows implementation of fork() :P */
+	if (owner_processid)
+		return 0;
+
+        char module[MAX_PATH];
+        if(!GetModuleFileName(NULL, module, MAX_PATH))
+        {
+                printf("GetModuleFileName() failed.\n");
+                return false;
+        }
+
+        STARTUPINFO startupinfo;
+        PROCESS_INFORMATION procinfo;
+        ZeroMemory(&startupinfo, sizeof(STARTUPINFO));
+        ZeroMemory(&procinfo, sizeof(PROCESS_INFORMATION));
+
+        // Fill in the startup info struct
+        GetStartupInfo(&startupinfo);
+
+        /* Default creation flags create the processes suspended */
+        DWORD startupflags = CREATE_SUSPENDED;
+
+        /* On windows 2003/XP and above, we can use the value
+         * CREATE_PRESERVE_CODE_AUTHZ_LEVEL which gives more access
+         * to the process which we may require on these operating systems.
+         */
+        OSVERSIONINFO vi;
+        vi.dwOSVersionInfoSize = sizeof(OSVERSIONINFO);
+        GetVersionEx(&vi);
+        if ((vi.dwMajorVersion >= 5) && (vi.dwMinorVersion > 0))
+                startupflags |= CREATE_PRESERVE_CODE_AUTHZ_LEVEL;
+
+        // Launch our "forked" process.
+        BOOL bSuccess = CreateProcess ( module, // Module (exe) filename
+                strdup(GetCommandLine()),       // Command line (exe plus parameters from the OS)
+                                                // NOTE: We cannot return the direct value of the
+                                                // GetCommandLine function here, as the pointer is
+                                                // passed straight to the child process, and will be
+                                                // invalid once we exit as it goes out of context.
+                                                // strdup() seems ok, though.
+                0,                              // PROCESS_SECURITY_ATTRIBUTES
+                0,                              // THREAD_SECURITY_ATTRIBUTES
+                TRUE,                           // We went to inherit handles.
+                startupflags,                   // Allow us full access to the process and suspend it.
+                0,                              // ENVIRONMENT
+                0,                              // CURRENT_DIRECTORY
+                &startupinfo,                   // startup info
+                &procinfo);                     // process info
+
+        if(!bSuccess)
+        {
+                printf("CreateProcess() error: %s\n", dlerror());
+                return false;
+        }
+
+        // Set the owner process id in the target process.
+        SIZE_T written = 0;
+        DWORD pid = GetCurrentProcessId();
+        if(!WriteProcessMemory(procinfo.hProcess, &owner_processid, &pid, sizeof(DWORD), &written) || written != sizeof(DWORD))
+        {
+                printf("WriteProcessMemory() failed: %s\n", dlerror());
+                return false;
+        }
+
+        // Resume the other thread (let it start)
+        ResumeThread(procinfo.hThread);
+
+        // Wait for the new process to kill us. If there is some error, the new process will end and we will end up at the next line.
+        WaitForSingleObject(procinfo.hProcess, INFINITE);
+
+        // If we hit this it means startup failed, default to 14 if this fails.
+        DWORD ExitCode = 14;
+        GetExitCodeProcess(procinfo.hProcess, &ExitCode);
+        CloseHandle(procinfo.hThread);
+        CloseHandle(procinfo.hProcess);
+        return ExitCode;
+}
+
+void WindowsForkKillOwner(InspIRCd * Instance)
+{
+        HANDLE hProcess = OpenProcess(PROCESS_TERMINATE, FALSE, owner_processid);
+        if(!hProcess || !owner_processid)
+        {
+                printf("Could not open process id %u: %s.\n", owner_processid, dlerror());
+                Instance->Exit(14);
+        }
+
+        // die die die
+        if(!TerminateProcess(hProcess, 0))
+        {
+                printf("Could not TerminateProcess(): %s\n", dlerror());
+                Instance->Exit(14);
+        }
+
+        CloseHandle(hProcess);
+}
+
+
diff --git a/win/inspircd_win32wrapper.h b/win/inspircd_win32wrapper.h
index fb6882378..73969e6b5 100644
--- a/win/inspircd_win32wrapper.h
+++ b/win/inspircd_win32wrapper.h
@@ -175,9 +175,16 @@ void ::operator delete(void * ptr);
 /* IPC Handlers */
 class InspIRCd;
 
-void InitIPC();
-void CheckIPC(InspIRCd * Instance);
-void CloseIPC();
+class IPC
+{
+ private:
+	InspIRCd* Instance;
+	HANDLE hIPCPipe;
+ public:
+	void IPC();
+	void CheckIPC(InspIRCd* Srv);
+	void ~IPC();
+};
 
 /* Look up the nameserver in use from the registry on windows */
 std::string FindNameServerWin();
@@ -185,5 +192,9 @@ std::string FindNameServerWin();
 /* Clear a windows console */
 void ClearConsole();
 
+DWORD WindowsForkStart(InspIRCd* Instance);
+
+void WindowsForkKillOwner(InspIRCd* Instance);
+
 #endif