Add the feature "dmpstore -l" and "dmpstore -s".
authorRuiyu Ni <ruiyu.ni@intel.com>
Fri, 10 Jan 2014 01:52:24 +0000 (01:52 +0000)
committerniruiyu <niruiyu@6f19259b-4bc3-4df7-8a09-765794883524>
Fri, 10 Jan 2014 01:52:24 +0000 (01:52 +0000)
The file format is as below:
<File> := <Variable>+
<Variable> := <Name-Size> <Data-Size> <Name> <GUID> <Attributes> <Data> <Crc32>

Each variable representation in the file has a CRC32 value which can provide a mechanism to detect the file modification.
When any CRC32 is incorrect, dmpstore rejects to load the variables from the file.

Signed-off-by: Ruiyu Ni <ruiyu.ni@intel.com>
Reviewed-by: Jaben Carsey <jaben.carsey@intel.com>
git-svn-id: https://svn.code.sf.net/p/edk2/code/trunk/edk2@15083 6f19259b-4bc3-4df7-8a09-765794883524

ShellPkg/Library/UefiShellDebug1CommandsLib/DmpStore.c
ShellPkg/Library/UefiShellDebug1CommandsLib/UefiShellDebug1CommandsLib.uni

index b02048302e6d64cc387667892507a1259a1c646d..6332416c8815427adfe8a4e43c0d7c4ae63b6136 100644 (file)
 \r
 #include "UefiShellDebug1CommandsLib.h"\r
 \r
