[mirror_edk2.git] / ShellPkg / Library / UefiShellDebug1CommandsLib / DmpStore.c

/** @file\r
  Main file for DmpStore shell Debug1 function.\r
\r
  Copyright (c) 2005 - 2014, Intel Corporation. All rights reserved.<BR>\r
  This program and the accompanying materials\r
  are licensed and made available under the terms and conditions of the BSD License\r
  which accompanies this distribution.  The full text of the license may be found at\r
  http://opensource.org/licenses/bsd-license.php\r
\r
  THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
  WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
\r
**/\r
\r
#include "UefiShellDebug1CommandsLib.h"\r
\r
/**\r
  Base on the input attribute value to return the attribute string.\r
\r
  @param[in]     Atts           The input attribute value\r
\r
  @retval The attribute string info.\r
**/\r
CHAR16 *\r
EFIAPI\r
GetAttrType (\r
  IN CONST UINT32 Atts\r
  )\r
{\r
  UINTN  BufLen;\r
  CHAR16 *RetString;\r
\r
  BufLen      = 0;\r
  RetString   = NULL;\r
 \r
  if ((Atts & EFI_VARIABLE_NON_VOLATILE) != 0) {\r
    StrnCatGrow (&RetString, &BufLen, L"+NV", 0);\r
  }\r
  if ((Atts & EFI_VARIABLE_RUNTIME_ACCESS) != 0) {\r
    StrnCatGrow (&RetString, &BufLen, L"+RS+BS", 0);\r
  } else if ((Atts & EFI_VARIABLE_BOOTSERVICE_ACCESS) != 0) {\r
    StrnCatGrow (&RetString, &BufLen, L"+BS", 0);\r
  }\r
  if ((Atts & EFI_VARIABLE_HARDWARE_ERROR_RECORD) != 0) {\r
    StrnCatGrow (&RetString, &BufLen, L"+HR", 0);\r
  }\r
  if ((Atts & EFI_VARIABLE_AUTHENTICATED_WRITE_ACCESS) != 0) {\r
    StrnCatGrow (&RetString, &BufLen, L"+AW", 0);\r
  }\r
  if ((Atts & EFI_VARIABLE_TIME_BASED_AUTHENTICATED_WRITE_ACCESS) != 0) {\r
    StrnCatGrow (&RetString, &BufLen, L"+AT", 0);\r
  }\r
\r
  if (RetString == NULL) {\r
    RetString = StrnCatGrow(&RetString, &BufLen, L"Invalid", 0);\r
  }\r
\r
  if ((RetString != NULL) && (RetString[0] == L'+')) {\r
    CopyMem(RetString, RetString + 1, StrSize(RetString + 1));\r
  }\r
\r
  return RetString;\r
}\r
\r
/**\r
  Recursive function to display or delete variables.\r
\r
  This function will call itself to create a stack-based list of allt he variables to process, \r
  then fromt he last to the first, they will do either printing or deleting.\r
\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] Guid           The GUID of the variable set (or NULL).\r
  @param[in] Delete         TRUE to delete, FALSE otherwise.\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
                            deleted, then set this to TRUE, otherwise ignored.\r
\r
  @retval SHELL_SUCCESS           The operation was successful.\r
  @retval SHELL_OUT_OF_RESOURCES  A memorty allocation failed.\r
  @retval SHELL_ABORTED           The abort message was received.\r
  @retval SHELL_DEVICE_ERROR      UEFI Variable Services returned an error.\r
  @retval SHELL_NOT_FOUND         the Name/Guid pair could not be found.\r
**/\r
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
  )\r
{\r
  EFI_STATUS                Status;\r
  CHAR16                    *FoundVarName;\r
  UINT8                     *DataBuffer;\r
  UINTN                     DataSize;\r
  UINT32                    Atts;\r
  SHELL_STATUS              ShellStatus;\r
  UINTN                     NameSize;\r
  CHAR16                    *RetString;\r
\r
  if (ShellGetExecutionBreakFlag()) {\r
    return (SHELL_ABORTED);\r
  }\r
\r
  NameSize      = 0;\r
  FoundVarName  = NULL;\r
\r
  if (PrevName!=NULL) {\r
    StrnCatGrow(&FoundVarName, &NameSize, PrevName, 0);\r
  } else {\r
    FoundVarName = AllocateZeroPool(sizeof(CHAR16));\r
  }\r
\r
  Status = gRT->GetNextVariableName (&NameSize, FoundVarName, &FoundVarGuid);\r
  if (Status == EFI_BUFFER_TOO_SMALL) {\r
    SHELL_FREE_NON_NULL(FoundVarName);\r
    FoundVarName = AllocateZeroPool (NameSize);\r
    if (PrevName != NULL) {\r
      StrCpy(FoundVarName, PrevName);\r
    }\r
\r
    Status = gRT->GetNextVariableName (&NameSize, FoundVarName, &FoundVarGuid);\r
  }\r
\r
  //\r
  // No more is fine.\r
  //\r
  if (Status == EFI_NOT_FOUND) {\r
    SHELL_FREE_NON_NULL(FoundVarName);\r
    return (SHELL_SUCCESS);\r
  } else if (EFI_ERROR(Status)) {\r
    SHELL_FREE_NON_NULL(FoundVarName);\r
    return (SHELL_DEVICE_ERROR);\r
  }\r
\r
  //\r
  // Recurse to the next iteration.  We know "our" variable's name.\r
  //\r
  ShellStatus = CascadeProcessVariables(VariableName, Guid, Delete, 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
     && ( Guid == NULL \r
      || CompareGuid(&FoundVarGuid, Guid) )\r
      ) {\r
    DataSize      = 0;\r
    DataBuffer    = NULL;\r
    //\r
    // do the print or delete\r
    //\r
    *FoundOne = TRUE;\r
    Status = gRT->GetVariable (FoundVarName, &FoundVarGuid, &Atts, &DataSize, DataBuffer);\r
    if (Status == EFI_BUFFER_TOO_SMALL) {\r
      SHELL_FREE_NON_NULL (DataBuffer);\r
      DataBuffer = AllocatePool (DataSize);\r
      if (DataBuffer == NULL) {\r
        Status = EFI_OUT_OF_RESOURCES;\r
      } else {\r
        Status = gRT->GetVariable (FoundVarName, &FoundVarGuid, &Atts, &DataSize, DataBuffer);\r
      }\r
    }\r
    if (!Delete) {\r
      //\r
      // Last error check then print this variable out.\r
      //\r
      if (!EFI_ERROR(Status) && DataBuffer != NULL) {\r
        RetString = GetAttrType(Atts);\r
        ShellPrintHiiEx(\r
          -1,\r
          -1,\r
          NULL,\r
          STRING_TOKEN(STR_DMPSTORE_HEADER_LINE),\r
          gShellDebug1HiiHandle,\r
          RetString,\r
          &FoundVarGuid,\r
          FoundVarName,\r
          DataSize);\r
        DumpHex(2, 0, DataSize, DataBuffer);\r
        SHELL_FREE_NON_NULL(RetString);\r
      }\r
    } else {\r
      //\r
      // We only need name to delete it...\r
      //\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
    }\r
    SHELL_FREE_NON_NULL(DataBuffer);\r
  }\r
\r
  SHELL_FREE_NON_NULL(FoundVarName);\r
\r
  if (Status == EFI_DEVICE_ERROR) {\r
    ShellStatus = SHELL_DEVICE_ERROR;\r
  } else if (Status == EFI_SECURITY_VIOLATION) {\r
    ShellStatus = SHELL_SECURITY_VIOLATION;\r
  } else if (EFI_ERROR(Status)) {\r
    ShellStatus = SHELL_NOT_READY;\r
  }\r
\r
  return (ShellStatus);\r
}\r
\r
/**\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
\r
  @retval SHELL_SUCCESS           The operation was successful.\r
  @retval SHELL_OUT_OF_RESOURCES  A memorty allocation failed.\r
  @retval SHELL_ABORTED           The abort message was received.\r
  @retval SHELL_DEVICE_ERROR      UEFI Variable Services returned an error.\r
  @retval SHELL_NOT_FOUND         the Name/Guid pair could not be found.\r
**/\r
SHELL_STATUS\r
EFIAPI\r
ProcessVariables (\r
  IN CONST CHAR16   *VariableName OPTIONAL,\r
  IN CONST EFI_GUID *Guid OPTIONAL,\r
  IN BOOLEAN        Delete\r
  )\r
{\r
  SHELL_STATUS              ShellStatus;\r
  BOOLEAN                   Found;\r
  EFI_GUID                  FoundVarGuid;\r
\r
  Found         = FALSE;\r
  ShellStatus   = SHELL_SUCCESS;\r
  ZeroMem (&FoundVarGuid, sizeof(EFI_GUID));\r
\r
  ShellStatus = CascadeProcessVariables(VariableName, Guid, Delete, NULL, FoundVarGuid, &Found);\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
      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_DMPSTORE_NO_VAR_FOUND), gShellDebug1HiiHandle);\r
    } else if (VariableName == 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
  }\r
  return (ShellStatus);\r
}\r
\r
STATIC CONST SHELL_PARAM_ITEM ParamList[] = {\r
  {L"-d", TypeFlag},\r
  {L"-l", TypeFlag},\r
  {L"-s", TypeFlag},\r
  {L"-all", TypeFlag},\r
  {L"-guid", TypeValue},\r
  {NULL, TypeMax}\r
  };\r
