[mirror_edk2.git] / Nt32Pkg / WinNtGopDxe / WinNtGopInput.c

/** @file\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
  WinNtGopInput.c\r
\r
Abstract:\r
\r
  This file produces the Simple Text In for an Gop window.\r
\r
  This stuff is linked at the hip to the Window, since the window\r
  processing is done in a thread kicked off in WinNtGopImplementation.c\r
\r
  Since the window information is processed in an other thread we need\r
  a keyboard Queue to pass data about. The Simple Text In code just\r
  takes data off the Queue. The WinProc message loop takes keyboard input\r
  and places it in the Queue.\r
\r
\r
**/\r
\r
\r
#include "WinNtGop.h"\r
\r
\r
/**\r
  TODO: Add function description\r
\r
  @param  Private               TODO: add argument description\r
\r
  @retval EFI_SUCCESS           TODO: Add description for return value\r
\r
**/\r
EFI_STATUS\r
GopPrivateCreateQ (\r
  IN  GOP_PRIVATE_DATA    *Private\r
  )\r
{\r
  Private->WinNtThunk->InitializeCriticalSection (&Private->QCriticalSection);\r
\r
  Private->Queue.Front  = 0;\r
  Private->Queue.Rear   = MAX_Q - 1;\r
  Private->Queue.Count  = 0;\r
  return EFI_SUCCESS;\r
}\r
\r
\r
/**\r
  TODO: Add function description\r
\r
  @param  Private               TODO: add argument description\r
\r
  @retval EFI_SUCCESS           TODO: Add description for return value\r
\r
**/\r
EFI_STATUS\r
GopPrivateDestroyQ (\r
  IN  GOP_PRIVATE_DATA    *Private\r
  )\r
{\r
  Private->Queue.Count = 0;\r
  Private->WinNtThunk->DeleteCriticalSection (&Private->QCriticalSection);\r
  return EFI_SUCCESS;\r
}\r
\r
\r
/**\r
  TODO: Add function description\r
\r
  @param  Private               TODO: add argument description\r
  @param  Key                   TODO: add argument description\r
\r
  @retval EFI_NOT_READY         TODO: Add description for return value\r
  @retval EFI_SUCCESS           TODO: Add description for return value\r
\r
**/\r
EFI_STATUS\r
GopPrivateAddQ (\r
  IN  GOP_PRIVATE_DATA    *Private,\r
  IN  EFI_INPUT_KEY       Key\r
  )\r
{\r
  Private->WinNtThunk->EnterCriticalSection (&Private->QCriticalSection);\r
\r
  if (Private->Queue.Count == MAX_Q) {\r
    Private->WinNtThunk->LeaveCriticalSection (&Private->QCriticalSection);\r
    return EFI_NOT_READY;\r
  }\r
\r
  Private->Queue.Rear                   = (Private->Queue.Rear + 1) % MAX_Q;\r
  Private->Queue.Q[Private->Queue.Rear] = Key;\r
  Private->Queue.Count++;\r
\r
  Private->WinNtThunk->LeaveCriticalSection (&Private->QCriticalSection);\r
  return EFI_SUCCESS;\r
}\r
\r
\r
/**\r
  TODO: Add function description\r
\r
  @param  Private               TODO: add argument description\r
  @param  Key                   TODO: add argument description\r
\r
  @retval EFI_NOT_READY         TODO: Add description for return value\r
  @retval EFI_SUCCESS           TODO: Add description for return value\r
\r
**/\r
EFI_STATUS\r
GopPrivateDeleteQ (\r
  IN  GOP_PRIVATE_DATA    *Private,\r
  OUT EFI_INPUT_KEY       *Key\r
  )\r
{\r
  Private->WinNtThunk->EnterCriticalSection (&Private->QCriticalSection);\r
\r
  if (Private->Queue.Count == 0) {\r
    Private->WinNtThunk->LeaveCriticalSection (&Private->QCriticalSection);\r
    return EFI_NOT_READY;\r
  }\r
\r
  *Key                  = Private->Queue.Q[Private->Queue.Front];\r
  Private->Queue.Front  = (Private->Queue.Front + 1) % MAX_Q;\r
  Private->Queue.Count--;\r
\r
  Private->WinNtThunk->LeaveCriticalSection (&Private->QCriticalSection);\r
  return EFI_SUCCESS;\r
}\r
\r
\r
/**\r
  TODO: Add function description\r
\r
  @param  Private               TODO: add argument description\r
\r
  @retval EFI_NOT_READY         TODO: Add description for return value\r
  @retval EFI_SUCCESS           TODO: Add description for return value\r
\r
**/\r
EFI_STATUS\r
GopPrivateCheckQ (\r
  IN  GOP_PRIVATE_DATA    *Private\r
  )\r
{\r
  if (Private->Queue.Count == 0) {\r
    return EFI_NOT_READY;\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
STATIC\r
BOOLEAN\r
GopPrivateIsKeyRegistered (\r
  IN EFI_KEY_DATA  *RegsiteredData,\r
  IN EFI_KEY_DATA  *InputData\r
  )\r
/*++\r
\r
Routine Description:\r
\r
Arguments:\r
\r
  RegsiteredData    - A pointer to a buffer that is filled in with the keystroke \r
                      state data for the key that was registered.\r
  InputData         - A pointer to a buffer that is filled in with the keystroke \r
                      state data for the key that was pressed.\r
\r
Returns:\r
  TRUE              - Key be pressed matches a registered key.\r
  FLASE             - Match failed. \r
  \r
--*/\r
{\r
  ASSERT (RegsiteredData != NULL && InputData != NULL);\r
  \r
  if ((RegsiteredData->Key.ScanCode    != InputData->Key.ScanCode) ||\r
      (RegsiteredData->Key.UnicodeChar != InputData->Key.UnicodeChar)) {\r
    return FALSE;  \r
  }      \r
  \r
  //\r
  // Assume KeyShiftState/KeyToggleState = 0 in Registered key data means these state could be ignored.\r
  //\r
  if (RegsiteredData->KeyState.KeyShiftState != 0 &&\r
      RegsiteredData->KeyState.KeyShiftState != InputData->KeyState.KeyShiftState) {\r
    return FALSE;    \r
  }   \r
  if (RegsiteredData->KeyState.KeyToggleState != 0 &&\r
      RegsiteredData->KeyState.KeyToggleState != InputData->KeyState.KeyToggleState) {\r
    return FALSE;    \r
  }     \r
  \r
  return TRUE;\r
\r
}\r
\r
\r
STATIC\r
VOID\r
GopPrivateInvokeRegisteredFunction (\r
  IN GOP_PRIVATE_DATA                     *Private,\r
  IN EFI_KEY_DATA                         *KeyData\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  This function updates the status light of NumLock, ScrollLock and CapsLock.\r
\r
Arguments:\r
\r
  Private       - The private structure of WinNt Gop device.\r
  KeyData       - A pointer to a buffer that is filled in with the keystroke \r
                  state data for the key that was pressed.\r
\r
Returns:\r
\r
  EFI_SUCCESS   - The status light is updated successfully.\r
\r
--*/  \r
{ \r
  LIST_ENTRY                          *Link;\r
  WIN_NT_GOP_SIMPLE_TEXTIN_EX_NOTIFY  *CurrentNotify;\r
  \r
  for (Link = Private->NotifyList.ForwardLink; Link != &Private->NotifyList; Link = Link->ForwardLink) {\r
    CurrentNotify = CR (\r
                      Link, \r
                      WIN_NT_GOP_SIMPLE_TEXTIN_EX_NOTIFY, \r
                      NotifyEntry, \r
                      WIN_NT_GOP_SIMPLE_TEXTIN_EX_NOTIFY_SIGNATURE\r
                      );\r
    if (GopPrivateIsKeyRegistered (&CurrentNotify->KeyData, KeyData)) { \r
      CurrentNotify->KeyNotificationFn (KeyData);\r
    }\r
  }    \r
}\r
\r
STATIC\r
EFI_STATUS\r
GopPrivateUpdateStatusLight (\r
  IN GOP_PRIVATE_DATA                     *Private\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  This function updates the status light of NumLock, ScrollLock and CapsLock.\r
\r
Arguments:\r
\r
  Private       - The private structure of WinNt console In/Out.\r
\r
Returns:\r
\r
  EFI_SUCCESS   - The status light is updated successfully.\r
\r
--*/  \r
{ \r
  //\r
  // BUGBUG:Only SendInput/keybd_event function can toggle \r
  // NumLock, CapsLock and ScrollLock keys.\r
  // Neither of these functions is included in EFI_WIN_NT_THUNK_PROTOCOL.\r
  // Thus, return immediately without operation.\r
  //\r
  return EFI_SUCCESS;\r
  \r
}\r
\r
\r
STATIC\r
EFI_STATUS\r
GopPrivateResetWorker (\r
  IN GOP_PRIVATE_DATA                     *Private\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  This function is a worker function for SimpleTextIn/SimpleTextInEx.Reset().\r
\r
Arguments:\r
\r
  Private     - WinNT GOP private structure\r
\r
Returns:\r
\r
  EFI_SUCCESS - Reset successfully\r
\r
--*/\r
{\r
  EFI_INPUT_KEY     Key;\r
  EFI_TPL           OldTpl;\r
\r
  //\r
  // Enter critical section\r
  //\r
  OldTpl = gBS->RaiseTPL (TPL_NOTIFY);\r
\r
  //\r
  // A reset is draining the Queue\r
  //\r
  while (GopPrivateDeleteQ (Private, &Key) == EFI_SUCCESS)\r
    ;\r
\r
  Private->LeftShift               = FALSE;\r
  Private->RightShift              = FALSE;\r
  Private->LeftAlt                 = FALSE;\r
  Private->RightAlt                = FALSE;\r
  Private->LeftCtrl                = FALSE;\r
  Private->RightCtrl               = FALSE;\r
  Private->LeftLogo                = FALSE;\r
  Private->RightLogo               = FALSE;\r
  Private->Menu                    = FALSE;\r
  Private->SysReq                  = FALSE;\r
  \r
  Private->CapsLock                = FALSE;\r
  Private->NumLock                 = FALSE;\r
  Private->ScrollLock              = FALSE;\r
  \r
  Private->KeyState.KeyShiftState  = EFI_SHIFT_STATE_VALID;\r
  Private->KeyState.KeyToggleState = EFI_TOGGLE_STATE_VALID;\r
\r
  //\r
  // Leave critical section and return\r
  //\r
  gBS->RestoreTPL (OldTpl);\r
\r
  return EFI_SUCCESS;\r
}\r
\r
STATIC\r
EFI_STATUS\r
GopPrivateReadKeyStrokeWorker (\r
  IN GOP_PRIVATE_DATA                   *Private,\r
  OUT EFI_KEY_DATA                      *KeyData\r
  )\r
/*++\r
\r
  Routine Description:\r
    Reads the next keystroke from the input device. The WaitForKey Event can \r
    be used to test for existance of a keystroke via WaitForEvent () call.\r
\r
  Arguments:\r
    Private    - The private structure of WinNt Gop device.\r
    KeyData    - A pointer to a buffer that is filled in with the keystroke \r
                 state data for the key that was pressed.\r
\r
  Returns:\r
    EFI_SUCCESS           - The keystroke information was returned.\r
    EFI_NOT_READY         - There was no keystroke data availiable.\r
    EFI_DEVICE_ERROR      - The keystroke information was not returned due to \r
                            hardware errors.\r
    EFI_INVALID_PARAMETER - KeyData is NULL.                        \r
\r
--*/\r
{\r
  EFI_STATUS                      Status;\r
  EFI_TPL                         OldTpl;  \r
\r
  if (KeyData == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  //\r
  // Enter critical section\r
  //\r
  OldTpl  = gBS->RaiseTPL (TPL_NOTIFY);\r
\r
  Status  = GopPrivateCheckQ (Private);\r
  if (!EFI_ERROR (Status)) {\r
    //\r
    // If a Key press exists try and read it.\r
    //\r
    Status = GopPrivateDeleteQ (Private, &KeyData->Key);\r
    if (!EFI_ERROR (Status)) {\r
      //\r
      // Record Key shift state and toggle state\r
      //      \r
      if (Private->LeftCtrl) {\r
        Private->KeyState.KeyShiftState  |= EFI_LEFT_CONTROL_PRESSED;\r
      }                                    \r
      if (Private->RightCtrl) {\r
        Private->KeyState.KeyShiftState  |= EFI_RIGHT_CONTROL_PRESSED;\r
      }                                    \r
      if (Private->LeftAlt) {                \r
        Private->KeyState.KeyShiftState  |= EFI_LEFT_ALT_PRESSED;\r
      }                                    \r
      if (Private->RightAlt) {                \r
        Private->KeyState.KeyShiftState  |= EFI_RIGHT_ALT_PRESSED;\r
      }                                      \r
      if (Private->LeftShift) {          \r
        Private->KeyState.KeyShiftState  |= EFI_LEFT_SHIFT_PRESSED;\r
      }                                    \r
      if (Private->RightShift) {         \r
        Private->KeyState.KeyShiftState  |= EFI_RIGHT_SHIFT_PRESSED;\r
      }                                    \r
      if (Private->LeftLogo) {           \r
        Private->KeyState.KeyShiftState  |= EFI_LEFT_LOGO_PRESSED;\r
      }                                    \r
      if (Private->RightLogo) {          \r
        Private->KeyState.KeyShiftState  |= EFI_RIGHT_LOGO_PRESSED;\r
      }                                    \r
      if (Private->Menu) {               \r
        Private->KeyState.KeyShiftState  |= EFI_MENU_KEY_PRESSED;\r
      }                                    \r
      if (Private->SysReq) {             \r
        Private->KeyState.KeyShiftState  |= EFI_SYS_REQ_PRESSED;\r
      }  \r
      if (Private->CapsLock) {\r
        Private->KeyState.KeyToggleState |= EFI_CAPS_LOCK_ACTIVE;\r
      }\r
      if (Private->NumLock) {\r
        Private->KeyState.KeyToggleState |= EFI_NUM_LOCK_ACTIVE;\r
      }\r
      if (Private->ScrollLock) {\r
        Private->KeyState.KeyToggleState |= EFI_SCROLL_LOCK_ACTIVE;\r
      }\r
      \r
      KeyData->KeyState.KeyShiftState  = Private->KeyState.KeyShiftState;\r
      KeyData->KeyState.KeyToggleState = Private->KeyState.KeyToggleState;\r
      \r
      Private->KeyState.KeyShiftState  = EFI_SHIFT_STATE_VALID;\r
      Private->KeyState.KeyToggleState = EFI_TOGGLE_STATE_VALID;\r
  \r
      //\r
      // Leave critical section and return\r
      //\r
      gBS->RestoreTPL (OldTpl);\r
      \r
      GopPrivateInvokeRegisteredFunction (Private, KeyData);\r
      \r
      return EFI_SUCCESS;\r
    }\r
  }\r
\r
  //\r
  // Leave critical section and return\r
  //\r
  gBS->RestoreTPL (OldTpl);\r
\r
  return Status;\r
\r
}\r
\r
\r
//\r
// Simple Text In implementation.\r
//\r
\r
\r
/**\r
  TODO: Add function description\r
\r
  @param  This                  TODO: add argument description\r
  @param  ExtendedVerification  TODO: add argument description\r
\r
  @retval EFI_SUCCESS           TODO: Add description for return value\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
WinNtGopSimpleTextInReset (\r
  IN EFI_SIMPLE_TEXT_INPUT_PROTOCOL          *This,\r
  IN BOOLEAN                              ExtendedVerification\r
  )\r
{\r
  GOP_PRIVATE_DATA  *Private;\r
\r
  Private = GOP_PRIVATE_DATA_FROM_TEXT_IN_THIS (This);\r
\r
  return GopPrivateResetWorker (Private);\r
}\r
\r
\r
/**\r
  TODO: Add function description\r
\r
  @param  This                  TODO: add argument description\r
  @param  Key                   TODO: add argument description\r
\r
  @return TODO: add return values\r
\r
**/\r
STATIC\r
EFI_STATUS\r
EFIAPI\r
WinNtGopSimpleTextInReadKeyStroke (\r
  IN EFI_SIMPLE_TEXT_INPUT_PROTOCOL          *This,\r
  OUT EFI_INPUT_KEY                       *Key\r
  )\r
{\r
  GOP_PRIVATE_DATA  *Private;\r
  EFI_STATUS        Status;\r
  EFI_KEY_DATA      KeyData;\r
\r
  Private = GOP_PRIVATE_DATA_FROM_TEXT_IN_THIS (This);\r
\r
  Status = GopPrivateReadKeyStrokeWorker (Private, &KeyData);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  CopyMem (Key, &KeyData.Key, sizeof (EFI_INPUT_KEY));\r
  \r
  return EFI_SUCCESS;  \r
}\r
\r
\r
/**\r
  TODO: Add function description\r
\r
  @param  Event                 TODO: add argument description\r
  @param  Context               TODO: add argument description\r
\r
  @return TODO: add return values\r
\r
**/\r
STATIC\r
VOID\r
EFIAPI\r
WinNtGopSimpleTextInWaitForKey (\r
  IN EFI_EVENT          Event,\r
  IN VOID               *Context\r
  )\r
{\r
  GOP_PRIVATE_DATA  *Private;\r
  EFI_STATUS        Status;\r
  EFI_TPL           OldTpl;\r
\r
  Private = (GOP_PRIVATE_DATA *) Context;\r
\r
  //\r
  // Enter critical section\r
  //\r
  OldTpl  = gBS->RaiseTPL (TPL_NOTIFY);\r
\r
  Status  = GopPrivateCheckQ (Private);\r
  if (!EFI_ERROR (Status)) {\r
    //\r
    // If a there is a key in the queue signal our event.\r
    //\r
    gBS->SignalEvent (Event);\r
  } else {\r
    //\r
    // We need to sleep or NT will schedule this thread with such high\r
    // priority that WinProc thread will never run and we will not see\r
    // keyboard input. This Sleep makes the syste run 10x faster, so don't\r
    // remove it.\r
    //\r
    Private->WinNtThunk->Sleep (1);\r
  }\r
\r
  //\r
  // Leave critical section and return\r
  //\r
  gBS->RestoreTPL (OldTpl);\r
}\r
\r
//\r
// Simple Text Input Ex protocol functions\r
//\r
\r
STATIC\r
EFI_STATUS\r
EFIAPI\r
WinNtGopSimpleTextInExResetEx (\r
  IN EFI_SIMPLE_TEXT_INPUT_EX_PROTOCOL  *This,\r
  IN BOOLEAN                            ExtendedVerification\r
  )\r
/*++\r
\r
  Routine Description:\r
    Reset the input device and optionaly run diagnostics\r
\r
  Arguments:\r
    This                 - Protocol instance pointer.\r
    ExtendedVerification - Driver may perform diagnostics on reset.\r
\r
  Returns:\r
    EFI_SUCCESS           - The device was reset.\r
\r
--*/\r
{\r
  GOP_PRIVATE_DATA *Private;\r
\r
  Private = GOP_PRIVATE_DATA_FROM_TEXT_IN_EX_THIS (This);\r
  \r
  return GopPrivateResetWorker (Private);\r
}\r
\r
STATIC\r
EFI_STATUS\r
EFIAPI\r
WinNtGopSimpleTextInExReadKeyStrokeEx (\r
  IN  EFI_SIMPLE_TEXT_INPUT_EX_PROTOCOL *This,\r
  OUT EFI_KEY_DATA                      *KeyData\r
  )\r
/*++\r
\r
  Routine Description:\r
    Reads the next keystroke from the input device. The WaitForKey Event can \r
    be used to test for existance of a keystroke via WaitForEvent () call.\r
\r
  Arguments:\r
    This       - Protocol instance pointer.\r
    KeyData    - A pointer to a buffer that is filled in with the keystroke \r
                 state data for the key that was pressed.\r
\r
  Returns:\r
    EFI_SUCCESS           - The keystroke information was returned.\r
    EFI_NOT_READY         - There was no keystroke data availiable.\r
    EFI_DEVICE_ERROR      - The keystroke information was not returned due to \r
                            hardware errors.\r
    EFI_INVALID_PARAMETER - KeyData is NULL.                        \r
\r
--*/\r
{\r
  GOP_PRIVATE_DATA *Private;\r
\r
  if (KeyData == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  Private = GOP_PRIVATE_DATA_FROM_TEXT_IN_EX_THIS (This);\r
  \r
  return GopPrivateReadKeyStrokeWorker (Private, KeyData);\r
\r
}\r
\r
EFI_STATUS\r
EFIAPI\r
WinNtGopSimpleTextInExSetState (\r
  IN EFI_SIMPLE_TEXT_INPUT_EX_PROTOCOL  *This,\r
  IN EFI_KEY_TOGGLE_STATE               *KeyToggleState\r
  )\r
/*++\r
\r
  Routine Description:\r
    Set certain state for the input device.\r
\r
  Arguments:\r
    This                  - Protocol instance pointer.\r
    KeyToggleState        - A pointer to the EFI_KEY_TOGGLE_STATE to set the \r
                            state for the input device.\r
                          \r
  Returns:                \r
    EFI_SUCCESS           - The device state was set successfully.\r
    EFI_DEVICE_ERROR      - The device is not functioning correctly and could \r
                            not have the setting adjusted.\r
    EFI_UNSUPPORTED       - The device does not have the ability to set its state.\r
    EFI_INVALID_PARAMETER - KeyToggleState is NULL.                       \r
\r
--*/   \r
{\r
  EFI_STATUS                      Status;\r
  GOP_PRIVATE_DATA                *Private;\r
\r
  if (KeyToggleState == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  Private = GOP_PRIVATE_DATA_FROM_TEXT_IN_EX_THIS (This);\r
\r
  if (((Private->KeyState.KeyToggleState & EFI_TOGGLE_STATE_VALID) != EFI_TOGGLE_STATE_VALID) ||\r
      ((*KeyToggleState & EFI_TOGGLE_STATE_VALID) != EFI_TOGGLE_STATE_VALID)) {\r
    return EFI_UNSUPPORTED;  \r
  }\r
\r
  Private->ScrollLock = FALSE;\r
  Private->NumLock    = FALSE;\r
  Private->CapsLock   = FALSE;\r
\r
  if ((*KeyToggleState & EFI_SCROLL_LOCK_ACTIVE) == EFI_SCROLL_LOCK_ACTIVE) {\r
    Private->ScrollLock = TRUE;\r
  } \r
  if ((*KeyToggleState & EFI_NUM_LOCK_ACTIVE) == EFI_NUM_LOCK_ACTIVE) {\r
    Private->NumLock = TRUE;\r
  }\r
  if ((*KeyToggleState & EFI_CAPS_LOCK_ACTIVE) == EFI_CAPS_LOCK_ACTIVE) {\r
    Private->CapsLock = TRUE;\r
  }\r
\r
  Status = GopPrivateUpdateStatusLight (Private);  \r
  if (EFI_ERROR (Status)) {\r
    return EFI_DEVICE_ERROR;\r
  }\r
\r
  Private->KeyState.KeyToggleState = *KeyToggleState;\r
  return EFI_SUCCESS;\r
  \r
}\r
\r
EFI_STATUS\r
EFIAPI\r
WinNtGopSimpleTextInExRegisterKeyNotify (\r
  IN EFI_SIMPLE_TEXT_INPUT_EX_PROTOCOL  *This,\r
  IN EFI_KEY_DATA                       *KeyData,\r
  IN EFI_KEY_NOTIFY_FUNCTION            KeyNotificationFunction,\r
  OUT EFI_HANDLE                        *NotifyHandle\r
  )\r
/*++\r
\r
  Routine Description:\r
    Register a notification function for a particular keystroke for the input device.\r
\r
  Arguments:\r
    This                    - Protocol instance pointer.\r
    KeyData                 - A pointer to a buffer that is filled in with the keystroke \r
                              information data for the key that was pressed.\r
    KeyNotificationFunction - Points to the function to be called when the key \r
                              sequence is typed specified by KeyData.                        \r
    NotifyHandle            - Points to the unique handle assigned to the registered notification.                          \r
\r
  Returns:\r
    EFI_SUCCESS             - The notification function was registered successfully.\r
    EFI_OUT_OF_RESOURCES    - Unable to allocate resources for necesssary data structures.\r
    EFI_INVALID_PARAMETER   - KeyData or NotifyHandle is NULL.                       \r
                              \r
--*/   \r
{\r
  EFI_STATUS                         Status;\r
  GOP_PRIVATE_DATA                   *Private;\r
  WIN_NT_GOP_SIMPLE_TEXTIN_EX_NOTIFY *CurrentNotify;\r
  LIST_ENTRY                         *Link;\r
  WIN_NT_GOP_SIMPLE_TEXTIN_EX_NOTIFY *NewNotify;      \r
\r
  if (KeyData == NULL || NotifyHandle == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }  \r
\r
  Private = GOP_PRIVATE_DATA_FROM_TEXT_IN_EX_THIS (This);\r
\r
  //\r
  // Return EFI_SUCCESS if the (KeyData, NotificationFunction) is already registered.\r
  //\r
  for (Link = Private->NotifyList.ForwardLink; Link != &Private->NotifyList; Link = Link->ForwardLink) {\r
    CurrentNotify = CR (\r
                      Link, \r
                      WIN_NT_GOP_SIMPLE_TEXTIN_EX_NOTIFY, \r
                      NotifyEntry, \r
                      WIN_NT_GOP_SIMPLE_TEXTIN_EX_NOTIFY_SIGNATURE\r
                      );\r
    if (GopPrivateIsKeyRegistered (&CurrentNotify->KeyData, KeyData)) { \r
      if (CurrentNotify->KeyNotificationFn == KeyNotificationFunction) {\r
        *NotifyHandle = CurrentNotify->NotifyHandle;\r
        return EFI_SUCCESS;\r
      }\r
    }\r
  }    \r
  \r
  //\r
  // Allocate resource to save the notification function\r
  //  \r
  NewNotify = (WIN_NT_GOP_SIMPLE_TEXTIN_EX_NOTIFY *) AllocateZeroPool (sizeof (WIN_NT_GOP_SIMPLE_TEXTIN_EX_NOTIFY));\r
  if (NewNotify == NULL) {\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  NewNotify->Signature         = WIN_NT_GOP_SIMPLE_TEXTIN_EX_NOTIFY_SIGNATURE;     \r
  NewNotify->KeyNotificationFn = KeyNotificationFunction;\r
  CopyMem (&NewNotify->KeyData, KeyData, sizeof (KeyData));\r
  InsertTailList (&Private->NotifyList, &NewNotify->NotifyEntry);\r
\r
  //\r
  // Use gSimpleTextInExNotifyGuid to get a valid EFI_HANDLE\r
  //  \r
  Status = gBS->InstallMultipleProtocolInterfaces (\r
                  &NewNotify->NotifyHandle,\r
                  &gSimpleTextInExNotifyGuid,\r
                  NULL,\r
                  NULL\r
                  );\r
  ASSERT_EFI_ERROR (Status);\r
  \r
  *NotifyHandle = NewNotify->NotifyHandle;  \r
  \r
  return EFI_SUCCESS;\r
  \r
}\r
\r
EFI_STATUS\r
EFIAPI\r
WinNtGopSimpleTextInExUnregisterKeyNotify (\r
  IN EFI_SIMPLE_TEXT_INPUT_EX_PROTOCOL  *This,\r
  IN EFI_HANDLE                         NotificationHandle\r
  )\r
/*++\r
\r
  Routine Description:\r
    Remove a registered notification function from a particular keystroke.\r
\r
  Arguments:\r
    This                    - Protocol instance pointer.    \r
    NotificationHandle      - The handle of the notification function being unregistered.\r
\r
  Returns:\r
    EFI_SUCCESS             - The notification function was unregistered successfully.\r
    EFI_INVALID_PARAMETER   - The NotificationHandle is invalid.\r
    EFI_NOT_FOUND           - Can not find the matching entry in database.  \r
                              \r
--*/   \r
{\r
  EFI_STATUS                         Status;\r
  GOP_PRIVATE_DATA                   *Private;\r
  LIST_ENTRY                         *Link;\r
  WIN_NT_GOP_SIMPLE_TEXTIN_EX_NOTIFY *CurrentNotify;\r
\r
  if (NotificationHandle == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  } \r
  \r
  Status = gBS->OpenProtocol (\r
                  NotificationHandle,\r
                  &gSimpleTextInExNotifyGuid,\r
                  NULL,\r
                  NULL,\r
                  NULL,\r
                  EFI_OPEN_PROTOCOL_TEST_PROTOCOL\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  Private = GOP_PRIVATE_DATA_FROM_TEXT_IN_EX_THIS (This);\r
\r
  for (Link = Private->NotifyList.ForwardLink; Link != &Private->NotifyList; Link = Link->ForwardLink) {\r
    CurrentNotify = CR (\r
                      Link, \r
                      WIN_NT_GOP_SIMPLE_TEXTIN_EX_NOTIFY, \r
                      NotifyEntry, \r
                      WIN_NT_GOP_SIMPLE_TEXTIN_EX_NOTIFY_SIGNATURE\r
                      );       \r
    if (CurrentNotify->NotifyHandle == NotificationHandle) {\r
      //\r
      // Remove the notification function from NotifyList and free resources\r
      //\r
      RemoveEntryList (&CurrentNotify->NotifyEntry);      \r
      Status = gBS->UninstallMultipleProtocolInterfaces (\r
                      CurrentNotify->NotifyHandle,\r
                      &gSimpleTextInExNotifyGuid,\r
                      NULL,\r
                      NULL\r
                      );\r
      ASSERT_EFI_ERROR (Status);\r
      gBS->FreePool (CurrentNotify);            \r
      return EFI_SUCCESS;\r
    }\r
  }\r
\r
  //\r
  // Can not find the specified Notification Handle\r
  //\r
  return EFI_NOT_FOUND;\r
}\r
\r
/**\r
  TODO: Add function description\r
\r
  @param  Private               TODO: add argument description\r
\r
  @return TODO: add return values\r
\r
**/\r
EFI_STATUS\r
WinNtGopInitializeSimpleTextInForWindow (\r
  IN  GOP_PRIVATE_DATA    *Private\r
  )\r
{\r
  EFI_STATUS  Status;\r
\r
  GopPrivateCreateQ (Private);\r
\r
  //\r
  // Initialize Simple Text In protoocol\r
  //\r
  Private->SimpleTextIn.Reset         = WinNtGopSimpleTextInReset;\r
  Private->SimpleTextIn.ReadKeyStroke = WinNtGopSimpleTextInReadKeyStroke;\r
\r
  Status = gBS->CreateEvent (\r
                  EVT_NOTIFY_WAIT,\r
                  TPL_NOTIFY,\r
                  WinNtGopSimpleTextInWaitForKey,\r
                  Private,\r
                  &Private->SimpleTextIn.WaitForKey\r
                  );\r
\r
  \r
  Private->SimpleTextInEx.Reset               = WinNtGopSimpleTextInExResetEx;\r
  Private->SimpleTextInEx.ReadKeyStrokeEx     = WinNtGopSimpleTextInExReadKeyStrokeEx;\r
  Private->SimpleTextInEx.SetState            = WinNtGopSimpleTextInExSetState;\r
  Private->SimpleTextInEx.RegisterKeyNotify   = WinNtGopSimpleTextInExRegisterKeyNotify;\r
  Private->SimpleTextInEx.UnregisterKeyNotify = WinNtGopSimpleTextInExUnregisterKeyNotify;\r
\r
  Private->SimpleTextInEx.Reset (&Private->SimpleTextInEx, FALSE);\r
  \r
  InitializeListHead (&Private->NotifyList);\r
\r
  Status = gBS->CreateEvent (\r
                  EVT_NOTIFY_WAIT,\r
                  TPL_NOTIFY,\r
                  WinNtGopSimpleTextInWaitForKey,\r
                  Private,\r
                  &Private->SimpleTextInEx.WaitForKeyEx\r
                  );\r
  ASSERT_EFI_ERROR (Status);\r
\r
\r
  return Status;\r
}\r
\r
\r
\r
/**\r
  TODO: Add function description\r
\r
  @param  Private               TODO: add argument description\r
\r
  @retval EFI_SUCCESS           TODO: Add description for return value\r
\r
**/\r
EFI_STATUS\r
WinNtGopDestroySimpleTextInForWindow (\r
  IN  GOP_PRIVATE_DATA    *Private\r
  )\r
{\r
  GopPrivateDestroyQ (Private);\r
  return EFI_SUCCESS;\r
}\r