+typedef enum {\r
+  DmpStoreDisplay,\r
+  DmpStoreDelete,\r
+  DmpStoreSave,\r
+  DmpStoreLoad\r
+} DMP_STORE_TYPE;\r
+\r
+typedef struct {\r
+  UINT32     Signature;\r
+  CHAR16     *Name;\r
+  EFI_GUID   Guid;\r
+  UINT32     Attributes;\r
+  UINT32     DataSize;\r
+  UINT8      *Data;\r
+  LIST_ENTRY Link;\r
+} DMP_STORE_VARIABLE;\r
+\r
+#define DMP_STORE_VARIABLE_SIGNATURE  SIGNATURE_32 ('_', 'd', 's', 's')\r
+\r
 /**\r
   Base on the input attribute value to return the attribute string.\r
 \r
@@ -62,6 +81,265 @@ GetAttrType (
   return RetString;\r
 }\r
 \r
+/**\r
+  Load the variable data from file and set to variable data base.\r
+\r
+  @param[in]  FileHandle     The file to be read.\r
+  @param[in]  Name           The name of the variables to be loaded.\r
+  @param[in]  Guid           The guid of the variables to be loaded.\r
+  @param[out] Found          TRUE when at least one variable was loaded and set.\r
+\r
+  @retval EFI_VOLUME_CORRUPTED  The file is in bad format.\r
+  @retval EFI_OUT_OF_RESOURCES  There is not enough memory to perform the operation.\r
+  @retval EFI_SUCCESS           Successfully load and set the variables.\r
+**/\r
+EFI_STATUS\r
+LoadVariablesFromFile (\r
+  IN SHELL_FILE_HANDLE FileHandle,\r
+  IN CONST CHAR16      *Name,\r
+  IN CONST EFI_GUID    *Guid,\r
+  OUT BOOLEAN          *Found\r
+  )\r
+{\r
+  EFI_STATUS           Status;\r
+  UINT32               NameSize;\r
+  UINT32               DataSize;\r
+  UINTN                BufferSize;\r
+  UINTN                RemainingSize;\r
+  UINT64               Position;\r
+  UINT64               FileSize;\r
+  LIST_ENTRY           List;\r
+  DMP_STORE_VARIABLE   *Variable;\r
+  LIST_ENTRY           *Link;\r
+  CHAR16               *Attributes;\r
+  UINT8                *Buffer;\r
+  UINT32               Crc32;\r
+\r
+  Status = ShellGetFileSize (FileHandle, &FileSize);\r
+  if (EFI_ERROR (Status)) {\r
+    return Status;\r
+  }\r
+\r
+  InitializeListHead (&List);\r
+  \r
+  Position = 0;\r
+  while (Position < FileSize) {\r
+    //\r
+    // NameSize\r
+    //\r
+    BufferSize = sizeof (NameSize);\r
+    Status = ShellReadFile (FileHandle, &BufferSize, &NameSize);\r
+    if (EFI_ERROR (Status) || (BufferSize != sizeof (NameSize))) {\r
+      Status = EFI_VOLUME_CORRUPTED;\r
+      break;\r
+    }\r
+\r
+    //\r
+    // DataSize\r
+    //\r
+    BufferSize = sizeof (DataSize);\r
+    Status = ShellReadFile (FileHandle, &BufferSize, &DataSize);\r
+    if (EFI_ERROR (Status) || (BufferSize != sizeof (DataSize))) {\r
+      Status = EFI_VOLUME_CORRUPTED;\r
+      break;\r
+    }\r
+\r
+    //\r
+    // Name, Guid, Attributes, Data, Crc32\r
+    //\r
+    RemainingSize = NameSize + sizeof (EFI_GUID) + sizeof (UINT32) + DataSize + sizeof (Crc32);\r
+    BufferSize    = sizeof (NameSize) + sizeof (DataSize) + RemainingSize;\r
+    Buffer        = AllocatePool (BufferSize);\r
+    if (Buffer == NULL) {\r
+      Status = EFI_OUT_OF_RESOURCES;\r
+      break;\r
+    }\r
+    BufferSize    = RemainingSize;\r
+    Status = ShellReadFile (FileHandle, &BufferSize, (UINT32 *) Buffer + 2);\r
+    if (EFI_ERROR (Status) || (BufferSize != RemainingSize)) {\r
+      Status = EFI_VOLUME_CORRUPTED;\r
+      FreePool (Buffer);\r
+      break;\r
+    }\r
+\r
+    //\r
+    // Check Crc32\r
+    //\r
+    * (UINT32 *) Buffer       = NameSize;\r
+    * ((UINT32 *) Buffer + 1) = DataSize;\r
+    BufferSize = RemainingSize + sizeof (NameSize) + sizeof (DataSize) - sizeof (Crc32);\r
+    gBS->CalculateCrc32 (\r
+           Buffer,\r
+           BufferSize,\r
+           &Crc32\r
+           );\r
+    if (Crc32 != * (UINT32 *) (Buffer + BufferSize)) {\r
+      FreePool (Buffer);\r
+      Status = EFI_VOLUME_CORRUPTED;\r
+      break;\r
+    }\r
+\r
+    Position += BufferSize + sizeof (Crc32);\r
+    \r
+    Variable = AllocateZeroPool (sizeof (*Variable) + NameSize + DataSize);\r
+    if (Variable == NULL) {\r
+      FreePool (Buffer);\r
+      Status = EFI_OUT_OF_RESOURCES;\r
+      break;\r
+    }\r
+    Variable->Signature = DMP_STORE_VARIABLE_SIGNATURE;\r
+    Variable->Name      = (CHAR16 *) (Variable + 1);\r
+    Variable->DataSize  = DataSize;\r
+    Variable->Data      = (UINT8 *) Variable->Name + NameSize;\r
+    CopyMem (Variable->Name,        Buffer + sizeof (NameSize) + sizeof (DataSize),                                                  NameSize);\r
+    CopyMem (&Variable->Guid,       Buffer + sizeof (NameSize) + sizeof (DataSize) + NameSize,                                       sizeof (EFI_GUID));\r
+    CopyMem (&Variable->Attributes, Buffer + sizeof (NameSize) + sizeof (DataSize) + NameSize + sizeof (EFI_GUID),                   sizeof (UINT32));\r
+    CopyMem (Variable->Data,        Buffer + sizeof (NameSize) + sizeof (DataSize) + NameSize + sizeof (EFI_GUID) + sizeof (UINT32), DataSize);\r
+\r
+    InsertTailList (&List, &Variable->Link);\r
+    FreePool (Buffer);\r
+  }\r
+    \r
+  if ((Position != FileSize) || EFI_ERROR (Status)) {\r
+    ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_DMPSTORE_LOAD_BAD_FILE), gShellDebug1HiiHandle);\r
+    Status = EFI_VOLUME_CORRUPTED;\r
+  }\r
+  \r
+  for ( Link = GetFirstNode (&List)\r
+      ; !IsNull (&List, Link) && !EFI_ERROR (Status)\r
+      ; Link = GetNextNode (&List, Link)\r
+      ) {\r
+    Variable = CR (Link, DMP_STORE_VARIABLE, Link, DMP_STORE_VARIABLE_SIGNATURE);\r
+    \r
+    if (((Name == NULL) || gUnicodeCollation->MetaiMatch (gUnicodeCollation, Variable->Name, (CHAR16 *) Name)) &&\r
+        ((Guid == NULL) || CompareGuid (&Variable->Guid, Guid))\r
+       ) {\r
+      Attributes = GetAttrType (Variable->Attributes);\r
+      ShellPrintHiiEx (\r
+        -1, -1, NULL, STRING_TOKEN(STR_DMPSTORE_HEADER_LINE), gShellDebug1HiiHandle,\r
+        Attributes, &Variable->Guid, Variable->Name, Variable->DataSize\r
+        );\r
+      SHELL_FREE_NON_NULL(Attributes);\r
+\r
+      *Found = TRUE;\r
+      Status = gRT->SetVariable (\r
+                      Variable->Name,\r
+                      &Variable->Guid,\r
+                      Variable->Attributes,\r
+                      Variable->DataSize,\r
+                      Variable->Data\r
+                      );\r
+      if (EFI_ERROR (Status)) {\r
+        ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_DMPSTORE_LOAD_GEN_FAIL), gShellDebug1HiiHandle, Variable->Name, Status);\r
+        //\r
+        // continue set variable upon failure\r
+        //\r
+        Status = EFI_SUCCESS;\r
+      }\r
+    }\r
+  }\r
+\r
+  for (Link = GetFirstNode (&List); !IsNull (&List, Link); ) {\r
+    Variable = CR (Link, DMP_STORE_VARIABLE, Link, DMP_STORE_VARIABLE_SIGNATURE);\r
+    Link = RemoveEntryList (&Variable->Link);\r
+    FreePool (Variable);\r
+  }\r
+\r
+  return Status;\r
+}\r
+\r
+/**\r
+  Append one variable to file.\r
+\r
+  @param[in] FileHandle        The file to be appended.\r
+  @param[in] Name              The variable name.\r
+  @param[in] Guid              The variable GUID.\r
+  @param[in] Attributes        The variable attributes.\r
+  @param[in] DataSize          The variable data size.\r
+  @param[in] Data              The variable data.\r
+\r
+  @retval EFI_OUT_OF_RESOURCES  There is not enough memory to perform the operation.\r
+  @retval EFI_SUCCESS           The variable is appended to file successfully.\r
+  @retval others                Failed to append the variable to file.\r
+**/\r
+EFI_STATUS\r
+AppendSingleVariableToFile (\r
+  IN SHELL_FILE_HANDLE FileHandle,\r
+  IN CONST CHAR16      *Name,\r
+  IN CONST EFI_GUID    *Guid,\r
+  IN UINT32            Attributes,\r
+  IN UINT32            DataSize,\r
+  IN CONST UINT8       *Data\r
+  )\r
+{\r
+  UINT32              NameSize;\r
+  UINT8               *Buffer;\r
+  UINT8               *Ptr;\r
+  UINTN               BufferSize;\r
+  EFI_STATUS          Status;\r
+\r
+  NameSize   = (UINT32) StrSize (Name);\r
+  BufferSize = sizeof (NameSize) + sizeof (DataSize)\r
+             + sizeof (*Guid)\r
+             + sizeof (Attributes)\r
+             + NameSize + DataSize\r
+             + sizeof (UINT32);\r
+\r
+  Buffer = AllocatePool (BufferSize);\r
+  if (Buffer == NULL) {\r
+    return EFI_OUT_OF_RESOURCES;\r
+  }\r
+\r
+  Ptr = Buffer;\r
+  //\r
+  // NameSize and DataSize\r
+  //\r
+  * (UINT32 *) Ptr = NameSize;\r
+  Ptr += sizeof (NameSize);\r
+  *(UINT32 *) Ptr = DataSize;\r
+  Ptr += sizeof (DataSize);\r
+\r
+  //\r
+  // Name\r
+  //\r
+  CopyMem (Ptr, Name, NameSize);\r
+  Ptr += NameSize;\r
+\r
+  //\r
+  // Guid\r
+  //\r
+  CopyMem (Ptr, Guid, sizeof (*Guid));\r
+  Ptr += sizeof (*Guid);\r
+\r
+  //\r
+  // Attributes\r
+  //\r
+  * (UINT32 *) Ptr = Attributes;\r
+  Ptr += sizeof (Attributes);\r
+\r
+  //\r
+  // Data\r
+  //\r
+  CopyMem (Ptr, Data, DataSize);\r
+  Ptr += DataSize;\r
+\r
+  //\r
+  // Crc32\r
+  //\r
+  gBS->CalculateCrc32 (Buffer, (UINTN) (Ptr - Buffer), (UINT32 *) Ptr);\r
+\r
+  Status = ShellWriteFile (FileHandle, &BufferSize, Buffer);\r
+  FreePool (Buffer);\r
+\r
+  if (!EFI_ERROR (Status) && \r
+      (BufferSize != sizeof (NameSize) + sizeof (DataSize) + sizeof (*Guid) + sizeof (Attributes) + NameSize + DataSize)\r
+    ) {\r
+    Status = EFI_DEVICE_ERROR;\r
+  }\r
+  \r
+  return Status;\r
+}\r
+\r
 /**\r
   Recursive function to display or delete variables.\r
 \r
@@ -70,9 +348,10 @@ GetAttrType (
 \r
   This is necessary since once a delete happens GetNextVariableName() will work.\r
 \r
-  @param[in] VariableName   The variable name of the EFI variable (or NULL).\r
+  @param[in] Name           The variable name of the EFI variable (or NULL).\r
   @param[in] Guid           The GUID of the variable set (or NULL).\r
-  @param[in] Delete         TRUE to delete, FALSE otherwise.\r
+  @param[in] Type           The operation type.\r
+  @param[in] FileHandle     The file to operate on (or NULL).\r
   @param[in] PrevName       The previous variable name from GetNextVariableName. L"" to start.\r
   @param[in] FoundVarGuid   The previous GUID from GetNextVariableName. ignored at start.\r
   @param[in] FoundOne       If a VariableName or Guid was specified and one was printed or\r
@@ -87,12 +366,13 @@ GetAttrType (
 SHELL_STATUS\r
 EFIAPI\r
 CascadeProcessVariables (\r
-  IN CONST CHAR16   *VariableName OPTIONAL,\r
-  IN CONST EFI_GUID *Guid OPTIONAL,\r
-  IN BOOLEAN        Delete,\r
-  IN CONST CHAR16   * CONST PrevName,\r
-  IN EFI_GUID       FoundVarGuid,\r
-  IN BOOLEAN        *FoundOne\r
+  IN CONST CHAR16      *Name        OPTIONAL,\r
+  IN CONST EFI_GUID    *Guid        OPTIONAL,\r
+  IN DMP_STORE_TYPE    Type,\r
+  IN EFI_FILE_PROTOCOL *FileHandle  OPTIONAL,\r
+  IN CONST CHAR16      * CONST PrevName,\r
+  IN EFI_GUID          FoundVarGuid,\r
+  IN BOOLEAN           *FoundOne\r
   )\r
 {\r
   EFI_STATUS                Status;\r
@@ -146,14 +426,14 @@ CascadeProcessVariables (
   //\r
   // Recurse to the next iteration.  We know "our" variable's name.\r
   //\r
-  ShellStatus = CascadeProcessVariables(VariableName, Guid, Delete, FoundVarName, FoundVarGuid, FoundOne);\r
+  ShellStatus = CascadeProcessVariables(Name, Guid, Type, FileHandle, FoundVarName, FoundVarGuid, FoundOne);\r
 \r
   //\r
   // No matter what happened we process our own variable\r
   // Only continue if Guid and VariableName are each either NULL or a match\r
   //\r
-  if ( ( VariableName == NULL \r
-      || gUnicodeCollation->MetaiMatch(gUnicodeCollation, FoundVarName, (CHAR16*)VariableName) )\r
+  if ( ( Name == NULL \r
+      || gUnicodeCollation->MetaiMatch(gUnicodeCollation, FoundVarName, (CHAR16*) Name) )\r
      && ( Guid == NULL \r
       || CompareGuid(&FoundVarGuid, Guid) )\r
       ) {\r
@@ -173,7 +453,7 @@ CascadeProcessVariables (
         Status = gRT->GetVariable (FoundVarName, &FoundVarGuid, &Atts, &DataSize, DataBuffer);\r
       }\r
     }\r
-    if (!Delete) {\r
+    if ((Type == DmpStoreDisplay) || (Type == DmpStoreSave)) {\r
       //\r
       // Last error check then print this variable out.\r
       //\r
@@ -189,28 +469,34 @@ CascadeProcessVariables (
           &FoundVarGuid,\r
           FoundVarName,\r
           DataSize);\r
-        DumpHex(2, 0, DataSize, DataBuffer);\r
+        if (Type == DmpStoreDisplay) {\r
+          DumpHex(2, 0, DataSize, DataBuffer);\r
+        } else {\r
+          Status = AppendSingleVariableToFile (\r
+                     FileHandle,\r
+                     FoundVarName,\r
+                     &FoundVarGuid,\r
+                     Atts,\r
+                     (UINT32) DataSize,\r
+                     DataBuffer\r
+                     );\r
+        }\r
         SHELL_FREE_NON_NULL(RetString);\r
       }\r
-    } else {\r
+    } else if (Type == DmpStoreDelete) {\r
       //\r
       // We only need name to delete it...\r
       //\r
-      ShellPrintHiiEx(\r
+      ShellPrintHiiEx (\r
         -1,\r
         -1,\r
         NULL,\r
         STRING_TOKEN(STR_DMPSTORE_DELETE_LINE),\r
         gShellDebug1HiiHandle,\r
         &FoundVarGuid,\r
-        FoundVarName);\r
-      ShellPrintHiiEx(\r
-        -1,\r
-        -1,\r
-        NULL,\r
-        STRING_TOKEN(STR_DMPSTORE_DELETE_DONE),\r
-        gShellDebug1HiiHandle,\r
-        gRT->SetVariable(FoundVarName, &FoundVarGuid, Atts, 0, NULL));\r
+        FoundVarName,\r
+        gRT->SetVariable (FoundVarName, &FoundVarGuid, Atts, 0, NULL)\r
+        );\r
     }\r
     SHELL_FREE_NON_NULL(DataBuffer);\r
   }\r
@@ -231,9 +517,10 @@ CascadeProcessVariables (
 /**\r
   Function to display or delete variables.  This will set up and call into the recursive function.\r
 \r
-  @param[in] VariableName   The variable name of the EFI variable (or NULL).\r
-  @param[in] Guid           The GUID of the variable set (or NULL).\r
-  @param[in] Delete         TRUE to delete, FALSE otherwise.\r
+  @param[in] Name        The variable name of the EFI variable (or NULL).\r
+  @param[in] Guid        The GUID of the variable set (or NULL).\r
+  @param[in] Type        The operation type.\r
+  @param[in] FileHandle  The file to save or load variables.\r
 \r
   @retval SHELL_SUCCESS           The operation was successful.\r
   @retval SHELL_OUT_OF_RESOURCES  A memorty allocation failed.\r
@@ -244,9 +531,10 @@ CascadeProcessVariables (
 SHELL_STATUS\r
 EFIAPI\r
 ProcessVariables (\r
-  IN CONST CHAR16   *VariableName OPTIONAL,\r
-  IN CONST EFI_GUID *Guid OPTIONAL,\r
-  IN BOOLEAN        Delete\r
+  IN CONST CHAR16      *Name      OPTIONAL,\r
+  IN CONST EFI_GUID    *Guid      OPTIONAL,\r
+  IN DMP_STORE_TYPE    Type,\r
+  IN SHELL_FILE_HANDLE FileHandle OPTIONAL\r
   )\r
 {\r
   SHELL_STATUS              ShellStatus;\r
@@ -257,19 +545,23 @@ ProcessVariables (
   ShellStatus   = SHELL_SUCCESS;\r
   ZeroMem (&FoundVarGuid, sizeof(EFI_GUID));\r
 \r
-  ShellStatus = CascadeProcessVariables(VariableName, Guid, Delete, NULL, FoundVarGuid, &Found);\r
+  if (Type == DmpStoreLoad) {\r
+    ShellStatus = LoadVariablesFromFile (FileHandle, Name, Guid, &Found);\r
+  } else {\r
+    ShellStatus = CascadeProcessVariables(Name, Guid, Type, FileHandle, NULL, FoundVarGuid, &Found);\r
+  }\r
 \r
   if (!Found) {\r
     if (ShellStatus == SHELL_OUT_OF_RESOURCES) {\r
       ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_OUT_MEM), gShellDebug1HiiHandle);\r
       return (ShellStatus);\r
-    } else if (VariableName != NULL && Guid == NULL) {\r
-      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_DMPSTORE_NO_VAR_FOUND_N), gShellDebug1HiiHandle, VariableName);\r
-    } else if (VariableName != NULL && Guid != NULL) {\r
-      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_DMPSTORE_NO_VAR_FOUND_GN), gShellDebug1HiiHandle, Guid, VariableName);\r
-    } else if (VariableName == NULL && Guid == NULL) {\r
+    } else if (Name != NULL && Guid == NULL) {\r
+      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_DMPSTORE_NO_VAR_FOUND_N), gShellDebug1HiiHandle, Name);\r
+    } else if (Name != NULL && Guid != NULL) {\r
+      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_DMPSTORE_NO_VAR_FOUND_GN), gShellDebug1HiiHandle, Guid, Name);\r
+    } else if (Name == NULL && Guid == NULL) {\r
       ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_DMPSTORE_NO_VAR_FOUND), gShellDebug1HiiHandle);\r
-    } else if (VariableName == NULL && Guid != NULL) {\r
+    } else if (Name == NULL && Guid != NULL) {\r
       ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_DMPSTORE_NO_VAR_FOUND_G), gShellDebug1HiiHandle, Guid);\r
     } \r
     return (SHELL_NOT_FOUND);\r
@@ -279,8 +571,8 @@ ProcessVariables (
 \r
 STATIC CONST SHELL_PARAM_ITEM ParamList[] = {\r
   {L"-d", TypeFlag},\r
-  {L"-l", TypeFlag},\r
-  {L"-s", TypeFlag},\r
+  {L"-l", TypeValue},\r
+  {L"-s", TypeValue},\r
   {L"-all", TypeFlag},\r
   {L"-guid", TypeValue},\r
   {NULL, TypeMax}\r
@@ -299,17 +591,24 @@ ShellCommandRunDmpStore (
   IN EFI_SYSTEM_TABLE  *SystemTable\r
   )\r
 {\r
-  EFI_STATUS    Status;\r
-  LIST_ENTRY    *Package;\r
-  CHAR16        *ProblemParam;\r
-  SHELL_STATUS  ShellStatus;\r
-  CONST CHAR16  *Temp;\r
-  EFI_GUID      *Guid;\r
-  EFI_GUID      GuidData;\r
-  CONST CHAR16  *VariableName;\r
+  EFI_STATUS        Status;\r
+  LIST_ENTRY        *Package;\r
+  CHAR16            *ProblemParam;\r
+  SHELL_STATUS      ShellStatus;\r
+  CONST CHAR16      *GuidStr;\r
+  CONST CHAR16      *File;\r
+  EFI_GUID          *Guid;\r
+  EFI_GUID          GuidData;\r
+  CONST CHAR16      *Name;\r
+  DMP_STORE_TYPE    Type;\r
+  SHELL_FILE_HANDLE FileHandle;\r
+  EFI_FILE_INFO     *FileInfo;\r
 \r
   ShellStatus   = SHELL_SUCCESS;\r
   Package       = NULL;\r
+  FileHandle    = NULL;\r
+  File          = NULL;\r
+  Type          = DmpStoreDisplay;\r
 \r
   Status = ShellCommandLineParse (ParamList, &Package, &ProblemParam, TRUE);\r
   if (EFI_ERROR(Status)) {\r
@@ -327,34 +626,121 @@ ShellCommandRunDmpStore (
     } else if (ShellCommandLineGetFlag(Package, L"-all") && ShellCommandLineGetFlag(Package, L"-guid")) {\r
       ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_CONFLICT), gShellDebug1HiiHandle, L"-all", L"-guid");\r
       ShellStatus = SHELL_INVALID_PARAMETER;\r
+    } else if (ShellCommandLineGetFlag(Package, L"-s") && ShellCommandLineGetFlag(Package, L"-l")) {\r
+      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_CONFLICT), gShellDebug1HiiHandle, L"-l", L"-s");\r
+      ShellStatus = SHELL_INVALID_PARAMETER;\r
     } else if ((ShellCommandLineGetFlag(Package, L"-s") || ShellCommandLineGetFlag(Package, L"-l")) && ShellCommandLineGetFlag(Package, L"-d")) {\r
       ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_CONFLICT), gShellDebug1HiiHandle, L"-l or -s", L"-d");\r
       ShellStatus = SHELL_INVALID_PARAMETER;\r
     } else {\r
       if (!ShellCommandLineGetFlag(Package, L"-all")) {\r
-        Temp = ShellCommandLineGetValue(Package, L"-guid");\r
-        if (Temp != NULL) {\r
-          Status = ConvertStringToGuid(Temp, &GuidData);\r
+        GuidStr = ShellCommandLineGetValue(Package, L"-guid");\r
+        if (GuidStr != NULL) {\r
+          Status = ConvertStringToGuid(GuidStr, &GuidData);\r
           if (EFI_ERROR(Status)) {\r
-            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, Temp);\r
+            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, GuidStr);\r
             ShellStatus = SHELL_INVALID_PARAMETER;\r
           }\r
           Guid = &GuidData;\r
         } else  {\r
           Guid = &gEfiGlobalVariableGuid;\r
         }\r
-        VariableName = ShellCommandLineGetRawValue(Package, 1);\r
+        Name = ShellCommandLineGetRawValue(Package, 1);\r
       } else {\r
-        VariableName  = NULL;\r
-        Guid          = NULL;\r
+        Name  = NULL;\r
+        Guid  = NULL;\r
       }\r
       if (ShellStatus == SHELL_SUCCESS) {\r
-        if (ShellCommandLineGetFlag(Package, L"-s") || ShellCommandLineGetFlag(Package, L"-l")) {\r
-          ///@todo fix this after lib ready...\r
-          ShellPrintEx(-1, -1, L"Not implemeneted yet.\r\n");\r
-          ShellStatus = SHELL_UNSUPPORTED;\r
-        } else {\r
-          ShellStatus = ProcessVariables (VariableName, Guid, ShellCommandLineGetFlag(Package, L"-d"));\r
+        if (ShellCommandLineGetFlag(Package, L"-s")) {\r
+          Type = DmpStoreSave;\r
+          File = ShellCommandLineGetValue(Package, L"-s");\r
+          if (File == NULL) {\r
+            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_VALUE), gShellDebug1HiiHandle, L"-s");\r
+            ShellStatus = SHELL_INVALID_PARAMETER;\r
+          } else {\r
+            Status = ShellOpenFileByName (File, &FileHandle, EFI_FILE_MODE_WRITE | EFI_FILE_MODE_READ, 0);\r
+            if (!EFI_ERROR (Status)) {\r
+              //\r
+              // Delete existing file, but do not delete existing directory\r
+              //\r
+              FileInfo = ShellGetFileInfo (FileHandle);\r
+              if (FileInfo == NULL) {\r
+                ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_OPEN_FAIL), gShellDebug1HiiHandle, File);\r
+                Status = EFI_DEVICE_ERROR;\r
+              } else {\r
+                if ((FileInfo->Attribute & EFI_FILE_DIRECTORY) == EFI_FILE_DIRECTORY) {\r
+                  ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_IS_DIRECTORY), gShellDebug1HiiHandle, File);\r
+                  Status = EFI_INVALID_PARAMETER;\r
+                } else {\r
+                  Status = ShellDeleteFile (&FileHandle);\r
+                  if (EFI_ERROR (Status)) {\r
+                    ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_DELETE_FAIL), gShellDebug1HiiHandle, File);\r
+                  }\r
+                }\r
+                FreePool (FileInfo);\r
+              }\r
+            } else if (Status == EFI_NOT_FOUND) {\r
+              //\r
+              // Good when file doesn't exist\r
+              //\r
+              Status = EFI_SUCCESS;\r
+            } else {\r
+              //\r
+              // Otherwise it's bad.\r
+              //\r
+              ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_OPEN_FAIL), gShellDebug1HiiHandle, File);\r
+            }\r
+\r
+            if (!EFI_ERROR (Status)) {\r
+              Status = ShellOpenFileByName (File, &FileHandle, EFI_FILE_MODE_CREATE | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_READ, 0);\r
+              if (EFI_ERROR (Status)) {\r
+                ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_OPEN_FAIL), gShellDebug1HiiHandle, File);\r
+              }\r
+            }\r
+\r
+            if (EFI_ERROR (Status)) {\r
+              ShellStatus = SHELL_INVALID_PARAMETER;\r
+            }\r
+          }\r
+        } else if (ShellCommandLineGetFlag(Package, L"-l")) {\r
+          Type = DmpStoreLoad;\r
+          File = ShellCommandLineGetValue(Package, L"-l");\r
+          if (File == NULL) {\r
+            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_VALUE), gShellDebug1HiiHandle, L"-l");\r
+            ShellStatus = SHELL_INVALID_PARAMETER;\r
+          } else {\r
+            Status = ShellOpenFileByName (File, &FileHandle, EFI_FILE_MODE_READ, 0);\r
+            if (EFI_ERROR (Status)) {\r
+              ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_OPEN_FAIL), gShellDebug1HiiHandle, File);\r
+              ShellStatus = SHELL_INVALID_PARAMETER;\r
+            } else {\r
+              FileInfo = ShellGetFileInfo (FileHandle);\r
+              if (FileInfo == NULL) {\r
+                ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_OPEN_FAIL), gShellDebug1HiiHandle, File);\r
+                ShellStatus = SHELL_DEVICE_ERROR;\r
+              } else {\r
+                if ((FileInfo->Attribute & EFI_FILE_DIRECTORY) == EFI_FILE_DIRECTORY) {\r
+                  ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_IS_DIRECTORY), gShellDebug1HiiHandle, File);\r
+                  ShellStatus = SHELL_INVALID_PARAMETER;\r
+                }\r
+                FreePool (FileInfo);\r
+              }\r
+            }\r
+          }\r
+        } else if (ShellCommandLineGetFlag(Package, L"-d")) {\r
+          Type = DmpStoreDelete;\r
+        }\r
+      }\r
+\r
+      if (ShellStatus == SHELL_SUCCESS) {\r
+        if (Type == DmpStoreSave) {\r
+          ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_DMPSTORE_SAVE), gShellDebug1HiiHandle, File);\r
+        } else if (Type == DmpStoreLoad) {\r
+          ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_DMPSTORE_LOAD), gShellDebug1HiiHandle, File);\r
+        }\r
+        ShellStatus = ProcessVariables (Name, Guid, Type, FileHandle);\r
+        if ((Type == DmpStoreLoad) || (Type == DmpStoreSave)) {\r
+          ShellCloseFile (&FileHandle);\r
         }\r
       }\r
     }\r
index ae8383492f208fb9cc5d159cffee58613b14f6e2..ba4c470aa28ea9513de60d5879806d6018f737da 100644 (file)
Binary files a/ShellPkg/Library/UefiShellDebug1CommandsLib/UefiShellDebug1CommandsLib.uni and b/ShellPkg/Library/UefiShellDebug1CommandsLib/UefiShellDebug1CommandsLib.uni differ