]> git.proxmox.com Git - mirror_edk2.git/blame - ShellPkg/Library/UefiShellDebug1CommandsLib/SetVar.c
projects / mirror_edk2.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
Update all the code to consume the ConvertDevicePathToText, ConvertDevicePathNodeToTe...
[mirror_edk2.git] / ShellPkg / Library / UefiShellDebug1CommandsLib / SetVar.c
CommitLineData
5d73d92f 1/** @file\r
2 Main file for SetVar shell Debug1 function.\r
3\r
863986b3 4 Copyright (c) 2010 - 2013, 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 CONST SHELL_PARAM_ITEM ParamList[] = {\r
18 {L"-guid", TypeValue},\r
19 {L"-bs", TypeFlag},\r
20 {L"-rt", TypeFlag},\r
21 {L"-nv", TypeFlag},\r
22 {NULL, TypeMax}\r
23 };\r
24\r
3737ac2b 25/**\r
26 Function for 'setvar' command.\r
27\r
28 @param[in] ImageHandle Handle to the Image (NULL if Internal).\r
29 @param[in] SystemTable Pointer to the System Table (NULL if Internal).\r
30**/\r
5d73d92f 31SHELL_STATUS\r
32EFIAPI\r
33ShellCommandRunSetVar (\r
34 IN EFI_HANDLE ImageHandle,\r
35 IN EFI_SYSTEM_TABLE *SystemTable\r
36 )\r
37{\r
38 EFI_STATUS Status;\r
39 LIST_ENTRY *Package;\r
40 CHAR16 *ProblemParam;\r
41 SHELL_STATUS ShellStatus;\r
42 CONST CHAR16 *VariableName;\r
43 CONST CHAR16 *Data;\r
44 EFI_GUID Guid;\r
45 CONST CHAR16 *StringGuid;\r
46 UINT32 Attributes;\r
3737ac2b 47 UINT32 Attributes2;\r
5d73d92f 48 VOID *Buffer;\r
49 UINTN Size;\r
50 UINTN LoopVar;\r
51 EFI_DEVICE_PATH_PROTOCOL *DevPath;\r
5d73d92f 52\r
53 ShellStatus = SHELL_SUCCESS;\r
54 Status = EFI_SUCCESS;\r
55 Buffer = NULL;\r
56 Size = 0;\r
57 Attributes = 0;\r
58 DevPath = NULL;\r
59\r
60 //\r
61 // initialize the shell lib (we must be in non-auto-init...)\r
62 //\r
63 Status = ShellInitialize();\r
64 ASSERT_EFI_ERROR(Status);\r
65\r
66 Status = CommandInit();\r
67 ASSERT_EFI_ERROR(Status);\r
68\r
69 //\r
70 // parse the command line\r
71 //\r
72 Status = ShellCommandLineParse (ParamList, &Package, &ProblemParam, TRUE);\r
73 if (EFI_ERROR(Status)) {\r
74 if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {\r
75 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, ProblemParam);\r
76 FreePool(ProblemParam);\r
77 ShellStatus = SHELL_INVALID_PARAMETER;\r
78 } else {\r
79 ASSERT(FALSE);\r
80 }\r
81 } else {\r
82 if (ShellCommandLineGetCount(Package) < 2) {\r
83 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle);\r
84 ShellStatus = SHELL_INVALID_PARAMETER;\r
85 } else if (ShellCommandLineGetCount(Package) > 3) {\r
86 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle);\r
87 ShellStatus = SHELL_INVALID_PARAMETER;\r
88 } else {\r
89 VariableName = ShellCommandLineGetRawValue(Package, 1);\r
90 Data = ShellCommandLineGetRawValue(Package, 2);\r
91 if (!ShellCommandLineGetFlag(Package, L"-guid")){\r
92 CopyGuid(&Guid, &gEfiGlobalVariableGuid);\r
93 } else {\r
94 StringGuid = ShellCommandLineGetValue(Package, L"-guid");\r
95 Status = ConvertStringToGuid(StringGuid, &Guid);\r
96 if (EFI_ERROR(Status)) {\r
3737ac2b 97 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, StringGuid);\r
98 ShellStatus = SHELL_INVALID_PARAMETER;\r
5d73d92f 99 }\r
100 }\r
101 if (Data == NULL) {\r
102 //\r
103 // Display what's there\r
104 //\r
105 Status = gRT->GetVariable((CHAR16*)VariableName, &Guid, &Attributes, &Size, Buffer);\r
106 if (Status == EFI_BUFFER_TOO_SMALL) {\r
3737ac2b 107 Buffer = AllocateZeroPool(Size);\r
5d73d92f 108 Status = gRT->GetVariable((CHAR16*)VariableName, &Guid, &Attributes, &Size, Buffer);\r
109 }\r
110 if (!EFI_ERROR(Status)&& Buffer != NULL) {\r
111 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_PRINT), gShellDebug1HiiHandle, &Guid, VariableName, Size);\r
112 for (LoopVar = 0 ; LoopVar < Size ; LoopVar++) {\r
113 ShellPrintEx(-1, -1, L"%02x ", ((UINT8*)Buffer)[LoopVar]);\r
114 }\r
115 ShellPrintEx(-1, -1, L"\r\n");\r
116 } else {\r
117 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_GET), gShellDebug1HiiHandle, &Guid, VariableName, Status);\r
118 ShellStatus = SHELL_ACCESS_DENIED;\r
119 }\r
120 } else if (StrCmp(Data, L"=") == 0) {\r
121 //\r
122 // Delete what's there!\r
123 //\r
124 Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, 0, NULL);\r
125 if (EFI_ERROR(Status)) {\r
126 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, &Guid, VariableName, Status);\r
127 ShellStatus = SHELL_ACCESS_DENIED;\r
128 } else {\r
129 ASSERT(ShellStatus == SHELL_SUCCESS);\r
130 }\r
131 } else {\r
132 if (Data[0] == L'=') {\r
133 Data++;\r
134 }\r
135 //\r
136 // Change what's there\r
137 //\r
138 if (ShellCommandLineGetFlag(Package, L"-bs")) {\r
139 Attributes |= EFI_VARIABLE_BOOTSERVICE_ACCESS;\r
140 }\r
141 if (ShellCommandLineGetFlag(Package, L"-rt")) {\r
142 Attributes |= EFI_VARIABLE_RUNTIME_ACCESS;\r
143 }\r
144 if (ShellCommandLineGetFlag(Package, L"-nv")) {\r
145 Attributes |= EFI_VARIABLE_NON_VOLATILE;\r
146 }\r
147 if (ShellIsHexOrDecimalNumber(Data, TRUE, FALSE)) {\r
3737ac2b 148 if (StrLen(Data) % 2 != 0) {\r
149 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM_VAL), gShellDebug1HiiHandle, Data);\r
150 ShellStatus = SHELL_INVALID_PARAMETER;\r
5d73d92f 151 } else {\r
3737ac2b 152 //\r
153 // arbitrary buffer\r
154 //\r
155 Buffer = AllocateZeroPool((StrLen(Data) / 2));\r
33c031ee 156 if (Buffer == NULL) {\r
157 Status = EFI_OUT_OF_RESOURCES;\r
158 } else {\r
159 for (LoopVar = 0 ; LoopVar < (StrLen(Data) / 2) ; LoopVar++) {\r
160 ((UINT8*)Buffer)[LoopVar] = (UINT8)(HexCharToUintn(Data[LoopVar*2]) * 16);\r
161 ((UINT8*)Buffer)[LoopVar] = (UINT8)(((UINT8*)Buffer)[LoopVar] + HexCharToUintn(Data[LoopVar*2+1]));\r
162 }\r
163 Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, StrLen(Data) / 2, Buffer);\r
3737ac2b 164 }\r
3737ac2b 165 if (EFI_ERROR(Status)) {\r
166 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, &Guid, VariableName, Status);\r
167 ShellStatus = SHELL_ACCESS_DENIED;\r
168 } else {\r
169 ASSERT(ShellStatus == SHELL_SUCCESS);\r
170 }\r
5d73d92f 171 }\r
172 } else if (StrnCmp(Data, L"\"", 1) == 0) {\r
3737ac2b 173 Size = 0;\r
174 Attributes2 = 0;\r
175 Status = gRT->GetVariable((CHAR16*)VariableName, &Guid, &Attributes2, &Size, Buffer);\r
176 if (Status == EFI_BUFFER_TOO_SMALL) {\r
177 Buffer = AllocateZeroPool(Size);\r
178 Status = gRT->GetVariable((CHAR16*)VariableName, &Guid, &Attributes2, &Size, Buffer);\r
179 FreePool(Buffer);\r
180 Attributes = Attributes2;\r
181 } \r
5d73d92f 182 //\r
183 // ascii text\r
184 //\r
185 Data++;\r
186 Buffer = AllocateZeroPool(StrSize(Data) / 2);\r
33c031ee 187 if (Buffer == NULL) {\r
188 Status = EFI_OUT_OF_RESOURCES;\r
189 } else {\r
190 AsciiSPrint(Buffer, StrSize(Data) / 2, "%s", Data);\r
191 ((CHAR8*)Buffer)[AsciiStrLen(Buffer)-1] = CHAR_NULL;\r
192 Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, AsciiStrSize(Buffer)-sizeof(CHAR8), Buffer);\r
193 }\r
5d73d92f 194 if (EFI_ERROR(Status)) {\r
195 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, &Guid, VariableName, Status);\r
196 ShellStatus = SHELL_ACCESS_DENIED;\r
197 } else {\r
198 ASSERT(ShellStatus == SHELL_SUCCESS);\r
199 }\r
200 } else if (StrnCmp(Data, L"L\"", 2) == 0) {\r
201 //\r
202 // ucs2 text\r
203 //\r
204 Data++;\r
205 Data++;\r
206 Buffer = AllocateZeroPool(StrSize(Data));\r
cbdd109b 207 if (Buffer == NULL) {\r
208 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_OUT_MEM), gShellDebug1HiiHandle);\r
209 ShellStatus = SHELL_OUT_OF_RESOURCES;\r
5d73d92f 210 } else {\r
cbdd109b 211 UnicodeSPrint(Buffer, StrSize(Data), L"%s", Data);\r
212 ((CHAR16*)Buffer)[StrLen(Buffer)-1] = CHAR_NULL;\r
213\r
214 Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, StrSize(Buffer)-sizeof(CHAR16), Buffer);\r
215 if (EFI_ERROR(Status)) {\r
216 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, &Guid, VariableName, Status);\r
217 ShellStatus = SHELL_ACCESS_DENIED;\r
218 } else {\r
219 ASSERT(ShellStatus == SHELL_SUCCESS);\r
220 }\r
5d73d92f 221 }\r
222 } else if (StrnCmp(Data, L"--", 2) == 0) {\r
223 //\r
224 // device path in text format\r
225 //\r
226 Data++;\r
227 Data++;\r
863986b3 228 DevPath = ConvertTextToDevicePath(Data);\r
5d73d92f 229 if (DevPath == NULL) {\r
230 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_DPFT), gShellDebug1HiiHandle, Status);\r
231 ShellStatus = SHELL_INVALID_PARAMETER;\r
232 } else {\r
233 Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, GetDevicePathSize(DevPath), DevPath);\r
234 if (EFI_ERROR(Status)) {\r
235 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, &Guid, VariableName, Status);\r
236 ShellStatus = SHELL_ACCESS_DENIED;\r
237 } else {\r
238 ASSERT(ShellStatus == SHELL_SUCCESS);\r
239 }\r
240 }\r
241 } else {\r
242 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, Data);\r
243 ShellStatus = SHELL_INVALID_PARAMETER;\r
244 }\r
245 }\r
246 }\r
247 ShellCommandLineFreeVarList (Package);\r
248 }\r
249\r
250 if (Buffer != NULL) {\r
251 FreePool(Buffer);\r
252 }\r
253\r
254 if (DevPath != NULL) {\r
255 FreePool(DevPath);\r
256 }\r
257\r
258 return (ShellStatus);\r
259}\r
EFI Development Kit II mirror for PVE