[mirror_edk2.git] / ArmPlatformPkg / Bds / BootMenu.c

/** @file\r
*\r
*  Copyright (c) 2011 - 2014, ARM Limited. All rights reserved.\r
*\r
*  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
**/\r
\r
#include "BdsInternal.h"\r
\r
#include <Guid/ArmGlobalVariableHob.h>\r
\r
extern EFI_HANDLE mImageHandle;\r
extern BDS_LOAD_OPTION_SUPPORT *BdsLoadOptionSupportList;\r
\r
\r
EFI_STATUS\r
SelectBootDevice (\r
  OUT BDS_SUPPORTED_DEVICE** SupportedBootDevice\r
  )\r
{\r
  EFI_STATUS  Status;\r
  LIST_ENTRY  SupportedDeviceList;\r
  UINTN       SupportedDeviceCount;\r
  LIST_ENTRY* Entry;\r
  UINTN       SupportedDeviceSelected;\r
  UINTN       Index;\r
\r
  //\r
  // List the Boot Devices supported\r
  //\r
\r
  // Start all the drivers first\r
  BdsConnectAllDrivers ();\r
\r
  // List the supported devices\r
  Status = BootDeviceListSupportedInit (&SupportedDeviceList);\r
  ASSERT_EFI_ERROR(Status);\r
\r
  SupportedDeviceCount = 0;\r
  for (Entry = GetFirstNode (&SupportedDeviceList);\r
       !IsNull (&SupportedDeviceList,Entry);\r
       Entry = GetNextNode (&SupportedDeviceList,Entry)\r
       )\r
  {\r
    *SupportedBootDevice = SUPPORTED_BOOT_DEVICE_FROM_LINK(Entry);\r
    Print(L"[%d] %s\n",SupportedDeviceCount+1,(*SupportedBootDevice)->Description);\r
\r
    DEBUG_CODE_BEGIN();\r
      CHAR16*                           DevicePathTxt;\r
      EFI_DEVICE_PATH_TO_TEXT_PROTOCOL* DevicePathToTextProtocol;\r
\r
      Status = gBS->LocateProtocol (&gEfiDevicePathToTextProtocolGuid, NULL, (VOID **)&DevicePathToTextProtocol);\r
      ASSERT_EFI_ERROR(Status);\r
      DevicePathTxt = DevicePathToTextProtocol->ConvertDevicePathToText ((*SupportedBootDevice)->DevicePathProtocol,TRUE,TRUE);\r
\r
      Print(L"\t- %s\n",DevicePathTxt);\r
\r
      FreePool(DevicePathTxt);\r
    DEBUG_CODE_END();\r
\r
    SupportedDeviceCount++;\r
  }\r
\r
  if (SupportedDeviceCount == 0) {\r
    Print(L"There is no supported device.\n");\r
    Status = EFI_ABORTED;\r
    goto EXIT;\r
  }\r
\r
  //\r
  // Select the Boot Device\r
  //\r
  SupportedDeviceSelected = 0;\r
  while (SupportedDeviceSelected == 0) {\r
    Print(L"Select the Boot Device: ");\r
    Status = GetHIInputInteger (&SupportedDeviceSelected);\r
    if (EFI_ERROR(Status)) {\r
      Status = EFI_ABORTED;\r
      goto EXIT;\r
    } else if ((SupportedDeviceSelected == 0) || (SupportedDeviceSelected > SupportedDeviceCount)) {\r
      Print(L"Invalid input (max %d)\n",SupportedDeviceCount);\r
      SupportedDeviceSelected = 0;\r
    }\r
  }\r
\r
  //\r
  // Get the Device Path for the selected boot device\r
  //\r
  Index = 1;\r
  for (Entry = GetFirstNode (&SupportedDeviceList);\r
       !IsNull (&SupportedDeviceList,Entry);\r
       Entry = GetNextNode (&SupportedDeviceList,Entry)\r
       )\r
  {\r
    if (Index == SupportedDeviceSelected) {\r
      *SupportedBootDevice = SUPPORTED_BOOT_DEVICE_FROM_LINK(Entry);\r
      break;\r
    }\r
    Index++;\r
  }\r
\r
EXIT:\r
  BootDeviceListSupportedFree (&SupportedDeviceList, *SupportedBootDevice);\r
  return Status;\r
}\r
\r
EFI_STATUS\r
BootMenuAddBootOption (\r
  IN LIST_ENTRY *BootOptionsList\r
  )\r
{\r
  EFI_STATUS                Status;\r
  BDS_SUPPORTED_DEVICE*     SupportedBootDevice;\r
  ARM_BDS_LOADER_ARGUMENTS* BootArguments;\r
  CHAR16                    BootDescription[BOOT_DEVICE_DESCRIPTION_MAX];\r
  CHAR8                     AsciiCmdLine[BOOT_DEVICE_OPTION_MAX];\r
  CHAR16                    CmdLine[BOOT_DEVICE_OPTION_MAX];\r
  UINT32                    Attributes;\r
  ARM_BDS_LOADER_TYPE       BootType;\r
  BDS_LOAD_OPTION_ENTRY     *BdsLoadOptionEntry;\r
  EFI_DEVICE_PATH           *DevicePath;\r
  EFI_DEVICE_PATH_PROTOCOL  *DevicePathNodes;\r
  EFI_DEVICE_PATH_PROTOCOL  *InitrdPathNodes;\r
  EFI_DEVICE_PATH_PROTOCOL  *InitrdPath;\r
  UINTN                     CmdLineSize;\r
  BOOLEAN                   InitrdSupport;\r
  UINTN                     InitrdSize;\r
  UINT8*                    OptionalData;\r
  UINTN                     OptionalDataSize;\r
  BOOLEAN                   RequestBootType;\r
\r
  Attributes                = 0;\r
  SupportedBootDevice = NULL;\r
\r
  // List the Boot Devices supported\r
  Status = SelectBootDevice (&SupportedBootDevice);\r
  if (EFI_ERROR(Status)) {\r
    Status = EFI_ABORTED;\r
    goto EXIT;\r
  }\r
\r
  // Create the specific device path node\r
  RequestBootType = TRUE;\r
  Status = SupportedBootDevice->Support->CreateDevicePathNode (L"EFI Application or the kernel", &DevicePathNodes, &RequestBootType);\r
  if (EFI_ERROR(Status)) {\r
    Status = EFI_ABORTED;\r
    goto EXIT;\r
  }\r
  // Append the Device Path to the selected device path\r
  DevicePath = AppendDevicePath (SupportedBootDevice->DevicePathProtocol, (CONST EFI_DEVICE_PATH_PROTOCOL *)DevicePathNodes);\r
  if (DevicePath == NULL) {\r
    Status = EFI_OUT_OF_RESOURCES;\r
    goto EXIT;\r
  }\r
\r
  if (RequestBootType) {\r
    Status = BootDeviceGetType (DevicePath, &BootType, &Attributes);\r
    if (EFI_ERROR(Status)) {\r
      Status = EFI_ABORTED;\r
      goto EXIT;\r
    }\r
  } else {\r
    BootType = BDS_LOADER_EFI_APPLICATION;\r
  }\r
\r
  if ((BootType == BDS_LOADER_KERNEL_LINUX_ATAG) || (BootType == BDS_LOADER_KERNEL_LINUX_FDT)) {\r
    Print(L"Add an initrd: ");\r
    Status = GetHIInputBoolean (&InitrdSupport);\r
    if (EFI_ERROR(Status)) {\r
      Status = EFI_ABORTED;\r
      goto EXIT;\r
    }\r
\r
    if (InitrdSupport) {\r
      // Create the specific device path node\r
      Status = SupportedBootDevice->Support->CreateDevicePathNode (L"initrd", &InitrdPathNodes, NULL);\r
      if (EFI_ERROR(Status) && Status != EFI_NOT_FOUND) { // EFI_NOT_FOUND is returned on empty input string, but we can boot without an initrd\r
        Status = EFI_ABORTED;\r
        goto EXIT;\r
      }\r
\r
      if (InitrdPathNodes != NULL) {\r
        // Append the Device Path to the selected device path\r
        InitrdPath = AppendDevicePath (SupportedBootDevice->DevicePathProtocol, (CONST EFI_DEVICE_PATH_PROTOCOL *)InitrdPathNodes);\r
        if (InitrdPath == NULL) {\r
          Status = EFI_OUT_OF_RESOURCES;\r
          goto EXIT;\r
        }\r
      } else {\r
        InitrdPath = NULL;\r
      }\r
    } else {\r
      InitrdPath = NULL;\r
    }\r
\r
    Print(L"Arguments to pass to the binary: ");\r
    Status = GetHIInputAscii (AsciiCmdLine, BOOT_DEVICE_OPTION_MAX);\r
    if (EFI_ERROR(Status)) {\r
      Status = EFI_ABORTED;\r
      goto FREE_DEVICE_PATH;\r
    }\r
\r
    CmdLineSize = AsciiStrSize (AsciiCmdLine);\r
    InitrdSize = GetDevicePathSize (InitrdPath);\r
\r
    OptionalDataSize = sizeof(ARM_BDS_LOADER_ARGUMENTS) + CmdLineSize + InitrdSize;\r
    BootArguments = (ARM_BDS_LOADER_ARGUMENTS*)AllocatePool (OptionalDataSize);\r
\r
    BootArguments->LinuxArguments.CmdLineSize = CmdLineSize;\r
    BootArguments->LinuxArguments.InitrdSize = InitrdSize;\r
    CopyMem ((VOID*)(&BootArguments->LinuxArguments + 1), AsciiCmdLine, CmdLineSize);\r
    CopyMem ((VOID*)((UINTN)(&BootArguments->LinuxArguments + 1) + CmdLineSize), InitrdPath, InitrdSize);\r
\r
    OptionalData = (UINT8*)BootArguments;\r
  } else {\r
    Print (L"Arguments to pass to the EFI Application: ");\r
    Status = GetHIInputStr (CmdLine, BOOT_DEVICE_OPTION_MAX);\r
    if (EFI_ERROR (Status)) {\r
      Status = EFI_ABORTED;\r
      goto EXIT;\r
    }\r
\r
    OptionalData = (UINT8*)CmdLine;\r
    OptionalDataSize = StrSize (CmdLine);\r
  }\r
\r
  Print(L"Description for this new Entry: ");\r
  Status = GetHIInputStr (BootDescription, BOOT_DEVICE_DESCRIPTION_MAX);\r
  if (EFI_ERROR(Status)) {\r
    Status = EFI_ABORTED;\r
    goto FREE_DEVICE_PATH;\r
  }\r
\r
  // Create new entry\r
  BdsLoadOptionEntry = (BDS_LOAD_OPTION_ENTRY*)AllocatePool (sizeof(BDS_LOAD_OPTION_ENTRY));\r
  Status = BootOptionCreate (Attributes, BootDescription, DevicePath, BootType, OptionalData, OptionalDataSize, &BdsLoadOptionEntry->BdsLoadOption);\r
  if (!EFI_ERROR(Status)) {\r
    InsertTailList (BootOptionsList, &BdsLoadOptionEntry->Link);\r
  }\r
\r
FREE_DEVICE_PATH:\r
  FreePool (DevicePath);\r
\r
EXIT:\r
  if (Status == EFI_ABORTED) {\r
    Print(L"\n");\r
  }\r
  FreePool(SupportedBootDevice);\r
  return Status;\r
}\r
\r
STATIC\r
EFI_STATUS\r
BootMenuSelectBootOption (\r
  IN  LIST_ENTRY*               BootOptionsList,\r
  IN  CONST CHAR16*             InputStatement,\r
  OUT BDS_LOAD_OPTION_ENTRY**   BdsLoadOptionEntry\r
  )\r
{\r
  EFI_STATUS                    Status;\r
  LIST_ENTRY*                   Entry;\r
  BDS_LOAD_OPTION*              BdsLoadOption;\r
  UINTN                         BootOptionSelected;\r
  UINTN                         BootOptionCount;\r
  UINTN                         Index;\r
  BOOLEAN                       IsUnicode;\r
\r
  // Display the list of supported boot devices\r
  BootOptionCount = 0;\r
  for (Entry = GetFirstNode (BootOptionsList);\r
       !IsNull (BootOptionsList,Entry);\r
       Entry = GetNextNode (BootOptionsList, Entry)\r
       )\r
  {\r
    BdsLoadOption = LOAD_OPTION_FROM_LINK(Entry);\r
\r
    Print (L"[%d] %s\n", (BootOptionCount + 1), BdsLoadOption->Description);\r
\r
    DEBUG_CODE_BEGIN();\r
      CHAR16*                           DevicePathTxt;\r
      EFI_DEVICE_PATH_TO_TEXT_PROTOCOL* DevicePathToTextProtocol;\r
      ARM_BDS_LOADER_TYPE               LoaderType;\r
      ARM_BDS_LOADER_OPTIONAL_DATA*     OptionalData;\r
\r
      Status = gBS->LocateProtocol(&gEfiDevicePathToTextProtocolGuid, NULL, (VOID **)&DevicePathToTextProtocol);\r
      ASSERT_EFI_ERROR(Status);\r
      DevicePathTxt = DevicePathToTextProtocol->ConvertDevicePathToText(BdsLoadOption->FilePathList,TRUE,TRUE);\r
\r
      Print(L"\t- %s\n",DevicePathTxt);\r
      OptionalData = BdsLoadOption->OptionalData;\r
      if (IS_ARM_BDS_BOOTENTRY (BdsLoadOption)) {\r
        LoaderType = (ARM_BDS_LOADER_TYPE)ReadUnaligned32 ((CONST UINT32*)&OptionalData->Header.LoaderType);\r
        if ((LoaderType == BDS_LOADER_KERNEL_LINUX_ATAG) || (LoaderType == BDS_LOADER_KERNEL_LINUX_FDT)) {\r
          Print (L"\t- Arguments: %a\n",&OptionalData->Arguments.LinuxArguments + 1);\r
        }\r
      } else if (OptionalData != NULL) {\r
        if (IsPrintableString (OptionalData, &IsUnicode)) {\r
          if (IsUnicode) {\r
            Print (L"\t- Arguments: %s\n", OptionalData);\r
          } else {\r
            AsciiPrint ("\t- Arguments: %a\n", OptionalData);\r
          }\r
        }\r
      }\r
\r
      FreePool(DevicePathTxt);\r
    DEBUG_CODE_END();\r
\r
    BootOptionCount++;\r
  }\r
\r
  // Check if a valid boot option(s) is found\r
  if (BootOptionCount == 0) {\r
    if (StrCmp (InputStatement, DELETE_BOOT_ENTRY) == 0) {\r
      Print (L"Nothing to remove!\n");\r
    } else if (StrCmp (InputStatement, UPDATE_BOOT_ENTRY) == 0) {\r
      Print (L"Couldn't find valid boot entries\n");\r
    } else{\r
      Print (L"No supported Boot Entry.\n");\r
    }\r
\r
    return EFI_NOT_FOUND;\r
  }\r
\r
  // Get the index of the boot device to delete\r
  BootOptionSelected = 0;\r
  while (BootOptionSelected == 0) {\r
    Print(InputStatement);\r
    Status = GetHIInputInteger (&BootOptionSelected);\r
    if (EFI_ERROR(Status)) {\r
      return Status;\r
    } else if ((BootOptionSelected == 0) || (BootOptionSelected > BootOptionCount)) {\r
      Print(L"Invalid input (max %d)\n",BootOptionCount);\r
      BootOptionSelected = 0;\r
    }\r
  }\r
\r
  // Get the structure of the Boot device to delete\r
  Index = 1;\r
  for (Entry = GetFirstNode (BootOptionsList);\r
       !IsNull (BootOptionsList, Entry);\r
       Entry = GetNextNode (BootOptionsList,Entry)\r
       )\r
  {\r
    if (Index == BootOptionSelected) {\r
      *BdsLoadOptionEntry = LOAD_OPTION_ENTRY_FROM_LINK(Entry);\r
      break;\r
    }\r
    Index++;\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
EFI_STATUS\r
BootMenuRemoveBootOption (\r
  IN LIST_ENTRY *BootOptionsList\r
  )\r
{\r
  EFI_STATUS                    Status;\r
  BDS_LOAD_OPTION_ENTRY*        BootOptionEntry;\r
\r
  Status = BootMenuSelectBootOption (BootOptionsList, DELETE_BOOT_ENTRY, &BootOptionEntry);\r
  if (EFI_ERROR(Status)) {\r
    return Status;\r
  }\r
\r
  // If the Boot Option was attached to a list remove it\r
  if (!IsListEmpty (&BootOptionEntry->Link)) {\r
    // Remove the entry from the list\r
    RemoveEntryList (&BootOptionEntry->Link);\r
  }\r
\r
  // Delete the BDS Load option structures\r
  BootOptionDelete (BootOptionEntry->BdsLoadOption);\r
\r
  return EFI_SUCCESS;\r
}\r
\r
EFI_STATUS\r
BootMenuUpdateBootOption (\r
  IN LIST_ENTRY *BootOptionsList\r
  )\r
{\r
  EFI_STATUS                    Status;\r
  BDS_LOAD_OPTION_ENTRY         *BootOptionEntry;\r
  BDS_LOAD_OPTION               *BootOption;\r
  BDS_LOAD_OPTION_SUPPORT*      DeviceSupport;\r
  ARM_BDS_LOADER_ARGUMENTS*     BootArguments;\r
  CHAR16                        BootDescription[BOOT_DEVICE_DESCRIPTION_MAX];\r
  CHAR8                         CmdLine[BOOT_DEVICE_OPTION_MAX];\r
  CHAR16                        UnicodeCmdLine[BOOT_DEVICE_OPTION_MAX];\r
  EFI_DEVICE_PATH               *DevicePath;\r
  EFI_DEVICE_PATH               *TempInitrdPath;\r
  ARM_BDS_LOADER_TYPE           BootType;\r
  ARM_BDS_LOADER_OPTIONAL_DATA* LoaderOptionalData;\r
  ARM_BDS_LINUX_ARGUMENTS*      LinuxArguments;\r
  EFI_DEVICE_PATH               *InitrdPathNodes;\r
  EFI_DEVICE_PATH               *InitrdPath;\r
  UINTN                         InitrdSize;\r
  UINTN                         CmdLineSize;\r
  BOOLEAN                       InitrdSupport;\r
  UINT8*                        OptionalData;\r
  UINTN                         OptionalDataSize;\r
  BOOLEAN                       RequestBootType;\r
  BOOLEAN                       IsPrintable;\r
  BOOLEAN                       IsUnicode;\r
\r
  Status = BootMenuSelectBootOption (BootOptionsList, UPDATE_BOOT_ENTRY, &BootOptionEntry);\r
  if (EFI_ERROR(Status)) {\r
    return Status;\r
  }\r
  BootOption = BootOptionEntry->BdsLoadOption;\r
\r
  // Get the device support for this Boot Option\r
  Status = BootDeviceGetDeviceSupport (BootOption->FilePathList, &DeviceSupport);\r
  if (EFI_ERROR(Status)) {\r
    Print(L"Not possible to retrieve the supported device for the update\n");\r
    return EFI_UNSUPPORTED;\r
  }\r
\r
  RequestBootType = TRUE;\r
  Status = DeviceSupport->UpdateDevicePathNode (BootOption->FilePathList, L"EFI Application or the kernel", &DevicePath, &RequestBootType);\r
  if (EFI_ERROR(Status)) {\r
    Status = EFI_ABORTED;\r
    goto EXIT;\r
  }\r
\r
  if (RequestBootType) {\r
    Status = BootDeviceGetType (DevicePath, &BootType, &BootOption->Attributes);\r
    if (EFI_ERROR(Status)) {\r
      Status = EFI_ABORTED;\r
      goto EXIT;\r
    }\r
  }\r
\r
  LoaderOptionalData = BootOption->OptionalData;\r
  if (LoaderOptionalData != NULL) {\r
    BootType = (ARM_BDS_LOADER_TYPE)ReadUnaligned32 ((UINT32 *)(&LoaderOptionalData->Header.LoaderType));\r
  } else {\r
    BootType = BDS_LOADER_EFI_APPLICATION;\r
  }\r
\r
  if ((BootType == BDS_LOADER_KERNEL_LINUX_ATAG) || (BootType == BDS_LOADER_KERNEL_LINUX_FDT)) {\r
    LinuxArguments = &LoaderOptionalData->Arguments.LinuxArguments;\r
\r
    CmdLineSize = ReadUnaligned16 ((CONST UINT16*)&LinuxArguments->CmdLineSize);\r
\r
    InitrdSize = ReadUnaligned16 ((CONST UINT16*)&LinuxArguments->InitrdSize);\r
    if (InitrdSize > 0) {\r
      Print(L"Keep the initrd: ");\r
    } else {\r
      Print(L"Add an initrd: ");\r
    }\r
    Status = GetHIInputBoolean (&InitrdSupport);\r
    if (EFI_ERROR(Status)) {\r
      Status = EFI_ABORTED;\r
      goto EXIT;\r
    }\r
\r
    if (InitrdSupport) {\r
      if (InitrdSize > 0) {\r
        // Case we update the initrd device path\r
        Status = DeviceSupport->UpdateDevicePathNode ((EFI_DEVICE_PATH*)((UINTN)(LinuxArguments + 1) + CmdLineSize), L"initrd", &InitrdPath, NULL);\r
        if (EFI_ERROR(Status) && Status != EFI_NOT_FOUND) {// EFI_NOT_FOUND is returned on empty input string, but we can boot without an initrd\r
          Status = EFI_ABORTED;\r
          goto EXIT;\r
        }\r
        InitrdSize = GetDevicePathSize (InitrdPath);\r
      } else {\r
        // Case we create the initrd device path\r
\r
        Status = DeviceSupport->CreateDevicePathNode (L"initrd", &InitrdPathNodes, NULL);\r
        if (EFI_ERROR(Status) && Status != EFI_NOT_FOUND) { // EFI_NOT_FOUND is returned on empty input string, but we can boot without an initrd\r
          Status = EFI_ABORTED;\r
          goto EXIT;\r
        }\r
\r
        if (InitrdPathNodes != NULL) {\r
          // Duplicate Linux kernel Device Path\r
          TempInitrdPath = DuplicateDevicePath (BootOption->FilePathList);\r
          // Replace Linux kernel Node by EndNode\r
          SetDevicePathEndNode (GetLastDevicePathNode (TempInitrdPath));\r
          // Append the Device Path to the selected device path\r
          InitrdPath = AppendDevicePath (TempInitrdPath, (CONST EFI_DEVICE_PATH_PROTOCOL *)InitrdPathNodes);\r
          FreePool (TempInitrdPath);\r
          if (InitrdPath == NULL) {\r
            Status = EFI_OUT_OF_RESOURCES;\r
            goto EXIT;\r
          }\r
          InitrdSize = GetDevicePathSize (InitrdPath);\r
        } else {\r
          InitrdPath = NULL;\r
        }\r
      }\r
    } else {\r
      InitrdSize = 0;\r
    }\r
\r
    Print(L"Arguments to pass to the binary: ");\r
    if (CmdLineSize > 0) {\r
      AsciiStrnCpy(CmdLine, (CONST CHAR8*)(LinuxArguments + 1), CmdLineSize);\r
    } else {\r
      CmdLine[0] = '\0';\r
    }\r
    Status = EditHIInputAscii (CmdLine, BOOT_DEVICE_OPTION_MAX);\r
    if (EFI_ERROR(Status)) {\r
      Status = EFI_ABORTED;\r
      goto FREE_DEVICE_PATH;\r
    }\r
\r
    CmdLineSize = AsciiStrSize (CmdLine);\r
\r
    OptionalDataSize = sizeof(ARM_BDS_LOADER_ARGUMENTS) + CmdLineSize + InitrdSize;\r
    BootArguments = (ARM_BDS_LOADER_ARGUMENTS*)AllocatePool (OptionalDataSize);\r
    BootArguments->LinuxArguments.CmdLineSize = CmdLineSize;\r
    BootArguments->LinuxArguments.InitrdSize = InitrdSize;\r
    CopyMem (&BootArguments->LinuxArguments + 1, CmdLine, CmdLineSize);\r
    CopyMem ((VOID*)((UINTN)(&BootArguments->LinuxArguments + 1) + CmdLineSize), InitrdPath, InitrdSize);\r
\r
    OptionalData = (UINT8*)BootArguments;\r
  } else {\r
    Print (L"Arguments to pass to the EFI Application: ");\r
\r
    if (BootOption->OptionalDataSize > 0) {\r
      IsPrintable = IsPrintableString (BootOption->OptionalData, &IsUnicode);\r
      if (IsPrintable) {\r
        if (IsUnicode) {\r
          StrnCpy (UnicodeCmdLine, BootOption->OptionalData, BootOption->OptionalDataSize / 2);\r
        } else {\r
          AsciiStrnCpy (CmdLine, BootOption->OptionalData, BootOption->OptionalDataSize);\r
        }\r
      }\r
    } else {\r
      UnicodeCmdLine[0] = L'\0';\r
      IsPrintable = TRUE;\r
      IsUnicode = TRUE;\r
    }\r
\r
    // We do not request arguments for OptionalData that cannot be printed\r
    if (IsPrintable) {\r
      if (IsUnicode) {\r
        Status = EditHIInputStr (UnicodeCmdLine, BOOT_DEVICE_OPTION_MAX);\r
        if (EFI_ERROR (Status)) {\r
          Status = EFI_ABORTED;\r
          goto FREE_DEVICE_PATH;\r
        }\r
\r
        OptionalData = (UINT8*)UnicodeCmdLine;\r
        OptionalDataSize = StrSize (UnicodeCmdLine);\r
      } else {\r
        Status = EditHIInputAscii (CmdLine, BOOT_DEVICE_OPTION_MAX);\r
        if (EFI_ERROR (Status)) {\r
          Status = EFI_ABORTED;\r
          goto FREE_DEVICE_PATH;\r
        }\r
\r
        OptionalData = (UINT8*)CmdLine;\r
        OptionalDataSize = AsciiStrSize (CmdLine);\r
      }\r
    } else {\r
      // We keep the former OptionalData\r
      OptionalData = BootOption->OptionalData;\r
      OptionalDataSize = BootOption->OptionalDataSize;\r
    }\r
  }\r
\r
  Print(L"Description for this new Entry: ");\r
  StrnCpy (BootDescription, BootOption->Description, BOOT_DEVICE_DESCRIPTION_MAX);\r
  Status = EditHIInputStr (BootDescription, BOOT_DEVICE_DESCRIPTION_MAX);\r
  if (EFI_ERROR(Status)) {\r
    Status = EFI_ABORTED;\r
    goto FREE_DEVICE_PATH;\r
  }\r
\r
  // Update the entry\r
  Status = BootOptionUpdate (BootOption, BootOption->Attributes, BootDescription, DevicePath, BootType, OptionalData, OptionalDataSize);\r
\r
FREE_DEVICE_PATH:\r
  FreePool (DevicePath);\r
\r
EXIT:\r
  if (Status == EFI_ABORTED) {\r
    Print(L"\n");\r
  }\r
  return Status;\r
}\r
\r
EFI_STATUS\r
UpdateFdtPath (\r
  IN LIST_ENTRY *BootOptionsList\r
  )\r
{\r
  EFI_STATUS                Status;\r
  UINTN                     FdtDevicePathSize;\r
  BDS_SUPPORTED_DEVICE      *SupportedBootDevice;\r
  EFI_DEVICE_PATH_PROTOCOL  *FdtDevicePathNodes;\r
  EFI_DEVICE_PATH_PROTOCOL  *FdtDevicePath;\r
\r
  Status = SelectBootDevice (&SupportedBootDevice);\r
  if (EFI_ERROR(Status)) {\r
    Status = EFI_ABORTED;\r
    goto EXIT;\r
  }\r
\r
  // Create the specific device path node\r
  Status = SupportedBootDevice->Support->CreateDevicePathNode (L"FDT blob", &FdtDevicePathNodes, NULL);\r
  if (EFI_ERROR(Status)) {\r
    Status = EFI_ABORTED;\r
    goto EXIT;\r
  }\r
\r
  if (FdtDevicePathNodes != NULL) {\r
    // Append the Device Path node to the select device path\r
    FdtDevicePath = AppendDevicePath (SupportedBootDevice->DevicePathProtocol, FdtDevicePathNodes);\r
    FdtDevicePathSize = GetDevicePathSize (FdtDevicePath);\r
    Status = gRT->SetVariable (\r
                    (CHAR16*)L"Fdt",\r
                    &gArmGlobalVariableGuid,\r
                    EFI_VARIABLE_RUNTIME_ACCESS | EFI_VARIABLE_NON_VOLATILE | EFI_VARIABLE_BOOTSERVICE_ACCESS,\r
                    FdtDevicePathSize,\r
                    FdtDevicePath\r
                    );\r
    ASSERT_EFI_ERROR(Status);\r
  } else {\r
    gRT->SetVariable (\r
           (CHAR16*)L"Fdt",\r
           &gArmGlobalVariableGuid,\r
           EFI_VARIABLE_RUNTIME_ACCESS | EFI_VARIABLE_NON_VOLATILE | EFI_VARIABLE_BOOTSERVICE_ACCESS,\r
           0,\r
           NULL\r
           );\r
    ASSERT_EFI_ERROR(Status);\r
  }\r
\r
EXIT:\r
  if (Status == EFI_ABORTED) {\r
    Print(L"\n");\r
  }\r
  FreePool(SupportedBootDevice);\r
  return Status;\r
}\r
\r
struct BOOT_MANAGER_ENTRY {\r
  CONST CHAR16* Description;\r
  EFI_STATUS (*Callback) (IN LIST_ENTRY *BootOptionsList);\r
} BootManagerEntries[] = {\r
    { L"Add Boot Device Entry", BootMenuAddBootOption },\r
    { L"Update Boot Device Entry", BootMenuUpdateBootOption },\r
    { L"Remove Boot Device Entry", BootMenuRemoveBootOption },\r
    { L"Update FDT path", UpdateFdtPath },\r
};\r
\r
EFI_STATUS\r
BootMenuManager (\r
  IN LIST_ENTRY *BootOptionsList\r
  )\r
{\r
  UINTN Index;\r
  UINTN OptionSelected;\r
  UINTN BootManagerEntryCount;\r
  EFI_STATUS Status;\r
\r
  BootManagerEntryCount = sizeof(BootManagerEntries) / sizeof(struct BOOT_MANAGER_ENTRY);\r
\r
  while (TRUE) {\r
    // Display Boot Manager menu\r
    for (Index = 0; Index < BootManagerEntryCount; Index++) {\r
      Print(L"[%d] %s\n",Index+1,BootManagerEntries[Index]);\r
    }\r
    Print(L"[%d] Return to main menu\n",Index+1);\r
\r
    // Select which entry to call\r
    Print(L"Choice: ");\r
    Status = GetHIInputInteger (&OptionSelected);\r
    if (EFI_ERROR(Status) || (OptionSelected == (BootManagerEntryCount+1))) {\r
      if (EFI_ERROR(Status)) {\r
        Print(L"\n");\r
      }\r
      return EFI_SUCCESS;\r
    } else if ((OptionSelected > 0) && (OptionSelected <= BootManagerEntryCount))  {\r
      BootManagerEntries[OptionSelected-1].Callback (BootOptionsList);\r
    }\r
  }\r
  // Should never go here\r
}\r
\r
EFI_STATUS\r
BootShell (\r
  IN LIST_ENTRY *BootOptionsList\r
  )\r
{\r
  EFI_STATUS Status;\r
\r
  // Start EFI Shell\r
  Status = BdsLoadApplication (mImageHandle, L"Shell", 0, NULL);\r
  if (Status == EFI_NOT_FOUND) {\r
    Print (L"Error: EFI Application not found.\n");\r
  } else if (EFI_ERROR(Status)) {\r
    Print (L"Error: Status Code: 0x%X\n",(UINT32)Status);\r
  }\r
\r
  return Status;\r
}\r
\r
struct BOOT_MAIN_ENTRY {\r
  CONST CHAR16* Description;\r
  EFI_STATUS (*Callback) (IN LIST_ENTRY *BootOptionsList);\r
} BootMainEntries[] = {\r
    { L"Shell", BootShell },\r
    { L"Boot Manager", BootMenuManager },\r
};\r
\r
\r
EFI_STATUS\r
BootMenuMain (\r
  VOID\r
  )\r
{\r
  LIST_ENTRY                    BootOptionsList;\r
  UINTN                         OptionCount;\r
  UINTN                         BootOptionCount;\r
  EFI_STATUS                    Status;\r
  LIST_ENTRY*                   Entry;\r
  BDS_LOAD_OPTION*              BootOption;\r
  UINTN                         BootOptionSelected;\r
  UINTN                         Index;\r
  UINTN                         BootMainEntryCount;\r
  BOOLEAN                       IsUnicode;\r
\r
  BootOption         = NULL;\r
  BootMainEntryCount = sizeof(BootMainEntries) / sizeof(struct BOOT_MAIN_ENTRY);\r
\r
  while (TRUE) {\r
    // Get Boot#### list\r
    BootOptionList (&BootOptionsList);\r
\r
    OptionCount = 1;\r
\r
    // Display the Boot options\r
    for (Entry = GetFirstNode (&BootOptionsList);\r
         !IsNull (&BootOptionsList,Entry);\r
         Entry = GetNextNode (&BootOptionsList,Entry)\r
         )\r
    {\r
      BootOption = LOAD_OPTION_FROM_LINK(Entry);\r
\r
      Print(L"[%d] %s\n", OptionCount, BootOption->Description);\r
\r
      DEBUG_CODE_BEGIN();\r
        CHAR16*                           DevicePathTxt;\r
        EFI_DEVICE_PATH_TO_TEXT_PROTOCOL* DevicePathToTextProtocol;\r
        ARM_BDS_LOADER_OPTIONAL_DATA*     OptionalData;\r
        UINTN                             CmdLineSize;\r
        ARM_BDS_LOADER_TYPE               LoaderType;\r
\r
        Status = gBS->LocateProtocol (&gEfiDevicePathToTextProtocolGuid, NULL, (VOID **)&DevicePathToTextProtocol);\r
        if (EFI_ERROR(Status)) {\r
          // You must provide an implementation of DevicePathToTextProtocol in your firmware (eg: DevicePathDxe)\r
          DEBUG((EFI_D_ERROR,"Error: Bds requires DevicePathToTextProtocol\n"));\r
          return Status;\r
        }\r
        DevicePathTxt = DevicePathToTextProtocol->ConvertDevicePathToText (BootOption->FilePathList, TRUE, TRUE);\r
\r
        Print(L"\t- %s\n",DevicePathTxt);\r
\r
        // If it is a supported BootEntry then print its details\r
        if (IS_ARM_BDS_BOOTENTRY (BootOption)) {\r
          OptionalData = BootOption->OptionalData;\r
          LoaderType = (ARM_BDS_LOADER_TYPE)ReadUnaligned32 ((CONST UINT32*)&OptionalData->Header.LoaderType);\r
          if ((LoaderType == BDS_LOADER_KERNEL_LINUX_ATAG) || (LoaderType == BDS_LOADER_KERNEL_LINUX_FDT)) {\r
            if (ReadUnaligned16 (&OptionalData->Arguments.LinuxArguments.InitrdSize) > 0) {\r
              CmdLineSize = ReadUnaligned16 (&OptionalData->Arguments.LinuxArguments.CmdLineSize);\r
              DevicePathTxt = DevicePathToTextProtocol->ConvertDevicePathToText (\r
                  GetAlignedDevicePath ((EFI_DEVICE_PATH*)((UINTN)(&OptionalData->Arguments.LinuxArguments + 1) + CmdLineSize)), TRUE, TRUE);\r
              Print(L"\t- Initrd: %s\n", DevicePathTxt);\r
            }\r
            if (ReadUnaligned16 (&OptionalData->Arguments.LinuxArguments.CmdLineSize) > 0) {\r
              Print(L"\t- Arguments: %a\n", (&OptionalData->Arguments.LinuxArguments + 1));\r
            }\r
          }\r
\r
          switch (LoaderType) {\r
            case BDS_LOADER_EFI_APPLICATION:\r
              Print(L"\t- LoaderType: EFI Application\n");\r
              break;\r
\r
            case BDS_LOADER_KERNEL_LINUX_ATAG:\r
              Print(L"\t- LoaderType: Linux kernel with ATAG support\n");\r
              break;\r
\r
            case BDS_LOADER_KERNEL_LINUX_FDT:\r
              Print(L"\t- LoaderType: Linux kernel with FDT support\n");\r
              break;\r
\r
            default:\r
              Print(L"\t- LoaderType: Not recognized (%d)\n", LoaderType);\r
          }\r
        } else if (BootOption->OptionalData != NULL) {\r
          if (IsPrintableString (BootOption->OptionalData, &IsUnicode)) {\r
            if (IsUnicode) {\r
              Print (L"\t- Arguments: %s\n", BootOption->OptionalData);\r
            } else {\r
              AsciiPrint ("\t- Arguments: %a\n", BootOption->OptionalData);\r
            }\r
          }\r
        }\r
        FreePool(DevicePathTxt);\r
      DEBUG_CODE_END();\r
\r
      OptionCount++;\r
    }\r
    BootOptionCount = OptionCount-1;\r
\r
    // Display the hardcoded Boot entries\r
    for (Index = 0; Index < BootMainEntryCount; Index++) {\r
      Print(L"[%d] %s\n",OptionCount,BootMainEntries[Index]);\r
      OptionCount++;\r
    }\r
\r
    // Request the boot entry from the user\r
    BootOptionSelected = 0;\r
    while (BootOptionSelected == 0) {\r
      Print(L"Start: ");\r
      Status = GetHIInputInteger (&BootOptionSelected);\r
      if (EFI_ERROR(Status) || (BootOptionSelected == 0) || (BootOptionSelected > OptionCount)) {\r
        Print(L"Invalid input (max %d)\n",(OptionCount-1));\r
        BootOptionSelected = 0;\r
      }\r
    }\r
\r
    // Start the selected entry\r
    if (BootOptionSelected > BootOptionCount) {\r
      // Start the hardcoded entry\r
      Status = BootMainEntries[BootOptionSelected - BootOptionCount - 1].Callback (&BootOptionsList);\r
    } else {\r
      // Find the selected entry from the Boot#### list\r
      Index = 1;\r
      for (Entry = GetFirstNode (&BootOptionsList);\r
           !IsNull (&BootOptionsList,Entry);\r
           Entry = GetNextNode (&BootOptionsList,Entry)\r
           )\r
      {\r
        if (Index == BootOptionSelected) {\r
          BootOption = LOAD_OPTION_FROM_LINK(Entry);\r
          break;\r
        }\r
        Index++;\r
      }\r
\r
      Status = BootOptionStart (BootOption);\r
    }\r
  }\r
  // Should never go here\r
}\r