Add "Debug1" profile (all but Edit and HexEdit commands)

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@11068 6f19259b-4bc3-4df7-8a09-765794883524

1 files changed, 228 insertions, 0 deletions

diff --git a/ShellPkg/Library/UefiShellDebug1CommandsLib/DmpStore.c b/ShellPkg/Library/UefiShellDebug1CommandsLib/DmpStore.c
new file mode 100644
index 0000000000..0a586591be
--- /dev/null
+++ b/ShellPkg/Library/UefiShellDebug1CommandsLib/DmpStore.c
@@ -0,0 +1,228 @@
+/** @file

+  Main file for DmpStore shell Debug1 function.

+

+  Copyright (c) 2005 - 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 "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

+};

+

+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;

+

+  ShellStatus   = SHELL_SUCCESS;

+  Size          = PcdGet16(PcdShellFileOperationSize);

+  FoundVarName  = AllocatePool(Size);

+

+  if (FoundVarName == NULL) {

+    return (SHELL_OUT_OF_RESOURCES);

+  }

+  FoundVarName[0] = CHAR_NULL;

+

+  Status = gRT->QueryVariableInfo(EFI_VARIABLE_BOOTSERVICE_ACCESS|EFI_VARIABLE_RUNTIME_ACCESS|EFI_VARIABLE_NON_VOLATILE, &MaxStorSize, &RemStorSize, &MaxVarSize);

+  ASSERT_EFI_ERROR(Status);

+

+  DataSize = (UINTN)MaxVarSize;

+  DataBuffer = AllocatePool(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

+    //

+    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);

+  }

+

+  return (SHELL_UNSUPPORTED);

+}

+

+STATIC CONST SHELL_PARAM_ITEM ParamList[] = {

+  {L"-d", TypeFlag},

+  {L"-l", TypeFlag},

+  {L"-s", TypeFlag},

+  {L"-all", TypeFlag},

+  {L"-guid", TypeValue},

+  {NULL, TypeMax}

+  };

+

+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) < 1) {

+      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle);

+      ShellStatus = SHELL_INVALID_PARAMETER;

+    } 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, L"-guid");

+            ShellStatus = SHELL_INVALID_PARAMETER;

+          }

+          Guid = &GuidData;

+        } else  {

+          Guid = &gEfiGlobalVariableGuid;

+        }

+        VariableName = ShellCommandLineGetRawValue(Package, 2);

+      } 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 (ASSERT follows).\r\n");

+          ShellStatus = SHELL_INVALID_PARAMETER;

+          ASSERT(FALSE);

+        } else {

+          ShellStatus = ProcessVariables (VariableName, Guid, ShellCommandLineGetFlag(Package, L"-d"));

+        }

+      }

+    }

+  }

+

+  if (Package != NULL) {

+    ShellCommandLineFreeVarList (Package);

+  }

+  return ShellStatus;

+}

+