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

/** @file\r
  Main file for SetVar shell Debug1 function.\r
\r
  (C) Copyright 2015 Hewlett-Packard Development Company, L.P.<BR>\r
  Copyright (c) 2010 - 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
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
/**\r
  Function for 'setvar' 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
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
\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, L"setvar", 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, L"setvar");  \r
      ShellStatus = SHELL_INVALID_PARAMETER;\r
    } else if (ShellCommandLineGetCount(Package) > 3) {\r
      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle, L"setvar");  \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
          ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellDebug1HiiHandle, L"setvar", StringGuid);  \r
          ShellStatus = SHELL_INVALID_PARAMETER;\r
        }\r
      }\r
      if (Data == NULL || Data[0] !=  L'=') {\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 = AllocateZeroPool(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, L"setvar", &Guid, VariableName);  \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, L"setvar", &Guid, VariableName);  \r
          ShellStatus = SHELL_ACCESS_DENIED;\r
        } else {\r
          ASSERT(ShellStatus == SHELL_SUCCESS);\r
        }\r
      } else {\r
        //\r
        // Change what's there or create a new one.\r
        //\r
\r
        ASSERT(Data[0] == L'=');\r
        Data++;\r
\r
        //\r
        // Determine if the variable exists and get the attributes\r
        //\r
        Status = gRT->GetVariable((CHAR16*)VariableName, &Guid, &Attributes, &Size, Buffer);\r
        if (Status == EFI_BUFFER_TOO_SMALL) {\r
          Buffer = AllocateZeroPool(Size);\r
          Status = gRT->GetVariable((CHAR16*)VariableName, &Guid, &Attributes, &Size, Buffer);\r
        }\r
\r
        if (EFI_ERROR(Status) || Buffer == NULL) {\r
          //\r
          // Creating a new variable.  determine attributes from command line.\r
          //\r
          Attributes = 0;\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
                          EFI_VARIABLE_BOOTSERVICE_ACCESS;\r
          }\r
          if (ShellCommandLineGetFlag(Package, L"-nv")) {\r
            Attributes |= EFI_VARIABLE_NON_VOLATILE;\r
          }\r
        }\r
        SHELL_FREE_NON_NULL(Buffer);\r
\r
        //\r
        // What type is the new data.\r
        //\r
        if (ShellIsHexOrDecimalNumber(Data, TRUE, FALSE)) {\r
          if (StrLen(Data) % 2 != 0) {\r
            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellDebug1HiiHandle, L"setvar", Data);  \r
            ShellStatus = SHELL_INVALID_PARAMETER;\r
          } else {\r
            //\r
            // arbitrary buffer\r
            //\r
            Buffer = AllocateZeroPool((StrLen(Data) / 2));\r
            if (Buffer == NULL) {\r
              Status = EFI_OUT_OF_RESOURCES;\r
            } else {\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
            }\r
            if (EFI_ERROR(Status)) {\r
              ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, L"setvar", &Guid, VariableName);  \r
              ShellStatus = SHELL_ACCESS_DENIED;\r
            } else {\r
              ASSERT(ShellStatus == SHELL_SUCCESS);\r
            }\r
          }\r
        } else if (StrnCmp(Data, L"\"", 1) == 0) {\r
          //\r
          // ascii text\r
          //\r
          Data++;\r
          Buffer = AllocateZeroPool(StrSize(Data) / 2);\r
          if (Buffer == NULL) {\r
            Status = EFI_OUT_OF_RESOURCES;\r
          } else {\r
            AsciiSPrint(Buffer, StrSize(Data) / 2, "%s", Data);\r
            ((CHAR8*)Buffer)[AsciiStrLen(Buffer)-1] = CHAR_NULL;\r
            Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, AsciiStrSize(Buffer)-sizeof(CHAR8), Buffer);\r
          }\r
          if (EFI_ERROR(Status)) {\r
            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, L"setvar", &Guid, VariableName);  \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
          if (Buffer == NULL) {\r
            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_OUT_MEM), gShellDebug1HiiHandle, L"setvar");  \r
            ShellStatus = SHELL_OUT_OF_RESOURCES;\r
          } else {\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, L"setvar", &Guid, VariableName);  \r
              ShellStatus = SHELL_ACCESS_DENIED;\r
            } else {\r
              ASSERT(ShellStatus == SHELL_SUCCESS);\r
            }\r
          }\r
        } else if (StrnCmp(Data, L"--", 2) == 0) {\r
          //\r
          // device path in text format\r
          //\r
          Data++;\r
          Data++;\r
          DevPath = ConvertTextToDevicePath(Data);\r
          if (DevPath == NULL) {\r
            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_DPFT), gShellDebug1HiiHandle, L"setvar");  \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, L"setvar", &Guid, VariableName);  \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_PARAM_INV), gShellDebug1HiiHandle, L"setvar", 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
c011b6c9	4	(C) Copyright 2015 Hewlett-Packard Development Company, L.P.<BR>\r
475387e9	5	Copyright (c) 2010 - 2014, Intel Corporation. All rights reserved.<BR>\r
5d73d92f	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
	18	STATIC 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
3737ac2b	26	/**\r
	27	Function for 'setvar' command.\r
	28	\r
	29	@param[in] ImageHandle Handle to the Image (NULL if Internal).\r
	30	@param[in] SystemTable Pointer to the System Table (NULL if Internal).\r
	31	**/\r
