tyler.durden
/
audk
mirror of https://github.com/acidanthera/audk.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
audk/ShellPkg/Library/UefiShellDriver1CommandsLib/DrvCfg.c

1480 lines
48 KiB
C
Raw Blame History

/** @file
Main file for DrvCfg shell Driver1 function.
(C) Copyright 2015 Hewlett-Packard Development Company, L.P.<BR>
Copyright (c) 2010 - 2018, Intel Corporation. All rights reserved.<BR>
SPDX-License-Identifier: BSD-2-Clause-Patent
**/
#include "UefiShellDriver1CommandsLib.h"
#include <Protocol/HiiConfigAccess.h>
#include <Protocol/HiiDatabase.h>
STATIC CONST EFI_GUID *CfgGuidList[] = { &gEfiDriverConfigurationProtocolGuid, &gEfiDriverConfiguration2ProtocolGuid, NULL };
/**
Find the EFI_HII_HANDLE by device path.
@param[in] DevPath1 The Device Path to match.
@param[out] HiiHandle The EFI_HII_HANDLE after the converstion.
@param[in] HiiDb The Hii database protocol
@retval EFI_SUCCESS The operation was successful.
@retval EFI_NOT_FOUND There was no EFI_HII_HANDLE found for that deviec path.
**/
EFI_STATUS
FindHiiHandleViaDevPath (
IN CONST EFI_DEVICE_PATH_PROTOCOL *DevPath1,
OUT EFI_HII_HANDLE *HiiHandle,
IN EFI_HII_DATABASE_PROTOCOL *HiiDb
)
{
EFI_HII_HANDLE *HandleBuffer;
UINTN HandleBufferSize;
VOID *MainBuffer;
UINTN MainBufferSize;
EFI_HII_PACKAGE_LIST_HEADER *PackageListHeader;
EFI_HII_PACKAGE_HEADER *PackageHeader;
UINTN LoopVariable;
EFI_DEVICE_PATH_PROTOCOL *DevPath2;
EFI_STATUS Status;
ASSERT (DevPath1 != NULL);
ASSERT (HiiHandle != NULL);
ASSERT (*HiiHandle == NULL);
ASSERT (HiiDb != NULL);
HandleBufferSize = 0;
HandleBuffer = NULL;
Status = HiiDb->ListPackageLists (HiiDb, EFI_HII_PACKAGE_DEVICE_PATH, NULL, &HandleBufferSize, HandleBuffer);
if (Status == EFI_BUFFER_TOO_SMALL) {
HandleBuffer = AllocateZeroPool (HandleBufferSize);
if (HandleBuffer == NULL) {
Status = EFI_OUT_OF_RESOURCES;
} else {
Status = HiiDb->ListPackageLists (HiiDb, EFI_HII_PACKAGE_DEVICE_PATH, NULL, &HandleBufferSize, HandleBuffer);
}
}
if (EFI_ERROR (Status)) {
SHELL_FREE_NON_NULL (HandleBuffer);
return (Status);
}
if (HandleBuffer == NULL) {
return EFI_NOT_FOUND;
}
for (LoopVariable = 0; LoopVariable < (HandleBufferSize/sizeof (HandleBuffer[0])) && *HiiHandle == NULL; LoopVariable++) {
MainBufferSize = 0;
MainBuffer = NULL;
Status = HiiDb->ExportPackageLists (HiiDb, HandleBuffer[LoopVariable], &MainBufferSize, MainBuffer);
if (Status == EFI_BUFFER_TOO_SMALL) {
MainBuffer = AllocateZeroPool (MainBufferSize);
if (MainBuffer != NULL) {
Status = HiiDb->ExportPackageLists (HiiDb, HandleBuffer[LoopVariable], &MainBufferSize, MainBuffer);
}
}
if (EFI_ERROR (Status)) {
continue;
}
//
// Enumerate through the block of returned memory.
// This should actually be a small block, but we need to be sure.
//
for (PackageListHeader = (EFI_HII_PACKAGE_LIST_HEADER *)MainBuffer
; PackageListHeader != NULL && ((CHAR8 *)PackageListHeader) < (((CHAR8 *)MainBuffer)+MainBufferSize) && *HiiHandle == NULL
; PackageListHeader = (EFI_HII_PACKAGE_LIST_HEADER *)(((CHAR8 *)(PackageListHeader)) + PackageListHeader->PackageLength))
{
for (PackageHeader = (EFI_HII_PACKAGE_HEADER *)(((CHAR8 *)(PackageListHeader))+sizeof (EFI_HII_PACKAGE_LIST_HEADER))
; PackageHeader != NULL && ((CHAR8 *)PackageHeader) < (((CHAR8 *)MainBuffer)+MainBufferSize) && PackageHeader->Type != EFI_HII_PACKAGE_END && *HiiHandle == NULL
; PackageHeader = (EFI_HII_PACKAGE_HEADER *)(((CHAR8 *)(PackageHeader))+PackageHeader->Length))
{
if (PackageHeader->Type == EFI_HII_PACKAGE_DEVICE_PATH) {
DevPath2 = (EFI_DEVICE_PATH_PROTOCOL *)(((CHAR8 *)PackageHeader) + sizeof (EFI_HII_PACKAGE_HEADER));
if (DevicePathCompare (&DevPath1, &DevPath2) == 0) {
*HiiHandle = HandleBuffer[LoopVariable];
break;
}
}
}
}
SHELL_FREE_NON_NULL (MainBuffer);
}
SHELL_FREE_NON_NULL (HandleBuffer);
if (*HiiHandle == NULL) {
return (EFI_NOT_FOUND);
}
return (EFI_SUCCESS);
}
/**
Convert a EFI_HANDLE to a EFI_HII_HANDLE.
@param[in] Handle The EFI_HANDLE to convert.
@param[out] HiiHandle The EFI_HII_HANDLE after the converstion.
@param[in] HiiDb The Hii database protocol
@retval EFI_SUCCESS The operation was successful.
**/
EFI_STATUS
ConvertHandleToHiiHandle (
IN CONST EFI_HANDLE Handle,
OUT EFI_HII_HANDLE *HiiHandle,
IN EFI_HII_DATABASE_PROTOCOL *HiiDb
)
{
EFI_STATUS Status;
EFI_DEVICE_PATH_PROTOCOL *DevPath1;
if ((HiiHandle == NULL) || (HiiDb == NULL)) {
return (EFI_INVALID_PARAMETER);
}
*HiiHandle = NULL;
if (Handle == NULL) {
return (EFI_SUCCESS);
}
DevPath1 = NULL;
Status = gBS->OpenProtocol (Handle, &gEfiDevicePathProtocolGuid, (VOID **)&DevPath1, gImageHandle, NULL, EFI_OPEN_PROTOCOL_GET_PROTOCOL);
if (EFI_ERROR (Status) || (DevPath1 == NULL)) {
return (EFI_NOT_FOUND);
}
return (FindHiiHandleViaDevPath (DevPath1, HiiHandle, HiiDb));
}
/**
Function to print out all HII configuration information to a file.
@param[in] Handle The handle to get info on. NULL to do all handles.
@param[in] FileName The filename to rwite the info to.
**/
SHELL_STATUS
ConfigToFile (
IN CONST EFI_HANDLE Handle,
IN CONST CHAR16 *FileName
)
{
EFI_HII_DATABASE_PROTOCOL *HiiDatabase;
EFI_STATUS Status;
VOID *MainBuffer;
UINTN MainBufferSize;
EFI_HII_HANDLE HiiHandle;
SHELL_FILE_HANDLE FileHandle;
HiiDatabase = NULL;
MainBufferSize = 0;
MainBuffer = NULL;
FileHandle = NULL;
Status = ShellOpenFileByName (FileName, &FileHandle, EFI_FILE_MODE_READ|EFI_FILE_MODE_WRITE|EFI_FILE_MODE_CREATE, 0);
if (EFI_ERROR (Status)) {
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_GEN_FILE_OPEN_FAIL),
gShellDriver1HiiHandle,
L"drvcfg",
FileName,
Status
);
return (SHELL_DEVICE_ERROR);
}
//
// Locate HII Database protocol
//
Status = gBS->LocateProtocol (
&gEfiHiiDatabaseProtocolGuid,
NULL,
(VOID **)&HiiDatabase
);
if (EFI_ERROR (Status) || (HiiDatabase == NULL)) {
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_GEN_PROTOCOL_NF),
gShellDriver1HiiHandle,
L"drvcfg",
L"EfiHiiDatabaseProtocol",
&gEfiHiiDatabaseProtocolGuid
);
ShellCloseFile (&FileHandle);
return (SHELL_NOT_FOUND);
}
HiiHandle = NULL;
Status = ConvertHandleToHiiHandle (Handle, &HiiHandle, HiiDatabase);
if (EFI_ERROR (Status)) {
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_GEN_HANDLE_NOT),
gShellDriver1HiiHandle,
L"drvcfg",
ConvertHandleToHandleIndex (Handle),
L"Device"
);
ShellCloseFile (&FileHandle);
return (SHELL_DEVICE_ERROR);
}
Status = HiiDatabase->ExportPackageLists (HiiDatabase, HiiHandle, &MainBufferSize, MainBuffer);
if (Status == EFI_BUFFER_TOO_SMALL) {
MainBuffer = AllocateZeroPool (MainBufferSize);
Status = HiiDatabase->ExportPackageLists (HiiDatabase, HiiHandle, &MainBufferSize, MainBuffer);
}
Status = ShellWriteFile (FileHandle, &MainBufferSize, MainBuffer);
ShellCloseFile (&FileHandle);
SHELL_FREE_NON_NULL (MainBuffer);
if (EFI_ERROR (Status)) {
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_FILE_WRITE_FAIL),
gShellDriver1HiiHandle,
L"drvcfg",
FileName
);
return (SHELL_DEVICE_ERROR);
}
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_DRVCFG_COMP),
gShellDriver1HiiHandle
);
return (SHELL_SUCCESS);
}
/**
Function to read in HII configuration information from a file.
@param[in] Handle The handle to get info for.
@param[in] FileName The filename to read the info from.
**/
SHELL_STATUS
ConfigFromFile (
IN EFI_HANDLE Handle,
IN CONST CHAR16 *FileName
)
{
EFI_HII_DATABASE_PROTOCOL *HiiDatabase;
EFI_STATUS Status;
VOID *MainBuffer;
UINT64 Temp;
UINTN MainBufferSize;
EFI_HII_HANDLE HiiHandle;
SHELL_FILE_HANDLE FileHandle;
CHAR16 *TempDevPathString;
EFI_HII_PACKAGE_LIST_HEADER *PackageListHeader;
EFI_HII_PACKAGE_HEADER *PackageHeader;
EFI_DEVICE_PATH_PROTOCOL *DevPath;
UINTN HandleIndex;
HiiDatabase = NULL;
MainBufferSize = 0;
MainBuffer = NULL;
FileHandle = NULL;
Status = ShellOpenFileByName (FileName, &FileHandle, EFI_FILE_MODE_READ, 0);
if (EFI_ERROR (Status)) {
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_GEN_FILE_OPEN_FAIL),
gShellDriver1HiiHandle,
L"drvcfg",
FileName,
Status
);
return (SHELL_DEVICE_ERROR);
}
//
// Locate HII Database protocol
//
Status = gBS->LocateProtocol (
&gEfiHiiDatabaseProtocolGuid,
NULL,
(VOID **)&HiiDatabase
);
if (EFI_ERROR (Status) || (HiiDatabase == NULL)) {
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_GEN_PROTOCOL_NF),
gShellDriver1HiiHandle,
L"drvcfg",
L"EfiHiiDatabaseProtocol",
&gEfiHiiDatabaseProtocolGuid
);
ShellCloseFile (&FileHandle);
return (SHELL_NOT_FOUND);
}
Status = ShellGetFileSize (FileHandle, &Temp);
MainBufferSize = (UINTN)Temp;
if (EFI_ERROR (Status)) {
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_FILE_READ_FAIL),
gShellDriver1HiiHandle,
L"drvcfg",
FileName
);
ShellCloseFile (&FileHandle);
return (SHELL_DEVICE_ERROR);
}
MainBuffer = AllocateZeroPool ((UINTN)MainBufferSize);
if (EFI_ERROR (Status)) {
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_GEN_OUT_MEM),
gShellDriver1HiiHandle,
L"drvcfg"
);
ShellCloseFile (&FileHandle);
return (SHELL_DEVICE_ERROR);
}
Status = ShellReadFile (FileHandle, &MainBufferSize, MainBuffer);
if (EFI_ERROR (Status)) {
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_FILE_READ_FAIL),
gShellDriver1HiiHandle,
L"drvcfg",
FileName
);
ShellCloseFile (&FileHandle);
SHELL_FREE_NON_NULL (MainBuffer);
return (SHELL_DEVICE_ERROR);
}
ShellCloseFile (&FileHandle);
if (Handle != NULL) {
//
// User override in place. Just do it.
//
HiiHandle = NULL;
Status = ConvertHandleToHiiHandle (Handle, &HiiHandle, HiiDatabase);
if (EFI_ERROR (Status)) {
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_GEN_HANDLE_NOT),
gShellDriver1HiiHandle,
L"drvcfg",
ConvertHandleToHandleIndex (Handle),
L"Device"
);
ShellCloseFile (&FileHandle);
return (SHELL_DEVICE_ERROR);
}
Status = HiiDatabase->UpdatePackageList (HiiDatabase, HiiHandle, MainBuffer);
if (EFI_ERROR (Status)) {
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_GEN_UEFI_FUNC_WARN),
gShellDriver1HiiHandle,
L"drvcfg",
L"HiiDatabase->UpdatePackageList",
Status
);
return (SHELL_DEVICE_ERROR);
}
} else {
//
// we need to parse the buffer and try to match the device paths for each item to try to find it's device path.
//
for (PackageListHeader = (EFI_HII_PACKAGE_LIST_HEADER *)MainBuffer
; PackageListHeader != NULL && ((CHAR8 *)PackageListHeader) < (((CHAR8 *)MainBuffer)+MainBufferSize)
; PackageListHeader = (EFI_HII_PACKAGE_LIST_HEADER *)(((CHAR8 *)(PackageListHeader)) + PackageListHeader->PackageLength))
{
for (PackageHeader = (EFI_HII_PACKAGE_HEADER *)(((CHAR8 *)(PackageListHeader))+sizeof (EFI_HII_PACKAGE_LIST_HEADER))
; PackageHeader != NULL && ((CHAR8 *)PackageHeader) < (((CHAR8 *)MainBuffer)+MainBufferSize) && PackageHeader->Type != EFI_HII_PACKAGE_END
; PackageHeader = (EFI_HII_PACKAGE_HEADER *)(((CHAR8 *)(PackageHeader))+PackageHeader->Length))
{
if (PackageHeader->Type == EFI_HII_PACKAGE_DEVICE_PATH) {
HiiHandle = NULL;
Status = FindHiiHandleViaDevPath ((EFI_DEVICE_PATH_PROTOCOL *)(((CHAR8 *)PackageHeader) + sizeof (EFI_HII_PACKAGE_HEADER)), &HiiHandle, HiiDatabase);
if (EFI_ERROR (Status)) {
//
// print out an error.
//
TempDevPathString = ConvertDevicePathToText ((EFI_DEVICE_PATH_PROTOCOL *)(((CHAR8 *)PackageHeader) + sizeof (EFI_HII_PACKAGE_HEADER)), TRUE, TRUE);
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_DRVCFG_IN_FILE_NF),
gShellDriver1HiiHandle,
TempDevPathString
);
SHELL_FREE_NON_NULL (TempDevPathString);
} else {
Status = HiiDatabase->UpdatePackageList (HiiDatabase, HiiHandle, PackageListHeader);
if (EFI_ERROR (Status)) {
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_GEN_UEFI_FUNC_WARN),
gShellDriver1HiiHandle,
L"drvcfg",
L"HiiDatabase->UpdatePackageList",
Status
);
return (SHELL_DEVICE_ERROR);
} else {
DevPath = (EFI_DEVICE_PATH_PROTOCOL *)(((CHAR8 *)PackageHeader) + sizeof (EFI_HII_PACKAGE_HEADER));
gBS->LocateDevicePath (&gEfiHiiConfigAccessProtocolGuid, &DevPath, &Handle);
HandleIndex = ConvertHandleToHandleIndex (Handle);
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_DRVCFG_DONE_HII),
gShellDriver1HiiHandle,
HandleIndex
);
}
}
}
}
}
}
SHELL_FREE_NON_NULL (MainBuffer);
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_DRVCFG_COMP),
gShellDriver1HiiHandle
);
return (SHELL_SUCCESS);
}
/**
Present a requested action to the user.
@param[in] DriverImageHandle The handle for the driver to configure.
@param[in] ControllerHandle The handle of the device being managed by the Driver specified.
@param[in] ChildHandle The handle of a child device of the specified device.
@param[in] ActionRequired The required HII action.
@retval SHELL_INVALID_PARAMETER A parameter has a invalid value.
**/
EFI_STATUS
ShellCmdDriverConfigurationProcessActionRequired (
EFI_HANDLE DriverImageHandle,
EFI_HANDLE ControllerHandle,
EFI_HANDLE ChildHandle,
EFI_DRIVER_CONFIGURATION_ACTION_REQUIRED ActionRequired
)
{
EFI_HANDLE ConnectControllerContextOverride[2];
switch (ActionRequired) {
case EfiDriverConfigurationActionNone:
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_DRVCFG_NONE), gShellDriver1HiiHandle);
break;
case EfiDriverConfigurationActionStopController:
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_DRVCFG_STOP), gShellDriver1HiiHandle);
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_DRVCFG_ENTER_S), gShellDriver1HiiHandle, L"stop controller");
ShellPromptForResponse (ShellPromptResponseTypeEnterContinue, NULL, NULL);
gBS->DisconnectController (ControllerHandle, DriverImageHandle, ChildHandle);
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_DRVCFG_CTLR_S), gShellDriver1HiiHandle, L"stopped");
break;
case EfiDriverConfigurationActionRestartController:
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_DRVCFG_RESTART_S), gShellDriver1HiiHandle, L"controller");
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_DRVCFG_ENTER_S), gShellDriver1HiiHandle, L"restart controller");
ShellPromptForResponse (ShellPromptResponseTypeEnterContinue, NULL, NULL);
gBS->DisconnectController (ControllerHandle, DriverImageHandle, ChildHandle);
ConnectControllerContextOverride[0] = DriverImageHandle;
ConnectControllerContextOverride[1] = NULL;
gBS->ConnectController (ControllerHandle, ConnectControllerContextOverride, NULL, TRUE);
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_DRVCFG_CTLR_S), gShellDriver1HiiHandle, L"restarted");
break;
case EfiDriverConfigurationActionRestartPlatform:
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_DRVCFG_RESTART_S), gShellDriver1HiiHandle, L"platform");
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_DRVCFG_ENTER_S), gShellDriver1HiiHandle, L"restart platform");
ShellPromptForResponse (ShellPromptResponseTypeEnterContinue, NULL, NULL);
gRT->ResetSystem (EfiResetCold, EFI_SUCCESS, 0, NULL);
break;
default:
return (EFI_INVALID_PARAMETER);
}
return EFI_SUCCESS;
}
/**
Do the configuration in an environment without HII.
@param[in] Language The language code.
@param[in] ForceDefaults TRUE to force defaults, FALSE otherwise.
@param[in] DefaultType If ForceDefaults is TRUE, specifies the default type.
@param[in] AllChildren TRUE to configure all children, FALSE otherwise.
@param[in] ValidateOptions TRUE to validate existing options, FALSE otherwise.
@param[in] SetOptions TRUE to set options, FALSE otherwise.
@param[in] DriverImageHandle The handle for the driver to configure.
@param[in] DeviceHandle The handle of the device being managed by the Driver specified.
@param[in] ChildHandle The handle of a child device of the specified device.
@retval SHELL_NOT_FOUND A specified handle could not be found.
@retval SHELL_INVALID_PARAMETER A parameter has a invalid value.
**/
SHELL_STATUS
PreHiiDrvCfg (
IN CONST CHAR8 *Language,
IN BOOLEAN ForceDefaults,
IN UINT32 DefaultType,
IN BOOLEAN AllChildren,
IN BOOLEAN ValidateOptions,
IN BOOLEAN SetOptions,
IN EFI_HANDLE DriverImageHandle,
IN EFI_HANDLE DeviceHandle,
IN EFI_HANDLE ChildHandle
)
{
EFI_STATUS Status;
SHELL_STATUS ShellStatus;
UINTN OuterLoopCounter;
CHAR8 *BestLanguage;
UINTN DriverImageHandleCount;
EFI_HANDLE *DriverImageHandleBuffer;
UINTN HandleCount;
EFI_HANDLE *HandleBuffer;
UINTN *HandleType;
UINTN LoopCounter;
UINTN ChildIndex;
UINTN ChildHandleCount;
EFI_HANDLE *ChildHandleBuffer;
UINTN *ChildHandleType;
EFI_DRIVER_CONFIGURATION_ACTION_REQUIRED ActionRequired;
EFI_DRIVER_CONFIGURATION_PROTOCOL *DriverConfiguration;
BOOLEAN Iso639Language;
UINTN HandleIndex1;
UINTN HandleIndex2;
UINTN HandleIndex3;
ShellStatus = SHELL_SUCCESS;
if ((ChildHandle == NULL) && AllChildren) {
SetOptions = FALSE;
}
if (ForceDefaults) {
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_DRVCFG_FORCE_D),
gShellDriver1HiiHandle,
DefaultType
);
} else if (ValidateOptions) {
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_DRVCFG_VALIDATE),
gShellDriver1HiiHandle
);
} else if (SetOptions) {
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_DRVCFG_SET),
gShellDriver1HiiHandle
);
}
if (DriverImageHandle == 0) {
DriverImageHandleBuffer = GetHandleListByProtocolList (CfgGuidList);
if (DriverImageHandleBuffer == NULL) {
ShellStatus = SHELL_NOT_FOUND;
goto Done;
}
for (
HandleBuffer = DriverImageHandleBuffer, DriverImageHandleCount = 0
; HandleBuffer != NULL && *HandleBuffer != NULL
; HandleBuffer++, DriverImageHandleCount++)
{
}
} else {
DriverImageHandleCount = 1;
//
// Allocate buffer to hold the image handle so as to
// keep consistent with the above clause
//
DriverImageHandleBuffer = AllocatePool (sizeof (EFI_HANDLE));
ASSERT (DriverImageHandleBuffer);
DriverImageHandleBuffer[0] = DriverImageHandle;
}
for (OuterLoopCounter = 0; OuterLoopCounter < DriverImageHandleCount; OuterLoopCounter++) {
Iso639Language = FALSE;
Status = gBS->OpenProtocol (
DriverImageHandleBuffer[OuterLoopCounter],
&gEfiDriverConfiguration2ProtocolGuid,
(VOID **)&DriverConfiguration,
NULL,
NULL,
EFI_OPEN_PROTOCOL_GET_PROTOCOL
);
if (EFI_ERROR (Status)) {
Iso639Language = TRUE;
Status = gBS->OpenProtocol (
DriverImageHandleBuffer[OuterLoopCounter],
&gEfiDriverConfigurationProtocolGuid,
(VOID **)&DriverConfiguration,
NULL,
NULL,
EFI_OPEN_PROTOCOL_GET_PROTOCOL
);
}
if (EFI_ERROR (Status)) {
// ShellPrintHiiEx(
// -1,
// -1,
// NULL,
// STRING_TOKEN (STR_DRVCFG_NOT_SUPPORT),
// gShellDriver1HiiHandle,
// ConvertHandleToHandleIndex (DriverImageHandleBuffer[OuterLoopCounter])
// );
ShellStatus = SHELL_UNSUPPORTED;
continue;
}
BestLanguage = GetBestLanguage (
DriverConfiguration->SupportedLanguages,
Iso639Language,
Language != NULL ? Language : "",
DriverConfiguration->SupportedLanguages,
NULL
);
if (BestLanguage == NULL) {
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_GEN_NO_VALUE),
gShellDriver1HiiHandle,
L"drvcfg",
L"-l"
);
ShellStatus = SHELL_INVALID_PARAMETER;
continue;
}
Status = ParseHandleDatabaseByRelationshipWithType (
DriverImageHandleBuffer[OuterLoopCounter],
NULL,
&HandleCount,
&HandleBuffer,
&HandleType
);
if (EFI_ERROR (Status)) {
continue;
}
if (SetOptions && (DeviceHandle == NULL)) {
gST->ConOut->ClearScreen (gST->ConOut);
Status = DriverConfiguration->SetOptions (
DriverConfiguration,
NULL,
NULL,
BestLanguage,
&ActionRequired
);
gST->ConOut->ClearScreen (gST->ConOut);
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_DRVCFG_ALL_LANG),
gShellDriver1HiiHandle,
ConvertHandleToHandleIndex (DriverImageHandleBuffer[OuterLoopCounter]),
DriverConfiguration->SupportedLanguages
);
if (!EFI_ERROR (Status)) {
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_DRVCFG_OPTIONS_SET),
gShellDriver1HiiHandle
);
for (LoopCounter = 0; LoopCounter < HandleCount; LoopCounter++) {
if ((HandleType[LoopCounter] & HR_CONTROLLER_HANDLE) == HR_CONTROLLER_HANDLE) {
ShellCmdDriverConfigurationProcessActionRequired (
DriverImageHandleBuffer[OuterLoopCounter],
HandleBuffer[LoopCounter],
NULL,
ActionRequired
);
}
}
} else {
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_DRVCFG_NOT_SET),
gShellDriver1HiiHandle,
Status
);
}
continue;
}
for (LoopCounter = 0; LoopCounter < HandleCount; LoopCounter++) {
if ((HandleType[LoopCounter] & HR_CONTROLLER_HANDLE) != HR_CONTROLLER_HANDLE) {
continue;
}
if ((DeviceHandle != NULL) && (DeviceHandle != HandleBuffer[LoopCounter])) {
continue;
}
if (ChildHandle == NULL) {
HandleIndex1 = ConvertHandleToHandleIndex (DriverImageHandleBuffer[OuterLoopCounter]);
HandleIndex2 = ConvertHandleToHandleIndex (HandleBuffer[LoopCounter]);
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_DRVCFG_CTRL_LANG),
gShellDriver1HiiHandle,
HandleIndex1,
HandleIndex2,
DriverConfiguration->SupportedLanguages
);
if (ForceDefaults) {
Status = DriverConfiguration->ForceDefaults (
DriverConfiguration,
HandleBuffer[LoopCounter],
NULL,
DefaultType,
&ActionRequired
);
if (!EFI_ERROR (Status)) {
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_DRVCFG_DEF_FORCED),
gShellDriver1HiiHandle
);
ShellCmdDriverConfigurationProcessActionRequired (
DriverImageHandleBuffer[OuterLoopCounter],
HandleBuffer[LoopCounter],
NULL,
ActionRequired
);
} else {
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_DRVCFG_FORCE_FAILED),
gShellDriver1HiiHandle,
Status
);
ShellStatus = SHELL_DEVICE_ERROR;
}
} else if (ValidateOptions) {
Status = DriverConfiguration->OptionsValid (
DriverConfiguration,
HandleBuffer[LoopCounter],
NULL
);
if (!EFI_ERROR (Status)) {
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_DRVCFG_OPTIONS_VALID),
gShellDriver1HiiHandle
);
} else {
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_DRVCFG_OPTIONS_INV),
gShellDriver1HiiHandle,
Status
);
ShellStatus = SHELL_DEVICE_ERROR;
}
} else if (SetOptions) {
gST->ConOut->ClearScreen (gST->ConOut);
Status = DriverConfiguration->SetOptions (
DriverConfiguration,
HandleBuffer[LoopCounter],
NULL,
BestLanguage,
&ActionRequired
);
gST->ConOut->ClearScreen (gST->ConOut);
HandleIndex1 = ConvertHandleToHandleIndex (DriverImageHandleBuffer[OuterLoopCounter]);
HandleIndex2 = ConvertHandleToHandleIndex (HandleBuffer[LoopCounter]);
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_DRVCFG_CTRL_LANG),
gShellDriver1HiiHandle,
HandleIndex1,
HandleIndex2,
DriverConfiguration->SupportedLanguages
);
if (!EFI_ERROR (Status)) {
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_DRVCFG_OPTIONS_SET),
gShellDriver1HiiHandle
);
ShellCmdDriverConfigurationProcessActionRequired (
DriverImageHandleBuffer[OuterLoopCounter],
HandleBuffer[LoopCounter],
NULL,
ActionRequired
);
} else {
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_DRVCFG_NOT_SET),
gShellDriver1HiiHandle,
Status
);
ShellStatus = SHELL_DEVICE_ERROR;
}
} else {
Print (L"\n");
}
}
if ((ChildHandle == NULL) && !AllChildren) {
continue;
}
Status = ParseHandleDatabaseByRelationshipWithType (
DriverImageHandleBuffer[OuterLoopCounter],
HandleBuffer[LoopCounter],
&ChildHandleCount,
&ChildHandleBuffer,
&ChildHandleType
);
if (EFI_ERROR (Status)) {
continue;
}
for (ChildIndex = 0; ChildIndex < ChildHandleCount; ChildIndex++) {
if ((ChildHandleType[ChildIndex] & HR_CHILD_HANDLE) != HR_CHILD_HANDLE) {
continue;
}
if ((ChildHandle != NULL) && (ChildHandle != ChildHandleBuffer[ChildIndex])) {
continue;
}
HandleIndex1 = ConvertHandleToHandleIndex (DriverImageHandleBuffer[OuterLoopCounter]);
HandleIndex2 = ConvertHandleToHandleIndex (HandleBuffer[LoopCounter]);
HandleIndex3 = ConvertHandleToHandleIndex (ChildHandleBuffer[ChildIndex]);
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_DRVCFG_CHILD_LANG),
gShellDriver1HiiHandle,
HandleIndex1,
HandleIndex2,
HandleIndex3,
DriverConfiguration->SupportedLanguages
);
if (ForceDefaults) {
Status = DriverConfiguration->ForceDefaults (
DriverConfiguration,
HandleBuffer[LoopCounter],
ChildHandleBuffer[ChildIndex],
DefaultType,
&ActionRequired
);
if (!EFI_ERROR (Status)) {
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_DRVCFG_DEF_FORCED),
gShellDriver1HiiHandle
);
ShellCmdDriverConfigurationProcessActionRequired (
DriverImageHandleBuffer[OuterLoopCounter],
HandleBuffer[LoopCounter],
ChildHandleBuffer[ChildIndex],
ActionRequired
);
} else {
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_DRVCFG_FORCE_FAILED),
gShellDriver1HiiHandle,
Status
);
ShellStatus = SHELL_DEVICE_ERROR;
}
} else if (ValidateOptions) {
Status = DriverConfiguration->OptionsValid (
DriverConfiguration,
HandleBuffer[LoopCounter],
ChildHandleBuffer[ChildIndex]
);
if (!EFI_ERROR (Status)) {
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_DRVCFG_OPTIONS_VALID),
gShellDriver1HiiHandle
);
} else {
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_DRVCFG_OPTIONS_INV),
gShellDriver1HiiHandle,
Status
);
ShellStatus = SHELL_DEVICE_ERROR;
}
} else if (SetOptions) {
gST->ConOut->ClearScreen (gST->ConOut);
Status = DriverConfiguration->SetOptions (
DriverConfiguration,
HandleBuffer[LoopCounter],
ChildHandleBuffer[ChildIndex],
BestLanguage,
&ActionRequired
);
gST->ConOut->ClearScreen (gST->ConOut);
HandleIndex1 = ConvertHandleToHandleIndex (DriverImageHandleBuffer[OuterLoopCounter]);
HandleIndex2 = ConvertHandleToHandleIndex (HandleBuffer[LoopCounter]);
HandleIndex3 = ConvertHandleToHandleIndex (ChildHandleBuffer[ChildIndex]);
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_DRVCFG_CHILD_LANG),
gShellDriver1HiiHandle,
HandleIndex1,
HandleIndex2,
HandleIndex3,
DriverConfiguration->SupportedLanguages
);
if (!EFI_ERROR (Status)) {
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_DRVCFG_OPTIONS_SET),
gShellDriver1HiiHandle
);
ShellCmdDriverConfigurationProcessActionRequired (
DriverImageHandleBuffer[OuterLoopCounter],
HandleBuffer[LoopCounter],
ChildHandleBuffer[ChildIndex],
ActionRequired
);
} else {
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_DRVCFG_NOT_SET),
gShellDriver1HiiHandle,
Status
);
ShellStatus = SHELL_DEVICE_ERROR;
}
} else {
Print (L"\n");
}
}
FreePool (ChildHandleBuffer);
FreePool (ChildHandleType);
}
FreePool (BestLanguage);
FreePool (HandleBuffer);
FreePool (HandleType);
}
if ((DriverImageHandle != NULL) && (DriverImageHandleCount != 0)) {
FreePool (DriverImageHandleBuffer);
}
Done:
return ShellStatus;
}
/**
Function to print out configuration information on all configurable handles.
@param[in] ChildrenToo TRUE to tewst for children.
@param[in] Language ASCII string for language code.
@param[in] UseHii TRUE to check for Hii and DPC, FALSE for DCP only.
@retval SHELL_SUCCESS The operation was successful.
**/
SHELL_STATUS
PrintConfigInfoOnAll (
IN CONST BOOLEAN ChildrenToo,
IN CONST CHAR8 *Language,
IN CONST BOOLEAN UseHii
)
{
EFI_HANDLE *HandleList;
EFI_HANDLE *CurrentHandle;
BOOLEAN Found;
UINTN Index2;
Found = FALSE;
HandleList = NULL;
CurrentHandle = NULL;
if (UseHii) {
//
// HII method
//
HandleList = GetHandleListByProtocol (&gEfiHiiConfigAccessProtocolGuid);
for (CurrentHandle = HandleList; CurrentHandle != NULL && *CurrentHandle != NULL; CurrentHandle++) {
Found = TRUE;
Index2 = *CurrentHandle == NULL ? 0 : ConvertHandleToHandleIndex (*CurrentHandle);
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_DRVCFG_LINE_HII),
gShellDriver1HiiHandle,
Index2
);
}
SHELL_FREE_NON_NULL (HandleList);
}
if (PreHiiDrvCfg (
Language,
FALSE,
0,
ChildrenToo,
FALSE,
FALSE,
0,
0,
0
) == SHELL_SUCCESS)
{
Found = TRUE;
}
if (!Found) {
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_DRVCFG_NONE_FOUND), gShellDriver1HiiHandle);
return (SHELL_SUCCESS);
}
return (SHELL_SUCCESS);
}
STATIC CONST SHELL_PARAM_ITEM ParamListHii[] = {
{ L"-s", TypeFlag },
{ L"-l", TypeValue },
{ L"-f", TypeValue },
{ L"-o", TypeValue },
{ L"-i", TypeValue },
{ NULL, TypeMax }
};
STATIC CONST SHELL_PARAM_ITEM ParamListPreHii[] = {
{ L"-c", TypeFlag },
{ L"-s", TypeFlag },
{ L"-v", TypeFlag },
{ L"-l", TypeValue },
{ L"-f", TypeValue },
{ NULL, TypeMax }
};
/**
Function for 'drvcfg' 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
ShellCommandRunDrvCfg (
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;
CONST CHAR16 *HandleIndex1;
CONST CHAR16 *HandleIndex2;
CONST CHAR16 *HandleIndex3;
CONST CHAR16 *ForceTypeString;
BOOLEAN Force;
BOOLEAN Set;
BOOLEAN Validate;
BOOLEAN InFromFile;
BOOLEAN OutToFile;
BOOLEAN AllChildren;
BOOLEAN UseHii;
UINT32 ForceType;
UINT64 Intermediate;
EFI_HANDLE Handle1;
EFI_HANDLE Handle2;
EFI_HANDLE Handle3;
CONST CHAR16 *FileName;
ShellStatus = SHELL_SUCCESS;
Status = EFI_SUCCESS;
Language = NULL;
UseHii = TRUE;
ProblemParam = 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 (ParamListHii, &Package, &ProblemParam, TRUE);
if (EFI_ERROR (Status) || (ShellCommandLineGetCount (Package) > 2)) {
UseHii = FALSE;
if (Package != NULL) {
ShellCommandLineFreeVarList (Package);
}
SHELL_FREE_NON_NULL (ProblemParam);
Status = ShellCommandLineParse (ParamListPreHii, &Package, &ProblemParam, TRUE);
if (EFI_ERROR (Status)) {
if ((Status == EFI_VOLUME_CORRUPTED) && (ProblemParam != NULL)) {
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDriver1HiiHandle, L"drvcfg", ProblemParam);
FreePool (ProblemParam);
ShellStatus = SHELL_INVALID_PARAMETER;
goto Done;
} else {
ASSERT (FALSE);
}
}
}
if (ShellStatus == SHELL_SUCCESS) {
if (ShellCommandLineGetCount (Package) > 4) {
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDriver1HiiHandle, L"drvcfg");
ShellStatus = SHELL_INVALID_PARAMETER;
goto Done;
}
Lang = ShellCommandLineGetValue (Package, L"-l");
if (Lang != NULL) {
Language = AllocateZeroPool (StrSize (Lang));
AsciiSPrint (Language, StrSize (Lang), "%S", Lang);
} else if (ShellCommandLineGetFlag (Package, L"-l")) {
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_VALUE), gShellDriver1HiiHandle, L"drvcfg", L"-l");
ShellStatus = SHELL_INVALID_PARAMETER;
goto Done;
}
Set = ShellCommandLineGetFlag (Package, L"-s");
Validate = ShellCommandLineGetFlag (Package, L"-v");
InFromFile = ShellCommandLineGetFlag (Package, L"-i");
OutToFile = ShellCommandLineGetFlag (Package, L"-o");
AllChildren = ShellCommandLineGetFlag (Package, L"-c");
Force = ShellCommandLineGetFlag (Package, L"-f");
ForceTypeString = ShellCommandLineGetValue (Package, L"-f");
if (OutToFile) {
FileName = ShellCommandLineGetValue (Package, L"-o");
} else if (InFromFile) {
FileName = ShellCommandLineGetValue (Package, L"-i");
} else {
FileName = NULL;
}
if (InFromFile && EFI_ERROR (ShellFileExists (FileName))) {
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_FIND_FAIL), gShellDriver1HiiHandle, L"drvcfg", FileName);
ShellStatus = SHELL_INVALID_PARAMETER;
goto Done;
}
if (OutToFile && !EFI_ERROR (ShellFileExists (FileName))) {
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_EXIST), gShellDriver1HiiHandle, L"drvcfg", FileName);
ShellStatus = SHELL_INVALID_PARAMETER;
goto Done;
}
if (Force && (ForceTypeString == NULL)) {
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_VALUE), gShellDriver1HiiHandle, L"drvcfg", L"-f");
ShellStatus = SHELL_INVALID_PARAMETER;
goto Done;
}
if (Force) {
Status = ShellConvertStringToUint64 (ForceTypeString, &Intermediate, FALSE, FALSE);
if (EFI_ERROR (Status)) {
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM_VAL), gShellDriver1HiiHandle, L"drvcfg", ForceTypeString, L"-f");
ShellStatus = SHELL_INVALID_PARAMETER;
goto Done;
}
ForceType = (UINT32)Intermediate;
} else {
ForceType = 0;
}
HandleIndex1 = ShellCommandLineGetRawValue (Package, 1);
Handle1 = NULL;
if ((HandleIndex1 != NULL) && !EFI_ERROR (ShellConvertStringToUint64 (HandleIndex1, &Intermediate, TRUE, FALSE))) {
Handle1 = ConvertHandleIndexToHandle ((UINTN)Intermediate);
if ((Handle1 == NULL) || ((UINT64)(UINTN)Intermediate != Intermediate)) {
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_INV_HANDLE), gShellDriver1HiiHandle, L"drvcfg", HandleIndex1);
ShellStatus = SHELL_INVALID_PARAMETER;
goto Done;
}
}
HandleIndex2 = ShellCommandLineGetRawValue (Package, 2);
Handle2 = NULL;
if ((HandleIndex2 != NULL) && !EFI_ERROR (ShellConvertStringToUint64 (HandleIndex2, &Intermediate, TRUE, FALSE))) {
Handle2 = ConvertHandleIndexToHandle ((UINTN)Intermediate);
if ((Handle2 == NULL) || ((UINT64)(UINTN)Intermediate != Intermediate)) {
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_INV_HANDLE), gShellDriver1HiiHandle, L"drvcfg", HandleIndex2);
ShellStatus = SHELL_INVALID_PARAMETER;
goto Done;
}
}
HandleIndex3 = ShellCommandLineGetRawValue (Package, 3);
Handle3 = NULL;
if ((HandleIndex3 != NULL) && !EFI_ERROR (ShellConvertStringToUint64 (HandleIndex3, &Intermediate, TRUE, FALSE))) {
Handle3 = ConvertHandleIndexToHandle ((UINTN)Intermediate);
if ((Handle3 == NULL) || ((UINT64)(UINTN)Intermediate != Intermediate)) {
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_INV_HANDLE), gShellDriver1HiiHandle, L"drvcfg", HandleIndex3);
ShellStatus = SHELL_INVALID_PARAMETER;
goto Done;
}
}
if ((InFromFile || OutToFile) && (FileName == NULL)) {
if (FileName == NULL) {
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_VALUE), gShellDriver1HiiHandle, L"drvcfg", InFromFile ? L"-i" : L"-o");
} else {
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_HANDLE_REQ), gShellDriver1HiiHandle, L"drvcfg");
}
ShellStatus = SHELL_INVALID_PARAMETER;
goto Done;
}
if (!UseHii && (InFromFile || OutToFile)) {
if (InFromFile) {
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellDriver1HiiHandle, L"drvcfg", L"-i");
ShellStatus = SHELL_INVALID_PARAMETER;
goto Done;
}
if (OutToFile) {
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellDriver1HiiHandle, L"drvcfg", L"-o");
ShellStatus = SHELL_INVALID_PARAMETER;
goto Done;
}
}
if (Validate && Force) {
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_CONFLICT), gShellDriver1HiiHandle, L"drvcfg", L"-v", L"-f");
ShellStatus = SHELL_INVALID_PARAMETER;
goto Done;
}
if (Validate && Set) {
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_CONFLICT), gShellDriver1HiiHandle, L"drvcfg", L"-v", L"-s");
ShellStatus = SHELL_INVALID_PARAMETER;
goto Done;
}
if (Set && Force) {
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_CONFLICT), gShellDriver1HiiHandle, L"drvcfg", L"-s", L"-f");
ShellStatus = SHELL_INVALID_PARAMETER;
goto Done;
}
if (OutToFile && InFromFile) {
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_CONFLICT), gShellDriver1HiiHandle, L"drvcfg", L"-i", L"-o");
ShellStatus = SHELL_INVALID_PARAMETER;
goto Done;
}
//
// We do HII first.
//
if (UseHii) {
if ((Handle1 != NULL) && EFI_ERROR (gBS->OpenProtocol (Handle1, &gEfiHiiConfigAccessProtocolGuid, NULL, gImageHandle, NULL, EFI_OPEN_PROTOCOL_TEST_PROTOCOL))) {
//
// no HII on this handle.
//
ShellStatus = SHELL_UNSUPPORTED;
} else if (Validate) {
} else if (Force) {
} else if (Set) {
} else if (InFromFile) {
ShellStatus = ConfigFromFile (Handle1, FileName);
if ((Handle1 != NULL) && (ShellStatus == SHELL_SUCCESS)) {
goto Done;
}
} else if (OutToFile) {
ShellStatus = ConfigToFile (Handle1, FileName);
if ((Handle1 != NULL) && (ShellStatus == SHELL_SUCCESS)) {
goto Done;
}
} else if (HandleIndex1 == NULL) {
//
// display all that are configurable
//
ShellStatus = PrintConfigInfoOnAll (AllChildren, Language, UseHii);
goto Done;
} else {
if (!EFI_ERROR (gBS->OpenProtocol (Handle1, &gEfiHiiConfigAccessProtocolGuid, NULL, gImageHandle, NULL, EFI_OPEN_PROTOCOL_TEST_PROTOCOL))) {
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_DRVCFG_LINE_HII),
gShellDriver1HiiHandle,
ConvertHandleToHandleIndex (Handle1)
);
goto Done;
}
}
}
//
// We allways need to do this one since it does both by default.
//
if (!InFromFile && !OutToFile) {
ShellStatus = PreHiiDrvCfg (
Language,
Force,
ForceType,
AllChildren,
Validate,
Set,
Handle1,
Handle2,
Handle3
);
}
if (ShellStatus == SHELL_UNSUPPORTED) {
ShellPrintHiiEx (
-1,
-1,
NULL,
STRING_TOKEN (STR_DRVCFG_NOT_SUPPORT),
gShellDriver1HiiHandle,
ConvertHandleToHandleIndex (Handle1)
);
}
}
Done:
ShellCommandLineFreeVarList (Package);
SHELL_FREE_NON_NULL (Language);
return (ShellStatus);
}
Reference in New Issue View Git Blame Copy Permalink