4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync/** @file
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Main file for DrvDiag shell Driver1 function.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Copyright (c) 2010 - 2011, Intel Corporation. All rights reserved.<BR>
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync This program and the accompanying materials
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync are licensed and made available under the terms and conditions of the BSD License
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync which accompanies this distribution. The full text of the license may be found at
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync http://opensource.org/licenses/bsd-license.php
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync**/
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync#include "UefiShellDriver1CommandsLib.h"
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncSTATIC CONST EFI_GUID *DiagGuidList[] = {&gEfiDriverDiagnosticsProtocolGuid, &gEfiDriverDiagnostics2ProtocolGuid, NULL};
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync//
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync// We need 1 more item on the list...
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync//
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsynctypedef enum {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync TestModeStandard = EfiDriverDiagnosticTypeStandard,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync TestModeExtended = EfiDriverDiagnosticTypeExtended,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync TestModeManufacturing = EfiDriverDiagnosticTypeManufacturing,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync TestModeList,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync TestModeMax
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync} DRV_DIAG_TEST_MODE;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync/**
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Do the diagnostics call for some set of handles.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @param[in] Mode The type of diagnostic test to run.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @param[in] Lang The language code to use.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @param[in] AllChilds Should the test be on all children.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @param[in] DriverHandle The driver handle to test with.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @param[in] ControllerHandle The specific controller handle to test.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @param[in] ChildHandle The specific child handle to test.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @retval EFI_SUCCESS The operation was successful.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @retval EFI_INVALID_PARAMETER A parameter had an invalid value.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @retval EFI_NOT_FOUND No diagnostic handle could be found.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync**/
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncEFI_STATUS
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncEFIAPI
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncDoDiagnostics (
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync IN CONST DRV_DIAG_TEST_MODE Mode,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync IN CONST CHAR8 *Lang,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync IN CONST BOOLEAN AllChilds,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync IN CONST EFI_HANDLE DriverHandle,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync IN CONST EFI_HANDLE ControllerHandle,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync IN CONST EFI_HANDLE ChildHandle
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync )
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync{
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_DRIVER_DIAGNOSTICS_PROTOCOL *DriverDiagnostics;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_DRIVER_DIAGNOSTICS2_PROTOCOL *DriverDiagnostics2;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_HANDLE *DriverHandleList;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_HANDLE *ControllerHandleList;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_HANDLE *ChildHandleList;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_HANDLE *Walker;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync UINTN DriverHandleListCount;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync UINTN ControllerHandleListCount;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync UINTN ChildHandleListCount;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync UINTN DriverHandleListLoop;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync UINTN ControllerHandleListLoop;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync UINTN ChildHandleListLoop;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_STATUS Status;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_STATUS Status2;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_GUID *ErrorType;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync UINTN OutBufferSize;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync CHAR16 *OutBuffer;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync UINTN HandleIndex1;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync UINTN HandleIndex2;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync CHAR8 *Language;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync CHAR8 *TempChar;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync BOOLEAN Found;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if ((ChildHandle != NULL && AllChilds) || (Mode >= TestModeMax)){
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return (EFI_INVALID_PARAMETER);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DriverDiagnostics = NULL;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DriverDiagnostics2 = NULL;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Status = EFI_SUCCESS;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Status2 = EFI_SUCCESS;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DriverHandleList = NULL;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ControllerHandleList = NULL;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ChildHandleList = NULL;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Language = NULL;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync OutBuffer = NULL;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ErrorType = NULL;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DriverHandleListCount = 0;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ControllerHandleListCount = 0;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ChildHandleListCount = 0;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (DriverHandle != NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DriverHandleList = AllocateZeroPool(2*sizeof(EFI_HANDLE));
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ASSERT(DriverHandleList!=NULL);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DriverHandleList[0] = DriverHandle;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DriverHandleListCount = 1;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DriverHandleList = GetHandleListByProtocolList(DiagGuidList);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (DriverHandleList == NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROTOCOL_NF), gShellDriver1HiiHandle, L"gEfiDriverDiagnosticsProtocolGuid", &gEfiDriverDiagnosticsProtocolGuid);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROTOCOL_NF), gShellDriver1HiiHandle, L"gEfiDriverDiagnostics2ProtocolGuid", &gEfiDriverDiagnostics2ProtocolGuid);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return (EFI_NOT_FOUND);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync for (Walker = DriverHandleList ; Walker != NULL && *Walker != NULL ; DriverHandleListCount++, Walker++);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (ControllerHandle != NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ControllerHandleList = AllocateZeroPool(2*sizeof(EFI_HANDLE));
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ASSERT(ControllerHandleList!=NULL);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ControllerHandleList[0] = ControllerHandle;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ControllerHandleListCount = 1;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ControllerHandleList = NULL;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (ChildHandle != NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ChildHandleList = AllocateZeroPool(2*sizeof(EFI_HANDLE));
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ASSERT(ChildHandleList!=NULL);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ChildHandleList[0] = ChildHandle;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ChildHandleListCount = 1;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else if (AllChilds) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ChildHandleList = NULL;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // This gets handled in the loop below.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ChildHandleList = NULL;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (Mode == TestModeList) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_DRVDIAG_HEADER), gShellDriver1HiiHandle);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync for (DriverHandleListLoop = 0
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ; DriverHandleListLoop < DriverHandleListCount
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ; DriverHandleListLoop++
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ){
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (Mode == TestModeList) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_DRVDIAG_DRIVER_HEADER), gShellDriver1HiiHandle, ConvertHandleToHandleIndex(DriverHandleList[DriverHandleListLoop]));
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (ControllerHandle == NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync PARSE_HANDLE_DATABASE_DEVICES(DriverHandleList[DriverHandleListLoop], &ControllerHandleListCount, &ControllerHandleList);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (ControllerHandleListCount == 0) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (Mode == TestModeList) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_DRVDIAG_DRIVER_NO_HANDLES), gShellDriver1HiiHandle);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (Mode == TestModeList) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ShellPrintEx(-1, -1, L"\r\n");
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync for (ControllerHandleListLoop = 0
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ; ControllerHandleListLoop < ControllerHandleListCount
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ; ControllerHandleListLoop++
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ){
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (AllChilds) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ASSERT(ChildHandleList == NULL);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync PARSE_HANDLE_DATABASE_MANAGED_CHILDREN(
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DriverHandleList[DriverHandleListLoop],
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ControllerHandleList[ControllerHandleListLoop],
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync &ChildHandleListCount,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync &ChildHandleList);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync for (ChildHandleListLoop = 0
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ; (ChildHandleListLoop < ChildHandleListCount || ChildHandleList == NULL)
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ; ChildHandleListLoop++
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ){
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Found = FALSE;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (Mode != TestModeList) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (Lang == NULL || Lang[2] == '-') {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Get the protocol pointer and call the function
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Status = gBS->OpenProtocol(
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DriverHandleList[DriverHandleListLoop],
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync &gEfiDriverDiagnostics2ProtocolGuid,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync (VOID**)&DriverDiagnostics2,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync gImageHandle,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync NULL,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_OPEN_PROTOCOL_GET_PROTOCOL);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (!EFI_ERROR(Status)) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (Lang == NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Language = AllocateZeroPool(AsciiStrSize(DriverDiagnostics2->SupportedLanguages));
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (Language == NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return (EFI_OUT_OF_RESOURCES);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync AsciiStrCpy(Language, DriverDiagnostics2->SupportedLanguages);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync TempChar = AsciiStrStr(Language, ";");
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (TempChar != NULL){
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync *TempChar = CHAR_NULL;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Language = AllocateZeroPool(AsciiStrSize(Lang));
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (Language == NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return (EFI_OUT_OF_RESOURCES);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync AsciiStrCpy(Language, Lang);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Found = TRUE;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Status = DriverDiagnostics2->RunDiagnostics(
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DriverDiagnostics2,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ControllerHandleList[ControllerHandleListLoop],
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ChildHandleList == NULL?NULL:ChildHandleList[ChildHandleListLoop],
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync (EFI_DRIVER_DIAGNOSTIC_TYPE)Mode,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Language,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync &ErrorType,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync &OutBufferSize,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync &OutBuffer);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync FreePool(Language);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (!Found && (Lang == NULL||(Lang!=NULL&&(Lang[2]!='-')))){
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Status = gBS->OpenProtocol(
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DriverHandleList[DriverHandleListLoop],
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync &gEfiDriverDiagnosticsProtocolGuid,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync (VOID**)&DriverDiagnostics,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync gImageHandle,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync NULL,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_OPEN_PROTOCOL_GET_PROTOCOL);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (!EFI_ERROR(Status)) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (Lang == NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Language = AllocateZeroPool(AsciiStrSize(DriverDiagnostics2->SupportedLanguages));
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (Language == NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return (EFI_OUT_OF_RESOURCES);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync AsciiStrCpy(Language, DriverDiagnostics2->SupportedLanguages);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync TempChar = AsciiStrStr(Language, ";");
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (TempChar != NULL){
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync *TempChar = CHAR_NULL;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Language = AllocateZeroPool(AsciiStrSize(Lang));
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (Language == NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return (EFI_OUT_OF_RESOURCES);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync AsciiStrCpy(Language, Lang);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Status = DriverDiagnostics->RunDiagnostics(
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DriverDiagnostics,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ControllerHandleList[ControllerHandleListLoop],
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ChildHandleList == NULL?NULL:ChildHandleList[ChildHandleListLoop],
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync (EFI_DRIVER_DIAGNOSTIC_TYPE)Mode,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Language,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync &ErrorType,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync &OutBufferSize,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync &OutBuffer);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync FreePool(Language);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (EFI_ERROR(Status)) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Status2 = Status;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HandleIndex1 = ConvertHandleToHandleIndex(DriverHandleList[DriverHandleListLoop]);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HandleIndex2 = ConvertHandleToHandleIndex(ControllerHandleList[ControllerHandleListLoop]);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ShellPrintHiiEx(
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync -1,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync -1,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync NULL,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync STRING_TOKEN (STR_3P_RESULT),
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync gShellDriver1HiiHandle,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync L"DrvDiag",
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HandleIndex1,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HandleIndex2,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ChildHandleList == NULL?0:ConvertHandleToHandleIndex(ChildHandleList[ChildHandleListLoop]),
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Status);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (OutBuffer!=NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync FreePool(OutBuffer);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync OutBuffer = NULL;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (ErrorType!=NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync FreePool(ErrorType);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ErrorType = NULL;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HandleIndex1 = ConvertHandleToHandleIndex(DriverHandleList[DriverHandleListLoop]);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HandleIndex2 = ConvertHandleToHandleIndex(ControllerHandleList[ControllerHandleListLoop]);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Print out the information that this set can be tested
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ShellPrintHiiEx(
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync -1,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync -1,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync NULL,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync STRING_TOKEN (STR_DRV_DIAG_ITEM_LINE),
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync gShellDriver1HiiHandle,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HandleIndex1,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync HandleIndex2,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ChildHandleList == NULL?0:ConvertHandleToHandleIndex(ChildHandleList[ChildHandleListLoop])
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync );
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // If we are doing a single pass with NULL child jump out after a single loop
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (ChildHandleList == NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync break;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (AllChilds) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync SHELL_FREE_NON_NULL(ChildHandleList);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ChildHandleList = NULL;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ChildHandleListCount = 0;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (ControllerHandle == NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync SHELL_FREE_NON_NULL(ControllerHandleList);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ControllerHandleList = NULL;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ControllerHandleListCount = 0;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (DriverHandleList != NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync FreePool(DriverHandleList);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (ControllerHandleList != NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync FreePool(ControllerHandleList);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (ChildHandleList != NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync FreePool(ChildHandleList);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return (Status2);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync}
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncSTATIC CONST SHELL_PARAM_ITEM ParamList[] = {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync {L"-c", TypeFlag},
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync {L"-s", TypeFlag},
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync {L"-e", TypeFlag},
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync {L"-m", TypeFlag},
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync {L"-l", TypeValue},
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync {NULL, TypeMax}
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync };
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync/**
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Function for 'drvdiag' command.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @param[in] ImageHandle Handle to the Image (NULL if Internal).
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync @param[in] SystemTable Pointer to the System Table (NULL if Internal).
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync**/
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncSHELL_STATUS
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncEFIAPI
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncShellCommandRunDrvDiag (
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync IN EFI_HANDLE ImageHandle,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync IN EFI_SYSTEM_TABLE *SystemTable
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync )
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync{
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_STATUS Status;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync LIST_ENTRY *Package;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync CHAR16 *ProblemParam;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync SHELL_STATUS ShellStatus;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DRV_DIAG_TEST_MODE Mode;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync CHAR8 *Language;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync CONST CHAR16 *DriverHandleStr;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync CONST CHAR16 *ControllerHandleStr;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync CONST CHAR16 *ChildHandleStr;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync CONST CHAR16 *Lang;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_HANDLE Handle1;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_HANDLE Handle2;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync EFI_HANDLE Handle3;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync UINT64 Intermediate;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ShellStatus = SHELL_SUCCESS;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Mode = TestModeMax;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Language = NULL;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // initialize the shell lib (we must be in non-auto-init...)
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Status = ShellInitialize();
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ASSERT_EFI_ERROR(Status);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Status = CommandInit();
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ASSERT_EFI_ERROR(Status);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // parse the command line
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Status = ShellCommandLineParse (ParamList, &Package, &ProblemParam, TRUE);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (EFI_ERROR(Status)) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDriver1HiiHandle, ProblemParam);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync FreePool(ProblemParam);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ShellStatus = SHELL_INVALID_PARAMETER;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ASSERT(FALSE);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // if more than 3 'value' parameters (plus the name one) or we have any 2 mode flags
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if ((ShellCommandLineGetCount(Package) > 4)
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ||(ShellCommandLineGetFlag(Package, L"-s") && ShellCommandLineGetFlag(Package, L"-e"))
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ||(ShellCommandLineGetFlag(Package, L"-s") && ShellCommandLineGetFlag(Package, L"-m"))
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ||(ShellCommandLineGetFlag(Package, L"-e") && ShellCommandLineGetFlag(Package, L"-m"))
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ){
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // error for too many parameters
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDriver1HiiHandle);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ShellStatus = SHELL_INVALID_PARAMETER;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else if ((ShellCommandLineGetFlag(Package, L"-s"))
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync || (ShellCommandLineGetFlag(Package, L"-e"))
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync || (ShellCommandLineGetFlag(Package, L"-m"))
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ){
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Run the apropriate test
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (ShellCommandLineGetFlag(Package, L"-s")) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Mode = TestModeStandard;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else if (ShellCommandLineGetFlag(Package, L"-e")) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Mode = TestModeExtended;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else if (ShellCommandLineGetFlag(Package, L"-m")) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Mode = TestModeManufacturing;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ASSERT(FALSE);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync // Do a listing of what's available to test
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync //
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Mode = TestModeList;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Lang = ShellCommandLineGetValue(Package, L"-l");
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (ShellCommandLineGetFlag(Package, L"-l") && Lang == NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ASSERT(Language == NULL);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_VALUE), gShellDriver1HiiHandle, L"-l");
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ShellCommandLineFreeVarList (Package);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return (SHELL_INVALID_PARAMETER);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else if (Lang != NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Language = AllocateZeroPool(StrSize(Lang));
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync AsciiSPrint(Language, StrSize(Lang), "%S", Lang);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync DriverHandleStr = ShellCommandLineGetRawValue(Package, 1);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ControllerHandleStr = ShellCommandLineGetRawValue(Package, 2);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ChildHandleStr = ShellCommandLineGetRawValue(Package, 3);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (DriverHandleStr == NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Handle1 = NULL;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ShellConvertStringToUint64(DriverHandleStr, &Intermediate, TRUE, FALSE);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Handle1 = ConvertHandleIndexToHandle((UINTN)Intermediate);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (ControllerHandleStr == NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Handle2 = NULL;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ShellConvertStringToUint64(ControllerHandleStr, &Intermediate, TRUE, FALSE);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Handle2 = ConvertHandleIndexToHandle((UINTN)Intermediate);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (ChildHandleStr == NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Handle3 = NULL;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ShellConvertStringToUint64(ChildHandleStr, &Intermediate, TRUE, FALSE);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Handle3 = ConvertHandleIndexToHandle((UINTN)Intermediate);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Status = DoDiagnostics (
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Mode,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Language,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ShellCommandLineGetFlag(Package, L"-c"),
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Handle1,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Handle2,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Handle3
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync );
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync SHELL_FREE_NON_NULL(Language);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ShellCommandLineFreeVarList (Package);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (ShellStatus == SHELL_SUCCESS) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (Status == EFI_SECURITY_VIOLATION) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ShellStatus = SHELL_SECURITY_VIOLATION;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else if (Status == EFI_INVALID_PARAMETER) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ShellStatus = SHELL_INVALID_PARAMETER;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else if (Status == EFI_NOT_FOUND) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ShellStatus = SHELL_NOT_FOUND;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else if (EFI_ERROR(Status)) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ShellStatus = SHELL_NOT_FOUND;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return (ShellStatus);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync}