]> git.proxmox.com Git - mirror_edk2.git/blame_incremental - ShellPkg/Library/UefiShellDebug1CommandsLib/SetVar.c
projects / mirror_edk2.git / blame_incremental
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (non-incremental) | history | HEAD
ShellPkg/pci: Fix extended register dumping for MFVC capability
[mirror_edk2.git] / ShellPkg / Library / UefiShellDebug1CommandsLib / SetVar.c
... / ...
CommitLineData
1/** @file\r
2 Main file for SetVar shell Debug1 function.\r
3\r
4 (C) Copyright 2015 Hewlett-Packard Development Company, L.P.<BR>\r
5 Copyright (c) 2010 - 2014, Intel Corporation. All rights reserved.<BR>\r
6 This program and the accompanying materials\r
7 are licensed and made available under the terms and conditions of the BSD License\r
8 which accompanies this distribution. The full text of the license may be found at\r
9 http://opensource.org/licenses/bsd-license.php\r
10\r
11 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
12 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
13\r
14**/\r
15\r
16#include "UefiShellDebug1CommandsLib.h"\r
17\r
18STATIC CONST SHELL_PARAM_ITEM ParamList[] = {\r
19 {L"-guid", TypeValue},\r
20 {L"-bs", TypeFlag},\r
21 {L"-rt", TypeFlag},\r
22 {L"-nv", TypeFlag},\r
23 {NULL, TypeMax}\r
24 };\r
25\r
26\r
27/**\r
28 Check if the input is a (potentially empty) string of hexadecimal nibbles.\r
29\r
30 @param[in] String The CHAR16 string to check.\r
31\r
32 @retval FALSE A character has been found in String for which\r
33 ShellIsHexaDecimalDigitCharacter() returned FALSE.\r
34\r
35 @retval TRUE Otherwise. (Note that this covers the case when String is\r
36 empty.)\r
37**/\r
38BOOLEAN\r
39IsStringOfHexNibbles (\r
40 IN CONST CHAR16 *String\r
41 )\r
42{\r
43 CONST CHAR16 *Pos;\r
44\r
45 for (Pos = String; *Pos != L'\0'; ++Pos) {\r
46 if (!ShellIsHexaDecimalDigitCharacter (*Pos)) {\r
47 return FALSE;\r
48 }\r
49 }\r
50 return TRUE;\r
51}\r
52\r
53\r
54/**\r
55 Function for 'setvar' command.\r
56\r
57 @param[in] ImageHandle Handle to the Image (NULL if Internal).\r
58 @param[in] SystemTable Pointer to the System Table (NULL if Internal).\r
59**/\r
60SHELL_STATUS\r
61EFIAPI\r
62ShellCommandRunSetVar (\r
63 IN EFI_HANDLE ImageHandle,\r
64 IN EFI_SYSTEM_TABLE *SystemTable\r
65 )\r
66{\r
67 EFI_STATUS Status;\r
68 LIST_ENTRY *Package;\r
69 CHAR16 *ProblemParam;\r
70 SHELL_STATUS ShellStatus;\r
71 CONST CHAR16 *VariableName;\r
72 CONST CHAR16 *Data;\r
73 EFI_GUID Guid;\r
74 CONST CHAR16 *StringGuid;\r
75 UINT32 Attributes;\r
76 VOID *Buffer;\r
77 UINTN Size;\r
78 UINTN LoopVar;\r
79 EFI_DEVICE_PATH_PROTOCOL *DevPath;\r
80\r
81 ShellStatus = SHELL_SUCCESS;\r
82 Status = EFI_SUCCESS;\r
83 Buffer = NULL;\r
84 Size = 0;\r
85 Attributes = 0;\r
86 DevPath = NULL;\r
87\r
88 //\r
89 // initialize the shell lib (we must be in non-auto-init...)\r
90 //\r
91 Status = ShellInitialize();\r
92 ASSERT_EFI_ERROR(Status);\r
93\r
94 Status = CommandInit();\r
95 ASSERT_EFI_ERROR(Status);\r
96\r
97 //\r
98 // parse the command line\r
99 //\r
100 Status = ShellCommandLineParse (ParamList, &Package, &ProblemParam, TRUE);\r
101 if (EFI_ERROR(Status)) {\r
102 if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {\r
103 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, L"setvar", ProblemParam); \r
104 FreePool(ProblemParam);\r
105 ShellStatus = SHELL_INVALID_PARAMETER;\r
106 } else {\r
107 ASSERT(FALSE);\r
108 }\r
109 } else {\r
110 if (ShellCommandLineGetCount(Package) < 2) {\r
111 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle, L"setvar"); \r
112 ShellStatus = SHELL_INVALID_PARAMETER;\r
113 } else if (ShellCommandLineGetCount(Package) > 3) {\r
114 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle, L"setvar"); \r
115 ShellStatus = SHELL_INVALID_PARAMETER;\r
116 } else {\r
117 VariableName = ShellCommandLineGetRawValue(Package, 1);\r
118 Data = ShellCommandLineGetRawValue(Package, 2);\r
119 if (!ShellCommandLineGetFlag(Package, L"-guid")){\r
120 CopyGuid(&Guid, &gEfiGlobalVariableGuid);\r
121 } else {\r
122 StringGuid = ShellCommandLineGetValue(Package, L"-guid");\r
123 Status = ConvertStringToGuid(StringGuid, &Guid);\r
124 if (EFI_ERROR(Status)) {\r
125 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellDebug1HiiHandle, L"setvar", StringGuid); \r
126 ShellStatus = SHELL_INVALID_PARAMETER;\r
127 }\r
128 }\r
129 if (Data == NULL || Data[0] != L'=') {\r
130 //\r
131 // Display what's there\r
132 //\r
133 Status = gRT->GetVariable((CHAR16*)VariableName, &Guid, &Attributes, &Size, Buffer);\r
134 if (Status == EFI_BUFFER_TOO_SMALL) {\r
135 Buffer = AllocateZeroPool(Size);\r
136 Status = gRT->GetVariable((CHAR16*)VariableName, &Guid, &Attributes, &Size, Buffer);\r
137 }\r
138 if (!EFI_ERROR(Status)&& Buffer != NULL) {\r
139 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_PRINT), gShellDebug1HiiHandle, &Guid, VariableName, Size);\r
140 for (LoopVar = 0 ; LoopVar < Size ; LoopVar++) {\r
141 ShellPrintEx(-1, -1, L"%02x ", ((UINT8*)Buffer)[LoopVar]);\r
142 }\r
143 ShellPrintEx(-1, -1, L"\r\n");\r
144 } else {\r
145 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_GET), gShellDebug1HiiHandle, L"setvar", &Guid, VariableName); \r
146 ShellStatus = SHELL_ACCESS_DENIED;\r
147 }\r
148 } else if (StrCmp(Data, L"=") == 0) {\r
149 //\r
150 // Delete what's there!\r
151 //\r
152 Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, 0, NULL);\r
153 if (EFI_ERROR(Status)) {\r
154 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, L"setvar", &Guid, VariableName); \r
155 ShellStatus = SHELL_ACCESS_DENIED;\r
156 } else {\r
157 ASSERT(ShellStatus == SHELL_SUCCESS);\r
158 }\r
159 } else {\r
160 //\r
161 // Change what's there or create a new one.\r
162 //\r
163\r
164 ASSERT(Data[0] == L'=');\r
165 Data++;\r
166 ASSERT(Data[0] != L'\0');\r
167\r
168 //\r
169 // Determine if the variable exists and get the attributes\r
170 //\r
171 Status = gRT->GetVariable((CHAR16*)VariableName, &Guid, &Attributes, &Size, Buffer);\r
172 if (Status == EFI_BUFFER_TOO_SMALL) {\r
173 Buffer = AllocateZeroPool(Size);\r
174 Status = gRT->GetVariable((CHAR16*)VariableName, &Guid, &Attributes, &Size, Buffer);\r
175 }\r
176\r
177 if (EFI_ERROR(Status) || Buffer == NULL) {\r
178 //\r
179 // Creating a new variable. determine attributes from command line.\r
180 //\r
181 Attributes = 0;\r
182 if (ShellCommandLineGetFlag(Package, L"-bs")) {\r
183 Attributes |= EFI_VARIABLE_BOOTSERVICE_ACCESS;\r
184 }\r
185 if (ShellCommandLineGetFlag(Package, L"-rt")) {\r
186 Attributes |= EFI_VARIABLE_RUNTIME_ACCESS |\r
187 EFI_VARIABLE_BOOTSERVICE_ACCESS;\r
188 }\r
189 if (ShellCommandLineGetFlag(Package, L"-nv")) {\r
190 Attributes |= EFI_VARIABLE_NON_VOLATILE;\r
191 }\r
192 }\r
193 SHELL_FREE_NON_NULL(Buffer);\r
194\r
195 //\r
196 // What type is the new data.\r
197 //\r
198 if (IsStringOfHexNibbles(Data)) {\r
199 if (StrLen(Data) % 2 != 0) {\r
200 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellDebug1HiiHandle, L"setvar", Data); \r
201 ShellStatus = SHELL_INVALID_PARAMETER;\r
202 } else {\r
203 //\r
204 // arbitrary buffer\r
205 //\r
206 Buffer = AllocateZeroPool((StrLen(Data) / 2));\r
207 if (Buffer == NULL) {\r
208 Status = EFI_OUT_OF_RESOURCES;\r
209 } else {\r
210 for (LoopVar = 0 ; LoopVar < (StrLen(Data) / 2) ; LoopVar++) {\r
211 ((UINT8*)Buffer)[LoopVar] = (UINT8)(HexCharToUintn(Data[LoopVar*2]) * 16);\r
212 ((UINT8*)Buffer)[LoopVar] = (UINT8)(((UINT8*)Buffer)[LoopVar] + HexCharToUintn(Data[LoopVar*2+1]));\r
213 }\r
214 Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, StrLen(Data) / 2, Buffer);\r
215 }\r
216 if (EFI_ERROR(Status)) {\r
217 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, L"setvar", &Guid, VariableName); \r
218 ShellStatus = SHELL_ACCESS_DENIED;\r
219 } else {\r
220 ASSERT(ShellStatus == SHELL_SUCCESS);\r
221 }\r
222 }\r
223 } else if (StrnCmp(Data, L"\"", 1) == 0) {\r
224 //\r
225 // ascii text\r
226 //\r
227 Data++;\r
228 Buffer = AllocateZeroPool(StrSize(Data) / 2);\r
229 if (Buffer == NULL) {\r
230 Status = EFI_OUT_OF_RESOURCES;\r
231 } else {\r
232 AsciiSPrint(Buffer, StrSize(Data) / 2, "%s", Data);\r
233 ((CHAR8*)Buffer)[AsciiStrLen(Buffer)-1] = CHAR_NULL;\r
234 Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, AsciiStrSize(Buffer)-sizeof(CHAR8), Buffer);\r
235 }\r
236 if (EFI_ERROR(Status)) {\r
237 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, L"setvar", &Guid, VariableName); \r
238 ShellStatus = SHELL_ACCESS_DENIED;\r
239 } else {\r
240 ASSERT(ShellStatus == SHELL_SUCCESS);\r
241 }\r
242 } else if (StrnCmp(Data, L"L\"", 2) == 0) {\r
243 //\r
244 // ucs2 text\r
245 //\r
246 Data++;\r
247 Data++;\r
248 Buffer = AllocateZeroPool(StrSize(Data));\r
249 if (Buffer == NULL) {\r
250 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_OUT_MEM), gShellDebug1HiiHandle, L"setvar"); \r
251 ShellStatus = SHELL_OUT_OF_RESOURCES;\r
252 } else {\r
253 UnicodeSPrint(Buffer, StrSize(Data), L"%s", Data);\r
254 ((CHAR16*)Buffer)[StrLen(Buffer)-1] = CHAR_NULL;\r
255\r
256 Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, StrSize(Buffer)-sizeof(CHAR16), Buffer);\r
257 if (EFI_ERROR(Status)) {\r
258 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, L"setvar", &Guid, VariableName); \r
259 ShellStatus = SHELL_ACCESS_DENIED;\r
260 } else {\r
261 ASSERT(ShellStatus == SHELL_SUCCESS);\r
262 }\r
263 }\r
264 } else if (StrnCmp(Data, L"--", 2) == 0) {\r
265 //\r
266 // device path in text format\r
267 //\r
268 Data++;\r
269 Data++;\r
270 DevPath = ConvertTextToDevicePath(Data);\r
271 if (DevPath == NULL) {\r
272 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_DPFT), gShellDebug1HiiHandle, L"setvar"); \r
273 ShellStatus = SHELL_INVALID_PARAMETER;\r
274 } else {\r
275 Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, GetDevicePathSize(DevPath), DevPath);\r
276 if (EFI_ERROR(Status)) {\r
277 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, L"setvar", &Guid, VariableName); \r
278 ShellStatus = SHELL_ACCESS_DENIED;\r
279 } else {\r
280 ASSERT(ShellStatus == SHELL_SUCCESS);\r
281 }\r
282 }\r
283 } else {\r
284 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellDebug1HiiHandle, L"setvar", Data); \r
285 ShellStatus = SHELL_INVALID_PARAMETER;\r
286 }\r
287 }\r
288 }\r
289 ShellCommandLineFreeVarList (Package);\r
290 }\r
291\r
292 if (Buffer != NULL) {\r
293 FreePool(Buffer);\r
294 }\r
295\r
296 if (DevPath != NULL) {\r
297 FreePool(DevPath);\r
298 }\r
299\r
300 return (ShellStatus);\r
301}\r
EFI Development Kit II mirror for PVE