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

/** @file\r
  Main file for SetVar shell Debug1 function.\r
\r
  Copyright (c) 2010, 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
STATIC CONST SHELL_PARAM_ITEM ParamList[] = {\r
  {L"-guid", TypeValue},\r
  {L"-bs", TypeFlag},\r
  {L"-rt", TypeFlag},\r
  {L"-nv", TypeFlag},\r
  {NULL, TypeMax}\r
  };\r
\r
SHELL_STATUS\r
EFIAPI\r
ShellCommandRunSetVar (\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        *VariableName;\r
  CONST CHAR16        *Data;\r
  EFI_GUID            Guid;\r
  CONST CHAR16        *StringGuid;\r
  UINT32              Attributes;\r
  VOID                *Buffer;\r
  UINTN               Size;\r
  UINTN               LoopVar;\r
  EFI_DEVICE_PATH_PROTOCOL           *DevPath;\r
  EFI_DEVICE_PATH_FROM_TEXT_PROTOCOL *DevPathFromText;\r
\r
  ShellStatus         = SHELL_SUCCESS;\r
  Status              = EFI_SUCCESS;\r
  Buffer              = NULL;\r
  Size                = 0;\r
  Attributes          = 0;\r
  DevPath             = NULL;\r
\r
  //\r
  // initialize the shell lib (we must be in non-auto-init...)\r
  //\r
  Status = ShellInitialize();\r
  ASSERT_EFI_ERROR(Status);\r
\r
  Status = CommandInit();\r
  ASSERT_EFI_ERROR(Status);\r
\r
  //\r
  // parse the command line\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_FEW), gShellDebug1HiiHandle);\r
      ShellStatus = SHELL_INVALID_PARAMETER;\r
    } else if (ShellCommandLineGetCount(Package) > 3) {\r
      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle);\r
      ShellStatus = SHELL_INVALID_PARAMETER;\r
    } else {\r
      VariableName  = ShellCommandLineGetRawValue(Package, 1);\r
      Data          = ShellCommandLineGetRawValue(Package, 2);\r
      if (!ShellCommandLineGetFlag(Package, L"-guid")){\r
        CopyGuid(&Guid, &gEfiGlobalVariableGuid);\r
      } else {\r
        StringGuid = ShellCommandLineGetValue(Package, L"-guid");\r
        Status = ConvertStringToGuid(StringGuid, &Guid);\r
        if (EFI_ERROR(Status)) {\r
          ShellStatus = SHELL_NOT_FOUND;\r
        }\r
      }\r
      if (Data == NULL) {\r
        //\r
        // Display what's there\r
        //\r
        Status = gRT->GetVariable((CHAR16*)VariableName, &Guid, &Attributes, &Size, Buffer);\r
        if (Status == EFI_BUFFER_TOO_SMALL) {\r
          Buffer = AllocatePool(Size);\r
          Status = gRT->GetVariable((CHAR16*)VariableName, &Guid, &Attributes, &Size, Buffer);\r
        }\r
        if (!EFI_ERROR(Status)&& Buffer != NULL) {\r
          ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_PRINT), gShellDebug1HiiHandle, &Guid, VariableName, Size);\r
          for (LoopVar = 0 ; LoopVar < Size ; LoopVar++) {\r
            ShellPrintEx(-1, -1, L"%02x ", ((UINT8*)Buffer)[LoopVar]);\r
          }\r
          ShellPrintEx(-1, -1, L"\r\n");\r
        } else {\r
          ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_GET), gShellDebug1HiiHandle, &Guid, VariableName, Status);\r
          ShellStatus = SHELL_ACCESS_DENIED;\r
        }\r
      } else if (StrCmp(Data, L"=") == 0) {\r
        //\r
        // Delete what's there!\r
        //\r
        Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, 0, NULL);\r
        if (EFI_ERROR(Status)) {\r
          ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, &Guid, VariableName, Status);\r
          ShellStatus = SHELL_ACCESS_DENIED;\r
        } else {\r
          ASSERT(ShellStatus == SHELL_SUCCESS);\r
        }\r
      } else {\r
        if (Data[0] == L'=') {\r
          Data++;\r
        }\r
        //\r
        // Change what's there\r
        //\r
        if (ShellCommandLineGetFlag(Package, L"-bs")) {\r
          Attributes |= EFI_VARIABLE_BOOTSERVICE_ACCESS;\r
        }\r
        if (ShellCommandLineGetFlag(Package, L"-rt")) {\r
          Attributes |= EFI_VARIABLE_RUNTIME_ACCESS;\r
        }\r
        if (ShellCommandLineGetFlag(Package, L"-nv")) {\r
          Attributes |= EFI_VARIABLE_NON_VOLATILE;\r
        }\r
        if (ShellIsHexOrDecimalNumber(Data, TRUE, FALSE)) {\r
          //\r
          // arbitrary buffer\r
          //\r
          Buffer = AllocateZeroPool((StrLen(Data) / 2));\r
          for (LoopVar = 0 ; LoopVar < (StrLen(Data) / 2) ; LoopVar++) {\r
            ((UINT8*)Buffer)[LoopVar] = (UINT8)(HexCharToUintn(Data[LoopVar*2]) * 16);\r
            ((UINT8*)Buffer)[LoopVar] = (UINT8)(((UINT8*)Buffer)[LoopVar] + HexCharToUintn(Data[LoopVar*2+1]));\r
          }\r
          Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, StrLen(Data) / 2, Buffer);\r
          if (EFI_ERROR(Status)) {\r
            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, &Guid, VariableName, Status);\r
            ShellStatus = SHELL_ACCESS_DENIED;\r
          } else {\r
            ASSERT(ShellStatus == SHELL_SUCCESS);\r
          }\r
        } else if (StrnCmp(Data, L"\"", 1) == 0) {\r
          //\r
          // ascii text\r
          //\r
          Data++;\r
          Buffer = AllocateZeroPool(StrSize(Data) / 2);\r
          AsciiSPrint(Buffer, StrSize(Data) / 2, "%s", Data);\r
          ((CHAR8*)Buffer)[AsciiStrLen(Buffer)-1] = CHAR_NULL;\r
\r
          Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, AsciiStrSize(Buffer)-sizeof(CHAR8), Buffer);\r
          if (EFI_ERROR(Status)) {\r
            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, &Guid, VariableName, Status);\r
            ShellStatus = SHELL_ACCESS_DENIED;\r
          } else {\r
            ASSERT(ShellStatus == SHELL_SUCCESS);\r
          }\r
        } else if (StrnCmp(Data, L"L\"", 2) == 0) {\r
          //\r
          // ucs2 text\r
          //\r
          Data++;\r
          Data++;\r
          Buffer = AllocateZeroPool(StrSize(Data));\r
          UnicodeSPrint(Buffer, StrSize(Data), L"%s", Data);\r
          ((CHAR16*)Buffer)[StrLen(Buffer)-1] = CHAR_NULL;\r
\r
          Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, StrSize(Buffer)-sizeof(CHAR16), Buffer);\r
          if (EFI_ERROR(Status)) {\r
            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, &Guid, VariableName, Status);\r
            ShellStatus = SHELL_ACCESS_DENIED;\r
          } else {\r
            ASSERT(ShellStatus == SHELL_SUCCESS);\r
          }\r
        } else if (StrnCmp(Data, L"--", 2) == 0) {\r
          //\r
          // device path in text format\r
          //\r
          Data++;\r
          Data++;\r
          Status = gBS->LocateProtocol(&gEfiDevicePathFromTextProtocolGuid, NULL, (VOID**)&DevPathFromText);\r
          ASSERT_EFI_ERROR(Status);\r
          DevPath = DevPathFromText->ConvertTextToDevicePath(Data);\r
          if (DevPath == NULL) {\r
            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_DPFT), gShellDebug1HiiHandle, Status);\r
            ShellStatus = SHELL_INVALID_PARAMETER;\r
          } else {\r
            Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, GetDevicePathSize(DevPath), DevPath);\r
            if (EFI_ERROR(Status)) {\r
              ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, &Guid, VariableName, Status);\r
              ShellStatus = SHELL_ACCESS_DENIED;\r
            } else {\r
              ASSERT(ShellStatus == SHELL_SUCCESS);\r
            }\r
          }\r
        } else {\r
          ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, Data);\r
          ShellStatus = SHELL_INVALID_PARAMETER;\r
        }\r
      }\r
    }\r
    ShellCommandLineFreeVarList (Package);\r
  }\r
\r
  if (Buffer != NULL) {\r
    FreePool(Buffer);\r
  }\r
\r
  if (DevPath != NULL) {\r
    FreePool(DevPath);\r
  }\r
\r
  return (ShellStatus);\r
}\r
Commit	Line	Data
5d73d92f	1	/** @file\r
	2	Main file for SetVar shell Debug1 function.\r
	3	\r
	4	Copyright (c) 2010, Intel Corporation. All rights reserved.<BR>\r
	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
	17	STATIC 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
	25	SHELL_STATUS\r
	26	EFIAPI\r
	27	ShellCommandRunSetVar (\r
	28	IN EFI_HANDLE ImageHandle,\r
	29	IN EFI_SYSTEM_TABLE *SystemTable\r
	30	)\r
	31	{\r
	32	EFI_STATUS Status;\r
	33	LIST_ENTRY *Package;\r
	34	CHAR16 *ProblemParam;\r
	35	SHELL_STATUS ShellStatus;\r
	36	CONST CHAR16 *VariableName;\r
	37	CONST CHAR16 *Data;\r
	38	EFI_GUID Guid;\r
	39	CONST CHAR16 *StringGuid;\r
	40	UINT32 Attributes;\r
	41	VOID *Buffer;\r
	42	UINTN Size;\r
	43	UINTN LoopVar;\r
	44	EFI_DEVICE_PATH_PROTOCOL *DevPath;\r
	45	EFI_DEVICE_PATH_FROM_TEXT_PROTOCOL *DevPathFromText;\r
	46	\r
	47	ShellStatus = SHELL_SUCCESS;\r
	48	Status = EFI_SUCCESS;\r
	49	Buffer = NULL;\r
	50	Size = 0;\r
	51	Attributes = 0;\r
	52	DevPath = NULL;\r
	53	\r
	54	//\r
	55	// initialize the shell lib (we must be in non-auto-init...)\r
	56	//\r
	57	Status = ShellInitialize();\r
	58	ASSERT_EFI_ERROR(Status);\r
	59	\r
	60	Status = CommandInit();\r
	61	ASSERT_EFI_ERROR(Status);\r
	62	\r
	63	//\r
	64	// parse the command line\r
65	//\r
66	Status = ShellCommandLineParse (ParamList, &Package, &ProblemParam, TRUE);\r
67	if (EFI_ERROR(Status)) {\r
68	if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {\r
69	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, ProblemParam);\r
70	FreePool(ProblemParam);\r
71	ShellStatus = SHELL_INVALID_PARAMETER;\r
72	} else {\r
73	ASSERT(FALSE);\r
74	}\r
75	} else {\r
76	if (ShellCommandLineGetCount(Package) < 2) {\r
77	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle);\r
78	ShellStatus = SHELL_INVALID_PARAMETER;\r
79	} else if (ShellCommandLineGetCount(Package) > 3) {\r
80	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle);\r
81	ShellStatus = SHELL_INVALID_PARAMETER;\r
82	} else {\r
83	VariableName = ShellCommandLineGetRawValue(Package, 1);\r
84	Data = ShellCommandLineGetRawValue(Package, 2);\r
85	if (!ShellCommandLineGetFlag(Package, L"-guid")){\r
86	CopyGuid(&Guid, &gEfiGlobalVariableGuid);\r
87	} else {\r
88	StringGuid = ShellCommandLineGetValue(Package, L"-guid");\r
89	Status = ConvertStringToGuid(StringGuid, &Guid);\r
90	if (EFI_ERROR(Status)) {\r
91	ShellStatus = SHELL_NOT_FOUND;\r
92	}\r
93	}\r
94	if (Data == NULL) {\r
95	//\r
96	// Display what's there\r
97	//\r
98	Status = gRT->GetVariable((CHAR16*)VariableName, &Guid, &Attributes, &Size, Buffer);\r
99	if (Status == EFI_BUFFER_TOO_SMALL) {\r
100	Buffer = AllocatePool(Size);\r
101	Status = gRT->GetVariable((CHAR16*)VariableName, &Guid, &Attributes, &Size, Buffer);\r
102	}\r
103	if (!EFI_ERROR(Status)&& Buffer != NULL) {\r
104	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_PRINT), gShellDebug1HiiHandle, &Guid, VariableName, Size);\r
105	for (LoopVar = 0 ; LoopVar < Size ; LoopVar++) {\r
106	ShellPrintEx(-1, -1, L"%02x ", ((UINT8*)Buffer)[LoopVar]);\r
107	}\r
108	ShellPrintEx(-1, -1, L"\r\n");\r
109	} else {\r
110	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_GET), gShellDebug1HiiHandle, &Guid, VariableName, Status);\r
111	ShellStatus = SHELL_ACCESS_DENIED;\r
112	}\r
113	} else if (StrCmp(Data, L"=") == 0) {\r
114	//\r
115	// Delete what's there!\r
116	//\r
117	Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, 0, NULL);\r
118	if (EFI_ERROR(Status)) {\r
119	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, &Guid, VariableName, Status);\r
120	ShellStatus = SHELL_ACCESS_DENIED;\r
121	} else {\r
122	ASSERT(ShellStatus == SHELL_SUCCESS);\r
123	}\r
124	} else {\r
125	if (Data[0] == L'=') {\r
126	Data++;\r
127	}\r
128	//\r
129	// Change what's there\r
130	//\r
131	if (ShellCommandLineGetFlag(Package, L"-bs")) {\r
132	Attributes \|= EFI_VARIABLE_BOOTSERVICE_ACCESS;\r
133	}\r
134	if (ShellCommandLineGetFlag(Package, L"-rt")) {\r
135	Attributes \|= EFI_VARIABLE_RUNTIME_ACCESS;\r
136	}\r
137	if (ShellCommandLineGetFlag(Package, L"-nv")) {\r
138	Attributes \|= EFI_VARIABLE_NON_VOLATILE;\r
139	}\r
140	if (ShellIsHexOrDecimalNumber(Data, TRUE, FALSE)) {\r
141	//\r
142	// arbitrary buffer\r
143	//\r
144	Buffer = AllocateZeroPool((StrLen(Data) / 2));\r
145	for (LoopVar = 0 ; LoopVar < (StrLen(Data) / 2) ; LoopVar++) {\r
146	((UINT8)Buffer)[LoopVar] = (UINT8)(HexCharToUintn(Data[LoopVar2]) * 16);\r
147	((UINT8)Buffer)[LoopVar] = (UINT8)(((UINT8)Buffer)[LoopVar] + HexCharToUintn(Data[LoopVar*2+1]));\r
148	}\r
149	Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, StrLen(Data) / 2, Buffer);\r
150	if (EFI_ERROR(Status)) {\r
151	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, &Guid, VariableName, Status);\r
152	ShellStatus = SHELL_ACCESS_DENIED;\r
153	} else {\r
154	ASSERT(ShellStatus == SHELL_SUCCESS);\r
155	}\r
156	} else if (StrnCmp(Data, L"\"", 1) == 0) {\r
157	//\r
158	// ascii text\r
159	//\r
160	Data++;\r
161	Buffer = AllocateZeroPool(StrSize(Data) / 2);\r
162	AsciiSPrint(Buffer, StrSize(Data) / 2, "%s", Data);\r
163	((CHAR8*)Buffer)[AsciiStrLen(Buffer)-1] = CHAR_NULL;\r
164	\r
165	Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, AsciiStrSize(Buffer)-sizeof(CHAR8), Buffer);\r
166	if (EFI_ERROR(Status)) {\r
167	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, &Guid, VariableName, Status);\r
168	ShellStatus = SHELL_ACCESS_DENIED;\r
169	} else {\r
170	ASSERT(ShellStatus == SHELL_SUCCESS);\r
171	}\r
172	} else if (StrnCmp(Data, L"L\"", 2) == 0) {\r
173	//\r
174	// ucs2 text\r
175	//\r
176	Data++;\r
177	Data++;\r
178	Buffer = AllocateZeroPool(StrSize(Data));\r
179	UnicodeSPrint(Buffer, StrSize(Data), L"%s", Data);\r
180	((CHAR16*)Buffer)[StrLen(Buffer)-1] = CHAR_NULL;\r
181	\r
182	Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, StrSize(Buffer)-sizeof(CHAR16), Buffer);\r
183	if (EFI_ERROR(Status)) {\r
184	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, &Guid, VariableName, Status);\r
185	ShellStatus = SHELL_ACCESS_DENIED;\r
186	} else {\r
187	ASSERT(ShellStatus == SHELL_SUCCESS);\r
188	}\r
189	} else if (StrnCmp(Data, L"--", 2) == 0) {\r
190	//\r
191	// device path in text format\r
192	//\r
193	Data++;\r
194	Data++;\r
195	Status = gBS->LocateProtocol(&gEfiDevicePathFromTextProtocolGuid, NULL, (VOID**)&DevPathFromText);\r
196	ASSERT_EFI_ERROR(Status);\r
197	DevPath = DevPathFromText->ConvertTextToDevicePath(Data);\r
198	if (DevPath == NULL) {\r
199	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_DPFT), gShellDebug1HiiHandle, Status);\r
200	ShellStatus = SHELL_INVALID_PARAMETER;\r
201	} else {\r
202	Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, GetDevicePathSize(DevPath), DevPath);\r
203	if (EFI_ERROR(Status)) {\r
204	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, &Guid, VariableName, Status);\r
205	ShellStatus = SHELL_ACCESS_DENIED;\r
206	} else {\r
207	ASSERT(ShellStatus == SHELL_SUCCESS);\r
208	}\r
209	}\r
210	} else {\r
211	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, Data);\r
212	ShellStatus = SHELL_INVALID_PARAMETER;\r
213	}\r
214	}\r
215	}\r
216	ShellCommandLineFreeVarList (Package);\r
217	}\r
218	\r
219	if (Buffer != NULL) {\r
220	FreePool(Buffer);\r
221	}\r
222	\r
223	if (DevPath != NULL) {\r
224	FreePool(DevPath);\r
225	}\r
226	\r
227	return (ShellStatus);\r
228	}\r