[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
/**\r
  Check if the input is a (potentially empty) string of hexadecimal nibbles.\r
\r
  @param[in] String  The CHAR16 string to check.\r
\r
  @retval FALSE  A character has been found in String for which\r
                 ShellIsHexaDecimalDigitCharacter() returned FALSE.\r
\r
  @retval TRUE   Otherwise. (Note that this covers the case when String is\r
                 empty.)\r
**/\r
BOOLEAN\r
EFIAPI\r
IsStringOfHexNibbles (\r
  IN CONST CHAR16  *String\r
  )\r
{\r
  CONST CHAR16 *Pos;\r
\r
  for (Pos = String; *Pos != L'\0'; ++Pos) {\r
    if (!ShellIsHexaDecimalDigitCharacter (*Pos)) {\r
      return FALSE;\r
    }\r
  }\r
  return TRUE;\r
}\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
        ASSERT(Data[0] != L'\0');\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 (IsStringOfHexNibbles(Data)) {\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
bbf57d4f LE	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
	38	BOOLEAN\r
	39	EFIAPI\r
	40	IsStringOfHexNibbles (\r
	41	IN CONST CHAR16 *String\r
	42	)\r
	43	{\r
	44	CONST CHAR16 *Pos;\r
	45	\r
	46	for (Pos = String; *Pos != L'\0'; ++Pos) {\r
	47	if (!ShellIsHexaDecimalDigitCharacter (*Pos)) {\r
	48	return FALSE;\r
	49	}\r
	50	}\r
	51	return TRUE;\r
	52	}\r
	53	\r
	54	\r
3737ac2b	55	/**\r
	56	Function for 'setvar' command.\r
	57	\r
	58	@param[in] ImageHandle Handle to the Image (NULL if Internal).\r
	59	@param[in] SystemTable Pointer to the System Table (NULL if Internal).\r
	60	**/\r
5d73d92f	61	SHELL_STATUS\r
	62	EFIAPI\r
	63	ShellCommandRunSetVar (\r
	64	IN EFI_HANDLE ImageHandle,\r
	65	IN EFI_SYSTEM_TABLE *SystemTable\r
	66	)\r
	67	{\r
	68	EFI_STATUS Status;\r
	69	LIST_ENTRY *Package;\r
	70	CHAR16 *ProblemParam;\r
	71	SHELL_STATUS ShellStatus;\r
	72	CONST CHAR16 *VariableName;\r
	73	CONST CHAR16 *Data;\r
	74	EFI_GUID Guid;\r
	75	CONST CHAR16 *StringGuid;\r
	76	UINT32 Attributes;\r
	77	VOID *Buffer;\r
	78	UINTN Size;\r
	79	UINTN LoopVar;\r
	80	EFI_DEVICE_PATH_PROTOCOL *DevPath;\r
5d73d92f	81	\r
	82	ShellStatus = SHELL_SUCCESS;\r
	83	Status = EFI_SUCCESS;\r
	84	Buffer = NULL;\r
	85	Size = 0;\r
	86	Attributes = 0;\r
	87	DevPath = NULL;\r
	88	\r
	89	//\r
	90	// initialize the shell lib (we must be in non-auto-init...)\r
	91	//\r
	92	Status = ShellInitialize();\r
	93	ASSERT_EFI_ERROR(Status);\r
	94	\r
	95	Status = CommandInit();\r
	96	ASSERT_EFI_ERROR(Status);\r
	97	\r
	98	//\r
	99	// parse the command line\r
	100	//\r
	101	Status = ShellCommandLineParse (ParamList, &Package, &ProblemParam, TRUE);\r
	102	if (EFI_ERROR(Status)) {\r
	103	if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {\r
4092a8f6	104	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, L"setvar", ProblemParam); \r
5d73d92f	105	FreePool(ProblemParam);\r
	106	ShellStatus = SHELL_INVALID_PARAMETER;\r
	107	} else {\r
	108	ASSERT(FALSE);\r
	109	}\r
	110	} else {\r
	111	if (ShellCommandLineGetCount(Package) < 2) {\r
4092a8f6	112	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle, L"setvar"); \r
5d73d92f	113	ShellStatus = SHELL_INVALID_PARAMETER;\r
5d73d92f	114	} else if (ShellCommandLineGetCount(Package) > 3) {\r
4092a8f6	115	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle, L"setvar"); \r
5d73d92f	116	ShellStatus = SHELL_INVALID_PARAMETER;\r
	117	} else {\r
	118	VariableName = ShellCommandLineGetRawValue(Package, 1);\r
	119	Data = ShellCommandLineGetRawValue(Package, 2);\r
	120	if (!ShellCommandLineGetFlag(Package, L"-guid")){\r
	121	CopyGuid(&Guid, &gEfiGlobalVariableGuid);\r
	122	} else {\r
	123	StringGuid = ShellCommandLineGetValue(Package, L"-guid");\r
	124	Status = ConvertStringToGuid(StringGuid, &Guid);\r
	125	if (EFI_ERROR(Status)) {\r
4092a8f6	126	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellDebug1HiiHandle, L"setvar", StringGuid); \r
3737ac2b	127	ShellStatus = SHELL_INVALID_PARAMETER;\r
5d73d92f	128	}\r
5d73d92f	129	}\r
475387e9	130	if (Data == NULL \|\| Data[0] != L'=') {\r
5d73d92f	131	//\r
	132	// Display what's there\r
	133	//\r
	134	Status = gRT->GetVariable((CHAR16*)VariableName, &Guid, &Attributes, &Size, Buffer);\r
	135	if (Status == EFI_BUFFER_TOO_SMALL) {\r
3737ac2b	136	Buffer = AllocateZeroPool(Size);\r
5d73d92f	137	Status = gRT->GetVariable((CHAR16*)VariableName, &Guid, &Attributes, &Size, Buffer);\r
	138	}\r
	139	if (!EFI_ERROR(Status)&& Buffer != NULL) {\r
	140	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_PRINT), gShellDebug1HiiHandle, &Guid, VariableName, Size);\r
	141	for (LoopVar = 0 ; LoopVar < Size ; LoopVar++) {\r
	142	ShellPrintEx(-1, -1, L"%02x ", ((UINT8*)Buffer)[LoopVar]);\r
	143	}\r
	144	ShellPrintEx(-1, -1, L"\r\n");\r
	145	} else {\r
4092a8f6	146	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_GET), gShellDebug1HiiHandle, L"setvar", &Guid, VariableName); \r
5d73d92f	147	ShellStatus = SHELL_ACCESS_DENIED;\r
	148	}\r
	149	} else if (StrCmp(Data, L"=") == 0) {\r
	150	//\r
	151	// Delete what's there!\r
	152	//\r
	153	Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, 0, NULL);\r
	154	if (EFI_ERROR(Status)) {\r
4092a8f6	155	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, L"setvar", &Guid, VariableName); \r
5d73d92f	156	ShellStatus = SHELL_ACCESS_DENIED;\r
	157	} else {\r
	158	ASSERT(ShellStatus == SHELL_SUCCESS);\r
	159	}\r
	160	} else {\r
5d73d92f	161	//\r
475387e9	162	// Change what's there or create a new one.\r
5d73d92f	163	//\r
475387e9 JC	164	\r
	165	ASSERT(Data[0] == L'=');\r
	166	Data++;\r
bbf57d4f	167	ASSERT(Data[0] != L'\0');\r
475387e9 JC	168	\r
	169	//\r
	170	// Determine if the variable exists and get the attributes\r
	171	//\r
	172	Status = gRT->GetVariable((CHAR16*)VariableName, &Guid, &Attributes, &Size, Buffer);\r
	173	if (Status == EFI_BUFFER_TOO_SMALL) {\r
	174	Buffer = AllocateZeroPool(Size);\r
	175	Status = gRT->GetVariable((CHAR16*)VariableName, &Guid, &Attributes, &Size, Buffer);\r
5d73d92f	176	}\r
475387e9 JC	177	\r
	178	if (EFI_ERROR(Status) \|\| Buffer == NULL) {\r
	179	//\r
	180	// Creating a new variable. determine attributes from command line.\r
	181	//\r
	182	Attributes = 0;\r
	183	if (ShellCommandLineGetFlag(Package, L"-bs")) {\r
	184	Attributes \|= EFI_VARIABLE_BOOTSERVICE_ACCESS;\r
	185	}\r
	186	if (ShellCommandLineGetFlag(Package, L"-rt")) {\r
	187	Attributes \|= EFI_VARIABLE_RUNTIME_ACCESS \|\r
	188	EFI_VARIABLE_BOOTSERVICE_ACCESS;\r
	189	}\r
	190	if (ShellCommandLineGetFlag(Package, L"-nv")) {\r
	191	Attributes \|= EFI_VARIABLE_NON_VOLATILE;\r
	192	}\r
5d73d92f	193	}\r
475387e9 JC	194	SHELL_FREE_NON_NULL(Buffer);\r
	195	\r
	196	//\r
	197	// What type is the new data.\r
	198	//\r
bbf57d4f	199	if (IsStringOfHexNibbles(Data)) {\r
3737ac2b	200	if (StrLen(Data) % 2 != 0) {\r
4092a8f6	201	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellDebug1HiiHandle, L"setvar", Data); \r
3737ac2b	202	ShellStatus = SHELL_INVALID_PARAMETER;\r
5d73d92f	203	} else {\r
3737ac2b	204	//\r
	205	// arbitrary buffer\r
	206	//\r
	207	Buffer = AllocateZeroPool((StrLen(Data) / 2));\r
33c031ee	208	if (Buffer == NULL) {\r
	209	Status = EFI_OUT_OF_RESOURCES;\r
	210	} else {\r
	211	for (LoopVar = 0 ; LoopVar < (StrLen(Data) / 2) ; LoopVar++) {\r
	212	((UINT8)Buffer)[LoopVar] = (UINT8)(HexCharToUintn(Data[LoopVar2]) * 16);\r
	213	((UINT8)Buffer)[LoopVar] = (UINT8)(((UINT8)Buffer)[LoopVar] + HexCharToUintn(Data[LoopVar*2+1]));\r
	214	}\r
	215	Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, StrLen(Data) / 2, Buffer);\r
3737ac2b	216	}\r
3737ac2b	217	if (EFI_ERROR(Status)) {\r
4092a8f6	218	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, L"setvar", &Guid, VariableName); \r
3737ac2b	219	ShellStatus = SHELL_ACCESS_DENIED;\r
	220	} else {\r
	221	ASSERT(ShellStatus == SHELL_SUCCESS);\r
	222	}\r
