Fix VS 2005 compatibility issue

[mirror_edk2.git] / EdkNt32Pkg / Dxe / WinNtThunk / Bus / Console / ConsoleIn.c

/*++\r
\r
Copyright (c) 2006, Intel Corporation                                                         \r
All rights reserved. This program and the accompanying materials                          \r
are licensed and made available under the terms and conditions of the BSD License         \r
which accompanies this distribution.  The full text of the license may be found at        \r
http://opensource.org/licenses/bsd-license.php                                            \r
                                                                                          \r
THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,                     \r
WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.             \r
\r
Module Name:\r
\r
  ConsoleIn.c\r
\r
Abstract:\r
\r
  Console based on Win32 APIs. \r
\r
  This file attaches a SimpleTextIn protocol to a previously open window.\r
  \r
  The constructor for this protocol depends on an open window. Currently\r
  the SimpleTextOut protocol creates a window when it's constructor is called.\r
  Thus this code must run after the constructor for the SimpleTextOut \r
  protocol\r
  \r
--*/\r
\r
#include "Console.h"\r
\r
//\r
// Private worker functions\r
//\r
STATIC\r
EFI_STATUS\r
WinNtSimpleTextInCheckKey (\r
  WIN_NT_SIMPLE_TEXT_PRIVATE_DATA  *Private\r
  );\r
