[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
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
bbf57d4f LE	39	IsStringOfHexNibbles (\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
3737ac2b	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
5d73d92f	60	SHELL_STATUS\r
	61	EFIAPI\r
	62	ShellCommandRunSetVar (\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
5d73d92f	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
4092a8f6	103	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, L"setvar", ProblemParam); \r
5d73d92f	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
4092a8f6	111	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle, L"setvar"); \r
5d73d92f	112	ShellStatus = SHELL_INVALID_PARAMETER;\r
5d73d92f	113	} else if (ShellCommandLineGetCount(Package) > 3) {\r
4092a8f6	114	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle, L"setvar"); \r
5d73d92f	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
4092a8f6	125	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellDebug1HiiHandle, L"setvar", StringGuid); \r
3737ac2b	126	ShellStatus = SHELL_INVALID_PARAMETER;\r
5d73d92f	127	}\r
5d73d92f	128	}\r
475387e9	129	if (Data == NULL \|\| Data[0] != L'=') {\r
5d73d92f	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
3737ac2b	135	Buffer = AllocateZeroPool(Size);\r
5d73d92f	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
4092a8f6	145	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_GET), gShellDebug1HiiHandle, L"setvar", &Guid, VariableName); \r
5d73d92f	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
4092a8f6	154	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, L"setvar", &Guid, VariableName); \r
5d73d92f	155	ShellStatus = SHELL_ACCESS_DENIED;\r
	156	} else {\r
	157	ASSERT(ShellStatus == SHELL_SUCCESS);\r
	158	}\r
	159	} else {\r
5d73d92f	160	//\r
475387e9	161	// Change what's there or create a new one.\r
5d73d92f	162	//\r
475387e9 JC	163	\r
	164	ASSERT(Data[0] == L'=');\r
	165	Data++;\r
bbf57d4f	166	ASSERT(Data[0] != L'\0');\r
475387e9 JC	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
5d73d92f	175	}\r
475387e9 JC	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
5d73d92f	192	}\r
475387e9 JC	193	SHELL_FREE_NON_NULL(Buffer);\r
	194	\r
	195	//\r
	196	// What type is the new data.\r
	197	//\r
bbf57d4f	198	if (IsStringOfHexNibbles(Data)) {\r
3737ac2b	199	if (StrLen(Data) % 2 != 0) {\r
4092a8f6	200	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellDebug1HiiHandle, L"setvar", Data); \r
3737ac2b	201	ShellStatus = SHELL_INVALID_PARAMETER;\r
5d73d92f	202	} else {\r
3737ac2b	203	//\r
	204	// arbitrary buffer\r
	205	//\r
	206	Buffer = AllocateZeroPool((StrLen(Data) / 2));\r
33c031ee	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[LoopVar2]) * 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
3737ac2b	215	}\r
3737ac2b	216	if (EFI_ERROR(Status)) {\r
4092a8f6	217	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, L"setvar", &Guid, VariableName); \r
3737ac2b	218	ShellStatus = SHELL_ACCESS_DENIED;\r
	219	} else {\r
	220	ASSERT(ShellStatus == SHELL_SUCCESS);\r
	221	}\r
5d73d92f	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
33c031ee	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
5d73d92f	236	if (EFI_ERROR(Status)) {\r
4092a8f6	237	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, L"setvar", &Guid, VariableName); \r
5d73d92f	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
cbdd109b	249	if (Buffer == NULL) {\r
4092a8f6	250	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_OUT_MEM), gShellDebug1HiiHandle, L"setvar"); \r
cbdd109b	251	ShellStatus = SHELL_OUT_OF_RESOURCES;\r
5d73d92f	252	} else {\r
cbdd109b	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
4092a8f6	258	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, L"setvar", &Guid, VariableName); \r
cbdd109b	259	ShellStatus = SHELL_ACCESS_DENIED;\r
	260	} else {\r
	261	ASSERT(ShellStatus == SHELL_SUCCESS);\r
	262	}\r
5d73d92f	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
863986b3	270	DevPath = ConvertTextToDevicePath(Data);\r
5d73d92f	271	if (DevPath == NULL) {\r
4092a8f6	272	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_DPFT), gShellDebug1HiiHandle, L"setvar"); \r
5d73d92f	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
4092a8f6	277	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, L"setvar", &Guid, VariableName); \r
5d73d92f	278	ShellStatus = SHELL_ACCESS_DENIED;\r
	279	} else {\r
	280	ASSERT(ShellStatus == SHELL_SUCCESS);\r
	281	}\r
	282	}\r
	283	} else {\r
4092a8f6	284	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellDebug1HiiHandle, L"setvar", Data); \r
5d73d92f	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