5d73d92f	223	}\r
	224	} else if (StrnCmp(Data, L"\"", 1) == 0) {\r
	225	//\r
	226	// ascii text\r
	227	//\r
	228	Data++;\r
	229	Buffer = AllocateZeroPool(StrSize(Data) / 2);\r
33c031ee	230	if (Buffer == NULL) {\r
	231	Status = EFI_OUT_OF_RESOURCES;\r
	232	} else {\r
	233	AsciiSPrint(Buffer, StrSize(Data) / 2, "%s", Data);\r
	234	((CHAR8*)Buffer)[AsciiStrLen(Buffer)-1] = CHAR_NULL;\r
	235	Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, AsciiStrSize(Buffer)-sizeof(CHAR8), Buffer);\r
	236	}\r
5d73d92f	237	if (EFI_ERROR(Status)) {\r
4092a8f6	238	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, L"setvar", &Guid, VariableName); \r
5d73d92f	239	ShellStatus = SHELL_ACCESS_DENIED;\r
	240	} else {\r
	241	ASSERT(ShellStatus == SHELL_SUCCESS);\r
	242	}\r
	243	} else if (StrnCmp(Data, L"L\"", 2) == 0) {\r
	244	//\r
	245	// ucs2 text\r
	246	//\r
	247	Data++;\r
	248	Data++;\r
	249	Buffer = AllocateZeroPool(StrSize(Data));\r
cbdd109b	250	if (Buffer == NULL) {\r
4092a8f6	251	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_OUT_MEM), gShellDebug1HiiHandle, L"setvar"); \r
cbdd109b	252	ShellStatus = SHELL_OUT_OF_RESOURCES;\r
5d73d92f	253	} else {\r
cbdd109b	254	UnicodeSPrint(Buffer, StrSize(Data), L"%s", Data);\r
	255	((CHAR16*)Buffer)[StrLen(Buffer)-1] = CHAR_NULL;\r
	256	\r
	257	Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, StrSize(Buffer)-sizeof(CHAR16), Buffer);\r
	258	if (EFI_ERROR(Status)) {\r
4092a8f6	259	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, L"setvar", &Guid, VariableName); \r
cbdd109b	260	ShellStatus = SHELL_ACCESS_DENIED;\r
	261	} else {\r
	262	ASSERT(ShellStatus == SHELL_SUCCESS);\r
	263	}\r