\r
/**\r
  Function for 'dmpstore' command.\r
\r
  @param[in] ImageHandle  Handle to the Image (NULL if Internal).\r
  @param[in] SystemTable  Pointer to the System Table (NULL if Internal).\r
**/\r
SHELL_STATUS\r
EFIAPI\r
ShellCommandRunDmpStore (\r
  IN EFI_HANDLE        ImageHandle,\r
  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
\r
  ShellStatus   = SHELL_SUCCESS;\r
  Package       = NULL;\r
\r
  Status = ShellCommandLineParse (ParamList, &Package, &ProblemParam, TRUE);\r
  if (EFI_ERROR(Status)) {\r
    if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {\r
      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, ProblemParam);\r
      FreePool(ProblemParam);\r
      ShellStatus = SHELL_INVALID_PARAMETER;\r
    } else {\r
      ASSERT(FALSE);\r
    }\r
  } else {\r
    if (ShellCommandLineGetCount(Package) > 2) {\r
      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle);\r
      ShellStatus = SHELL_INVALID_PARAMETER;\r
    } 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")) && 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
          if (EFI_ERROR(Status)) {\r
            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, Temp);\r
            ShellStatus = SHELL_INVALID_PARAMETER;\r
          }\r
          Guid = &GuidData;\r
        } else  {\r
          Guid = &gEfiGlobalVariableGuid;\r
        }\r
        VariableName = ShellCommandLineGetRawValue(Package, 1);\r
      } else {\r
        VariableName  = 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
        }\r
      }\r
    }\r
  }\r
