[sir_richard] 48780: Patch by Anton Yarotsky: [NTOSKRNL]: Fix support for non Headless scenarios as well.

Author: sir_richard
Date: Thu Sep 16 02:22:10 2010
New Revision: 48780
URL: http://svn.reactos.org/svn/reactos?rev=48780&view=rev
Log:
Patch by Anton Yarotsky:
[NTOSKRNL]: Fix support for non Headless scenarios as well.
Modified:
    trunk/reactos/ntoskrnl/ex/hdlsterm.c
Modified: trunk/reactos/ntoskrnl/ex/hdlsterm.c
URL:
http://svn.reactos.org/svn/reactos/trunk/reactos/ntoskrnl/ex/hdlsterm.c?rev…
==============================================================================
--- trunk/reactos/ntoskrnl/ex/hdlsterm.c [iso-8859-1] (original)
+++ trunk/reactos/ntoskrnl/ex/hdlsterm.c [iso-8859-1] Thu Sep 16 02:22:10 2010
@@ -194,25 +194,22 @@
        HdlspSendStringAtBaud(HeadlessGlobals->TmpBuffer);
 }
-/*
- * @unimplemented
- */
 NTSTATUS
 NTAPI
-HeadlessDispatch(
-    IN HEADLESS_CMD Command,
-    IN PVOID InputBuffer,
-    IN SIZE_T InputBufferSize,
-    OUT PVOID OutputBuffer,
-    OUT PSIZE_T OutputBufferSize
+HdlspDispatch(
+       IN HEADLESS_CMD Command,
+       IN PVOID InputBuffer,
+       IN SIZE_T InputBufferSize,
+       OUT PVOID OutputBuffer,
+       OUT PSIZE_T OutputBufferSize
        )
 {
        NTSTATUS Status = STATUS_NOT_IMPLEMENTED;
        ASSERT(HeadlessGlobals != NULL);
 //     ASSERT(HeadlessGlobals->PageLockHandle != NULL);
-
+
        /* FIXME: This should be using the headless spinlock */
-
+
        /* Ignore non-reentrant commands */
        if ((Command != HeadlessCmdAddLogEntry) &&
                (Command != HeadlessCmdStartBugCheck) &&
@@ -224,7 +221,7 @@
                /* Don't allow these commands next time */
                HeadlessGlobals->ProcessingCmd = TRUE;
        }
-
+
        /* Handle each command */
        switch (Command)
        {
@@ -241,7 +238,7 @@
                                Status = STATUS_INVALID_PARAMETER;
                                goto Reset;
                        }
-
+
                        /* Terminal should be on */
                        if (HeadlessGlobals->TerminalEnabled)
                        {
@@ -249,7 +246,7 @@
                                PHEADLESS_CMD_PUT_STRING PutString = (PVOID)InputBuffer;
                                HdlspPutString(PutString->String);
                        }
-
+
                        /* Return success either way */
                        Status = STATUS_SUCCESS;
                        break;
@@ -304,10 +301,51 @@
        {
                ASSERT(HeadlessGlobals->ProcessingCmd == TRUE);
                HeadlessGlobals->ProcessingCmd = FALSE;
-    }
+       }
        //UNIMPLEMENTED;
        return STATUS_SUCCESS;
 }
+/*
+ * @unimplemented
+ */
+NTSTATUS
+NTAPI
+HeadlessDispatch(
+       IN HEADLESS_CMD Command,
+       IN PVOID InputBuffer,
+       IN SIZE_T InputBufferSize,
+       OUT PVOID OutputBuffer,
+       OUT PSIZE_T OutputBufferSize
+       )
+{
+       /* Check for stubs that will expect something even with headless off */
+       if (!HeadlessGlobals)
+       {
+               /* Don't allow the SAC to connect */
+               if (Command == HeadlessCmdEnableTerminal) return STATUS_UNSUCCESSFUL;
+
+               /* Send bogus reply */
+               if ((Command == HeadlessCmdQueryInformation) ||
+                       (Command == HeadlessCmdGetByte) ||
+                       (Command == HeadlessCmdGetLine) ||
+                       (Command == HeadlessCmdCheckForReboot) ||
+                       (Command == HeadlessCmdTerminalPoll))
+               {
+                       if (!(OutputBuffer) || !(OutputBufferSize)) return
STATUS_INVALID_PARAMETER;
+                       RtlZeroMemory(OutputBuffer, *OutputBufferSize);
+               }
+               return STATUS_SUCCESS;
+       }
+
+       /* Do the real work */
+       return HdlspDispatch(
+               Command,
+               InputBuffer,
+               InputBufferSize,
+               OutputBuffer,
+               OutputBufferSize);
+}
+
 /* EOF */