5d73d92f	32	SHELL_STATUS\r
	33	EFIAPI\r
	34	ShellCommandRunSetVar (\r
	35	IN EFI_HANDLE ImageHandle,\r
	36	IN EFI_SYSTEM_TABLE *SystemTable\r
	37	)\r
	38	{\r
	39	EFI_STATUS Status;\r
	40	LIST_ENTRY *Package;\r
	41	CHAR16 *ProblemParam;\r
	42	SHELL_STATUS ShellStatus;\r
	43	CONST CHAR16 *VariableName;\r
	44	CONST CHAR16 *Data;\r
	45	EFI_GUID Guid;\r
	46	CONST CHAR16 *StringGuid;\r
	47	UINT32 Attributes;\r
	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
4092a8f6	75	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, L"setvar", ProblemParam); \r
5d73d92f	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
4092a8f6	83	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle, L"setvar"); \r
5d73d92f	84	ShellStatus = SHELL_INVALID_PARAMETER;\r
5d73d92f	85	} else if (ShellCommandLineGetCount(Package) > 3) {\r
4092a8f6	86	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle, L"setvar"); \r
5d73d92f	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
4092a8f6	97	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellDebug1HiiHandle, L"setvar", StringGuid); \r
3737ac2b	98	ShellStatus = SHELL_INVALID_PARAMETER;\r
5d73d92f	99	}\r
5d73d92f	100	}\r
475387e9	101	if (Data == NULL \|\| Data[0] != L'=') {\r
5d73d92f	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
4092a8f6	117	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_GET), gShellDebug1HiiHandle, L"setvar", &Guid, VariableName); \r
5d73d92f	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
4092a8f6	126	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, L"setvar", &Guid, VariableName); \r
5d73d92f	127	ShellStatus = SHELL_ACCESS_DENIED;\r
	128	} else {\r
	129	ASSERT(ShellStatus == SHELL_SUCCESS);\r
	130	}\r
	131	} else {\r
5d73d92f	132	//\r
475387e9	133	// Change what's there or create a new one.\r
5d73d92f	134	//\r
475387e9 JC	135	\r
	136	ASSERT(Data[0] == L'=');\r
	137	Data++;\r
	138	\r
	139	//\r
	140	// Determine if the variable exists and get the attributes\r
	141	//\r
	142	Status = gRT->GetVariable((CHAR16*)VariableName, &Guid, &Attributes, &Size, Buffer);\r
	143	if (Status == EFI_BUFFER_TOO_SMALL) {\r
	144	Buffer = AllocateZeroPool(Size);\r
	145	Status = gRT->GetVariable((CHAR16*)VariableName, &Guid, &Attributes, &Size, Buffer);\r
5d73d92f	146	}\r
475387e9 JC	147	\r
	148	if (EFI_ERROR(Status) \|\| Buffer == NULL) {\r
	149	//\r
	150	// Creating a new variable. determine attributes from command line.\r
	151	//\r
	152	Attributes = 0;\r
	153	if (ShellCommandLineGetFlag(Package, L"-bs")) {\r
	154	Attributes \|= EFI_VARIABLE_BOOTSERVICE_ACCESS;\r
	155	}\r
	156	if (ShellCommandLineGetFlag(Package, L"-rt")) {\r
	157	Attributes \|= EFI_VARIABLE_RUNTIME_ACCESS \|\r
	158	EFI_VARIABLE_BOOTSERVICE_ACCESS;\r
	159	}\r
	160	if (ShellCommandLineGetFlag(Package, L"-nv")) {\r
	161	Attributes \|= EFI_VARIABLE_NON_VOLATILE;\r
	162	}\r
5d73d92f	163	}\r
475387e9 JC	164	SHELL_FREE_NON_NULL(Buffer);\r
	165	\r
	166	//\r
	167	// What type is the new data.\r
	168	//\r
5d73d92f	169	if (ShellIsHexOrDecimalNumber(Data, TRUE, FALSE)) {\r
3737ac2b	170	if (StrLen(Data) % 2 != 0) {\r
4092a8f6	171	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellDebug1HiiHandle, L"setvar", Data); \r
3737ac2b	172	ShellStatus = SHELL_INVALID_PARAMETER;\r
5d73d92f	173	} else {\r
3737ac2b	174	//\r
	175	// arbitrary buffer\r
	176	//\r
	177	Buffer = AllocateZeroPool((StrLen(Data) / 2));\r
33c031ee	178	if (Buffer == NULL) {\r
	179	Status = EFI_OUT_OF_RESOURCES;\r
	180	} else {\r
	181	for (LoopVar = 0 ; LoopVar < (StrLen(Data) / 2) ; LoopVar++) {\r
	182	((UINT8)Buffer)[LoopVar] = (UINT8)(HexCharToUintn(Data[LoopVar2]) * 16);\r
	183	((UINT8)Buffer)[LoopVar] = (UINT8)(((UINT8)Buffer)[LoopVar] + HexCharToUintn(Data[LoopVar*2+1]));\r
	184	}\r
	185	Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, StrLen(Data) / 2, Buffer);\r
3737ac2b	186	}\r
3737ac2b	187	if (EFI_ERROR(Status)) {\r
4092a8f6	188	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, L"setvar", &Guid, VariableName); \r
3737ac2b	189	ShellStatus = SHELL_ACCESS_DENIED;\r
	190	} else {\r
	191	ASSERT(ShellStatus == SHELL_SUCCESS);\r
	192	}\r