\r
  if (Package != NULL) {\r
    ShellCommandLineFreeVarList (Package);\r
  }\r
  return ShellStatus;\r
}\r
\r
Commit	Line	Data
5d73d92f	1	/** @file\r
	2	Main file for DmpStore shell Debug1 function.\r
	3	\r
de4ebdcf	4	Copyright (c) 2005 - 2014, Intel Corporation. All rights reserved.<BR>\r
5d73d92f	5	This program and the accompanying materials\r
	6	are licensed and made available under the terms and conditions of the BSD License\r
	7	which accompanies this distribution. The full text of the license may be found at\r
	8	http://opensource.org/licenses/bsd-license.php\r
	9	\r
	10	THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
	11	WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
	12	\r
	13	**/\r
	14	\r
	15	#include "UefiShellDebug1CommandsLib.h"\r
	16	\r
cd0842dc ED	17	/**\r
	18	Base on the input attribute value to return the attribute string.\r
	19	\r
	20	@param[in] Atts The input attribute value\r
cd0842dc ED	21	\r
	22	@retval The attribute string info.\r
	23	**/\r
d6972185	24	CHAR16 *\r
73c82041 ED	25	EFIAPI\r
73c82041 ED	26	GetAttrType (\r
d6972185	27	IN CONST UINT32 Atts\r
73c82041 ED	28	)\r
73c82041 ED	29	{\r
99bb0f8d	30	UINTN BufLen;\r
d6972185	31	CHAR16 *RetString;\r
73c82041	32	\r
d6972185 ED	33	BufLen = 0;\r
	34	RetString = NULL;\r
	35	\r
cd0842dc	36	if ((Atts & EFI_VARIABLE_NON_VOLATILE) != 0) {\r
d6972185	37	StrnCatGrow (&RetString, &BufLen, L"+NV", 0);\r
73c82041	38	}\r
cd0842dc	39	if ((Atts & EFI_VARIABLE_RUNTIME_ACCESS) != 0) {\r
d6972185	40	StrnCatGrow (&RetString, &BufLen, L"+RS+BS", 0);\r
cd0842dc	41	} else if ((Atts & EFI_VARIABLE_BOOTSERVICE_ACCESS) != 0) {\r
d6972185	42	StrnCatGrow (&RetString, &BufLen, L"+BS", 0);\r
73c82041	43	}\r
cd0842dc	44	if ((Atts & EFI_VARIABLE_HARDWARE_ERROR_RECORD) != 0) {\r
d6972185	45	StrnCatGrow (&RetString, &BufLen, L"+HR", 0);\r
73c82041	46	}\r
cd0842dc	47	if ((Atts & EFI_VARIABLE_AUTHENTICATED_WRITE_ACCESS) != 0) {\r
d6972185	48	StrnCatGrow (&RetString, &BufLen, L"+AW", 0);\r
73c82041	49	}\r
cd0842dc	50	if ((Atts & EFI_VARIABLE_TIME_BASED_AUTHENTICATED_WRITE_ACCESS) != 0) {\r
d6972185	51	StrnCatGrow (&RetString, &BufLen, L"+AT", 0);\r
73c82041 ED	52	}\r
73c82041 ED	53	\r
d6972185 ED	54	if (RetString == NULL) {\r
d6972185 ED	55	RetString = StrnCatGrow(&RetString, &BufLen, L"Invalid", 0);\r
73c82041	56	}\r
d6972185	57	\r
44890dbd	58	if ((RetString != NULL) && (RetString[0] == L'+')) {\r
d6972185	59	CopyMem(RetString, RetString + 1, StrSize(RetString + 1));\r
73c82041	60	}\r
d6972185 ED	61	\r
d6972185 ED	62	return RetString;\r
73c82041	63	}\r
5d73d92f	64	\r
3737ac2b	65	/**\r
5bd12b05 JC	66	Recursive function to display or delete variables.\r
	67	\r
	68	This function will call itself to create a stack-based list of allt he variables to process, \r
	69	then fromt he last to the first, they will do either printing or deleting.\r
	70	\r
	71	This is necessary since once a delete happens GetNextVariableName() will work.\r
3737ac2b	72	\r
	73	@param[in] VariableName The variable name of the EFI variable (or NULL).\r
	74	@param[in] Guid The GUID of the variable set (or NULL).\r
	75	@param[in] Delete TRUE to delete, FALSE otherwise.\r
5bd12b05 JC	76	@param[in] PrevName The previous variable name from GetNextVariableName. L"" to start.\r
	77	@param[in] FoundVarGuid The previous GUID from GetNextVariableName. ignored at start.\r
	78	@param[in] FoundOne If a VariableName or Guid was specified and one was printed or\r
	79	deleted, then set this to TRUE, otherwise ignored.\r
3737ac2b	80	\r
	81	@retval SHELL_SUCCESS The operation was successful.\r
	82	@retval SHELL_OUT_OF_RESOURCES A memorty allocation failed.\r
	83	@retval SHELL_ABORTED The abort message was received.\r
	84	@retval SHELL_DEVICE_ERROR UEFI Variable Services returned an error.\r
	85	@retval SHELL_NOT_FOUND the Name/Guid pair could not be found.\r
	86	**/\r
