bcfg - updated for bugs. compress - rename for coding standards. add comments. dblk - add comments, input verification, and a header line dmem - add comments, add input verification, add system table info dmpstore - add comments eficompress - add comments and add input verification efidecompress - add comments and add input verification loadpcirom - add comments and more output messages memmap - add more output to exceed the spec. mm - move functions, add comments, add input verification. mode - add comment pci - add input verification. SerMode - add comments and add input verification setsize - add comments and add input verification setvar - add comments and add input verification smbiosview - add input verification. clarify error messages. git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@11438 6f19259b-4bc3-4df7-8a09-765794883524
		
			
				
	
	
		
			261 lines
		
	
	
		
			8.3 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			261 lines
		
	
	
		
			8.3 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/** @file
 | 
						|
  Main file for DmpStore shell Debug1 function.
 | 
						|
 | 
						|
  Copyright (c) 2005 - 2011, 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"
 | 
						|
 | 
						|
STATIC CHAR16   *AttrType[] = {
 | 
						|
  L"invalid",   // 000
 | 
						|
  L"invalid",   // 001
 | 
						|
  L"BS",        // 010
 | 
						|
  L"NV+BS",     // 011
 | 
						|
  L"RT+BS",     // 100
 | 
						|
  L"NV+RT+BS",  // 101
 | 
						|
  L"RT+BS",     // 110
 | 
						|
  L"NV+RT+BS",  // 111
 | 
						|
};
 | 
						|
 | 
						|
/**
 | 
						|
  Function to display or delete variables.
 | 
						|
 | 
						|
  @param[in] VariableName   The variable name of the EFI variable (or NULL).
 | 
						|
  @param[in] Guid           The GUID of the variable set (or NULL).
 | 
						|
  @param[in] Delete         TRUE to delete, FALSE otherwise.
 | 
						|
 | 
						|
  @retval SHELL_SUCCESS           The operation was successful.
 | 
						|
  @retval SHELL_OUT_OF_RESOURCES  A memorty allocation failed.
 | 
						|
  @retval SHELL_ABORTED           The abort message was received.
 | 
						|
  @retval SHELL_DEVICE_ERROR      UEFI Variable Services returned an error.
 | 
						|
  @retval SHELL_NOT_FOUND         the Name/Guid pair could not be found.
 | 
						|
**/
 | 
						|
SHELL_STATUS
 | 
						|
EFIAPI
 | 
						|