5d73d92f	264	}\r
	265	} else if (StrnCmp(Data, L"--", 2) == 0) {\r
	266	//\r
	267	// device path in text format\r
	268	//\r
	269	Data++;\r
	270	Data++;\r
863986b3	271	DevPath = ConvertTextToDevicePath(Data);\r
5d73d92f	272	if (DevPath == NULL) {\r
4092a8f6	273	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_DPFT), gShellDebug1HiiHandle, L"setvar"); \r
5d73d92f	274	ShellStatus = SHELL_INVALID_PARAMETER;\r
	275	} else {\r
	276	Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, GetDevicePathSize(DevPath), DevPath);\r
	277	if (EFI_ERROR(Status)) {\r
4092a8f6	278	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, L"setvar", &Guid, VariableName); \r
5d73d92f	279	ShellStatus = SHELL_ACCESS_DENIED;\r
	280	} else {\r
	281	ASSERT(ShellStatus == SHELL_SUCCESS);\r
	282	}\r
	283	}\r
	284	} else {\r
4092a8f6	285	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellDebug1HiiHandle, L"setvar", Data); \r
5d73d92f	286	ShellStatus = SHELL_INVALID_PARAMETER;\r
	287	}\r
	288	}\r
	289	}\r
	290	ShellCommandLineFreeVarList (Package);\r
	291	}\r
	292	\r
	293	if (Buffer != NULL) {\r
	294	FreePool(Buffer);\r
	295	}\r
	296	\r
	297	if (DevPath != NULL) {\r
	298	FreePool(DevPath);\r
	299	}\r
	300	\r
	301	return (ShellStatus);\r
	302	}\r