mirror of
https://github.com/CloverHackyColor/CloverBootloader.git
synced 2024-12-05 13:33:33 +01:00
214 lines
7.3 KiB
C
214 lines
7.3 KiB
C
|
/** @file
|
||
|
Main file for Dblk shell Debug1 function.
|
||
|
|
||
|
(C) Copyright 2015 Hewlett-Packard Development Company, L.P.<BR>
|
||
|
Copyright (c) 2005 - 2017, Intel Corporation. All rights reserved.<BR>
|
||
|
This program and the accompanying materials
|
||
|
are licensed and made available under the terms and conditions of the BSD License
|
||
|
which accompanies this distribution. The full text of the license may be found at
|
||
|
http://opensource.org/licenses/bsd-license.php
|
||
|
|
||
|
THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
|
||
|
WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
|
||
|
|
||
|
**/
|
||
|
|
||
|
#include "UefiShellDebug1CommandsLib.h"
|
||
|
#include <Protocol/BlockIo.h>
|
||
|
|
||
|
/**
|
||
|
Display blocks to the screen.
|
||
|
|
||
|
@param[in] DevPath The device path to get the blocks from.
|
||
|
@param[in] Lba The Lba number to start from.
|
||
|
@param[in] BlockCount How many blocks to display.
|
||
|
|
||
|
@retval SHELL_SUCCESS The display was successful.
|
||
|
**/
|
||
|
SHELL_STATUS
|
||
|
DisplayTheBlocks(
|
||
|
IN CONST EFI_DEVICE_PATH_PROTOCOL *DevPath,
|
||
|
IN CONST UINT64 Lba,
|
||
|
IN CONST UINT8 BlockCount
|
||
|
)
|
||
|
{
|
||
|
EFI_BLOCK_IO_PROTOCOL *BlockIo;
|
||
|
EFI_HANDLE BlockIoHandle;
|
||
|
EFI_STATUS Status;
|
||
|
SHELL_STATUS ShellStatus;
|
||
|
UINT8 *Buffer;
|
||
|
UINT8 *OriginalBuffer;
|
||
|
UINTN BufferSize;
|
||
|
|
||
|
ShellStatus = SHELL_SUCCESS;
|
||
|
|
||
|
Status = gBS->LocateDevicePath(&gEfiBlockIoProtocolGuid, (EFI_DEVICE_PATH_PROTOCOL **)&DevPath, &BlockIoHandle);
|
||
|
if (EFI_ERROR(Status)) {
|
||
|
return (SHELL_NOT_FOUND);
|
||
|
}
|
||
|
|
||
|
Status = gBS->OpenProtocol(BlockIoHandle, &gEfiBlockIoProtocolGuid, (VOID**)&BlockIo, gImageHandle, NULL, EFI_OPEN_PROTOCOL_GET_PROTOCOL);
|
||
|
if (EFI_ERROR(Status)) {
|
||
|
return (SHELL_NOT_FOUND);
|
||
|
}
|
||
|
|
||
|
BufferSize = BlockIo->Media->BlockSize * BlockCount;
|
||
|
if(BlockIo->Media->IoAlign == 0) {
|
||
|
BlockIo->Media->IoAlign = 1;
|
||
|
}
|
||
|
|
||
|
if (BufferSize > 0) {
|
||
|
OriginalBuffer = AllocateZeroPool(BufferSize + BlockIo->Media->IoAlign);
|
||
|
Buffer = ALIGN_POINTER (OriginalBuffer,BlockIo->Media->IoAlign);
|
||
|
} else {
|
||
|
ShellPrintEx(-1,-1,L" BlockSize: 0x%08x, BlockCount: 0x%08x\r\n", BlockIo->Media->BlockSize, BlockCount);
|
||
|
OriginalBuffer = NULL;
|
||
|
Buffer = NULL;
|
||
|
}
|
||
|
|
||
|
Status = BlockIo->ReadBlocks(BlockIo, BlockIo->Media->MediaId, Lba, BufferSize, Buffer);
|
||
|
if (!EFI_ERROR(Status) && Buffer != NULL) {
|
||
|
ShellPrintHiiEx(
|
||
|
-1,
|
||
|
-1,
|
||
|
NULL,
|
||
|
STRING_TOKEN (STR_DBLK_HEADER),
|
||
|
gShellDebug1HiiHandle,
|
||
|
Lba,
|
||
|
BufferSize,
|
||
|
BlockIo
|
||
|
);
|
||
|
|
||
|
DumpHex(2,0,BufferSize,Buffer);
|
||
|
} else {
|
||
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_READ_FAIL), gShellDebug1HiiHandle, L"dblk", L"BlockIo");
|
||
|
ShellStatus = SHELL_DEVICE_ERROR;
|
||
|
}
|
||
|
|
||
|
if (OriginalBuffer != NULL) {
|
||
|
FreePool (OriginalBuffer);
|
||
|
}
|
||
|
|
||
|
gBS->CloseProtocol(BlockIoHandle, &gEfiBlockIoProtocolGuid, gImageHandle, NULL);
|
||
|
return (ShellStatus);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
Function for 'dblk' command.
|
||
|
|
||
|
@param[in] ImageHandle Handle to the Image (NULL if Internal).
|
||
|
@param[in] SystemTable Pointer to the System Table (NULL if Internal).
|
||
|
**/
|
||
|
SHELL_STATUS
|
||
|
EFIAPI
|
||
|
ShellCommandRunDblk (
|
||
|
IN EFI_HANDLE ImageHandle,
|
||
|
IN EFI_SYSTEM_TABLE *SystemTable
|
||
|
)
|
||
|
{
|
||
|
EFI_STATUS Status;
|
||
|
LIST_ENTRY *Package;
|
||
|
CHAR16 *ProblemParam;
|
||
|
SHELL_STATUS ShellStatus;
|
||
|
CONST CHAR16 *BlockName;
|
||
|
CONST CHAR16 *LbaString;
|
||
|
CONST CHAR16 *BlockCountString;
|
||
|
UINT64 Lba;
|
||
|
UINT64 BlockCount;
|
||
|
EFI_DEVICE_PATH_PROTOCOL *DevPath;
|
||
|
|
||
|
ShellStatus = SHELL_SUCCESS;
|
||
|
Status = EFI_SUCCESS;
|
||
|
|
||
|
//
|
||
|
// initialize the shell lib (we must be in non-auto-init...)
|
||
|
//
|
||
|
Status = ShellInitialize();
|
||
|
if (EFI_ERROR(Status)) {
|
||
|
|
||
|
return SHELL_UNSUPPORTED;
|
||
|
}
|
||
|
|
||
|
Status = CommandInit();
|
||
|
// ASSERT_EFI_ERROR(Status);
|
||
|
if (EFI_ERROR(Status)) {
|
||
|
return SHELL_UNSUPPORTED;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// parse the command line
|
||
|
//
|
||
|
Status = ShellCommandLineParse (EmptyParamList, &Package, &ProblemParam, TRUE);
|
||
|
if (EFI_ERROR(Status)) {
|
||
|
if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {
|
||
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, L"dblk", ProblemParam);
|
||
|
FreePool(ProblemParam);
|
||
|
ShellStatus = SHELL_INVALID_PARAMETER;
|
||
|
} /*else {
|
||
|
ASSERT(FALSE);
|
||
|
} */
|
||
|
} else {
|
||
|
if (ShellCommandLineGetCount(Package) > 4) {
|
||
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle, L"dblk");
|
||
|
ShellStatus = SHELL_INVALID_PARAMETER;
|
||
|
} else if (ShellCommandLineGetCount(Package) < 2) {
|
||
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle, L"dblk");
|
||
|
ShellStatus = SHELL_INVALID_PARAMETER;
|
||
|
} else {
|
||
|
//
|
||
|
// Parse the params
|
||
|
//
|
||
|
BlockName = ShellCommandLineGetRawValue(Package, 1);
|
||
|
LbaString = ShellCommandLineGetRawValue(Package, 2);
|
||
|
BlockCountString = ShellCommandLineGetRawValue(Package, 3);
|
||
|
|
||
|
if (LbaString == NULL) {
|
||
|
Lba = 0;
|
||
|
} else {
|
||
|
if (!ShellIsHexOrDecimalNumber(LbaString, TRUE, FALSE)) {
|
||
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellDebug1HiiHandle, L"dblk", LbaString);
|
||
|
ShellStatus = SHELL_INVALID_PARAMETER;
|
||
|
}
|
||
|
ShellConvertStringToUint64(LbaString, &Lba, TRUE, FALSE);
|
||
|
}
|
||
|
|
||
|
if (BlockCountString == NULL) {
|
||
|
BlockCount = 1;
|
||
|
} else {
|
||
|
if (!ShellIsHexOrDecimalNumber(BlockCountString, TRUE, FALSE)) {
|
||
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellDebug1HiiHandle, L"dblk", BlockCountString);
|
||
|
ShellStatus = SHELL_INVALID_PARAMETER;
|
||
|
}
|
||
|
ShellConvertStringToUint64(BlockCountString, &BlockCount, TRUE, FALSE);
|
||
|
if (BlockCount > 0x10) {
|
||
|
BlockCount = 0x10;
|
||
|
} else if (BlockCount == 0) {
|
||
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellDebug1HiiHandle, L"dblk", BlockCountString);
|
||
|
ShellStatus = SHELL_INVALID_PARAMETER;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (ShellStatus == SHELL_SUCCESS) {
|
||
|
//
|
||
|
// do the work if we have a valid block identifier
|
||
|
//
|
||
|
if (gEfiShellProtocol->GetDevicePathFromMap(BlockName) == NULL) {
|
||
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellDebug1HiiHandle, L"dblk", BlockName);
|
||
|
ShellStatus = SHELL_INVALID_PARAMETER;
|
||
|
} else {
|
||
|
DevPath = (EFI_DEVICE_PATH_PROTOCOL*)gEfiShellProtocol->GetDevicePathFromMap(BlockName);
|
||
|
if (gBS->LocateDevicePath(&gEfiBlockIoProtocolGuid, &DevPath, NULL) == EFI_NOT_FOUND) {
|
||
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_MAP_PROTOCOL), gShellDebug1HiiHandle, L"dblk", BlockName, L"BlockIo");
|
||
|
ShellStatus = SHELL_INVALID_PARAMETER;
|
||
|
} else {
|
||
|
ShellStatus = DisplayTheBlocks(gEfiShellProtocol->GetDevicePathFromMap(BlockName), Lba, (UINT8)BlockCount);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
ShellCommandLineFreeVarList (Package);
|
||
|
}
|
||
|
return (ShellStatus);
|
||
|
}
|