5d73d92f	87	SHELL_STATUS\r
5d73d92f	88	EFIAPI\r
5bd12b05	89	CascadeProcessVariables (\r
5d73d92f	90	IN CONST CHAR16 *VariableName OPTIONAL,\r
5d73d92f	91	IN CONST EFI_GUID *Guid OPTIONAL,\r
5bd12b05 JC	92	IN BOOLEAN Delete,\r
	93	IN CONST CHAR16 * CONST PrevName,\r
	94	IN EFI_GUID FoundVarGuid,\r
	95	IN BOOLEAN *FoundOne\r
5d73d92f	96	)\r
	97	{\r
	98	EFI_STATUS Status;\r
5d73d92f	99	CHAR16 *FoundVarName;\r
5d73d92f	100	UINT8 *DataBuffer;\r
	101	UINTN DataSize;\r
	102	UINT32 Atts;\r
	103	SHELL_STATUS ShellStatus;\r
73c82041	104	UINTN NameSize;\r
d6972185	105	CHAR16 *RetString;\r
5d73d92f	106	\r
5bd12b05 JC	107	if (ShellGetExecutionBreakFlag()) {\r
5bd12b05 JC	108	return (SHELL_ABORTED);\r
5d73d92f	109	}\r
5d73d92f	110	\r
5bd12b05 JC	111	NameSize = 0;\r
5bd12b05 JC	112	FoundVarName = NULL;\r
5d73d92f	113	\r
5bd12b05 JC	114	if (PrevName!=NULL) {\r
	115	StrnCatGrow(&FoundVarName, &NameSize, PrevName, 0);\r
	116	} else {\r
	117	FoundVarName = AllocateZeroPool(sizeof(CHAR16));\r
	118	}\r
	119	\r
	120	Status = gRT->GetNextVariableName (&NameSize, FoundVarName, &FoundVarGuid);\r
	121	if (Status == EFI_BUFFER_TOO_SMALL) {\r
	122	SHELL_FREE_NON_NULL(FoundVarName);\r
	123	FoundVarName = AllocateZeroPool (NameSize);\r
	124	if (PrevName != NULL) {\r
	125	StrCpy(FoundVarName, PrevName);\r
5d73d92f	126	}\r
5d73d92f	127	\r
5bd12b05 JC	128	Status = gRT->GetNextVariableName (&NameSize, FoundVarName, &FoundVarGuid);\r
	129	}\r
	130	\r
	131	//\r
	132	// No more is fine.\r
	133	//\r
	134	if (Status == EFI_NOT_FOUND) {\r
	135	SHELL_FREE_NON_NULL(FoundVarName);\r
	136	return (SHELL_SUCCESS);\r
	137	} else if (EFI_ERROR(Status)) {\r
	138	SHELL_FREE_NON_NULL(FoundVarName);\r
	139	return (SHELL_DEVICE_ERROR);\r
	140	}\r
	141	\r
	142	//\r
	143	// Recurse to the next iteration. We know "our" variable's name.\r
	144	//\r
	145	ShellStatus = CascadeProcessVariables(VariableName, Guid, Delete, FoundVarName, FoundVarGuid, FoundOne);\r
	146	\r
	147	//\r
	148	// No matter what happened we process our own variable\r
	149	// Only continue if Guid and VariableName are each either NULL or a match\r
	150	//\r
	151	if ( ( VariableName == NULL \r
	152	\|\| gUnicodeCollation->MetaiMatch(gUnicodeCollation, FoundVarName, (CHAR16*)VariableName) )\r
	153	&& ( Guid == NULL \r
	154	\|\| CompareGuid(&FoundVarGuid, Guid) )\r
	155	) {\r
	156	DataSize = 0;\r
	157	DataBuffer = NULL;\r
5d73d92f	158	//\r
5bd12b05	159	// do the print or delete\r
5d73d92f	160	//\r
5bd12b05 JC	161	*FoundOne = TRUE;\r
5bd12b05 JC	162	Status = gRT->GetVariable (FoundVarName, &FoundVarGuid, &Atts, &DataSize, DataBuffer);\r
73c82041	163	if (Status == EFI_BUFFER_TOO_SMALL) {\r
5bd12b05 JC	164	SHELL_FREE_NON_NULL (DataBuffer);\r
5bd12b05 JC	165	DataBuffer = AllocatePool (DataSize);\r
73c82041 ED	166	if (DataBuffer == NULL) {\r
73c82041 ED	167	Status = EFI_OUT_OF_RESOURCES;\r
5bd12b05 JC	168	} else {\r
5bd12b05 JC	169	Status = gRT->GetVariable (FoundVarName, &FoundVarGuid, &Atts, &DataSize, DataBuffer);\r
73c82041	170	}\r
5bd12b05	171	}\r
5d73d92f	172	if (!Delete) {\r
5bd12b05 JC	173	//\r
	174	// Last error check then print this variable out.\r
	175	//\r
de4ebdcf	176	if (!EFI_ERROR(Status) && DataBuffer != NULL) {\r
5bd12b05 JC	177	RetString = GetAttrType(Atts);\r
	178	ShellPrintHiiEx(\r
	179	-1,\r
	180	-1,\r
	181	NULL,\r
	182	STRING_TOKEN(STR_DMPSTORE_HEADER_LINE),\r
	183	gShellDebug1HiiHandle,\r
	184	RetString,\r
	185	&FoundVarGuid,\r
	186	FoundVarName,\r
	187	DataSize);\r
	188	DumpHex(2, 0, DataSize, DataBuffer);\r
	189	SHELL_FREE_NON_NULL(RetString);\r
	190	}\r
5d73d92f	191	} else {\r
5bd12b05 JC	192	//\r
	193	// We only need name to delete it...\r
	194	//\r
5d73d92f	195	ShellPrintHiiEx(\r
	196	-1,\r
	197	-1,\r
	198	NULL,\r
	199	STRING_TOKEN(STR_DMPSTORE_DELETE_LINE),\r
	200	gShellDebug1HiiHandle,\r
	201	&FoundVarGuid,\r
	202	FoundVarName);\r
	203	ShellPrintHiiEx(\r
	204	-1,\r
	205	-1,\r
	206	NULL,\r
	207	STRING_TOKEN(STR_DMPSTORE_DELETE_DONE),\r
	208	gShellDebug1HiiHandle,\r
	209	gRT->SetVariable(FoundVarName, &FoundVarGuid, Atts, 0, NULL));\r
d6972185	210	}\r
5bd12b05	211	SHELL_FREE_NON_NULL(DataBuffer);\r
5d73d92f	212	}\r
5d73d92f	213	\r
5bd12b05 JC	214	SHELL_FREE_NON_NULL(FoundVarName);\r
	215	\r
	216	if (Status == EFI_DEVICE_ERROR) {\r
	217	ShellStatus = SHELL_DEVICE_ERROR;\r
	218	} else if (Status == EFI_SECURITY_VIOLATION) {\r
	219	ShellStatus = SHELL_SECURITY_VIOLATION;\r
	220	} else if (EFI_ERROR(Status)) {\r
	221	ShellStatus = SHELL_NOT_READY;\r
5d73d92f	222	}\r
5bd12b05 JC	223	\r
	224	return (ShellStatus);\r
	225	}\r
	226	\r
	227	/**\r
	228	Function to display or delete variables. This will set up and call into the recursive function.\r
	229	\r
	230	@param[in] VariableName The variable name of the EFI variable (or NULL).\r
	231	@param[in] Guid The GUID of the variable set (or NULL).\r
	232	@param[in] Delete TRUE to delete, FALSE otherwise.\r
	233	\r
	234	@retval SHELL_SUCCESS The operation was successful.\r
	235	@retval SHELL_OUT_OF_RESOURCES A memorty allocation failed.\r
	236	@retval SHELL_ABORTED The abort message was received.\r
	237	@retval SHELL_DEVICE_ERROR UEFI Variable Services returned an error.\r
	238	@retval SHELL_NOT_FOUND the Name/Guid pair could not be found.\r
	239	**/\r
	240	SHELL_STATUS\r
	241	EFIAPI\r
	242	ProcessVariables (\r
	243	IN CONST CHAR16 *VariableName OPTIONAL,\r
	244	IN CONST EFI_GUID *Guid OPTIONAL,\r
	245	IN BOOLEAN Delete\r
	246	)\r
	247	{\r
	248	SHELL_STATUS ShellStatus;\r
	249	BOOLEAN Found;\r
	250	EFI_GUID FoundVarGuid;\r
	251	\r
	252	Found = FALSE;\r
	253	ShellStatus = SHELL_SUCCESS;\r
	254	ZeroMem (&FoundVarGuid, sizeof(EFI_GUID));\r
	255	\r
	256	ShellStatus = CascadeProcessVariables(VariableName, Guid, Delete, NULL, FoundVarGuid, &Found);\r
	257	\r
3737ac2b	258	if (!Found) {\r
5bd12b05	259	if (ShellStatus == SHELL_OUT_OF_RESOURCES) {\r
73c82041	260	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_OUT_MEM), gShellDebug1HiiHandle);\r
5bd12b05 JC	261	return (ShellStatus);\r
5bd12b05 JC	262	} else if (VariableName != NULL && Guid == NULL) {\r
3737ac2b	263	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_DMPSTORE_NO_VAR_FOUND_N), gShellDebug1HiiHandle, VariableName);\r
	264	} else if (VariableName != NULL && Guid != NULL) {\r
	265	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_DMPSTORE_NO_VAR_FOUND_GN), gShellDebug1HiiHandle, Guid, VariableName);\r
	266	} else if (VariableName == NULL && Guid == NULL) {\r
	267	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_DMPSTORE_NO_VAR_FOUND), gShellDebug1HiiHandle);\r
	268	} else if (VariableName == NULL && Guid != NULL) {\r
	269	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_DMPSTORE_NO_VAR_FOUND_G), gShellDebug1HiiHandle, Guid);\r
	270	} \r
	271	return (SHELL_NOT_FOUND);\r
	272	}\r
