Adding Driver1 profile commands to the UEFI Shell 2.0.
git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@11050 6f19259b-4bc3-4df7-8a09-765794883524
This commit is contained in:
jcarsey
235
ShellPkg/Library/UefiShellDriver1CommandsLib/DevTree.c
Normal file
235
ShellPkg/Library/UefiShellDriver1CommandsLib/DevTree.c
Normal file
@ -0,0 +1,235 @@
|
||||
/** @file
|
||||
Main file for DevTree shell Driver1 function.
|
||||
|
||||
Copyright (c) 2010, 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 "UefiShellDriver1CommandsLib.h"
|
||||
|
||||
STATIC CONST SHELL_PARAM_ITEM ParamList[] = {
|
||||
{L"-d", TypeFlag},
|
||||
{L"-l", TypeValue},
|
||||
{NULL, TypeMax}
|
||||
};
|
||||
|
||||
SHELL_STATUS
|
||||
EFIAPI
|
||||
DoDevTreeForHandle(
|
||||
IN CONST EFI_HANDLE TheHandle,
|
||||
IN CONST CHAR8 *Lang OPTIONAL,
|
||||
IN CONST BOOLEAN UseDevPaths,
|
||||
IN CONST UINTN IndentCharCount,
|
||||
IN CONST CHAR16 *HiiString
|
||||
)
|
||||
{
|
||||
SHELL_STATUS ShellStatus;
|
||||
EFI_STATUS Status;
|
||||
CHAR16 *FormatString;
|
||||
CHAR16 *Name;
|
||||
EFI_HANDLE *ChildHandleBuffer;
|
||||
UINTN ChildCount;
|
||||
UINTN LoopVar;
|
||||
|
||||
Status = EFI_SUCCESS;
|
||||
ShellStatus = SHELL_SUCCESS;
|
||||
Name = NULL;
|
||||
ChildHandleBuffer = NULL;
|
||||
ChildCount = 0;
|
||||
|
||||
ASSERT(TheHandle != NULL);
|
||||
//
|
||||
// We want controller handles. they will not have LoadedImage or DriverBinding (or others...)
|
||||
//
|
||||
Status = gBS->OpenProtocol (
|
||||
TheHandle,
|
||||
&gEfiDriverBindingProtocolGuid,
|
||||
NULL,
|
||||
NULL,
|
||||
NULL,
|
||||
EFI_OPEN_PROTOCOL_TEST_PROTOCOL
|
||||
);
|
||||
if (!EFI_ERROR (Status)) {
|
||||
return SHELL_SUCCESS;
|
||||
}
|
||||
|
||||
Status = gBS->OpenProtocol (
|
||||
TheHandle,
|
||||
&gEfiLoadedImageProtocolGuid,
|
||||
NULL,
|
||||
NULL,
|
||||
NULL,
|
||||
EFI_OPEN_PROTOCOL_TEST_PROTOCOL
|
||||
);
|
||||
if (!EFI_ERROR (Status)) {
|
||||
return SHELL_SUCCESS;
|
||||
}
|
||||
|
||||
//
|
||||
// If we are at the begining then we want root handles they have no parents and do have device path.
|
||||
//
|
||||
if (IndentCharCount == 0) {
|
||||
Status = gBS->OpenProtocol (
|
||||
TheHandle,
|
||||
&gEfiDevicePathProtocolGuid,
|
||||
NULL,
|
||||
NULL,
|
||||
NULL,
|
||||
EFI_OPEN_PROTOCOL_TEST_PROTOCOL
|
||||
);
|
||||
if (EFI_ERROR (Status)) {
|
||||
return SHELL_SUCCESS;
|
||||
}
|
||||
}
|
||||
|
||||
FormatString = AllocateZeroPool(StrSize(HiiString) + (10)*sizeof(FormatString[0]));
|
||||
|
||||
ASSERT(HiiString != NULL);
|
||||
ASSERT(FormatString != NULL);
|
||||
|
||||
//
|
||||
// we generate the format string on the fly so that we can control the
|
||||
// number of space characters that the first (empty) string has. this
|
||||
// handles the indenting.
|
||||
//
|
||||
|
||||
UnicodeSPrint(FormatString, StrSize(HiiString) + (10)*sizeof(FormatString[0]), L"%%%ds %s", IndentCharCount, HiiString);
|
||||
gEfiShellProtocol->GetDeviceName((EFI_HANDLE)TheHandle, !UseDevPaths?EFI_DEVICE_NAME_USE_COMPONENT_NAME|EFI_DEVICE_NAME_USE_DEVICE_PATH:EFI_DEVICE_NAME_USE_DEVICE_PATH, (CHAR8*)Lang, &Name);
|
||||
//
|
||||
// print out the information for ourselves
|
||||
//
|
||||
ShellPrintEx(
|
||||
-1,
|
||||
-1,
|
||||
FormatString,
|
||||
L"",
|
||||
ConvertHandleToHandleIndex(TheHandle),
|
||||
Name==NULL?L"Unknown":Name);
|
||||
|
||||
FreePool(FormatString);
|
||||
if (Name != NULL) {
|
||||
FreePool(Name);
|
||||
}
|
||||
|
||||
//
|
||||
// recurse on each child handle with IndentCharCount + 2
|
||||
//
|
||||
ParseHandleDatabaseForChildControllers(TheHandle, &ChildCount, &ChildHandleBuffer);
|
||||
for (LoopVar = 0 ; LoopVar < ChildCount && ShellStatus == SHELL_SUCCESS; LoopVar++){
|
||||
ShellStatus = DoDevTreeForHandle(ChildHandleBuffer[LoopVar], Lang, UseDevPaths, IndentCharCount+2, HiiString);
|
||||
}
|
||||
|
||||
if (ChildHandleBuffer != NULL) {
|
||||
FreePool(ChildHandleBuffer);
|
||||
}
|
||||
|
||||
return (ShellStatus);
|
||||
}
|
||||
|
||||
/**
|
||||
Function for 'devtree' 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
|
||||
ShellCommandRunDevTree (
|
||||
IN EFI_HANDLE ImageHandle,
|
||||
IN EFI_SYSTEM_TABLE *SystemTable
|
||||
)
|
||||
{
|
||||
EFI_STATUS Status;
|
||||
LIST_ENTRY *Package;
|
||||
CHAR16 *ProblemParam;
|
||||
SHELL_STATUS ShellStatus;
|
||||
CHAR8 *Language;
|
||||
CONST CHAR16 *Lang;
|
||||
CHAR16 *HiiString;
|
||||
UINTN LoopVar;
|
||||
EFI_HANDLE TheHandle;
|
||||
BOOLEAN FlagD;
|
||||
|
||||
ShellStatus = SHELL_SUCCESS;
|
||||
Status = EFI_SUCCESS;
|
||||
Language = NULL;
|
||||
|
||||
//
|
||||
// initialize the shell lib (we must be in non-auto-init...)
|
||||
//
|
||||
Status = ShellInitialize();
|
||||
ASSERT_EFI_ERROR(Status);
|
||||
|
||||
Status = CommandInit();
|
||||
ASSERT_EFI_ERROR(Status);
|
||||
|
||||
//
|
||||
// parse the command line
|
||||
//
|
||||
Status = ShellCommandLineParse (ParamList, &Package, &ProblemParam, TRUE);
|
||||
if (EFI_ERROR(Status)) {
|
||||
if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {
|
||||
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDriver1HiiHandle, ProblemParam);
|
||||
FreePool(ProblemParam);
|
||||
ShellStatus = SHELL_INVALID_PARAMETER;
|
||||
} else {
|
||||
ASSERT(FALSE);
|
||||
}
|
||||
} else {
|
||||
if (ShellCommandLineGetCount(Package) > 2) {
|
||||
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDriver1HiiHandle);
|
||||
ShellCommandLineFreeVarList (Package);
|
||||
return (SHELL_INVALID_PARAMETER);
|
||||
}
|
||||
Lang = ShellCommandLineGetValue(Package, L"-l");
|
||||
if (Lang != NULL) {
|
||||
Language = AllocateZeroPool(StrSize(Lang));
|
||||
AsciiSPrint(Language, StrSize(Lang), "%S", Lang);
|
||||
} else if (!ShellCommandLineGetFlag(Package, L"-l")){
|
||||
ASSERT(Language == NULL);
|
||||
// Language = AllocateZeroPool(10);
|
||||
// AsciiSPrint(Language, 10, "en-us");
|
||||
} else {
|
||||
ASSERT(Language == NULL);
|
||||
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_VALUE), gShellDriver1HiiHandle, L"-l");
|
||||
ShellCommandLineFreeVarList (Package);
|
||||
return (SHELL_INVALID_PARAMETER);
|
||||
}
|
||||
FlagD = ShellCommandLineGetFlag(Package, L"-d");
|
||||
|
||||
Lang = ShellCommandLineGetRawValue(Package, 1);
|
||||
HiiString = HiiGetString(gShellDriver1HiiHandle, STRING_TOKEN (STR_DEV_TREE_OUTPUT), Language);
|
||||
|
||||
if (Lang == NULL) {
|
||||
for (LoopVar = 1 ; ; LoopVar++){
|
||||
TheHandle = ConvertHandleIndexToHandle(LoopVar);
|
||||
if (TheHandle == NULL){
|
||||
break;
|
||||
}
|
||||
ShellStatus = DoDevTreeForHandle(TheHandle, Language, FlagD, 0, HiiString);
|
||||
}
|
||||
} else {
|
||||
if (!ShellIsHexOrDecimalNumber(Lang, TRUE, FALSE) || ConvertHandleIndexToHandle(StrHexToUintn(Lang)) == NULL) {
|
||||
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_INV_HANDLE), gShellDriver1HiiHandle, Lang);
|
||||
ShellStatus = SHELL_INVALID_PARAMETER;
|
||||
} else {
|
||||
ShellStatus = DoDevTreeForHandle(ConvertHandleIndexToHandle(StrHexToUintn(Lang)), Language, FlagD, 0, HiiString);
|
||||
}
|
||||
}
|
||||
|
||||
if (HiiString != NULL) {
|
||||
FreePool(HiiString);
|
||||
}
|
||||
SHELL_FREE_NON_NULL(Language);
|
||||
ShellCommandLineFreeVarList (Package);
|
||||
}
|
||||
|
||||
return (ShellStatus);
|
||||
}
|
||||
Reference in New Issue
Block a user