2 Main file for Dblk shell Debug1 function.
4 (C) Copyright 2015 Hewlett-Packard Development Company, L.P.<BR>
5 Copyright (c) 2005 - 2011, Intel Corporation. All rights reserved.<BR>
6 This program and the accompanying materials
7 are licensed and made available under the terms and conditions of the BSD License
8 which accompanies this distribution. The full text of the license may be found at
9 http://opensource.org/licenses/bsd-license.php
11 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
12 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
16 #include "UefiShellDebug1CommandsLib.h"
17 #include <Protocol/BlockIo.h>
20 Display blocks to the screen.
22 @param[in] DevPath The device path to get the blocks from.
23 @param[in] Lba The Lba number to start from.
24 @param[in] BlockCount How many blocks to display.
26 @retval SHELL_SUCCESS The display was successful.
31 IN CONST EFI_DEVICE_PATH_PROTOCOL
*DevPath
,
33 IN CONST UINT8 BlockCount
36 EFI_BLOCK_IO_PROTOCOL
*BlockIo
;
37 EFI_HANDLE BlockIoHandle
;
39 SHELL_STATUS ShellStatus
;
43 ShellStatus
= SHELL_SUCCESS
;
45 Status
= gBS
->LocateDevicePath(&gEfiBlockIoProtocolGuid
, (EFI_DEVICE_PATH_PROTOCOL
**)&DevPath
, &BlockIoHandle
);
46 if (EFI_ERROR(Status
)) {
47 return (SHELL_NOT_FOUND
);
50 Status
= gBS
->OpenProtocol(BlockIoHandle
, &gEfiBlockIoProtocolGuid
, (VOID
**)&BlockIo
, gImageHandle
, NULL
, EFI_OPEN_PROTOCOL_GET_PROTOCOL
);
51 if (EFI_ERROR(Status
)) {
52 return (SHELL_NOT_FOUND
);
55 BufferSize
= BlockIo
->Media
->BlockSize
* BlockCount
;
57 Buffer
= AllocateZeroPool(BufferSize
);
59 ShellPrintEx(-1,-1,L
" BlockSize: 0x%08x, BlockCount: 0x%08x\r\n", BlockIo
->Media
->BlockSize
, BlockCount
);
63 Status
= BlockIo
->ReadBlocks(BlockIo
, BlockIo
->Media
->MediaId
, Lba
, BufferSize
, Buffer
);
64 if (!EFI_ERROR(Status
) && Buffer
!= NULL
) {
69 STRING_TOKEN (STR_DBLK_HEADER
),
70 gShellDebug1HiiHandle
,
76 DumpHex(2,0,BufferSize
,Buffer
);
78 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_FILE_READ_FAIL
), gShellDebug1HiiHandle
, L
"dblk", L
"BlockIo");
79 ShellStatus
= SHELL_DEVICE_ERROR
;
86 gBS
->CloseProtocol(BlockIoHandle
, &gEfiBlockIoProtocolGuid
, gImageHandle
, NULL
);
91 Function for 'dblk' command.
93 @param[in] ImageHandle Handle to the Image (NULL if Internal).
94 @param[in] SystemTable Pointer to the System Table (NULL if Internal).
99 IN EFI_HANDLE ImageHandle
,
100 IN EFI_SYSTEM_TABLE
*SystemTable
105 CHAR16
*ProblemParam
;
106 SHELL_STATUS ShellStatus
;
107 CONST CHAR16
*BlockName
;
108 CONST CHAR16
*LbaString
;
109 CONST CHAR16
*BlockCountString
;
112 EFI_DEVICE_PATH_PROTOCOL
*DevPath
;
114 ShellStatus
= SHELL_SUCCESS
;
115 Status
= EFI_SUCCESS
;
118 // initialize the shell lib (we must be in non-auto-init...)
120 Status
= ShellInitialize();
121 ASSERT_EFI_ERROR(Status
);
123 Status
= CommandInit();
124 ASSERT_EFI_ERROR(Status
);
127 // parse the command line
129 Status
= ShellCommandLineParse (EmptyParamList
, &Package
, &ProblemParam
, TRUE
);
130 if (EFI_ERROR(Status
)) {
131 if (Status
== EFI_VOLUME_CORRUPTED
&& ProblemParam
!= NULL
) {
132 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_PROBLEM
), gShellDebug1HiiHandle
, L
"dblk", ProblemParam
);
133 FreePool(ProblemParam
);
134 ShellStatus
= SHELL_INVALID_PARAMETER
;
139 if (ShellCommandLineGetCount(Package
) > 4) {
140 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_TOO_MANY
), gShellDebug1HiiHandle
, L
"dblk");
141 ShellStatus
= SHELL_INVALID_PARAMETER
;
142 } else if (ShellCommandLineGetCount(Package
) < 2) {
143 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_TOO_FEW
), gShellDebug1HiiHandle
, L
"dblk");
144 ShellStatus
= SHELL_INVALID_PARAMETER
;
149 BlockName
= ShellCommandLineGetRawValue(Package
, 1);
150 LbaString
= ShellCommandLineGetRawValue(Package
, 2);
151 BlockCountString
= ShellCommandLineGetRawValue(Package
, 3);
153 if (LbaString
== NULL
) {
156 if (!ShellIsHexOrDecimalNumber(LbaString
, TRUE
, FALSE
)) {
157 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_PARAM_INV
), gShellDebug1HiiHandle
, L
"dblk", LbaString
);
158 ShellStatus
= SHELL_INVALID_PARAMETER
;
160 ShellConvertStringToUint64(LbaString
, &Lba
, TRUE
, FALSE
);
163 if (BlockCountString
== NULL
) {
166 if (!ShellIsHexOrDecimalNumber(BlockCountString
, TRUE
, FALSE
)) {
167 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_PARAM_INV
), gShellDebug1HiiHandle
, L
"dblk", BlockCountString
);
168 ShellStatus
= SHELL_INVALID_PARAMETER
;
170 ShellConvertStringToUint64(BlockCountString
, &BlockCount
, TRUE
, FALSE
);
171 if (BlockCount
> 0x10) {
173 } else if (BlockCount
== 0) {
174 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_PARAM_INV
), gShellDebug1HiiHandle
, L
"dblk", BlockCountString
);
175 ShellStatus
= SHELL_INVALID_PARAMETER
;
179 if (ShellStatus
== SHELL_SUCCESS
) {
181 // do the work if we have a valid block identifier
183 if (gEfiShellProtocol
->GetDevicePathFromMap(BlockName
) == NULL
) {
184 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_PARAM_INV
), gShellDebug1HiiHandle
, L
"dblk", BlockName
);
185 ShellStatus
= SHELL_INVALID_PARAMETER
;
187 DevPath
= (EFI_DEVICE_PATH_PROTOCOL
*)gEfiShellProtocol
->GetDevicePathFromMap(BlockName
);
188 if (gBS
->LocateDevicePath(&gEfiBlockIoProtocolGuid
, &DevPath
, NULL
) == EFI_NOT_FOUND
) {
189 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_MAP_PROTOCOL
), gShellDebug1HiiHandle
, L
"dblk", BlockName
, L
"BlockIo");
190 ShellStatus
= SHELL_INVALID_PARAMETER
;
192 ShellStatus
= DisplayTheBlocks(gEfiShellProtocol
->GetDevicePathFromMap(BlockName
), Lba
, (UINT8
)BlockCount
);
198 ShellCommandLineFreeVarList (Package
);
200 return (ShellStatus
);