224e8e2f	273	return (ShellStatus);\r
5d73d92f	274	}\r
	275	\r
	276	STATIC CONST SHELL_PARAM_ITEM ParamList[] = {\r
	277	{L"-d", TypeFlag},\r
	278	{L"-l", TypeFlag},\r
	279	{L"-s", TypeFlag},\r
	280	{L"-all", TypeFlag},\r
	281	{L"-guid", TypeValue},\r
	282	{NULL, TypeMax}\r
	283	};\r
	284	\r
3737ac2b	285	/**\r
	286	Function for 'dmpstore' command.\r
	287	\r
	288	@param[in] ImageHandle Handle to the Image (NULL if Internal).\r
	289	@param[in] SystemTable Pointer to the System Table (NULL if Internal).\r
	290	**/\r
5d73d92f	291	SHELL_STATUS\r
	292	EFIAPI\r
	293	ShellCommandRunDmpStore (\r
	294	IN EFI_HANDLE ImageHandle,\r
	295	IN EFI_SYSTEM_TABLE *SystemTable\r
	296	)\r
	297	{\r
	298	EFI_STATUS Status;\r
	299	LIST_ENTRY *Package;\r
	300	CHAR16 *ProblemParam;\r
	301	SHELL_STATUS ShellStatus;\r
	302	CONST CHAR16 *Temp;\r
	303	EFI_GUID *Guid;\r
	304	EFI_GUID GuidData;\r
	305	CONST CHAR16 *VariableName;\r
	306	\r
	307	ShellStatus = SHELL_SUCCESS;\r
	308	Package = NULL;\r
	309	\r
	310	Status = ShellCommandLineParse (ParamList, &Package, &ProblemParam, TRUE);\r
	311	if (EFI_ERROR(Status)) {\r
	312	if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {\r
	313	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, ProblemParam);\r
	314	FreePool(ProblemParam);\r
	315	ShellStatus = SHELL_INVALID_PARAMETER;\r
	316	} else {\r
	317	ASSERT(FALSE);\r
	318	}\r
	319	} else {\r
3737ac2b	320	if (ShellCommandLineGetCount(Package) > 2) {\r
5d73d92f	321	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle);\r
	322	ShellStatus = SHELL_INVALID_PARAMETER;\r
	323	} else if (ShellCommandLineGetFlag(Package, L"-all") && ShellCommandLineGetFlag(Package, L"-guid")) {\r
	324	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_CONFLICT), gShellDebug1HiiHandle, L"-all", L"-guid");\r
	325	ShellStatus = SHELL_INVALID_PARAMETER;\r
	326	} else if ((ShellCommandLineGetFlag(Package, L"-s") \|\| ShellCommandLineGetFlag(Package, L"-l")) && ShellCommandLineGetFlag(Package, L"-d")) {\r
	327	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_CONFLICT), gShellDebug1HiiHandle, L"-l or -s", L"-d");\r
	328	ShellStatus = SHELL_INVALID_PARAMETER;\r
	329	} else {\r
	330	if (!ShellCommandLineGetFlag(Package, L"-all")) {\r
	331	Temp = ShellCommandLineGetValue(Package, L"-guid");\r
	332	if (Temp != NULL) {\r
	333	Status = ConvertStringToGuid(Temp, &GuidData);\r
	334	if (EFI_ERROR(Status)) {\r
3737ac2b	335	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, Temp);\r
5d73d92f	336	ShellStatus = SHELL_INVALID_PARAMETER;\r
	337	}\r
	338	Guid = &GuidData;\r
	339	} else {\r
	340	Guid = &gEfiGlobalVariableGuid;\r
	341	}\r
3737ac2b	342	VariableName = ShellCommandLineGetRawValue(Package, 1);\r
5d73d92f	343	} else {\r
	344	VariableName = NULL;\r
	345	Guid = NULL;\r
	346	}\r
	347	if (ShellStatus == SHELL_SUCCESS) {\r
	348	if (ShellCommandLineGetFlag(Package, L"-s") \|\| ShellCommandLineGetFlag(Package, L"-l")) {\r
ea4fe84f	349	///@todo fix this after lib ready...\r
3737ac2b	350	ShellPrintEx(-1, -1, L"Not implemeneted yet.\r\n");\r
3737ac2b	351	ShellStatus = SHELL_UNSUPPORTED;\r
5d73d92f	352	} else {\r
	353	ShellStatus = ProcessVariables (VariableName, Guid, ShellCommandLineGetFlag(Package, L"-d"));\r
	354	}\r
	355	}\r
	356	}\r
	357	}\r
	358	\r
	359	if (Package != NULL) {\r
	360	ShellCommandLineFreeVarList (Package);\r
	361	}\r
	362	return ShellStatus;\r
	363	}\r
	364	\r