Add WinNtConsole driver into Nt32Pkg.

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@2779 6f19259b-4bc3-4df7-8a09-765794883524

1 files changed, 379 insertions, 0 deletions

diff --git a/Nt32Pkg/WinNtConsoleDxe/ConsoleIn.c b/Nt32Pkg/WinNtConsoleDxe/ConsoleIn.c
new file mode 100644
index 0000000000..4f00f8b15d
--- /dev/null
+++ b/Nt32Pkg/WinNtConsoleDxe/ConsoleIn.c
@@ -0,0 +1,379 @@
+/*++

+

+Copyright (c) 2006 - 2007, Intel Corporation

+All rights reserved. This program and the accompanying materials

+are licensed and made available under the terms and conditions of the BSD License

+which accompanies this distribution.  The full text of the license may be found at

+http://opensource.org/licenses/bsd-license.php

+

+THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,

+WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.

+

+Module Name:

+

+  ConsoleIn.c

+

+Abstract:

+

+  Console based on Win32 APIs.

+

+  This file attaches a SimpleTextIn protocol to a previously open window.

+

+  The constructor for this protocol depends on an open window. Currently

+  the SimpleTextOut protocol creates a window when it's constructor is called.

+  Thus this code must run after the constructor for the SimpleTextOut

+  protocol

+

+--*/

+//

+// The package level header files this module uses

+//

+#include <Uefi.h>

+#include <WinNtDxe.h>

+//

+// The protocols, PPI and GUID defintions for this module

+//

+#include <Protocol/SimpleTextIn.h>

+#include <Protocol/WinNtIo.h>

+#include <Protocol/SimpleTextOut.h>

+#include <Protocol/ComponentName.h>

+#include <Protocol/DriverBinding.h>

+//

+// The Library classes this module consumes

+//

+#include <Library/DebugLib.h>

+#include <Library/BaseLib.h>

+#include <Library/UefiDriverEntryPoint.h>

+#include <Library/UefiLib.h>

+#include <Library/BaseMemoryLib.h>

+#include <Library/UefiBootServicesTableLib.h>

+#include <Library/MemoryAllocationLib.h>

+#include "Console.h"

+

+//

+// Private worker functions

+//

+STATIC

+EFI_STATUS

+WinNtSimpleTextInCheckKey (

+  WIN_NT_SIMPLE_TEXT_PRIVATE_DATA  *Private

+  );

+

+EFI_STATUS

+EFIAPI

+WinNtSimpleTextInReset (

+  IN EFI_SIMPLE_TEXT_INPUT_PROTOCOL          *This,

+  IN BOOLEAN                              ExtendedVerification

+  )

+/*++

+

+Routine Description:

+

+  TODO: Add function description

+

+Arguments:

+

+  This                  - TODO: add argument description

+  ExtendedVerification  - TODO: add argument description

+

+Returns:

+

+  EFI_SUCCESS - TODO: Add description for return value

+

+--*/

+{

+  WIN_NT_SIMPLE_TEXT_PRIVATE_DATA *Private;

+

+  Private = WIN_NT_SIMPLE_TEXT_IN_PRIVATE_DATA_FROM_THIS (This);

+  return EFI_SUCCESS;

+}

+

+STATIC

+EFI_STATUS

+WinNtConvertInputRecordToEfiKey (

+  IN  INPUT_RECORD    *InputRecord,

+  OUT EFI_INPUT_KEY   *Key

+  )

+/*++

+

+Routine Description:

+

+  TODO: Add function description

+

+Arguments:

+

+  InputRecord - TODO: add argument description

+  Key         - TODO: add argument description

+

+Returns:

+

+  EFI_NOT_READY - TODO: Add description for return value

+  EFI_NOT_READY - TODO: Add description for return value

+  EFI_NOT_READY - TODO: Add description for return value

+  EFI_SUCCESS - TODO: Add description for return value

+

+--*/