ProcessVariables (
 | 
						|
  IN CONST CHAR16   *VariableName OPTIONAL,
 | 
						|
  IN CONST EFI_GUID *Guid OPTIONAL,
 | 
						|
  IN BOOLEAN        Delete
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS                Status;
 | 
						|
  UINT64                    MaxStorSize;
 | 
						|
  UINT64                    RemStorSize;
 | 
						|
  UINT64                    MaxVarSize;
 | 
						|
  CHAR16                    *FoundVarName;
 | 
						|
  UINTN                     Size;
 | 
						|
  EFI_GUID                  FoundVarGuid;
 | 
						|
  UINT8                     *DataBuffer;
 | 
						|
  UINTN                     DataSize;
 | 
						|
  UINT32                    Atts;
 | 
						|
  SHELL_STATUS              ShellStatus;
 | 
						|
  BOOLEAN                   Found;
 | 
						|
 | 
						|
  Status = gRT->QueryVariableInfo(EFI_VARIABLE_BOOTSERVICE_ACCESS|EFI_VARIABLE_RUNTIME_ACCESS|EFI_VARIABLE_NON_VOLATILE, &MaxStorSize, &RemStorSize, &MaxVarSize);
 | 
						|
  if (EFI_ERROR(Status)) {
 | 
						|
    return (SHELL_DEVICE_ERROR);
 | 
						|
  }
 | 
						|
 | 
						|
  Found         = FALSE;
 | 
						|
  ShellStatus   = SHELL_SUCCESS;
 | 
						|
  Size          = PcdGet16(PcdShellFileOperationSize);
 | 
						|
  FoundVarName  = AllocateZeroPool(Size);
 | 
						|
 | 
						|
  if (FoundVarName == NULL) {
 | 
						|
    return (SHELL_OUT_OF_RESOURCES);
 | 
						|
  }
 | 
						|
  FoundVarName[0] = CHAR_NULL;
 | 
						|
 | 
						|
 | 
						|
  DataSize = (UINTN)MaxVarSize;
 | 
						|
  DataBuffer = AllocateZeroPool(DataSize);
 | 
						|
  if (DataBuffer == NULL) {
 | 
						|
    FreePool(FoundVarName);
 | 
						|
    return (SHELL_OUT_OF_RESOURCES);
 | 
						|
  }
 | 
						|
 | 
						|
  for (;;){
 | 
						|
    if (ShellGetExecutionBreakFlag()) {
 | 
						|
      ShellStatus = SHELL_ABORTED;
 | 
						|
      break;
 | 
						|
    }
 | 
						|
    Size      = (UINTN)PcdGet16(PcdShellFileOperationSize);
 | 
						|
    DataSize  = (UINTN)MaxVarSize;
 | 
						|
 | 
						|
    Status = gRT->GetNextVariableName(&Size, FoundVarName, &FoundVarGuid);
 | 
						|
    if (Status == EFI_NOT_FOUND) {
 | 
						|
      break;
 | 
						|
    }
 | 
						|
    ASSERT_EFI_ERROR(Status);
 | 
						|
 | 
						|
    Status = gRT->GetVariable(FoundVarName, &FoundVarGuid, &Atts, &DataSize, DataBuffer);
 | 
						|
    ASSERT_EFI_ERROR(Status);
 | 
						|
 | 
						|
    //
 | 
						|
    // Check if it matches
 | 
						|
    //
 | 
						|
    if (VariableName != NULL) {
 | 
						|
      if (!gUnicodeCollation->MetaiMatch(gUnicodeCollation, FoundVarName, (CHAR16*)VariableName)) {
 | 
						|
        continue;
 | 
						|
      }
 | 
						|
    }
 | 
						|
    if (Guid != NULL) {
 | 
						|
      if (!CompareGuid(&FoundVarGuid, Guid)) {
 | 
						|
        continue;
 | 
						|
      }
 | 
						|
    }
 | 
						|
 | 
						|
    //
 | 
						|
    // do the print or delete
 | 
						|
    //
 | 
						|
    Found = TRUE;
 | 
						|
    if (!Delete) {
 | 
						|
      ShellPrintHiiEx(
 | 
						|
        -1,
 | 
						|
        -1,
 | 
						|
        NULL,
 | 
						|
        STRING_TOKEN(STR_DMPSTORE_HEADER_LINE),
 | 
						|
        gShellDebug1HiiHandle,
 | 
						|
        AttrType[Atts & 7],
 | 
						|
        &FoundVarGuid,
 | 
						|
        FoundVarName,
 | 
						|
        DataSize);
 | 
						|
      DumpHex(2, 0, DataSize, DataBuffer);
 | 
						|
    } else {
 | 
						|
      ShellPrintHiiEx(
 | 
						|
        -1,
 | 
						|
        -1,
 | 
						|
        NULL,
 | 
						|
        STRING_TOKEN(STR_DMPSTORE_DELETE_LINE),
 | 
						|
        gShellDebug1HiiHandle,
 | 
						|
        &FoundVarGuid,
 | 
						|
        FoundVarName);
 | 
						|
      ShellPrintHiiEx(
 | 
						|
        -1,
 | 
						|
        -1,
 | 
						|
        NULL,
 | 
						|
        STRING_TOKEN(STR_DMPSTORE_DELETE_DONE),
 | 
						|
        gShellDebug1HiiHandle,
 | 
						|
        gRT->SetVariable(FoundVarName, &FoundVarGuid, Atts, 0, NULL));
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  if (FoundVarName != NULL) {
 | 
						|
    FreePool(FoundVarName);
 | 
						|
  }
 | 
						|
  if (DataBuffer != NULL) {
 | 
						|
    FreePool(DataBuffer);
 | 
						|
  }
 | 
						|
  if (!Found) {
 | 
						|
    if (VariableName != NULL && Guid == NULL) {
 | 
						|
      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_DMPSTORE_NO_VAR_FOUND_N), gShellDebug1HiiHandle, VariableName);
 | 
						|
    } else if (VariableName != NULL && Guid != NULL) {
 | 
						|
      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_DMPSTORE_NO_VAR_FOUND_GN), gShellDebug1HiiHandle, Guid, VariableName);
 | 
						|
    } else if (VariableName == NULL && Guid == NULL) {
 | 
						|
      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_DMPSTORE_NO_VAR_FOUND), gShellDebug1HiiHandle);
 | 
						|
    } else if (VariableName == NULL && Guid != NULL) {
 | 
						|
      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_DMPSTORE_NO_VAR_FOUND_G), gShellDebug1HiiHandle, Guid);
 | 
						|
    } 
 | 
						|
    return (SHELL_NOT_FOUND);
 | 
						|
  }
 | 
						|
  return (SHELL_SUCCESS);
 | 
						|
}
 | 
						|
 | 
						|
