reactos/win32ss/user/winsrv/consrv/shutdown.c
Giannis Adamopoulos 4bef6f5913 [WINSRV] consrv: Implement terminating console processes on shutdown
Also stub handling non-console, non-gui processes.
2019-01-04 13:29:26 +02:00

113 lines
3.6 KiB
C

/*
* COPYRIGHT: See COPYING in the top level directory
* PROJECT: ReactOS Console Server DLL
* FILE: win32ss/user/winsrv/consrv/shutdown.c
* PURPOSE: Processes Shutdown
* PROGRAMMERS: Alex Ionescu
*/
/* INCLUDES *******************************************************************/
#include "consrv.h"
#include <psapi.h>
#define NDEBUG
#include <debug.h>
/* FUNCTIONS ******************************************************************/
static void
NotifyConsoleProcessForShutdown(IN PCSR_PROCESS CsrProcess,
IN PCONSOLE_PROCESS_DATA ProcessData,
IN ULONG Flags)
{
DPRINT1("ConsoleClientShutdown(0x%p, 0x%x) - Console process [0x%x, 0x%x]\n",
CsrProcess, Flags, CsrProcess->ClientId.UniqueProcess, CsrProcess->ClientId.UniqueThread);
/* Send a log-off event. In reality this should be way more complex */
ConSrvConsoleCtrlEventTimeout(CTRL_LOGOFF_EVENT, ProcessData,
ShutdownSettings.WaitToKillAppTimeout);
}
static BOOL
NotifyGenericProcessForShutdown(IN PCSR_PROCESS CsrProcess,
IN ULONG Flags)
{
/* FIXME: Implement the generic process shutdown handler */
UNIMPLEMENTED_ONCE;
return TRUE;
}
ULONG
NTAPI
NonConsoleProcessShutdown(IN PCSR_PROCESS Process,
IN ULONG Flags)
{
if (NotifyGenericProcessForShutdown(Process, Flags))
{
/* Terminate this process */
#if DBG
WCHAR buffer[MAX_PATH];
if (!GetProcessImageFileNameW(Process->ProcessHandle, buffer, MAX_PATH))
{
DPRINT1("Terminating process %x\n", Process->ClientId.UniqueProcess);
}
else
{
DPRINT1("Terminating process %x (%S)\n", Process->ClientId.UniqueProcess, buffer);
}
#endif
NtTerminateProcess(Process->ProcessHandle, 0);
WaitForSingleObject(Process->ProcessHandle, ShutdownSettings.ProcessTerminateTimeout);
}
CsrDereferenceProcess(Process);
return CsrShutdownCsrProcess;
}
// NOTE: See http://blogs.msdn.com/b/ntdebugging/archive/2007/06/09/how-windows-shuts-down.aspx
ULONG
NTAPI
ConsoleClientShutdown(IN PCSR_PROCESS CsrProcess,
IN ULONG Flags,
IN BOOLEAN FirstPhase)
{
PCONSOLE_PROCESS_DATA ProcessData = ConsoleGetPerProcessData(CsrProcess);
/* Do not kill system processes when a user is logging off */
if ((Flags & EWX_SHUTDOWN) == EWX_LOGOFF &&
(CsrProcess->ShutdownFlags & (CsrShutdownSystem | CsrShutdownOther)))
{
DPRINT("Do not kill a system process in a logoff request!\n");
return CsrShutdownNonCsrProcess;
}
/* Is it a console process? */
if ( ProcessData->ConsoleHandle != NULL ||
ProcessData->HandleTable != NULL )
{
NotifyConsoleProcessForShutdown(CsrProcess, ProcessData, Flags);
/* We are done with the process itself */
CsrDereferenceProcess(CsrProcess);
return CsrShutdownCsrProcess;
}
else
{
DPRINT("ConsoleClientShutdown(0x%p, 0x%x, %s) - Non-console process [0x%x, 0x%x]\n",
CsrProcess, Flags, FirstPhase ? "FirstPhase" : "LastPhase",
CsrProcess->ClientId.UniqueProcess, CsrProcess->ClientId.UniqueThread);
/* On first pass, let the gui server terminate all the processes that it owns */
if (FirstPhase) return CsrShutdownNonCsrProcess;
/* Use the generic handler since this isn't a gui process */
return NonConsoleProcessShutdown(CsrProcess, Flags);
}
return CsrShutdownNonCsrProcess;
}
/* EOF */