5d73d92f	193	}\r
	194	} else if (StrnCmp(Data, L"\"", 1) == 0) {\r
	195	//\r
	196	// ascii text\r
	197	//\r
	198	Data++;\r
	199	Buffer = AllocateZeroPool(StrSize(Data) / 2);\r
33c031ee	200	if (Buffer == NULL) {\r
	201	Status = EFI_OUT_OF_RESOURCES;\r
	202	} else {\r
	203	AsciiSPrint(Buffer, StrSize(Data) / 2, "%s", Data);\r
	204	((CHAR8*)Buffer)[AsciiStrLen(Buffer)-1] = CHAR_NULL;\r
	205	Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, AsciiStrSize(Buffer)-sizeof(CHAR8), Buffer);\r
	206	}\r
5d73d92f	207	if (EFI_ERROR(Status)) {\r
4092a8f6	208	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, L"setvar", &Guid, VariableName); \r
5d73d92f	209	ShellStatus = SHELL_ACCESS_DENIED;\r
	210	} else {\r
	211	ASSERT(ShellStatus == SHELL_SUCCESS);\r
	212	}\r
	213	} else if (StrnCmp(Data, L"L\"", 2) == 0) {\r
	214	//\r
	215	// ucs2 text\r
	216	//\r
	217	Data++;\r
	218	Data++;\r
	219	Buffer = AllocateZeroPool(StrSize(Data));\r
cbdd109b	220	if (Buffer == NULL) {\r
4092a8f6	221	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_OUT_MEM), gShellDebug1HiiHandle, L"setvar"); \r
cbdd109b	222	ShellStatus = SHELL_OUT_OF_RESOURCES;\r
5d73d92f	223	} else {\r
cbdd109b	224	UnicodeSPrint(Buffer, StrSize(Data), L"%s", Data);\r
	225	((CHAR16*)Buffer)[StrLen(Buffer)-1] = CHAR_NULL;\r
	226	\r
	227	Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, StrSize(Buffer)-sizeof(CHAR16), Buffer);\r
	228	if (EFI_ERROR(Status)) {\r
4092a8f6	229	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, L"setvar", &Guid, VariableName); \r
cbdd109b	230	ShellStatus = SHELL_ACCESS_DENIED;\r
	231	} else {\r
	232	ASSERT(ShellStatus == SHELL_SUCCESS);\r
	233	}\r
5d73d92f	234	}\r
	235	} else if (StrnCmp(Data, L"--", 2) == 0) {\r
	236	//\r
	237	// device path in text format\r
	238	//\r
	239	Data++;\r
	240	Data++;\r
863986b3	241	DevPath = ConvertTextToDevicePath(Data);\r
5d73d92f	242	if (DevPath == NULL) {\r
4092a8f6	243	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_DPFT), gShellDebug1HiiHandle, L"setvar"); \r
5d73d92f	244	ShellStatus = SHELL_INVALID_PARAMETER;\r
	245	} else {\r
	246	Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, GetDevicePathSize(DevPath), DevPath);\r
	247	if (EFI_ERROR(Status)) {\r
4092a8f6	248	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, L"setvar", &Guid, VariableName); \r
5d73d92f	249	ShellStatus = SHELL_ACCESS_DENIED;\r
	250	} else {\r
	251	ASSERT(ShellStatus == SHELL_SUCCESS);\r
	252	}\r
	253	}\r
	254	} else {\r
4092a8f6	255	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellDebug1HiiHandle, L"setvar", Data); \r
5d73d92f	256	ShellStatus = SHELL_INVALID_PARAMETER;\r
	257	}\r
	258	}\r
	259	}\r
	260	ShellCommandLineFreeVarList (Package);\r
	261	}\r
	262	\r
	263	if (Buffer != NULL) {\r
	264	FreePool(Buffer);\r
	265	}\r
	266	\r
	267	if (DevPath != NULL) {\r
	268	FreePool(DevPath);\r
	269	}\r
	270	\r
	271	return (ShellStatus);\r
	272	}\r