+{

+  //

+  // Make sure InputRecord is an event that represents a keypress

+  //

+  if (InputRecord->EventType == KEY_EVENT) {

+    if (!InputRecord->Event.KeyEvent.bKeyDown) {

+      return EFI_NOT_READY;

+    }

+  } else {

+    return EFI_NOT_READY;

+  }

+

+  //

+  // Check to see if we should return a scan code in place of Unicode character.

+  //

+  Key->ScanCode     = 0;

+  Key->UnicodeChar  = 0;

+  if ((InputRecord->Event.KeyEvent.dwControlKeyState & (NUMLOCK_ON | ENHANCED_KEY)) != NUMLOCK_ON) {

+    //

+    // Only check these scan codes if num lock is off.

+    //

+    switch (InputRecord->Event.KeyEvent.wVirtualScanCode) {

+      case 0x48: Key->ScanCode = SCAN_UP;         break;

+      case 0x50: Key->ScanCode = SCAN_DOWN;       break;

+      case 0x4d: Key->ScanCode = SCAN_RIGHT;      break;

+      case 0x4b: Key->ScanCode = SCAN_LEFT;       break;

+      case 0x47: Key->ScanCode = SCAN_HOME;       break;

+      case 0x4F: Key->ScanCode = SCAN_END;        break;

+      case 0x52: Key->ScanCode = SCAN_INSERT;     break;

+      case 0x53: Key->ScanCode = SCAN_DELETE;     break;

+      case 0x49: Key->ScanCode = SCAN_PAGE_UP;    break;

+      case 0x51: Key->ScanCode = SCAN_PAGE_DOWN;  break;

+    }

+  }

+

+  switch (InputRecord->Event.KeyEvent.wVirtualScanCode) {

+    case 0x3b: Key->ScanCode = SCAN_F1;   break;

+    case 0x3c: Key->ScanCode = SCAN_F2;   break;

+    case 0x3d: Key->ScanCode = SCAN_F3;   break;

+    case 0x3e: Key->ScanCode = SCAN_F4;   break;

+    case 0x3f: Key->ScanCode = SCAN_F5;   break;

+    case 0x40: Key->ScanCode = SCAN_F6;   break;

+    case 0x41: Key->ScanCode = SCAN_F7;   break;

+    case 0x42: Key->ScanCode = SCAN_F8;   break;

+    case 0x43: Key->ScanCode = SCAN_F9;   break;

+    case 0x44: Key->ScanCode = SCAN_F10;  break;

+    case 0x01: Key->ScanCode = SCAN_ESC;  break;

+  }

+

+  //

+  // If there's a scan code pass it, and don't pass the char code

+  //

+  if (Key->ScanCode == 0) {

+    Key->UnicodeChar = InputRecord->Event.KeyEvent.uChar.UnicodeChar;

+    if (Key->UnicodeChar == 0) {

+      return EFI_NOT_READY;

+    }

+  }

+

+  return EFI_SUCCESS;

+}

+

+STATIC

+EFI_STATUS

+EFIAPI

+WinNtSimpleTextInReadKeyStroke (

+  IN EFI_SIMPLE_TEXT_INPUT_PROTOCOL          *This,

+  OUT EFI_INPUT_KEY                       *Key

+  )

+/*++

+

+Routine Description:

+

+  TODO: Add function description

+

+Arguments:

+

+  This  - TODO: add argument description

+  Key   - TODO: add argument description

+

+Returns:

+

+  EFI_DEVICE_ERROR - TODO: Add description for return value

+  EFI_NOT_READY - TODO: Add description for return value

+

+--*/

+{

+  EFI_STATUS                      Status;

+  WIN_NT_SIMPLE_TEXT_PRIVATE_DATA *Private;

+  INPUT_RECORD                    InputRecord;

+  DWORD                           NtEventCount;

+

+  Private = WIN_NT_SIMPLE_TEXT_IN_PRIVATE_DATA_FROM_THIS (This);

+

+  Status  = WinNtSimpleTextInCheckKey (Private);

+  if (EFI_ERROR (Status)) {

+    return Status;

+  }

+

+  do {

+

+    if (!Private->WinNtThunk->ReadConsoleInput (Private->NtInHandle, &InputRecord, 1, &NtEventCount)) {

+      return EFI_DEVICE_ERROR;

+    }

+

+    if (NtEventCount == 0) {

+      return EFI_NOT_READY;

+    }

+

+    //

+    // Convert the Input Record to an EFI Keystroke.

+    //

+    Status = WinNtConvertInputRecordToEfiKey (&InputRecord, Key);

+  } while (EFI_ERROR (Status));

+

+  return Status;

+}