\r
EFI_STATUS\r
EFIAPI\r
WinNtSimpleTextInReset (\r
  IN EFI_SIMPLE_TEXT_IN_PROTOCOL          *This,\r
  IN BOOLEAN                              ExtendedVerification\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  TODO: Add function description\r
\r
Arguments:\r
\r
  This                  - TODO: add argument description\r
  ExtendedVerification  - TODO: add argument description\r
\r
Returns:\r
\r
  EFI_SUCCESS - TODO: Add description for return value\r
\r
--*/\r
{\r
  WIN_NT_SIMPLE_TEXT_PRIVATE_DATA *Private;\r
\r
  Private = WIN_NT_SIMPLE_TEXT_IN_PRIVATE_DATA_FROM_THIS (This);\r
  return EFI_SUCCESS;\r
}\r
\r
STATIC\r
EFI_STATUS\r
WinNtConvertInputRecordToEfiKey (\r
  IN  INPUT_RECORD    *InputRecord,\r
  OUT EFI_INPUT_KEY   *Key\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  TODO: Add function description\r
\r
Arguments:\r
\r
  InputRecord - TODO: add argument description\r
  Key         - TODO: add argument description\r
\r
Returns:\r
\r
  EFI_NOT_READY - TODO: Add description for return value\r
  EFI_NOT_READY - TODO: Add description for return value\r
  EFI_NOT_READY - TODO: Add description for return value\r
  EFI_SUCCESS - TODO: Add description for return value\r
\r
--*/\r
{\r
  //\r
  // Make sure InputRecord is an event that represents a keypress\r
  //\r
  if (InputRecord->EventType == KEY_EVENT) {\r
    if (!InputRecord->Event.KeyEvent.bKeyDown) {\r
      return EFI_NOT_READY;\r
    }\r
  } else {\r
    return EFI_NOT_READY;\r
  }\r
  \r
  //\r
  // Check to see if we should return a scan code in place of Unicode character.\r
  //\r
  Key->ScanCode     = 0;\r
  Key->UnicodeChar  = 0;\r
  if ((InputRecord->Event.KeyEvent.dwControlKeyState & (NUMLOCK_ON | ENHANCED_KEY)) != NUMLOCK_ON) {\r
    //\r
    // Only check these scan codes if num lock is off.\r
    //\r
    switch (InputRecord->Event.KeyEvent.wVirtualScanCode) {\r
      case 0x48: Key->ScanCode = SCAN_UP;         break;\r
      case 0x50: Key->ScanCode = SCAN_DOWN;       break;\r
      case 0x4d: Key->ScanCode = SCAN_RIGHT;      break;\r
      case 0x4b: Key->ScanCode = SCAN_LEFT;       break;\r
      case 0x47: Key->ScanCode = SCAN_HOME;       break;\r
      case 0x4F: Key->ScanCode = SCAN_END;        break;\r
      case 0x52: Key->ScanCode = SCAN_INSERT;     break;\r
      case 0x53: Key->ScanCode = SCAN_DELETE;     break;\r
      case 0x49: Key->ScanCode = SCAN_PAGE_UP;    break;\r
      case 0x51: Key->ScanCode = SCAN_PAGE_DOWN;  break;\r
    }\r
  }\r
\r
  switch (InputRecord->Event.KeyEvent.wVirtualScanCode) {\r
    case 0x3b: Key->ScanCode = SCAN_F1;   break;\r
    case 0x3c: Key->ScanCode = SCAN_F2;   break;\r
    case 0x3d: Key->ScanCode = SCAN_F3;   break;\r
    case 0x3e: Key->ScanCode = SCAN_F4;   break;\r
    case 0x3f: Key->ScanCode = SCAN_F5;   break;\r
    case 0x40: Key->ScanCode = SCAN_F6;   break;\r
    case 0x41: Key->ScanCode = SCAN_F7;   break;\r
    case 0x42: Key->ScanCode = SCAN_F8;   break;\r
    case 0x43: Key->ScanCode = SCAN_F9;   break;\r
    case 0x44: Key->ScanCode = SCAN_F10;  break;\r
    case 0x01: Key->ScanCode = SCAN_ESC;  break;\r
  }\r
\r
  //\r
  // If there's a scan code pass it, and don't pass the char code\r
  //\r
  if (Key->ScanCode == 0) {\r
    Key->UnicodeChar = InputRecord->Event.KeyEvent.uChar.UnicodeChar;\r
    if (Key->UnicodeChar == 0) {\r
      return EFI_NOT_READY;\r
    }\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
STATIC\r
EFI_STATUS\r
EFIAPI\r
WinNtSimpleTextInReadKeyStroke (\r
  IN EFI_SIMPLE_TEXT_IN_PROTOCOL          *This,\r
  OUT EFI_INPUT_KEY                       *Key\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  TODO: Add function description\r
\r
Arguments:\r
\r
  This  - TODO: add argument description\r
  Key   - TODO: add argument description\r
\r
Returns:\r
\r
  EFI_DEVICE_ERROR - TODO: Add description for return value\r
  EFI_NOT_READY - TODO: Add description for return value\r
\r
--*/\r
{\r
  EFI_STATUS                      Status;\r
  WIN_NT_SIMPLE_TEXT_PRIVATE_DATA *Private;\r
  INPUT_RECORD                    InputRecord;\r
  DWORD                           NtEventCount;\r
\r
  Private = WIN_NT_SIMPLE_TEXT_IN_PRIVATE_DATA_FROM_THIS (This);\r
\r
  Status  = WinNtSimpleTextInCheckKey (Private);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  do {\r
\r
    if (!Private->WinNtThunk->ReadConsoleInput (Private->NtInHandle, &InputRecord, 1, &NtEventCount)) {\r
      return EFI_DEVICE_ERROR;\r
    }\r
\r
    if (NtEventCount == 0) {\r
      return EFI_NOT_READY;\r
    }\r
\r
    //\r
    // Convert the Input Record to an EFI Keystroke.\r
    //\r
    Status = WinNtConvertInputRecordToEfiKey (&InputRecord, Key);\r
  } while (EFI_ERROR (Status));\r
\r
  return Status;\r
}\r
\r
STATIC\r
VOID\r
EFIAPI\r
WinNtSimpleTextInWaitForKey (\r
  IN EFI_EVENT          Event,\r
  IN VOID               *Context\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  TODO: Add function description\r
\r
Arguments:\r
\r
  Event   - TODO: add argument description\r
  Context - TODO: add argument description\r
\r
Returns:\r
\r
  TODO: add return values\r
\r
--*/\r
{\r
  WIN_NT_SIMPLE_TEXT_PRIVATE_DATA *Private;\r
  EFI_STATUS                      Status;\r
\r
  Private = (WIN_NT_SIMPLE_TEXT_PRIVATE_DATA *) Context;\r
  Status  = WinNtSimpleTextInCheckKey (Private);\r
  if (!EFI_ERROR (Status)) {\r
    gBS->SignalEvent (Event);\r
  }\r
}\r
\r
STATIC\r
EFI_STATUS\r
WinNtSimpleTextInCheckKey (\r
  WIN_NT_SIMPLE_TEXT_PRIVATE_DATA   *Private\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  TODO: Add function description\r
\r
Arguments:\r
\r
  Private - TODO: add argument description\r
\r
Returns:\r
\r
  TODO: add return values\r
\r
--*/\r
{\r
  INPUT_RECORD  *InputRecord;\r
  DWORD         NtEventCount;\r
  DWORD         ActualNtEventCount;\r
  EFI_STATUS    Status;\r
  BOOLEAN       Success;\r
  UINTN         Index;\r
  EFI_INPUT_KEY Key;\r
\r
  InputRecord   = NULL;\r
  NtEventCount  = 0;\r
  Private->WinNtThunk->GetNumberOfConsoleInputEvents (Private->NtInHandle, &NtEventCount);\r
  if (NtEventCount == 0) {\r
    Status = EFI_NOT_READY;\r
    goto Done;\r
  }\r
\r
  Status = gBS->AllocatePool (\r
                  EfiBootServicesData,\r
                  sizeof (INPUT_RECORD) * NtEventCount,\r
                  &InputRecord\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    Status = EFI_NOT_READY;\r
    goto Done;\r
  }\r
\r
  Success = (BOOLEAN) Private->WinNtThunk->PeekConsoleInput (\r
                                            Private->NtInHandle,\r
                                            InputRecord,\r
                                            NtEventCount,\r
                                            &ActualNtEventCount\r
                                            );\r
  if (!Success) {\r
    Status = EFI_NOT_READY;\r
    goto Done;\r
  }\r
\r
  Status = EFI_NOT_READY;\r
  for (Index = 0; Index < (UINTN) ActualNtEventCount; Index++) {\r
    //\r
    // Convert the Input Record to an EFI Keystroke.\r
    //\r
    Status = WinNtConvertInputRecordToEfiKey (&InputRecord[Index], &Key);\r
    if (!EFI_ERROR (Status)) {\r
      Status = EFI_SUCCESS;\r
      goto Done;\r
    }\r
  }\r
\r
Done:\r
  if (InputRecord != NULL) {\r
    gBS->FreePool (InputRecord);\r
  }\r
\r
  return Status;\r
}\r
\r
EFI_STATUS\r
WinNtSimpleTextInAttachToWindow (\r
  IN  WIN_NT_SIMPLE_TEXT_PRIVATE_DATA *Private\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  TODO: Add function description\r
\r
Arguments:\r
\r
  Private - TODO: add argument description\r
\r
Returns:\r
\r
  TODO: add return values\r
\r
--*/\r
{\r
  EFI_STATUS  Status;\r
\r
  Private->NtInHandle                 = Private->WinNtThunk->GetStdHandle (STD_INPUT_HANDLE);\r
\r
  Private->SimpleTextIn.Reset         = WinNtSimpleTextInReset;\r
  Private->SimpleTextIn.ReadKeyStroke = WinNtSimpleTextInReadKeyStroke;\r
\r
  Status = gBS->CreateEvent (\r
                  EFI_EVENT_NOTIFY_WAIT,\r
                  EFI_TPL_NOTIFY,\r
                  WinNtSimpleTextInWaitForKey,\r
                  Private,\r
                  &Private->SimpleTextIn.WaitForKey\r
                  );\r
  ASSERT_EFI_ERROR (Status);\r
\r
  return Status;\r
}\r