Dblk.c 6.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204
  1. /** @file
  2. Main file for Dblk shell Debug1 function.
  3. (C) Copyright 2015 Hewlett-Packard Development Company, L.P.<BR>
  4. Copyright (c) 2005 - 2018, Intel Corporation. All rights reserved.<BR>
  5. SPDX-License-Identifier: BSD-2-Clause-Patent
  6. **/
  7. #include "UefiShellDebug1CommandsLib.h"
  8. #include <Protocol/BlockIo.h>
  9. /**
  10. Display blocks to the screen.
  11. @param[in] DevPath The device path to get the blocks from.
  12. @param[in] Lba The Lba number to start from.
  13. @param[in] BlockCount How many blocks to display.
  14. @retval SHELL_SUCCESS The display was successful.
  15. **/
  16. SHELL_STATUS
  17. DisplayTheBlocks (
  18. IN CONST EFI_DEVICE_PATH_PROTOCOL *DevPath,
  19. IN CONST UINT64 Lba,
  20. IN CONST UINT8 BlockCount
  21. )
  22. {
  23. EFI_BLOCK_IO_PROTOCOL *BlockIo;
  24. EFI_HANDLE BlockIoHandle;
  25. EFI_STATUS Status;
  26. SHELL_STATUS ShellStatus;
  27. UINT8 *Buffer;
  28. UINT8 *OriginalBuffer;
  29. UINTN BufferSize;
  30. ShellStatus = SHELL_SUCCESS;
  31. Status = gBS->LocateDevicePath (&gEfiBlockIoProtocolGuid, (EFI_DEVICE_PATH_PROTOCOL **)&DevPath, &BlockIoHandle);
  32. if (EFI_ERROR (Status)) {
  33. return (SHELL_NOT_FOUND);
  34. }
  35. Status = gBS->OpenProtocol (BlockIoHandle, &gEfiBlockIoProtocolGuid, (VOID **)&BlockIo, gImageHandle, NULL, EFI_OPEN_PROTOCOL_GET_PROTOCOL);
  36. if (EFI_ERROR (Status)) {
  37. return (SHELL_NOT_FOUND);
  38. }
  39. BufferSize = BlockIo->Media->BlockSize * BlockCount;
  40. if (BlockIo->Media->IoAlign == 0) {
  41. BlockIo->Media->IoAlign = 1;
  42. }
  43. if (BufferSize > 0) {
  44. OriginalBuffer = AllocateZeroPool (BufferSize + BlockIo->Media->IoAlign);
  45. Buffer = ALIGN_POINTER (OriginalBuffer, BlockIo->Media->IoAlign);
  46. } else {
  47. ShellPrintEx (-1, -1, L" BlockSize: 0x%08x, BlockCount: 0x%08x\r\n", BlockIo->Media->BlockSize, BlockCount);
  48. OriginalBuffer = NULL;
  49. Buffer = NULL;
  50. }
  51. Status = BlockIo->ReadBlocks (BlockIo, BlockIo->Media->MediaId, Lba, BufferSize, Buffer);
  52. if (!EFI_ERROR (Status) && (Buffer != NULL)) {
  53. ShellPrintHiiEx (
  54. -1,
  55. -1,
  56. NULL,
  57. STRING_TOKEN (STR_DBLK_HEADER),
  58. gShellDebug1HiiHandle,
  59. Lba,
  60. BufferSize,
  61. BlockIo
  62. );
  63. DumpHex (2, 0, BufferSize, Buffer);
  64. } else {
  65. ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_FILE_READ_FAIL), gShellDebug1HiiHandle, L"dblk", L"BlockIo");
  66. ShellStatus = SHELL_DEVICE_ERROR;
  67. }
  68. if (OriginalBuffer != NULL) {
  69. FreePool (OriginalBuffer);
  70. }
  71. gBS->CloseProtocol (BlockIoHandle, &gEfiBlockIoProtocolGuid, gImageHandle, NULL);
  72. return (ShellStatus);
  73. }
  74. /**
  75. Function for 'dblk' command.
  76. @param[in] ImageHandle Handle to the Image (NULL if Internal).
  77. @param[in] SystemTable Pointer to the System Table (NULL if Internal).
  78. **/
  79. SHELL_STATUS
  80. EFIAPI
  81. ShellCommandRunDblk (
  82. IN EFI_HANDLE ImageHandle,
  83. IN EFI_SYSTEM_TABLE *SystemTable
  84. )
  85. {
  86. EFI_STATUS Status;
  87. LIST_ENTRY *Package;
  88. CHAR16 *ProblemParam;
  89. SHELL_STATUS ShellStatus;
  90. CONST CHAR16 *BlockName;
  91. CONST CHAR16 *LbaString;
  92. CONST CHAR16 *BlockCountString;
  93. UINT64 Lba;
  94. UINT64 BlockCount;
  95. EFI_DEVICE_PATH_PROTOCOL *DevPath;
  96. ShellStatus = SHELL_SUCCESS;
  97. Status = EFI_SUCCESS;
  98. //
  99. // initialize the shell lib (we must be in non-auto-init...)
  100. //
  101. Status = ShellInitialize ();
  102. ASSERT_EFI_ERROR (Status);
  103. Status = CommandInit ();
  104. ASSERT_EFI_ERROR (Status);
  105. //
  106. // parse the command line
  107. //
  108. Status = ShellCommandLineParse (EmptyParamList, &Package, &ProblemParam, TRUE);
  109. if (EFI_ERROR (Status)) {
  110. if ((Status == EFI_VOLUME_CORRUPTED) && (ProblemParam != NULL)) {
  111. ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, L"dblk", ProblemParam);
  112. FreePool (ProblemParam);
  113. ShellStatus = SHELL_INVALID_PARAMETER;
  114. } else {
  115. ASSERT (FALSE);
  116. }
  117. } else {
  118. if (ShellCommandLineGetCount (Package) > 4) {
  119. ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle, L"dblk");
  120. ShellStatus = SHELL_INVALID_PARAMETER;
  121. } else if (ShellCommandLineGetCount (Package) < 2) {
  122. ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle, L"dblk");
  123. ShellStatus = SHELL_INVALID_PARAMETER;
  124. } else {
  125. //
  126. // Parse the params
  127. //
  128. BlockName = ShellCommandLineGetRawValue (Package, 1);
  129. LbaString = ShellCommandLineGetRawValue (Package, 2);
  130. BlockCountString = ShellCommandLineGetRawValue (Package, 3);
  131. if (LbaString == NULL) {
  132. Lba = 0;
  133. } else {
  134. if (!ShellIsHexOrDecimalNumber (LbaString, TRUE, FALSE)) {
  135. ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellDebug1HiiHandle, L"dblk", LbaString);
  136. ShellStatus = SHELL_INVALID_PARAMETER;
  137. }
  138. ShellConvertStringToUint64 (LbaString, &Lba, TRUE, FALSE);
  139. }
  140. if (BlockCountString == NULL) {
  141. BlockCount = 1;
  142. } else {
  143. if (!ShellIsHexOrDecimalNumber (BlockCountString, TRUE, FALSE)) {
  144. ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellDebug1HiiHandle, L"dblk", BlockCountString);
  145. ShellStatus = SHELL_INVALID_PARAMETER;
  146. }
  147. ShellConvertStringToUint64 (BlockCountString, &BlockCount, TRUE, FALSE);
  148. if (BlockCount > 0x10) {
  149. BlockCount = 0x10;
  150. } else if (BlockCount == 0) {
  151. ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellDebug1HiiHandle, L"dblk", BlockCountString);
  152. ShellStatus = SHELL_INVALID_PARAMETER;
  153. }
  154. }
  155. if (ShellStatus == SHELL_SUCCESS) {
  156. //
  157. // do the work if we have a valid block identifier
  158. //
  159. if (gEfiShellProtocol->GetDevicePathFromMap (BlockName) == NULL) {
  160. ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellDebug1HiiHandle, L"dblk", BlockName);
  161. ShellStatus = SHELL_INVALID_PARAMETER;
  162. } else {
  163. DevPath = (EFI_DEVICE_PATH_PROTOCOL *)gEfiShellProtocol->GetDevicePathFromMap (BlockName);
  164. if (gBS->LocateDevicePath (&gEfiBlockIoProtocolGuid, &DevPath, NULL) == EFI_NOT_FOUND) {
  165. ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_MAP_PROTOCOL), gShellDebug1HiiHandle, L"dblk", BlockName, L"BlockIo");
  166. ShellStatus = SHELL_INVALID_PARAMETER;
  167. } else {
  168. ShellStatus = DisplayTheBlocks (gEfiShellProtocol->GetDevicePathFromMap (BlockName), Lba, (UINT8)BlockCount);
  169. }
  170. }
  171. }
  172. }
  173. ShellCommandLineFreeVarList (Package);
  174. }
  175. return (ShellStatus);
  176. }