STATIC CONST SHELL_PARAM_ITEM ParamList[] = {
 | 
						|
  {L"-d", TypeFlag},
 | 
						|
  {L"-l", TypeFlag},
 | 
						|
  {L"-s", TypeFlag},
 | 
						|
  {L"-all", TypeFlag},
 | 
						|
  {L"-guid", TypeValue},
 | 
						|
  {NULL, TypeMax}
 | 
						|
  };
 | 
						|
 | 
						|
/**
 | 
						|
  Function for 'dmpstore' 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
 | 
						|
ShellCommandRunDmpStore (
 | 
						|
  IN EFI_HANDLE        ImageHandle,
 | 
						|
  IN EFI_SYSTEM_TABLE  *SystemTable
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS    Status;
 | 
						|
  LIST_ENTRY    *Package;
 | 
						|
  CHAR16        *ProblemParam;
 | 
						|
  SHELL_STATUS  ShellStatus;
 | 
						|
  CONST CHAR16  *Temp;
 | 
						|
  EFI_GUID      *Guid;
 | 
						|
  EFI_GUID      GuidData;
 | 
						|
  CONST CHAR16  *VariableName;
 | 
						|
 | 
						|
  ShellStatus   = SHELL_SUCCESS;
 | 
						|
  Package       = NULL;
 | 
						|
 | 
						|
  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), gShellDebug1HiiHandle, 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), gShellDebug1HiiHandle);
 | 
						|
      ShellStatus = SHELL_INVALID_PARAMETER;
 | 
						|
    } else if (ShellCommandLineGetFlag(Package, L"-all") && ShellCommandLineGetFlag(Package, L"-guid")) {
 | 
						|
      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_CONFLICT), gShellDebug1HiiHandle, L"-all", L"-guid");
 | 
						|
      ShellStatus = SHELL_INVALID_PARAMETER;
 | 
						|
    } else if ((ShellCommandLineGetFlag(Package, L"-s") || ShellCommandLineGetFlag(Package, L"-l")) && ShellCommandLineGetFlag(Package, L"-d")) {
 | 
						|
      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_CONFLICT), gShellDebug1HiiHandle, L"-l or -s", L"-d");
 | 
						|
      ShellStatus = SHELL_INVALID_PARAMETER;
 | 
						|
    } else {
 | 
						|
      if (!ShellCommandLineGetFlag(Package, L"-all")) {
 | 
						|
        Temp = ShellCommandLineGetValue(Package, L"-guid");
 | 
						|
        if (Temp != NULL) {
 | 
						|
          Status = ConvertStringToGuid(Temp, &GuidData);
 | 
						|
          if (EFI_ERROR(Status)) {
 | 
						|
            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, Temp);
 | 
						|
            ShellStatus = SHELL_INVALID_PARAMETER;
 | 
						|
          }
 | 
						|
          Guid = &GuidData;
 | 
						|
        } else  {
 | 
						|
          Guid = &gEfiGlobalVariableGuid;
 | 
						|
        }
 | 
						|
        VariableName = ShellCommandLineGetRawValue(Package, 1);
 | 
						|
      } else {
 | 
						|
        VariableName  = NULL;
 | 
						|
        Guid          = NULL;
 | 
						|
      }
 | 
						|
      if (ShellStatus == SHELL_SUCCESS) {
 | 
						|
        if (ShellCommandLineGetFlag(Package, L"-s") || ShellCommandLineGetFlag(Package, L"-l")) {
 | 
						|
          ///@todo fix this after Jordan makes lib...
 | 
						|
          ShellPrintEx(-1, -1, L"Not implemeneted yet.\r\n");
 | 
						|
          ShellStatus = SHELL_UNSUPPORTED;
 | 
						|
        } else {
 | 
						|
          ShellStatus = ProcessVariables (VariableName, Guid, ShellCommandLineGetFlag(Package, L"-d"));
 | 
						|
        }
 | 
						|
      }
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  if (Package != NULL) {
 | 
						|
    ShellCommandLineFreeVarList (Package);
 | 
						|
  }
 | 
						|
  return ShellStatus;
 | 
						|
}
 | 
						|
 |