[mirror_edk2.git] / MdeModulePkg / Library / UefiBootManagerLib / BmHotkey.c

/** @file\r
  Hotkey library functions.\r
\r
Copyright (c) 2011 - 2018, Intel Corporation. All rights reserved.<BR>\r
(C) Copyright 2016 Hewlett Packard Enterprise Development LP<BR>\r
SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
\r
#include "InternalBm.h"\r
\r
//\r
// Lock for linked list\r
//\r
EFI_LOCK    mBmHotkeyLock           = EFI_INITIALIZE_LOCK_VARIABLE (TPL_NOTIFY);\r
LIST_ENTRY  mBmHotkeyList           = INITIALIZE_LIST_HEAD_VARIABLE (mBmHotkeyList);\r
EFI_EVENT   mBmHotkeyTriggered      = NULL;\r
BOOLEAN     mBmHotkeyServiceStarted = FALSE;\r
UINTN       mBmHotkeySupportCount   = 0;\r
\r
//\r
// Set OptionNumber as unassigned value to indicate the option isn't initialized\r
//\r
EFI_BOOT_MANAGER_LOAD_OPTION  mBmHotkeyBootOption = { LoadOptionNumberUnassigned };\r
\r
EFI_BOOT_MANAGER_KEY_OPTION  *mBmContinueKeyOption   = NULL;\r
VOID                         *mBmTxtInExRegistration = NULL;\r
\r
/**\r
  Return the buffer size of the EFI_BOOT_MANAGER_KEY_OPTION data.\r
\r
  @param   KeyOption            The input key option info.\r
\r
  @retval  The buffer size of the key option data.\r
**/\r
UINTN\r
BmSizeOfKeyOption (\r
  IN CONST EFI_BOOT_MANAGER_KEY_OPTION  *KeyOption\r
  )\r
{\r
  return OFFSET_OF (EFI_BOOT_MANAGER_KEY_OPTION, Keys)\r
         + KeyOption->KeyData.Options.InputKeyCount * sizeof (EFI_INPUT_KEY);\r
}\r
\r
/**\r
\r
  Check whether the input key option is valid.\r
\r
  @param   KeyOption          Key option.\r
  @param   KeyOptionSize      Size of the key option.\r
\r
  @retval  TRUE               Input key option is valid.\r
  @retval  FALSE              Input key option is not valid.\r
**/\r
BOOLEAN\r
BmIsKeyOptionValid (\r
  IN CONST EFI_BOOT_MANAGER_KEY_OPTION  *KeyOption,\r
  IN       UINTN                        KeyOptionSize\r
  )\r
{\r
  UINT16  OptionName[BM_OPTION_NAME_LEN];\r
  UINT8   *BootOption;\r
  UINTN   BootOptionSize;\r
  UINT32  Crc;\r
\r
  if (BmSizeOfKeyOption (KeyOption) != KeyOptionSize) {\r
    return FALSE;\r
  }\r
\r
  //\r
  // Check whether corresponding Boot Option exist\r
  //\r
  UnicodeSPrint (\r
    OptionName,\r
    sizeof (OptionName),\r
    L"%s%04x",\r
    mBmLoadOptionName[LoadOptionTypeBoot],\r
    KeyOption->BootOption\r
    );\r
  GetEfiGlobalVariable2 (OptionName, (VOID **)&BootOption, &BootOptionSize);\r
\r
  if (BootOption == NULL) {\r
    return FALSE;\r
  }\r
\r
  //\r
  // Check CRC for Boot Option\r
  //\r
  gBS->CalculateCrc32 (BootOption, BootOptionSize, &Crc);\r
  FreePool (BootOption);\r
\r
  return (BOOLEAN)(KeyOption->BootOptionCrc == Crc);\r
}\r
\r
/**\r
\r
  Check whether the input variable is an key option variable.\r
\r
  @param   Name               Input variable name.\r
  @param   Guid               Input variable guid.\r
  @param   OptionNumber       The option number of this key option variable.\r
\r
  @retval  TRUE               Input variable is a key option variable.\r
  @retval  FALSE              Input variable is not a key option variable.\r
**/\r
BOOLEAN\r
BmIsKeyOptionVariable (\r
  CHAR16    *Name,\r
  EFI_GUID  *Guid,\r
  UINT16    *OptionNumber\r
  )\r
{\r
  UINTN  Index;\r
  UINTN  Uint;\r
\r
  if (!CompareGuid (Guid, &gEfiGlobalVariableGuid) ||\r
      (StrSize (Name) != sizeof (L"Key####")) ||\r
      (StrnCmp (Name, L"Key", 3) != 0)\r
      )\r
  {\r
    return FALSE;\r
  }\r
\r
  *OptionNumber = 0;\r
  for (Index = 3; Index < 7; Index++) {\r
    Uint = BmCharToUint (Name[Index]);\r
    if (Uint == -1) {\r
      return FALSE;\r
    } else {\r
      *OptionNumber = (UINT16)Uint + *OptionNumber * 0x10;\r
    }\r
  }\r
\r
  return TRUE;\r
}\r
\r
typedef struct {\r
  EFI_BOOT_MANAGER_KEY_OPTION    *KeyOptions;\r
  UINTN                          KeyOptionCount;\r
} BM_COLLECT_KEY_OPTIONS_PARAM;\r
\r
/**\r
  Visitor function to collect the key options from NV storage.\r
\r
  @param Name    Variable name.\r
  @param Guid    Variable GUID.\r
  @param Context The same context passed to BmForEachVariable.\r
**/\r
VOID\r
BmCollectKeyOptions (\r
  CHAR16    *Name,\r
  EFI_GUID  *Guid,\r
  VOID      *Context\r
  )\r
{\r
  UINTN                         Index;\r
  BM_COLLECT_KEY_OPTIONS_PARAM  *Param;\r
  VOID                          *KeyOption;\r
  UINT16                        OptionNumber;\r
  UINTN                         KeyOptionSize;\r
\r
  Param = (BM_COLLECT_KEY_OPTIONS_PARAM *)Context;\r
\r
  if (BmIsKeyOptionVariable (Name, Guid, &OptionNumber)) {\r
    GetEfiGlobalVariable2 (Name, &KeyOption, &KeyOptionSize);\r
    ASSERT (KeyOption != NULL);\r
    if (BmIsKeyOptionValid (KeyOption, KeyOptionSize)) {\r
      Param->KeyOptions = ReallocatePool (\r
                            Param->KeyOptionCount * sizeof (EFI_BOOT_MANAGER_KEY_OPTION),\r
                            (Param->KeyOptionCount + 1) * sizeof (EFI_BOOT_MANAGER_KEY_OPTION),\r
                            Param->KeyOptions\r
                            );\r
      ASSERT (Param->KeyOptions != NULL);\r
      //\r
      // Insert the key option in order\r
      //\r
      for (Index = 0; Index < Param->KeyOptionCount; Index++) {\r
        if (OptionNumber < Param->KeyOptions[Index].OptionNumber) {\r
          break;\r
        }\r
      }\r
\r
      CopyMem (&Param->KeyOptions[Index + 1], &Param->KeyOptions[Index], (Param->KeyOptionCount - Index) * sizeof (EFI_BOOT_MANAGER_KEY_OPTION));\r
      CopyMem (&Param->KeyOptions[Index], KeyOption, KeyOptionSize);\r
      Param->KeyOptions[Index].OptionNumber = OptionNumber;\r
      Param->KeyOptionCount++;\r
    }\r
\r
    FreePool (KeyOption);\r
  }\r
}\r
\r
/**\r
  Return the array of key options.\r
\r
  @param Count  Return the number of key options.\r
\r
  @retval NULL  No key option.\r
  @retval Other Pointer to the key options.\r
**/\r
EFI_BOOT_MANAGER_KEY_OPTION *\r
BmGetKeyOptions (\r
  OUT UINTN  *Count\r
  )\r
{\r
  BM_COLLECT_KEY_OPTIONS_PARAM  Param;\r
\r
  if (Count == NULL) {\r
    return NULL;\r
  }\r
\r
  Param.KeyOptions     = NULL;\r
  Param.KeyOptionCount = 0;\r
\r
  BmForEachVariable (BmCollectKeyOptions, (VOID *)&Param);\r
\r
  *Count = Param.KeyOptionCount;\r
\r
  return Param.KeyOptions;\r
}\r
\r
/**\r
  Check whether the bit is set in the value.\r
\r
  @param   Value            The value need to be check.\r
  @param   Bit              The bit filed need to be check.\r
\r
  @retval  TRUE             The bit is set.\r
  @retval  FALSE            The bit is not set.\r
**/\r
BOOLEAN\r
BmBitSet (\r
  IN UINT32  Value,\r
  IN UINT32  Bit\r
  )\r
{\r
  return (BOOLEAN)((Value & Bit) != 0);\r
}\r
\r
/**\r
  Initialize the KeyData and Key[] in the EFI_BOOT_MANAGER_KEY_OPTION.\r
\r
  @param  Modifier   Input key info.\r
  @param  Args       Va_list info.\r
  @param  KeyOption  Key info which need to update.\r
\r
  @retval  EFI_SUCCESS             Succeed to initialize the KeyData and Key[].\r
  @return  EFI_INVALID_PARAMETER   Input parameter error.\r
**/\r
EFI_STATUS\r
BmInitializeKeyFields (\r
  IN UINT32                        Modifier,\r
  IN VA_LIST                       Args,\r
  OUT EFI_BOOT_MANAGER_KEY_OPTION  *KeyOption\r
  )\r
{\r
  EFI_INPUT_KEY  *Key;\r
\r
  if (KeyOption == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  Key = NULL;\r
  while (KeyOption->KeyData.Options.InputKeyCount < sizeof (KeyOption->Keys) / sizeof (KeyOption->Keys[0])) {\r
    Key = VA_ARG (Args, EFI_INPUT_KEY *);\r
    if (Key == NULL) {\r
      break;\r
    }\r
\r
    CopyMem (\r
      &KeyOption->Keys[KeyOption->KeyData.Options.InputKeyCount],\r
      Key,\r
      sizeof (EFI_INPUT_KEY)\r
      );\r
    KeyOption->KeyData.Options.InputKeyCount++;\r
  }\r
\r
  if (Key != NULL) {\r
    //\r
    // Too many keys\r
    //\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  if ((Modifier & ~(EFI_BOOT_MANAGER_SHIFT_PRESSED\r
                    | EFI_BOOT_MANAGER_CONTROL_PRESSED\r
                    | EFI_BOOT_MANAGER_ALT_PRESSED\r
                    | EFI_BOOT_MANAGER_LOGO_PRESSED\r
                    | EFI_BOOT_MANAGER_MENU_KEY_PRESSED\r
                    | EFI_BOOT_MANAGER_SYS_REQ_PRESSED\r
                    )) != 0)\r
  {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  if (BmBitSet (Modifier, EFI_BOOT_MANAGER_SHIFT_PRESSED)) {\r
    KeyOption->KeyData.Options.ShiftPressed = 1;\r
  }\r
\r
  if (BmBitSet (Modifier, EFI_BOOT_MANAGER_CONTROL_PRESSED)) {\r
    KeyOption->KeyData.Options.ControlPressed = 1;\r
  }\r
\r
  if (BmBitSet (Modifier, EFI_BOOT_MANAGER_ALT_PRESSED)) {\r
    KeyOption->KeyData.Options.AltPressed = 1;\r
  }\r
\r
  if (BmBitSet (Modifier, EFI_BOOT_MANAGER_LOGO_PRESSED)) {\r
    KeyOption->KeyData.Options.LogoPressed = 1;\r
  }\r
\r
  if (BmBitSet (Modifier, EFI_BOOT_MANAGER_MENU_KEY_PRESSED)) {\r
    KeyOption->KeyData.Options.MenuPressed = 1;\r
  }\r
\r
  if (BmBitSet (Modifier, EFI_BOOT_MANAGER_SYS_REQ_PRESSED)) {\r
    KeyOption->KeyData.Options.SysReqPressed = 1;\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Try to boot the boot option triggered by hot key.\r
**/\r
VOID\r
EFIAPI\r
EfiBootManagerHotkeyBoot (\r
  VOID\r
  )\r
{\r
  if (mBmHotkeyBootOption.OptionNumber != LoadOptionNumberUnassigned) {\r
    EfiBootManagerBoot (&mBmHotkeyBootOption);\r
    EfiBootManagerFreeLoadOption (&mBmHotkeyBootOption);\r
    mBmHotkeyBootOption.OptionNumber = LoadOptionNumberUnassigned;\r
  }\r
}\r
\r
/**\r
  This is the common notification function for HotKeys, it will be registered\r
  with SimpleTextInEx protocol interface - RegisterKeyNotify() of ConIn handle.\r
\r
  @param KeyData         A pointer to a buffer that is filled in with the keystroke\r
                         information for the key that was pressed.\r
\r
  @retval  EFI_SUCCESS   KeyData is successfully processed.\r
  @return  EFI_NOT_FOUND Fail to find boot option variable.\r
**/\r
EFI_STATUS\r
EFIAPI\r
BmHotkeyCallback (\r
  IN EFI_KEY_DATA  *KeyData\r
  )\r
{\r
  LIST_ENTRY    *Link;\r
  BM_HOTKEY     *Hotkey;\r
  CHAR16        OptionName[BM_OPTION_NAME_LEN];\r
  EFI_STATUS    Status;\r
  EFI_KEY_DATA  *HotkeyData;\r
\r
  if (mBmHotkeyBootOption.OptionNumber != LoadOptionNumberUnassigned) {\r
    //\r
    // Do not process sequential hotkey stroke until the current boot option returns\r
    //\r
    return EFI_SUCCESS;\r
  }\r
\r
  DEBUG ((DEBUG_INFO, "[Bds]BmHotkeyCallback: %04x:%04x\n", KeyData->Key.ScanCode, KeyData->Key.UnicodeChar));\r
\r
  EfiAcquireLock (&mBmHotkeyLock);\r
  for ( Link = GetFirstNode (&mBmHotkeyList)\r
        ; !IsNull (&mBmHotkeyList, Link)\r
        ; Link = GetNextNode (&mBmHotkeyList, Link)\r
        )\r
  {\r
    Hotkey = BM_HOTKEY_FROM_LINK (Link);\r
\r
    //\r
    // Is this Key Stroke we are waiting for?\r
    //\r
    ASSERT (Hotkey->WaitingKey < (sizeof (Hotkey->KeyData) / sizeof (Hotkey->KeyData[0])));\r
    HotkeyData = &Hotkey->KeyData[Hotkey->WaitingKey];\r
    if ((KeyData->Key.ScanCode == HotkeyData->Key.ScanCode) &&\r
        (KeyData->Key.UnicodeChar == HotkeyData->Key.UnicodeChar) &&\r
        (((KeyData->KeyState.KeyShiftState & EFI_SHIFT_STATE_VALID) != 0) ?\r
         (KeyData->KeyState.KeyShiftState == HotkeyData->KeyState.KeyShiftState) : TRUE\r
        )\r
        )\r
    {\r
      //\r
      // Receive an expecting key stroke, transit to next waiting state\r
      //\r
      Hotkey->WaitingKey++;\r
\r
      if (Hotkey->WaitingKey == Hotkey->CodeCount) {\r
        //\r
        // Reset to initial waiting state\r
        //\r
        Hotkey->WaitingKey = 0;\r
        //\r
        // Received the whole key stroke sequence\r
        //\r
        Status = gBS->SignalEvent (mBmHotkeyTriggered);\r
        ASSERT_EFI_ERROR (Status);\r
\r
        if (!Hotkey->IsContinue) {\r
          //\r
          // Launch its BootOption\r
          //\r
          UnicodeSPrint (\r
            OptionName,\r
            sizeof (OptionName),\r
            L"%s%04x",\r
            mBmLoadOptionName[LoadOptionTypeBoot],\r
            Hotkey->BootOption\r
            );\r
          Status = EfiBootManagerVariableToLoadOption (OptionName, &mBmHotkeyBootOption);\r
          DEBUG ((DEBUG_INFO, "[Bds]Hotkey for %s pressed - %r\n", OptionName, Status));\r
          if (EFI_ERROR (Status)) {\r
            mBmHotkeyBootOption.OptionNumber = LoadOptionNumberUnassigned;\r
          }\r
        } else {\r
          DEBUG ((DEBUG_INFO, "[Bds]Continue key pressed!\n"));\r
        }\r
      }\r
    } else {\r
      //\r
      // Receive an unexpected key stroke, reset to initial waiting state\r
      //\r
      Hotkey->WaitingKey = 0;\r
    }\r
  }\r
\r
  EfiReleaseLock (&mBmHotkeyLock);\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Return the active Simple Text Input Ex handle array.\r
  If the SystemTable.ConsoleInHandle is NULL, the function returns all\r
  founded Simple Text Input Ex handles.\r
  Otherwise, it just returns the ConsoleInHandle.\r
\r
  @param Count  Return the handle count.\r
\r
  @retval The active console handles.\r
**/\r
EFI_HANDLE *\r
BmGetActiveConsoleIn (\r
  OUT UINTN  *Count\r
  )\r
{\r
  EFI_STATUS  Status;\r
  EFI_HANDLE  *Handles;\r
\r
  Handles = NULL;\r
  *Count  = 0;\r
\r
  if (gST->ConsoleInHandle != NULL) {\r
    Status = gBS->OpenProtocol (\r
                    gST->ConsoleInHandle,\r
                    &gEfiSimpleTextInputExProtocolGuid,\r
                    NULL,\r
                    gImageHandle,\r
                    NULL,\r
                    EFI_OPEN_PROTOCOL_TEST_PROTOCOL\r
                    );\r
    if (!EFI_ERROR (Status)) {\r
      Handles = AllocateCopyPool (sizeof (EFI_HANDLE), &gST->ConsoleInHandle);\r
      if (Handles != NULL) {\r
        *Count = 1;\r
      }\r
    }\r
  } else {\r
    Status = gBS->LocateHandleBuffer (\r
                    ByProtocol,\r
                    &gEfiSimpleTextInputExProtocolGuid,\r
                    NULL,\r
                    Count,\r
                    &Handles\r
                    );\r
  }\r
\r
  return Handles;\r
}\r
\r
/**\r
  Unregister hotkey notify list.\r
\r
  @param    Hotkey                Hotkey list.\r
\r
  @retval   EFI_SUCCESS           Unregister hotkey notify success.\r
  @retval   Others                Unregister hotkey notify failed.\r
**/\r
EFI_STATUS\r
BmUnregisterHotkeyNotify (\r
  IN BM_HOTKEY  *Hotkey\r
  )\r
{\r
  EFI_STATUS                         Status;\r
  UINTN                              Index;\r
  UINTN                              KeyIndex;\r
  EFI_HANDLE                         *Handles;\r
  UINTN                              HandleCount;\r
  EFI_SIMPLE_TEXT_INPUT_EX_PROTOCOL  *TxtInEx;\r
  VOID                               *NotifyHandle;\r
\r
  Handles = BmGetActiveConsoleIn (&HandleCount);\r
  for (Index = 0; Index < HandleCount; Index++) {\r
    Status = gBS->HandleProtocol (Handles[Index], &gEfiSimpleTextInputExProtocolGuid, (VOID **)&TxtInEx);\r
    ASSERT_EFI_ERROR (Status);\r
    for (KeyIndex = 0; KeyIndex < Hotkey->CodeCount; KeyIndex++) {\r
      Status = TxtInEx->RegisterKeyNotify (\r
                          TxtInEx,\r
                          &Hotkey->KeyData[KeyIndex],\r
                          BmHotkeyCallback,\r
                          &NotifyHandle\r
                          );\r
      if (!EFI_ERROR (Status)) {\r
        Status = TxtInEx->UnregisterKeyNotify (TxtInEx, NotifyHandle);\r
        DEBUG ((DEBUG_INFO, "[Bds]UnregisterKeyNotify: %04x/%04x %r\n", Hotkey->KeyData[KeyIndex].Key.ScanCode, Hotkey->KeyData[KeyIndex].Key.UnicodeChar, Status));\r
      }\r
    }\r
  }\r
\r
  if (Handles != NULL) {\r
    FreePool (Handles);\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Register hotkey notify list.\r
\r
  @param    TxtInEx               Pointer to EFI_SIMPLE_TEXT_INPUT_EX_PROTOCOL protocol.\r
  @param    Hotkey                Hotkey list.\r
\r
  @retval   EFI_SUCCESS           Register hotkey notify success.\r
  @retval   Others                Register hotkey notify failed.\r
**/\r
EFI_STATUS\r
BmRegisterHotkeyNotify (\r
  IN EFI_SIMPLE_TEXT_INPUT_EX_PROTOCOL  *TxtInEx,\r
  IN BM_HOTKEY                          *Hotkey\r
  )\r
{\r
  EFI_STATUS  Status;\r
  UINTN       Index;\r
  VOID        *NotifyHandle;\r
\r
  for (Index = 0; Index < Hotkey->CodeCount; Index++) {\r
    Status = TxtInEx->RegisterKeyNotify (\r
                        TxtInEx,\r
                        &Hotkey->KeyData[Index],\r
                        BmHotkeyCallback,\r
                        &NotifyHandle\r
                        );\r
    DEBUG ((\r
      DEBUG_INFO,\r
      "[Bds]RegisterKeyNotify: %04x/%04x %08x/%02x %r\n",\r
      Hotkey->KeyData[Index].Key.ScanCode,\r
      Hotkey->KeyData[Index].Key.UnicodeChar,\r
      Hotkey->KeyData[Index].KeyState.KeyShiftState,\r
      Hotkey->KeyData[Index].KeyState.KeyToggleState,\r
      Status\r
      ));\r
    if (EFI_ERROR (Status)) {\r
      //\r
      // some of the hotkey registry failed\r
      // do not unregister all in case we have both CTRL-ALT-P and CTRL-ALT-P-R\r
      //\r
      break;\r
    }\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Generate key shift state base on the input key option info.\r
\r
  @param    Depth                 Which key is checked.\r
  @param    KeyOption             Input key option info.\r
  @param    KeyShiftState         Input key shift state.\r
  @param    KeyShiftStates        Return possible key shift state array.\r
  @param    KeyShiftStateCount    Possible key shift state count.\r
**/\r
VOID\r
BmGenerateKeyShiftState (\r
  IN UINTN                        Depth,\r
  IN EFI_BOOT_MANAGER_KEY_OPTION  *KeyOption,\r
  IN UINT32                       KeyShiftState,\r
  IN UINT32                       *KeyShiftStates,\r
  IN UINTN                        *KeyShiftStateCount\r
  )\r
{\r
  switch (Depth) {\r
    case 0:\r
      if (KeyOption->KeyData.Options.ShiftPressed) {\r
        BmGenerateKeyShiftState (Depth + 1, KeyOption, KeyShiftState | EFI_RIGHT_SHIFT_PRESSED, KeyShiftStates, KeyShiftStateCount);\r
        BmGenerateKeyShiftState (Depth + 1, KeyOption, KeyShiftState | EFI_LEFT_SHIFT_PRESSED, KeyShiftStates, KeyShiftStateCount);\r
      } else {\r
        BmGenerateKeyShiftState (Depth + 1, KeyOption, KeyShiftState, KeyShiftStates, KeyShiftStateCount);\r
      }\r
\r
      break;\r
\r
    case 1:\r
      if (KeyOption->KeyData.Options.ControlPressed) {\r
        BmGenerateKeyShiftState (Depth + 1, KeyOption, KeyShiftState | EFI_RIGHT_CONTROL_PRESSED, KeyShiftStates, KeyShiftStateCount);\r
        BmGenerateKeyShiftState (Depth + 1, KeyOption, KeyShiftState | EFI_LEFT_CONTROL_PRESSED, KeyShiftStates, KeyShiftStateCount);\r
      } else {\r
        BmGenerateKeyShiftState (Depth + 1, KeyOption, KeyShiftState, KeyShiftStates, KeyShiftStateCount);\r
      }\r
\r
      break;\r
\r
    case 2:\r
      if (KeyOption->KeyData.Options.AltPressed) {\r
        BmGenerateKeyShiftState (Depth + 1, KeyOption, KeyShiftState | EFI_RIGHT_ALT_PRESSED, KeyShiftStates, KeyShiftStateCount);\r
        BmGenerateKeyShiftState (Depth + 1, KeyOption, KeyShiftState | EFI_LEFT_ALT_PRESSED, KeyShiftStates, KeyShiftStateCount);\r
      } else {\r
        BmGenerateKeyShiftState (Depth + 1, KeyOption, KeyShiftState, KeyShiftStates, KeyShiftStateCount);\r
      }\r
\r
      break;\r
    case 3:\r
      if (KeyOption->KeyData.Options.LogoPressed) {\r
        BmGenerateKeyShiftState (Depth + 1, KeyOption, KeyShiftState | EFI_RIGHT_LOGO_PRESSED, KeyShiftStates, KeyShiftStateCount);\r
        BmGenerateKeyShiftState (Depth + 1, KeyOption, KeyShiftState | EFI_LEFT_LOGO_PRESSED, KeyShiftStates, KeyShiftStateCount);\r
      } else {\r
        BmGenerateKeyShiftState (Depth + 1, KeyOption, KeyShiftState, KeyShiftStates, KeyShiftStateCount);\r
      }\r
\r
      break;\r
    case 4:\r
      if (KeyOption->KeyData.Options.MenuPressed) {\r
        KeyShiftState |= EFI_MENU_KEY_PRESSED;\r
      }\r
\r
      if (KeyOption->KeyData.Options.SysReqPressed) {\r
        KeyShiftState |= EFI_SYS_REQ_PRESSED;\r
      }\r
\r
      KeyShiftStates[*KeyShiftStateCount] = KeyShiftState;\r
      (*KeyShiftStateCount)++;\r
      break;\r
  }\r
}\r
\r
/**\r
  Add it to hot key database, register it to existing TxtInEx.\r
  New TxtInEx will be automatically registered with all the hot key in dababase\r
\r
  @param    KeyOption  Input key option info.\r
**/\r
EFI_STATUS\r
BmProcessKeyOption (\r
  IN EFI_BOOT_MANAGER_KEY_OPTION  *KeyOption\r
  )\r
{\r
  EFI_STATUS                         Status;\r
  EFI_SIMPLE_TEXT_INPUT_EX_PROTOCOL  *TxtInEx;\r
  EFI_HANDLE                         *Handles;\r
  UINTN                              HandleCount;\r
  UINTN                              HandleIndex;\r
  UINTN                              Index;\r
  BM_HOTKEY                          *Hotkey;\r
  UINTN                              KeyIndex;\r
  //\r
  // 16 is enough to enumerate all the possible combination of LEFT_XXX and RIGHT_XXX\r
  //\r
  UINT32  KeyShiftStates[16];\r
  UINTN   KeyShiftStateCount;\r
\r
  if (KeyOption->KeyData.Options.InputKeyCount > mBmHotkeySupportCount) {\r
    return EFI_UNSUPPORTED;\r
  }\r
\r
  KeyShiftStateCount = 0;\r
  BmGenerateKeyShiftState (0, KeyOption, EFI_SHIFT_STATE_VALID, KeyShiftStates, &KeyShiftStateCount);\r
  ASSERT (KeyShiftStateCount <= ARRAY_SIZE (KeyShiftStates));\r
\r
  EfiAcquireLock (&mBmHotkeyLock);\r
\r
  Handles = BmGetActiveConsoleIn (&HandleCount);\r
\r
  for (Index = 0; Index < KeyShiftStateCount; Index++) {\r
    Hotkey = AllocateZeroPool (sizeof (BM_HOTKEY));\r
    ASSERT (Hotkey != NULL);\r
\r
    Hotkey->Signature  = BM_HOTKEY_SIGNATURE;\r
    Hotkey->BootOption = KeyOption->BootOption;\r
    Hotkey->IsContinue = (BOOLEAN)(KeyOption == mBmContinueKeyOption);\r
    Hotkey->CodeCount  = (UINT8)KeyOption->KeyData.Options.InputKeyCount;\r
\r
    for (KeyIndex = 0; KeyIndex < Hotkey->CodeCount; KeyIndex++) {\r
      CopyMem (&Hotkey->KeyData[KeyIndex].Key, &KeyOption->Keys[KeyIndex], sizeof (EFI_INPUT_KEY));\r
      Hotkey->KeyData[KeyIndex].KeyState.KeyShiftState = KeyShiftStates[Index];\r
    }\r
\r
    InsertTailList (&mBmHotkeyList, &Hotkey->Link);\r
\r
    for (HandleIndex = 0; HandleIndex < HandleCount; HandleIndex++) {\r
      Status = gBS->HandleProtocol (Handles[HandleIndex], &gEfiSimpleTextInputExProtocolGuid, (VOID **)&TxtInEx);\r
      ASSERT_EFI_ERROR (Status);\r
      BmRegisterHotkeyNotify (TxtInEx, Hotkey);\r
    }\r
  }\r
\r
  if (Handles != NULL) {\r
    FreePool (Handles);\r
  }\r
\r
  EfiReleaseLock (&mBmHotkeyLock);\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Callback function for SimpleTextInEx protocol install events\r
\r
  @param Event           the event that is signaled.\r
  @param Context         not used here.\r
\r
**/\r
VOID\r
EFIAPI\r
BmTxtInExCallback (\r
  IN EFI_EVENT  Event,\r
  IN VOID       *Context\r
  )\r
{\r
  EFI_STATUS                         Status;\r
  UINTN                              BufferSize;\r
  EFI_HANDLE                         Handle;\r
  EFI_SIMPLE_TEXT_INPUT_EX_PROTOCOL  *TxtInEx;\r
  LIST_ENTRY                         *Link;\r
\r
  while (TRUE) {\r
    BufferSize = sizeof (EFI_HANDLE);\r
    Status     = gBS->LocateHandle (\r
                        ByRegisterNotify,\r
                        NULL,\r
                        mBmTxtInExRegistration,\r
                        &BufferSize,\r
                        &Handle\r
                        );\r
    if (EFI_ERROR (Status)) {\r
      //\r
      // If no more notification events exist\r
      //\r
      return;\r
    }\r
\r
    Status = gBS->HandleProtocol (\r
                    Handle,\r
                    &gEfiSimpleTextInputExProtocolGuid,\r
                    (VOID **)&TxtInEx\r
                    );\r
    ASSERT_EFI_ERROR (Status);\r
\r
    //\r
    // Register the hot key notification for the existing items in the list\r
    //\r
    EfiAcquireLock (&mBmHotkeyLock);\r
    for (Link = GetFirstNode (&mBmHotkeyList); !IsNull (&mBmHotkeyList, Link); Link = GetNextNode (&mBmHotkeyList, Link)) {\r
      BmRegisterHotkeyNotify (TxtInEx, BM_HOTKEY_FROM_LINK (Link));\r
    }\r
\r
    EfiReleaseLock (&mBmHotkeyLock);\r
  }\r
}\r
\r
/**\r
  Free the key options returned from BmGetKeyOptions.\r
\r
  @param KeyOptions     Pointer to the key options.\r
  @param KeyOptionCount Number of the key options.\r
\r
  @retval EFI_SUCCESS   The key options are freed.\r
  @retval EFI_NOT_FOUND KeyOptions is NULL.\r
**/\r
EFI_STATUS\r
BmFreeKeyOptions (\r
  IN EFI_BOOT_MANAGER_KEY_OPTION  *KeyOptions,\r
  IN UINTN                        KeyOptionCount\r
  )\r
{\r
  if (KeyOptions != NULL) {\r
    FreePool (KeyOptions);\r
    return EFI_SUCCESS;\r
  } else {\r
    return EFI_NOT_FOUND;\r
  }\r
}\r
\r
/**\r
  Register the key option to exit the waiting of the Boot Manager timeout.\r
  Platform should ensure that the continue key option isn't conflict with\r
  other boot key options.\r
\r
  @param Modifier     Key shift state.\r
  @param  ...         Parameter list of pointer of EFI_INPUT_KEY.\r
\r
  @retval EFI_SUCCESS         Successfully register the continue key option.\r
  @retval EFI_ALREADY_STARTED The continue key option is already registered.\r
**/\r
EFI_STATUS\r
EFIAPI\r
EfiBootManagerRegisterContinueKeyOption (\r
  IN UINT32  Modifier,\r
  ...\r
  )\r
{\r
  EFI_STATUS                   Status;\r
  EFI_BOOT_MANAGER_KEY_OPTION  KeyOption;\r
  VA_LIST                      Args;\r
\r
  if (mBmContinueKeyOption != NULL) {\r
    return EFI_ALREADY_STARTED;\r
  }\r
\r
  ZeroMem (&KeyOption, sizeof (EFI_BOOT_MANAGER_KEY_OPTION));\r
  VA_START (Args, Modifier);\r
  Status = BmInitializeKeyFields (Modifier, Args, &KeyOption);\r
  VA_END (Args);\r
\r
  if (!EFI_ERROR (Status)) {\r
    mBmContinueKeyOption = AllocateCopyPool (sizeof (EFI_BOOT_MANAGER_KEY_OPTION), &KeyOption);\r
    ASSERT (mBmContinueKeyOption != NULL);\r
    if (mBmHotkeyServiceStarted) {\r
      BmProcessKeyOption (mBmContinueKeyOption);\r
    }\r
  }\r
\r
  return Status;\r
}\r
\r
/**\r
  Stop the hotkey processing.\r
\r
  @param    Event          Event pointer related to hotkey service.\r
  @param    Context        Context pass to this function.\r
**/\r
VOID\r
EFIAPI\r
BmStopHotkeyService (\r
  IN EFI_EVENT  Event,\r
  IN VOID       *Context\r
  )\r
{\r
  LIST_ENTRY  *Link;\r
  BM_HOTKEY   *Hotkey;\r
\r
  DEBUG ((DEBUG_INFO, "[Bds]Stop Hotkey Service!\n"));\r
  gBS->CloseEvent (Event);\r
\r
  EfiAcquireLock (&mBmHotkeyLock);\r
  for (Link = GetFirstNode (&mBmHotkeyList); !IsNull (&mBmHotkeyList, Link); ) {\r
    Hotkey = BM_HOTKEY_FROM_LINK (Link);\r
    BmUnregisterHotkeyNotify (Hotkey);\r
    Link = RemoveEntryList (Link);\r
    FreePool (Hotkey);\r
  }\r
\r
  EfiReleaseLock (&mBmHotkeyLock);\r
}\r
\r
/**\r
  Start the hot key service so that the key press can trigger the boot option.\r
\r
  @param HotkeyTriggered  Return the waitable event and it will be signaled\r
                          when a valid hot key is pressed.\r
\r
  @retval EFI_SUCCESS     The hot key service is started.\r
**/\r
EFI_STATUS\r
EFIAPI\r
EfiBootManagerStartHotkeyService (\r
  IN EFI_EVENT  *HotkeyTriggered\r
  )\r
{\r
  EFI_STATUS                   Status;\r
  EFI_BOOT_MANAGER_KEY_OPTION  *KeyOptions;\r
  UINTN                        KeyOptionCount;\r
  UINTN                        Index;\r
  EFI_EVENT                    Event;\r
  UINT32                       *BootOptionSupport;\r
\r
  GetEfiGlobalVariable2 (EFI_BOOT_OPTION_SUPPORT_VARIABLE_NAME, (VOID **)&BootOptionSupport, NULL);\r
  if (BootOptionSupport != NULL) {\r
    if ((*BootOptionSupport & EFI_BOOT_OPTION_SUPPORT_KEY)  != 0) {\r
      mBmHotkeySupportCount = ((*BootOptionSupport & EFI_BOOT_OPTION_SUPPORT_COUNT) >> LowBitSet32 (EFI_BOOT_OPTION_SUPPORT_COUNT));\r
    }\r
\r
    FreePool (BootOptionSupport);\r
  }\r
\r
  if (mBmHotkeySupportCount == 0) {\r
    DEBUG ((DEBUG_INFO, "Bds: BootOptionSupport NV variable forbids starting the hotkey service.\n"));\r
    return EFI_UNSUPPORTED;\r
  }\r
\r
  Status = gBS->CreateEvent (\r
                  EVT_NOTIFY_WAIT,\r
                  TPL_CALLBACK,\r
                  EfiEventEmptyFunction,\r
                  NULL,\r
                  &mBmHotkeyTriggered\r
                  );\r
  ASSERT_EFI_ERROR (Status);\r
\r
  if (HotkeyTriggered != NULL) {\r
    *HotkeyTriggered = mBmHotkeyTriggered;\r
  }\r
\r
  KeyOptions = BmGetKeyOptions (&KeyOptionCount);\r
  for (Index = 0; Index < KeyOptionCount; Index++) {\r
    BmProcessKeyOption (&KeyOptions[Index]);\r
  }\r
\r
  BmFreeKeyOptions (KeyOptions, KeyOptionCount);\r
\r
  if (mBmContinueKeyOption != NULL) {\r
    BmProcessKeyOption (mBmContinueKeyOption);\r
  }\r
\r
  //\r
  // Hook hotkey on every future SimpleTextInputEx instance when\r
  // SystemTable.ConsoleInHandle == NULL, which means the console\r
  // manager (ConSplitter) is absent.\r
  //\r
  if (gST->ConsoleInHandle == NULL) {\r
    EfiCreateProtocolNotifyEvent (\r
      &gEfiSimpleTextInputExProtocolGuid,\r
      TPL_CALLBACK,\r
      BmTxtInExCallback,\r
      NULL,\r
      &mBmTxtInExRegistration\r
      );\r
  }\r
\r
  Status = EfiCreateEventReadyToBootEx (\r
             TPL_CALLBACK,\r
             BmStopHotkeyService,\r
             NULL,\r
             &Event\r
             );\r
  ASSERT_EFI_ERROR (Status);\r
\r
  mBmHotkeyServiceStarted = TRUE;\r
  return Status;\r
}\r
\r
/**\r
  Add the key option.\r
  It adds the key option variable and the key option takes affect immediately.\r
\r
  @param AddedOption      Return the added key option.\r
  @param BootOptionNumber The boot option number for the key option.\r
  @param Modifier         Key shift state.\r
  @param ...              Parameter list of pointer of EFI_INPUT_KEY.\r
\r
  @retval EFI_SUCCESS         The key option is added.\r
  @retval EFI_ALREADY_STARTED The hot key is already used by certain key option.\r
**/\r
EFI_STATUS\r
EFIAPI\r
EfiBootManagerAddKeyOptionVariable (\r
  OUT EFI_BOOT_MANAGER_KEY_OPTION  *AddedOption    OPTIONAL,\r
  IN UINT16                        BootOptionNumber,\r
  IN UINT32                        Modifier,\r
  ...\r
  )\r
{\r
  EFI_STATUS                   Status;\r
  VA_LIST                      Args;\r
  VOID                         *BootOption;\r
  UINTN                        BootOptionSize;\r
  CHAR16                       BootOptionName[BM_OPTION_NAME_LEN];\r
  EFI_BOOT_MANAGER_KEY_OPTION  KeyOption;\r
  EFI_BOOT_MANAGER_KEY_OPTION  *KeyOptions;\r
  UINTN                        KeyOptionCount;\r
  UINTN                        Index;\r
  UINTN                        KeyOptionNumber;\r
  CHAR16                       KeyOptionName[sizeof ("Key####")];\r
\r
  UnicodeSPrint (\r
    BootOptionName,\r
    sizeof (BootOptionName),\r
    L"%s%04x",\r
    mBmLoadOptionName[LoadOptionTypeBoot],\r
    BootOptionNumber\r
    );\r
  GetEfiGlobalVariable2 (BootOptionName, &BootOption, &BootOptionSize);\r
\r
  if (BootOption == NULL) {\r
    return EFI_NOT_FOUND;\r
  }\r
\r
  ZeroMem (&KeyOption, sizeof (EFI_BOOT_MANAGER_KEY_OPTION));\r
  KeyOption.BootOption = BootOptionNumber;\r
  Status               = gBS->CalculateCrc32 (BootOption, BootOptionSize, &KeyOption.BootOptionCrc);\r
  ASSERT_EFI_ERROR (Status);\r
  FreePool (BootOption);\r
\r
  VA_START (Args, Modifier);\r
  Status = BmInitializeKeyFields (Modifier, Args, &KeyOption);\r
  VA_END (Args);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  KeyOptionNumber = LoadOptionNumberUnassigned;\r
  //\r
  // Check if the hot key sequence was defined already\r
  //\r
  KeyOptions = BmGetKeyOptions (&KeyOptionCount);\r
  for (Index = 0; Index < KeyOptionCount; Index++) {\r
    if ((KeyOptions[Index].KeyData.PackedValue == KeyOption.KeyData.PackedValue) &&\r
        (CompareMem (KeyOptions[Index].Keys, KeyOption.Keys, KeyOption.KeyData.Options.InputKeyCount * sizeof (EFI_INPUT_KEY)) == 0))\r
    {\r
      break;\r
    }\r
\r
    if ((KeyOptionNumber == LoadOptionNumberUnassigned) &&\r
        (KeyOptions[Index].OptionNumber > Index)\r
        )\r
    {\r
      KeyOptionNumber = Index;\r
    }\r
  }\r
\r
  BmFreeKeyOptions (KeyOptions, KeyOptionCount);\r
\r
  if (Index < KeyOptionCount) {\r
    return EFI_ALREADY_STARTED;\r
  }\r
\r
  if (KeyOptionNumber == LoadOptionNumberUnassigned) {\r
    KeyOptionNumber = KeyOptionCount;\r
  }\r
\r
  UnicodeSPrint (KeyOptionName, sizeof (KeyOptionName), L"Key%04x", KeyOptionNumber);\r
\r
  Status = gRT->SetVariable (\r
                  KeyOptionName,\r
                  &gEfiGlobalVariableGuid,\r
                  EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_RUNTIME_ACCESS | EFI_VARIABLE_NON_VOLATILE,\r
                  BmSizeOfKeyOption (&KeyOption),\r
                  &KeyOption\r
                  );\r
  if (!EFI_ERROR (Status)) {\r
    //\r
    // Return the Key Option in case needed by caller\r
    //\r
    if (AddedOption != NULL) {\r
      CopyMem (AddedOption, &KeyOption, sizeof (EFI_BOOT_MANAGER_KEY_OPTION));\r
    }\r
\r
    //\r
    // Register the newly added hot key\r
    // Calling this function before EfiBootManagerStartHotkeyService doesn't\r
    // need to call BmProcessKeyOption\r
    //\r
    if (mBmHotkeyServiceStarted) {\r
      BmProcessKeyOption (&KeyOption);\r
    }\r
  }\r
\r
  return Status;\r
}\r
\r
/**\r
  Delete the Key Option variable and unregister the hot key\r
\r
  @param DeletedOption  Return the deleted key options.\r
  @param Modifier       Key shift state.\r
  @param ...            Parameter list of pointer of EFI_INPUT_KEY.\r
\r
  @retval EFI_SUCCESS   The key option is deleted.\r
  @retval EFI_NOT_FOUND The key option cannot be found.\r
**/\r
EFI_STATUS\r
EFIAPI\r
EfiBootManagerDeleteKeyOptionVariable (\r
  IN EFI_BOOT_MANAGER_KEY_OPTION  *DeletedOption  OPTIONAL,\r
  IN UINT32                       Modifier,\r
  ...\r
  )\r
{\r
  EFI_STATUS                   Status;\r
  UINTN                        Index;\r
  VA_LIST                      Args;\r
  EFI_BOOT_MANAGER_KEY_OPTION  KeyOption;\r
  EFI_BOOT_MANAGER_KEY_OPTION  *KeyOptions;\r
  UINTN                        KeyOptionCount;\r
  LIST_ENTRY                   *Link;\r
  BM_HOTKEY                    *Hotkey;\r
  UINT32                       ShiftState;\r
  BOOLEAN                      Match;\r
  CHAR16                       KeyOptionName[sizeof ("Key####")];\r
\r
  ZeroMem (&KeyOption, sizeof (EFI_BOOT_MANAGER_KEY_OPTION));\r
  VA_START (Args, Modifier);\r
  Status = BmInitializeKeyFields (Modifier, Args, &KeyOption);\r
  VA_END (Args);\r
\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  EfiAcquireLock (&mBmHotkeyLock);\r
  //\r
  // Delete the key option from active hot key list\r
  // Could have multiple entries when modifier isn't 0 because we map the ShiftPressed to RIGHT_SHIFT and RIGHT_SHIFT\r
  //\r
  for (Link = GetFirstNode (&mBmHotkeyList); !IsNull (&mBmHotkeyList, Link); ) {\r
    Hotkey = BM_HOTKEY_FROM_LINK (Link);\r
    Match  = (BOOLEAN)(Hotkey->CodeCount == KeyOption.KeyData.Options.InputKeyCount);\r
\r
    for (Index = 0; Match && (Index < Hotkey->CodeCount); Index++) {\r
      ShiftState = Hotkey->KeyData[Index].KeyState.KeyShiftState;\r
      if (\r
          (BmBitSet (ShiftState, EFI_RIGHT_SHIFT_PRESSED | EFI_LEFT_SHIFT_PRESSED) != KeyOption.KeyData.Options.ShiftPressed) ||\r
          (BmBitSet (ShiftState, EFI_RIGHT_CONTROL_PRESSED | EFI_LEFT_CONTROL_PRESSED) != KeyOption.KeyData.Options.ControlPressed) ||\r
          (BmBitSet (ShiftState, EFI_RIGHT_ALT_PRESSED | EFI_LEFT_ALT_PRESSED) != KeyOption.KeyData.Options.AltPressed) ||\r
          (BmBitSet (ShiftState, EFI_RIGHT_LOGO_PRESSED | EFI_LEFT_LOGO_PRESSED) != KeyOption.KeyData.Options.LogoPressed) ||\r
          (BmBitSet (ShiftState, EFI_MENU_KEY_PRESSED) != KeyOption.KeyData.Options.MenuPressed) ||\r
          (BmBitSet (ShiftState, EFI_SYS_REQ_PRESSED) != KeyOption.KeyData.Options.SysReqPressed) ||\r
          (CompareMem (&Hotkey->KeyData[Index].Key, &KeyOption.Keys[Index], sizeof (EFI_INPUT_KEY)) != 0)\r
          )\r
      {\r
        //\r
        // Break when any field doesn't match\r
        //\r
        Match = FALSE;\r
        break;\r
      }\r
    }\r
\r
    if (Match) {\r
      Link = RemoveEntryList (Link);\r
      FreePool (Hotkey);\r
    } else {\r
      Link = GetNextNode (&mBmHotkeyList, Link);\r
    }\r
  }\r
\r
  //\r
  // Delete the key option from the variable\r
  //\r
  Status     = EFI_NOT_FOUND;\r
  KeyOptions = BmGetKeyOptions (&KeyOptionCount);\r
  for (Index = 0; Index < KeyOptionCount; Index++) {\r
    if ((KeyOptions[Index].KeyData.PackedValue == KeyOption.KeyData.PackedValue) &&\r
        (CompareMem (\r
           KeyOptions[Index].Keys,\r
           KeyOption.Keys,\r
           KeyOption.KeyData.Options.InputKeyCount * sizeof (EFI_INPUT_KEY)\r
           ) == 0)\r
        )\r
    {\r
      UnicodeSPrint (KeyOptionName, sizeof (KeyOptionName), L"Key%04x", KeyOptions[Index].OptionNumber);\r
      Status = gRT->SetVariable (\r
                      KeyOptionName,\r
                      &gEfiGlobalVariableGuid,\r
                      EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_RUNTIME_ACCESS | EFI_VARIABLE_NON_VOLATILE,\r
                      0,\r
                      NULL\r
                      );\r
      //\r
      // Return the deleted key option in case needed by caller\r
      //\r
      if (DeletedOption != NULL) {\r
        CopyMem (DeletedOption, &KeyOptions[Index], sizeof (EFI_BOOT_MANAGER_KEY_OPTION));\r
      }\r
\r
      break;\r
    }\r
  }\r
\r
  BmFreeKeyOptions (KeyOptions, KeyOptionCount);\r
\r
  EfiReleaseLock (&mBmHotkeyLock);\r
\r
  return Status;\r
}\r