]> git.proxmox.com Git - mirror_edk2.git/blame - ShellPkg/Library/UefiShellDebug1CommandsLib/DmpStore.c
Roll back the change which caused build fail
[mirror_edk2.git] / ShellPkg / Library / UefiShellDebug1CommandsLib / DmpStore.c
CommitLineData
5d73d92f 1/** @file\r
2 Main file for DmpStore shell Debug1 function.\r
3\r
ea4fe84f 4 Copyright (c) 2005 - 2012, 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
17STATIC CHAR16 *AttrType[] = {\r
18 L"invalid", // 000\r
19 L"invalid", // 001\r
20 L"BS", // 010\r
21 L"NV+BS", // 011\r
22 L"RT+BS", // 100\r
23 L"NV+RT+BS", // 101\r
24 L"RT+BS", // 110\r
25 L"NV+RT+BS", // 111\r
26};\r
27\r
3737ac2b 28/**\r
29 Function to display or delete variables.\r
30\r
31 @param[in] VariableName The variable name of the EFI variable (or NULL).\r
32 @param[in] Guid The GUID of the variable set (or NULL).\r
33 @param[in] Delete TRUE to delete, FALSE otherwise.\r
34\r
35 @retval SHELL_SUCCESS The operation was successful.\r
36 @retval SHELL_OUT_OF_RESOURCES A memorty allocation failed.\r
37 @retval SHELL_ABORTED The abort message was received.\r
38 @retval SHELL_DEVICE_ERROR UEFI Variable Services returned an error.\r
39 @retval SHELL_NOT_FOUND the Name/Guid pair could not be found.\r
40**/\r
5d73d92f 41SHELL_STATUS\r
42EFIAPI\r
43ProcessVariables (\r
44 IN CONST CHAR16 *VariableName OPTIONAL,\r
45 IN CONST EFI_GUID *Guid OPTIONAL,\r
46 IN BOOLEAN Delete\r
47 )\r
48{\r
49 EFI_STATUS Status;\r
50 UINT64 MaxStorSize;\r
51 UINT64 RemStorSize;\r
52 UINT64 MaxVarSize;\r
53 CHAR16 *FoundVarName;\r
54 UINTN Size;\r
55 EFI_GUID FoundVarGuid;\r
56 UINT8 *DataBuffer;\r
57 UINTN DataSize;\r
58 UINT32 Atts;\r
59 SHELL_STATUS ShellStatus;\r
3737ac2b 60 BOOLEAN Found;\r
61\r
62 Status = gRT->QueryVariableInfo(EFI_VARIABLE_BOOTSERVICE_ACCESS|EFI_VARIABLE_RUNTIME_ACCESS|EFI_VARIABLE_NON_VOLATILE, &MaxStorSize, &RemStorSize, &MaxVarSize);\r
63 if (EFI_ERROR(Status)) {\r
64 return (SHELL_DEVICE_ERROR);\r
65 }\r
5d73d92f 66\r
3737ac2b 67 Found = FALSE;\r
5d73d92f 68 ShellStatus = SHELL_SUCCESS;\r
69 Size = PcdGet16(PcdShellFileOperationSize);\r
3737ac2b 70 FoundVarName = AllocateZeroPool(Size);\r
5d73d92f 71\r
72 if (FoundVarName == NULL) {\r
73 return (SHELL_OUT_OF_RESOURCES);\r
74 }\r
75 FoundVarName[0] = CHAR_NULL;\r
76\r
5d73d92f 77\r
78 DataSize = (UINTN)MaxVarSize;\r
3737ac2b 79 DataBuffer = AllocateZeroPool(DataSize);\r
5d73d92f 80 if (DataBuffer == NULL) {\r
81 FreePool(FoundVarName);\r
82 return (SHELL_OUT_OF_RESOURCES);\r
83 }\r
84\r
85 for (;;){\r
86 if (ShellGetExecutionBreakFlag()) {\r
87 ShellStatus = SHELL_ABORTED;\r
88 break;\r
89 }\r
90 Size = (UINTN)PcdGet16(PcdShellFileOperationSize);\r
91 DataSize = (UINTN)MaxVarSize;\r
92\r
93 Status = gRT->GetNextVariableName(&Size, FoundVarName, &FoundVarGuid);\r
94 if (Status == EFI_NOT_FOUND) {\r
95 break;\r
96 }\r
97 ASSERT_EFI_ERROR(Status);\r
98\r
99 Status = gRT->GetVariable(FoundVarName, &FoundVarGuid, &Atts, &DataSize, DataBuffer);\r
100 ASSERT_EFI_ERROR(Status);\r
101\r
102 //\r
103 // Check if it matches\r
104 //\r
105 if (VariableName != NULL) {\r
106 if (!gUnicodeCollation->MetaiMatch(gUnicodeCollation, FoundVarName, (CHAR16*)VariableName)) {\r
107 continue;\r
108 }\r
109 }\r
110 if (Guid != NULL) {\r
111 if (!CompareGuid(&FoundVarGuid, Guid)) {\r
112 continue;\r
113 }\r
114 }\r
115\r
116 //\r
117 // do the print or delete\r
118 //\r
3737ac2b 119 Found = TRUE;\r
5d73d92f 120 if (!Delete) {\r
121 ShellPrintHiiEx(\r
122 -1,\r
123 -1,\r
124 NULL,\r
125 STRING_TOKEN(STR_DMPSTORE_HEADER_LINE),\r
126 gShellDebug1HiiHandle,\r
127 AttrType[Atts & 7],\r
128 &FoundVarGuid,\r
129 FoundVarName,\r
130 DataSize);\r
131 DumpHex(2, 0, DataSize, DataBuffer);\r
132 } else {\r
133 ShellPrintHiiEx(\r
134 -1,\r
135 -1,\r
136 NULL,\r
137 STRING_TOKEN(STR_DMPSTORE_DELETE_LINE),\r
138 gShellDebug1HiiHandle,\r
139 &FoundVarGuid,\r
140 FoundVarName);\r
141 ShellPrintHiiEx(\r
142 -1,\r
143 -1,\r
144 NULL,\r
145 STRING_TOKEN(STR_DMPSTORE_DELETE_DONE),\r
146 gShellDebug1HiiHandle,\r
147 gRT->SetVariable(FoundVarName, &FoundVarGuid, Atts, 0, NULL));\r
9d8fa791 148 FoundVarName[0] = CHAR_NULL;\r
5d73d92f 149 }\r
150 }\r
151\r
152 if (FoundVarName != NULL) {\r
153 FreePool(FoundVarName);\r
154 }\r
155 if (DataBuffer != NULL) {\r
156 FreePool(DataBuffer);\r
157 }\r
3737ac2b 158 if (!Found) {\r
159 if (VariableName != NULL && Guid == NULL) {\r
160 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_DMPSTORE_NO_VAR_FOUND_N), gShellDebug1HiiHandle, VariableName);\r
161 } else if (VariableName != NULL && Guid != NULL) {\r
162 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_DMPSTORE_NO_VAR_FOUND_GN), gShellDebug1HiiHandle, Guid, VariableName);\r
163 } else if (VariableName == NULL && Guid == NULL) {\r
164 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_DMPSTORE_NO_VAR_FOUND), gShellDebug1HiiHandle);\r
165 } else if (VariableName == NULL && Guid != NULL) {\r
166 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_DMPSTORE_NO_VAR_FOUND_G), gShellDebug1HiiHandle, Guid);\r
167 } \r
168 return (SHELL_NOT_FOUND);\r
169 }\r
170 return (SHELL_SUCCESS);\r
5d73d92f 171}\r
172\r
173STATIC CONST SHELL_PARAM_ITEM ParamList[] = {\r
174 {L"-d", TypeFlag},\r
175 {L"-l", TypeFlag},\r
176 {L"-s", TypeFlag},\r
177 {L"-all", TypeFlag},\r
178 {L"-guid", TypeValue},\r
179 {NULL, TypeMax}\r
180 };\r
181\r
3737ac2b 182/**\r
183 Function for 'dmpstore' command.\r
184\r
185 @param[in] ImageHandle Handle to the Image (NULL if Internal).\r
186 @param[in] SystemTable Pointer to the System Table (NULL if Internal).\r
187**/\r
5d73d92f 188SHELL_STATUS\r
189EFIAPI\r
190ShellCommandRunDmpStore (\r
191 IN EFI_HANDLE ImageHandle,\r
192 IN EFI_SYSTEM_TABLE *SystemTable\r
193 )\r
194{\r
195 EFI_STATUS Status;\r
196 LIST_ENTRY *Package;\r
197 CHAR16 *ProblemParam;\r
198 SHELL_STATUS ShellStatus;\r
199 CONST CHAR16 *Temp;\r
200 EFI_GUID *Guid;\r
201 EFI_GUID GuidData;\r
202 CONST CHAR16 *VariableName;\r
203\r
204 ShellStatus = SHELL_SUCCESS;\r
205 Package = NULL;\r
206\r
207 Status = ShellCommandLineParse (ParamList, &Package, &ProblemParam, TRUE);\r
208 if (EFI_ERROR(Status)) {\r
209 if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {\r
210 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, ProblemParam);\r
211 FreePool(ProblemParam);\r
212 ShellStatus = SHELL_INVALID_PARAMETER;\r
213 } else {\r
214 ASSERT(FALSE);\r
215 }\r
216 } else {\r
3737ac2b 217 if (ShellCommandLineGetCount(Package) > 2) {\r
5d73d92f 218 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle);\r
219 ShellStatus = SHELL_INVALID_PARAMETER;\r
220 } else if (ShellCommandLineGetFlag(Package, L"-all") && ShellCommandLineGetFlag(Package, L"-guid")) {\r
221 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_CONFLICT), gShellDebug1HiiHandle, L"-all", L"-guid");\r
222 ShellStatus = SHELL_INVALID_PARAMETER;\r
223 } else if ((ShellCommandLineGetFlag(Package, L"-s") || ShellCommandLineGetFlag(Package, L"-l")) && ShellCommandLineGetFlag(Package, L"-d")) {\r
224 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_CONFLICT), gShellDebug1HiiHandle, L"-l or -s", L"-d");\r
225 ShellStatus = SHELL_INVALID_PARAMETER;\r
226 } else {\r
227 if (!ShellCommandLineGetFlag(Package, L"-all")) {\r
228 Temp = ShellCommandLineGetValue(Package, L"-guid");\r
229 if (Temp != NULL) {\r
230 Status = ConvertStringToGuid(Temp, &GuidData);\r
231 if (EFI_ERROR(Status)) {\r
3737ac2b 232 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, Temp);\r
5d73d92f 233 ShellStatus = SHELL_INVALID_PARAMETER;\r
234 }\r
235 Guid = &GuidData;\r
236 } else {\r
237 Guid = &gEfiGlobalVariableGuid;\r
238 }\r
3737ac2b 239 VariableName = ShellCommandLineGetRawValue(Package, 1);\r
5d73d92f 240 } else {\r
241 VariableName = NULL;\r
242 Guid = NULL;\r
243 }\r
244 if (ShellStatus == SHELL_SUCCESS) {\r
245 if (ShellCommandLineGetFlag(Package, L"-s") || ShellCommandLineGetFlag(Package, L"-l")) {\r
ea4fe84f 246 ///@todo fix this after lib ready...\r
3737ac2b 247 ShellPrintEx(-1, -1, L"Not implemeneted yet.\r\n");\r
248 ShellStatus = SHELL_UNSUPPORTED;\r
5d73d92f 249 } else {\r
250 ShellStatus = ProcessVariables (VariableName, Guid, ShellCommandLineGetFlag(Package, L"-d"));\r
251 }\r
252 }\r
253 }\r
254 }\r
255\r
256 if (Package != NULL) {\r
257 ShellCommandLineFreeVarList (Package);\r
258 }\r
259 return ShellStatus;\r
260}\r
261\r