+

+STATIC

+VOID

+EFIAPI

+WinNtSimpleTextInWaitForKey (

+  IN EFI_EVENT          Event,

+  IN VOID               *Context

+  )

+/*++

+

+Routine Description:

+

+  TODO: Add function description

+

+Arguments:

+

+  Event   - TODO: add argument description

+  Context - TODO: add argument description

+

+Returns:

+

+  TODO: add return values

+

+--*/

+{

+  WIN_NT_SIMPLE_TEXT_PRIVATE_DATA *Private;

+  EFI_STATUS                      Status;

+

+  Private = (WIN_NT_SIMPLE_TEXT_PRIVATE_DATA *) Context;

+  Status  = WinNtSimpleTextInCheckKey (Private);

+  if (!EFI_ERROR (Status)) {

+    gBS->SignalEvent (Event);

+  }

+}

+

+STATIC

+EFI_STATUS

+WinNtSimpleTextInCheckKey (

+  WIN_NT_SIMPLE_TEXT_PRIVATE_DATA   *Private

+  )

+/*++

+

+Routine Description:

+

+  TODO: Add function description

+

+Arguments:

+

+  Private - TODO: add argument description

+

+Returns:

+

+  TODO: add return values

+

+--*/

+{

+  INPUT_RECORD  *InputRecord;

+  DWORD         NtEventCount;

+  DWORD         ActualNtEventCount;

+  EFI_STATUS    Status;

+  BOOLEAN       Success;

+  UINTN         Index;

+  EFI_INPUT_KEY Key;

+

+  InputRecord   = NULL;

+  NtEventCount  = 0;

+  Private->WinNtThunk->GetNumberOfConsoleInputEvents (Private->NtInHandle, &NtEventCount);

+  if (NtEventCount == 0) {

+    Status = EFI_NOT_READY;

+    goto Done;

+  }

+

+  InputRecord = AllocatePool (sizeof (INPUT_RECORD) * NtEventCount);

+  if (InputRecord == NULL) {

+    Status = EFI_NOT_READY;

+    goto Done;

+  }

+

+  Success = (BOOLEAN) Private->WinNtThunk->PeekConsoleInput (

+                                            Private->NtInHandle,

+                                            InputRecord,

+                                            NtEventCount,

+                                            &ActualNtEventCount

+                                            );

+  if (!Success) {

+    Status = EFI_NOT_READY;

+    goto Done;

+  }

+

+  Status = EFI_NOT_READY;

+  for (Index = 0; Index < (UINTN) ActualNtEventCount; Index++) {

+    //

+    // Convert the Input Record to an EFI Keystroke.

+    //

+    Status = WinNtConvertInputRecordToEfiKey (&InputRecord[Index], &Key);

+    if (!EFI_ERROR (Status)) {

+      Status = EFI_SUCCESS;

+      goto Done;

+    }

+  }

+

+Done:

+  if (InputRecord != NULL) {

+    FreePool (InputRecord);

+  }

+

+  return Status;

+}

+

+EFI_STATUS

+WinNtSimpleTextInAttachToWindow (

+  IN  WIN_NT_SIMPLE_TEXT_PRIVATE_DATA *Private

+  )

+/*++

+

+Routine Description:

+

+  TODO: Add function description

+

+Arguments:

+

+  Private - TODO: add argument description

+

+Returns:

+

+  TODO: add return values

+

+--*/

+{

+  EFI_STATUS  Status;

+

+  Private->NtInHandle                 = Private->WinNtThunk->GetStdHandle (STD_INPUT_HANDLE);

+

+  Private->SimpleTextIn.Reset         = WinNtSimpleTextInReset;

+  Private->SimpleTextIn.ReadKeyStroke = WinNtSimpleTextInReadKeyStroke;

+

+  Status = gBS->CreateEvent (

+                  EVT_NOTIFY_WAIT,

+                  TPL_NOTIFY,

+                  WinNtSimpleTextInWaitForKey,

+                  Private,

+                  &Private->SimpleTextIn.WaitForKey

+                  );

+  ASSERT_EFI_ERROR (Status);

+

+  return Status;

+}