[mirror_edk2.git] / ShellPkg / Application / Shell / ShellEnvVar.c

/** @file\r
  function declarations for shell environment functions.\r
\r
  Copyright (c) 2009 - 2015, 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 "Shell.h"\r
\r
#define INIT_NAME_BUFFER_SIZE  128\r
#define INIT_DATA_BUFFER_SIZE  1024\r
\r
/**\r
  Reports whether an environment variable is Volatile or Non-Volatile.\r
\r
  @param EnvVarName             The name of the environment variable in question\r
\r
  @retval TRUE                  This environment variable is Volatile\r
  @retval FALSE                 This environment variable is NON-Volatile\r
**/\r
BOOLEAN\r
EFIAPI\r
IsVolatileEnv (\r
  IN CONST CHAR16 *EnvVarName\r
  )\r
{\r
  EFI_STATUS  Status;\r
  UINTN       Size;\r
  VOID        *Buffer;\r
  UINT32      Attribs;\r
\r
  Size = 0;\r
  Buffer = NULL;\r
\r
  //\r
  // get the variable\r
  //\r
  Status = gRT->GetVariable((CHAR16*)EnvVarName,\r
                            &gShellVariableGuid,\r
                            &Attribs,\r
                            &Size,\r
                            Buffer);\r
  if (Status == EFI_BUFFER_TOO_SMALL) {\r
    Buffer = AllocateZeroPool(Size);\r
    ASSERT(Buffer != NULL);\r
    Status = gRT->GetVariable((CHAR16*)EnvVarName,\r
                              &gShellVariableGuid,\r
                              &Attribs,\r
                              &Size,\r
                              Buffer);\r
    FreePool(Buffer);\r
  }\r
  //\r
  // not found means volatile\r
  //\r
  if (Status == EFI_NOT_FOUND) {\r
    return (TRUE);\r
  }\r
  ASSERT_EFI_ERROR(Status);\r
\r
  //\r
  // check for the Non Volatile bit\r
  //\r
  if ((Attribs & EFI_VARIABLE_NON_VOLATILE) == EFI_VARIABLE_NON_VOLATILE) {\r
    return (FALSE);\r
  }\r
\r
  //\r
  // everything else is volatile\r
  //\r
  return (TRUE);\r
}\r
\r
/**\r
  free function for ENV_VAR_LIST objects.\r
\r
  @param[in] List               The pointer to pointer to list.\r
**/\r
VOID\r
EFIAPI\r
FreeEnvironmentVariableList(\r
  IN LIST_ENTRY *List\r
  )\r
{\r
  ENV_VAR_LIST *Node;\r
\r
  ASSERT (List != NULL);\r
  if (List == NULL) {\r
    return;\r
  }\r
\r
  for ( Node = (ENV_VAR_LIST*)GetFirstNode(List)\r
      ; !IsListEmpty(List)\r
      ; Node = (ENV_VAR_LIST*)GetFirstNode(List)\r
     ){\r
    ASSERT(Node != NULL);\r
    RemoveEntryList(&Node->Link);\r
    if (Node->Key != NULL) {\r
      FreePool(Node->Key);\r
    }\r
    if (Node->Val != NULL) {\r
      FreePool(Node->Val);\r
    }\r
    FreePool(Node);\r
  }\r
}\r
\r
/**\r
  Creates a list of all Shell-Guid-based environment variables.\r
\r
  @param[in, out] ListHead       The pointer to pointer to LIST ENTRY object for\r
                                 storing this list.\r
\r
  @retval EFI_SUCCESS           the list was created sucessfully.\r
**/\r
EFI_STATUS\r
EFIAPI\r
GetEnvironmentVariableList(\r
  IN OUT LIST_ENTRY *ListHead\r
  )\r
{\r
  CHAR16            *VariableName;\r
  UINTN             NameSize;\r
  UINTN             NameBufferSize;\r
  EFI_STATUS        Status;\r
  EFI_GUID          Guid;\r
  UINTN             ValSize;\r
  UINTN             ValBufferSize;\r
  ENV_VAR_LIST      *VarList;\r
\r
  if (ListHead == NULL) {\r
    return (EFI_INVALID_PARAMETER);\r
  }\r
  \r
  Status = EFI_SUCCESS;\r
  \r
  ValBufferSize = INIT_DATA_BUFFER_SIZE;\r
  NameBufferSize = INIT_NAME_BUFFER_SIZE;\r
  VariableName = AllocateZeroPool(NameBufferSize);\r
  if (VariableName == NULL) {\r
    return (EFI_OUT_OF_RESOURCES);\r
  }\r
  *VariableName = CHAR_NULL;\r
\r
  while (!EFI_ERROR(Status)) {\r
    NameSize = NameBufferSize;\r
    Status = gRT->GetNextVariableName(&NameSize, VariableName, &Guid);\r
    if (Status == EFI_NOT_FOUND){\r
      Status = EFI_SUCCESS;\r
      break;\r
    } else if (Status == EFI_BUFFER_TOO_SMALL) {\r
      NameBufferSize = NameSize > NameBufferSize * 2 ? NameSize : NameBufferSize * 2;\r
      SHELL_FREE_NON_NULL(VariableName);\r
      VariableName = AllocateZeroPool(NameBufferSize);\r
      if (VariableName == NULL) {\r
        Status = EFI_OUT_OF_RESOURCES;\r
        break;\r
      }\r
      NameSize = NameBufferSize;\r
      Status = gRT->GetNextVariableName(&NameSize, VariableName, &Guid);\r
    }\r
    \r
    if (!EFI_ERROR(Status) && CompareGuid(&Guid, &gShellVariableGuid)){\r
      VarList = AllocateZeroPool(sizeof(ENV_VAR_LIST));\r
      if (VarList == NULL) {\r
        Status = EFI_OUT_OF_RESOURCES;\r
      } else {\r
        ValSize = ValBufferSize;\r
        VarList->Val = AllocateZeroPool(ValSize);\r
        if (VarList->Val == NULL) {\r
            SHELL_FREE_NON_NULL(VarList);\r
            Status = EFI_OUT_OF_RESOURCES;\r
            break;\r
        }\r
        Status = SHELL_GET_ENVIRONMENT_VARIABLE_AND_ATTRIBUTES(VariableName, &VarList->Atts, &ValSize, VarList->Val);\r
        if (Status == EFI_BUFFER_TOO_SMALL){\r
          ValBufferSize = ValSize > ValBufferSize * 2 ? ValSize : ValBufferSize * 2;\r
          SHELL_FREE_NON_NULL (VarList->Val);\r
          VarList->Val = AllocateZeroPool(ValBufferSize);\r
          if (VarList->Val == NULL) {\r
            SHELL_FREE_NON_NULL(VarList);\r
            Status = EFI_OUT_OF_RESOURCES;\r
            break;\r
          }\r
          \r
          ValSize = ValBufferSize;\r
          Status = SHELL_GET_ENVIRONMENT_VARIABLE_AND_ATTRIBUTES(VariableName, &VarList->Atts, &ValSize, VarList->Val);\r
        }\r
        if (!EFI_ERROR(Status)) {\r
          VarList->Key = AllocateCopyPool(StrSize(VariableName), VariableName);\r
          if (VarList->Key == NULL) {\r
            SHELL_FREE_NON_NULL(VarList->Val);\r
            SHELL_FREE_NON_NULL(VarList);\r
            Status = EFI_OUT_OF_RESOURCES;\r
          } else {\r
            InsertTailList(ListHead, &VarList->Link);\r
          }\r
        } else {\r
          SHELL_FREE_NON_NULL(VarList->Val);\r
          SHELL_FREE_NON_NULL(VarList);\r
        }\r
      } // if (VarList == NULL) ... else ...\r
    } // compare guid\r
  } // while\r
  SHELL_FREE_NON_NULL (VariableName);\r
\r
  if (EFI_ERROR(Status)) {\r
    FreeEnvironmentVariableList(ListHead);\r
  }\r
\r
  return (Status);\r
}\r
\r
/**\r
  Sets a list of all Shell-Guid-based environment variables.  this will\r
  also eliminate all existing shell environment variables (even if they\r
  are not on the list).\r
\r
  This function will also deallocate the memory from List.\r
\r
  @param[in] ListHead           The pointer to LIST_ENTRY from\r
                                GetShellEnvVarList().\r
\r
  @retval EFI_SUCCESS           the list was Set sucessfully.\r
**/\r
EFI_STATUS\r
EFIAPI\r
SetEnvironmentVariableList(\r
  IN LIST_ENTRY *ListHead\r
  )\r
{\r
  ENV_VAR_LIST      VarList;\r
  ENV_VAR_LIST      *Node;\r
  EFI_STATUS        Status;\r
  UINTN             Size;\r
\r
  InitializeListHead(&VarList.Link);\r
\r
  //\r
  // Delete all the current environment variables\r
  //\r
  Status = GetEnvironmentVariableList(&VarList.Link);\r
  ASSERT_EFI_ERROR(Status);\r
\r
  for ( Node = (ENV_VAR_LIST*)GetFirstNode(&VarList.Link)\r
      ; !IsNull(&VarList.Link, &Node->Link)\r
      ; Node = (ENV_VAR_LIST*)GetNextNode(&VarList.Link, &Node->Link)\r
     ){\r
    if (Node->Key != NULL) {\r
      Status = SHELL_DELETE_ENVIRONMENT_VARIABLE(Node->Key);\r
    }\r
    ASSERT_EFI_ERROR(Status);\r
  }\r
\r
  FreeEnvironmentVariableList(&VarList.Link);\r
\r
  //\r
  // set all the variables fron the list\r
  //\r
  for ( Node = (ENV_VAR_LIST*)GetFirstNode(ListHead)\r
      ; !IsNull(ListHead, &Node->Link)\r
      ; Node = (ENV_VAR_LIST*)GetNextNode(ListHead, &Node->Link)\r
     ){\r
    Size = StrSize(Node->Val);\r
    if (Node->Atts & EFI_VARIABLE_NON_VOLATILE) {\r
      Status = SHELL_SET_ENVIRONMENT_VARIABLE_NV(Node->Key, Size, Node->Val);\r
    } else {\r
      Status = SHELL_SET_ENVIRONMENT_VARIABLE_V (Node->Key, Size, Node->Val);\r
    }\r
    ASSERT_EFI_ERROR(Status);\r
  }\r
  FreeEnvironmentVariableList(ListHead);\r
\r
  return (Status);\r
}\r
\r
/**\r
  sets a list of all Shell-Guid-based environment variables.\r
\r
  @param Environment        Points to a NULL-terminated array of environment\r
                            variables with the format 'x=y', where x is the\r
                            environment variable name and y is the value.\r
\r
  @retval EFI_SUCCESS       The command executed successfully.\r
  @retval EFI_INVALID_PARAMETER The parameter is invalid.\r
  @retval EFI_OUT_OF_RESOURCES Out of resources.\r
\r
  @sa SetEnvironmentVariableList\r
**/\r
EFI_STATUS\r
EFIAPI\r
SetEnvironmentVariables(\r
  IN CONST CHAR16 **Environment\r
  )\r
{\r
  CONST CHAR16  *CurrentString;\r
  UINTN         CurrentCount;\r
  ENV_VAR_LIST  *VarList;\r
  ENV_VAR_LIST  *Node;\r
\r
  VarList = NULL;\r
\r
  if (Environment == NULL) {\r
    return (EFI_INVALID_PARAMETER);\r
  }\r
\r
  //\r
  // Build a list identical to the ones used for get/set list functions above\r
  //\r
  for ( CurrentCount = 0\r
      ;\r
      ; CurrentCount++\r
     ){\r
    CurrentString = Environment[CurrentCount];\r
    if (CurrentString == NULL) {\r
      break;\r
    }\r
    ASSERT(StrStr(CurrentString, L"=") != NULL);\r
    Node = AllocateZeroPool(sizeof(ENV_VAR_LIST));\r
    if (Node == NULL) {\r
      SetEnvironmentVariableList(&VarList->Link);\r
      return (EFI_OUT_OF_RESOURCES);\r
    }\r
\r
    Node->Key = AllocateZeroPool((StrStr(CurrentString, L"=") - CurrentString + 1) * sizeof(CHAR16));\r
    if (Node->Key == NULL) {\r
      SHELL_FREE_NON_NULL(Node);\r
      SetEnvironmentVariableList(&VarList->Link);\r
      return (EFI_OUT_OF_RESOURCES);\r
    }\r
\r
    //\r
    // Copy the string into the Key, leaving the last character allocated as NULL to terminate\r
    //\r
    StrnCpy(Node->Key, CurrentString, StrStr(CurrentString, L"=") - CurrentString);\r
\r
    //\r
    // ValueSize = TotalSize - already removed size - size for '=' + size for terminator (the last 2 items cancel each other)\r
    //\r
    Node->Val = AllocateCopyPool(StrSize(CurrentString) - StrSize(Node->Key), CurrentString + StrLen(Node->Key) + 1);\r
    if (Node->Val == NULL) {\r
      SHELL_FREE_NON_NULL(Node->Key);\r
      SHELL_FREE_NON_NULL(Node);\r
      SetEnvironmentVariableList(&VarList->Link);\r
      return (EFI_OUT_OF_RESOURCES);\r
    }\r
\r
    Node->Atts = EFI_VARIABLE_BOOTSERVICE_ACCESS;\r
\r
    if (VarList == NULL) {\r
      VarList = AllocateZeroPool(sizeof(ENV_VAR_LIST));\r
      if (VarList == NULL) {\r
        SHELL_FREE_NON_NULL(Node->Key);\r
        SHELL_FREE_NON_NULL(Node->Val);\r
        SHELL_FREE_NON_NULL(Node);\r
        return (EFI_OUT_OF_RESOURCES);\r
      }\r
      InitializeListHead(&VarList->Link);\r
    }\r
    InsertTailList(&VarList->Link, &Node->Link);\r
\r
  } // for loop\r
\r
  //\r
  // set this new list as the set of all environment variables.\r
  // this function also frees the memory and deletes all pre-existing\r
  // shell-guid based environment variables.\r
  //\r
  return (SetEnvironmentVariableList(&VarList->Link));\r
}\r
Commit	Line	Data
	1	/** @file\r
	2	function declarations for shell environment functions.\r
	3	\r
	4	Copyright (c) 2009 - 2015, 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 "Shell.h"\r
	16	\r
	17	#define INIT_NAME_BUFFER_SIZE 128\r
	18	#define INIT_DATA_BUFFER_SIZE 1024\r
	19	\r
	20	/**\r
	21	Reports whether an environment variable is Volatile or Non-Volatile.\r
	22	\r
	23	@param EnvVarName The name of the environment variable in question\r
	24	\r
	25	@retval TRUE This environment variable is Volatile\r
	26	@retval FALSE This environment variable is NON-Volatile\r
	27	**/\r
	28	BOOLEAN\r
	29	EFIAPI\r
	30	IsVolatileEnv (\r
	31	IN CONST CHAR16 *EnvVarName\r
	32	)\r
	33	{\r
	34	EFI_STATUS Status;\r
	35	UINTN Size;\r
	36	VOID *Buffer;\r
	37	UINT32 Attribs;\r
	38	\r
	39	Size = 0;\r
	40	Buffer = NULL;\r
	41	\r
	42	//\r
	43	// get the variable\r
	44	//\r
	45	Status = gRT->GetVariable((CHAR16*)EnvVarName,\r
	46	&gShellVariableGuid,\r
	47	&Attribs,\r
	48	&Size,\r
	49	Buffer);\r
	50	if (Status == EFI_BUFFER_TOO_SMALL) {\r
	51	Buffer = AllocateZeroPool(Size);\r
	52	ASSERT(Buffer != NULL);\r
	53	Status = gRT->GetVariable((CHAR16*)EnvVarName,\r
	54	&gShellVariableGuid,\r
	55	&Attribs,\r
	56	&Size,\r
	57	Buffer);\r
	58	FreePool(Buffer);\r
	59	}\r
	60	//\r
	61	// not found means volatile\r
	62	//\r
	63	if (Status == EFI_NOT_FOUND) {\r
	64	return (TRUE);\r
	65	}\r
	66	ASSERT_EFI_ERROR(Status);\r
	67	\r
	68	//\r
	69	// check for the Non Volatile bit\r
	70	//\r
	71	if ((Attribs & EFI_VARIABLE_NON_VOLATILE) == EFI_VARIABLE_NON_VOLATILE) {\r
	72	return (FALSE);\r
	73	}\r
	74	\r
	75	//\r
	76	// everything else is volatile\r
	77	//\r
	78	return (TRUE);\r
	79	}\r
	80	\r
	81	/**\r
	82	free function for ENV_VAR_LIST objects.\r
	83	\r
	84	@param[in] List The pointer to pointer to list.\r
	85	**/\r
	86	VOID\r
	87	EFIAPI\r
	88	FreeEnvironmentVariableList(\r
	89	IN LIST_ENTRY *List\r
	90	)\r
	91	{\r
	92	ENV_VAR_LIST *Node;\r
	93	\r
	94	ASSERT (List != NULL);\r
	95	if (List == NULL) {\r
	96	return;\r
	97	}\r
	98	\r
	99	for ( Node = (ENV_VAR_LIST*)GetFirstNode(List)\r
	100	; !IsListEmpty(List)\r
	101	; Node = (ENV_VAR_LIST*)GetFirstNode(List)\r
	102	){\r
	103	ASSERT(Node != NULL);\r
	104	RemoveEntryList(&Node->Link);\r
	105	if (Node->Key != NULL) {\r
	106	FreePool(Node->Key);\r
	107	}\r
	108	if (Node->Val != NULL) {\r
	109	FreePool(Node->Val);\r
	110	}\r
	111	FreePool(Node);\r
	112	}\r
	113	}\r
	114	\r
	115	/**\r
	116	Creates a list of all Shell-Guid-based environment variables.\r
	117	\r
	118	@param[in, out] ListHead The pointer to pointer to LIST ENTRY object for\r
	119	storing this list.\r
	120	\r
	121	@retval EFI_SUCCESS the list was created sucessfully.\r
	122	**/\r
	123	EFI_STATUS\r
	124	EFIAPI\r
	125	GetEnvironmentVariableList(\r
	126	IN OUT LIST_ENTRY *ListHead\r
	127	)\r
	128	{\r
	129	CHAR16 *VariableName;\r
	130	UINTN NameSize;\r
	131	UINTN NameBufferSize;\r
	132	EFI_STATUS Status;\r
	133	EFI_GUID Guid;\r
	134	UINTN ValSize;\r
	135	UINTN ValBufferSize;\r
	136	ENV_VAR_LIST *VarList;\r
	137	\r
	138	if (ListHead == NULL) {\r
	139	return (EFI_INVALID_PARAMETER);\r
	140	}\r
	141	\r
	142	Status = EFI_SUCCESS;\r
	143	\r
	144	ValBufferSize = INIT_DATA_BUFFER_SIZE;\r
	145	NameBufferSize = INIT_NAME_BUFFER_SIZE;\r
	146	VariableName = AllocateZeroPool(NameBufferSize);\r
	147	if (VariableName == NULL) {\r
	148	return (EFI_OUT_OF_RESOURCES);\r
	149	}\r
	150	*VariableName = CHAR_NULL;\r
	151	\r
	152	while (!EFI_ERROR(Status)) {\r
	153	NameSize = NameBufferSize;\r
	154	Status = gRT->GetNextVariableName(&NameSize, VariableName, &Guid);\r
	155	if (Status == EFI_NOT_FOUND){\r
	156	Status = EFI_SUCCESS;\r
	157	break;\r
	158	} else if (Status == EFI_BUFFER_TOO_SMALL) {\r
	159	NameBufferSize = NameSize > NameBufferSize * 2 ? NameSize : NameBufferSize * 2;\r
	160	SHELL_FREE_NON_NULL(VariableName);\r
	161	VariableName = AllocateZeroPool(NameBufferSize);\r
	162	if (VariableName == NULL) {\r
	163	Status = EFI_OUT_OF_RESOURCES;\r
	164	break;\r
	165	}\r
	166	NameSize = NameBufferSize;\r
	167	Status = gRT->GetNextVariableName(&NameSize, VariableName, &Guid);\r
	168	}\r
	169	\r
	170	if (!EFI_ERROR(Status) && CompareGuid(&Guid, &gShellVariableGuid)){\r
	171	VarList = AllocateZeroPool(sizeof(ENV_VAR_LIST));\r
	172	if (VarList == NULL) {\r
	173	Status = EFI_OUT_OF_RESOURCES;\r
	174	} else {\r
	175	ValSize = ValBufferSize;\r
	176	VarList->Val = AllocateZeroPool(ValSize);\r
	177	if (VarList->Val == NULL) {\r
	178	SHELL_FREE_NON_NULL(VarList);\r
	179	Status = EFI_OUT_OF_RESOURCES;\r
	180	break;\r
	181	}\r
	182	Status = SHELL_GET_ENVIRONMENT_VARIABLE_AND_ATTRIBUTES(VariableName, &VarList->Atts, &ValSize, VarList->Val);\r
	183	if (Status == EFI_BUFFER_TOO_SMALL){\r
	184	ValBufferSize = ValSize > ValBufferSize * 2 ? ValSize : ValBufferSize * 2;\r
	185	SHELL_FREE_NON_NULL (VarList->Val);\r
	186	VarList->Val = AllocateZeroPool(ValBufferSize);\r
	187	if (VarList->Val == NULL) {\r
	188	SHELL_FREE_NON_NULL(VarList);\r
	189	Status = EFI_OUT_OF_RESOURCES;\r
	190	break;\r
	191	}\r
	192	\r
	193	ValSize = ValBufferSize;\r
	194	Status = SHELL_GET_ENVIRONMENT_VARIABLE_AND_ATTRIBUTES(VariableName, &VarList->Atts, &ValSize, VarList->Val);\r
	195	}\r
	196	if (!EFI_ERROR(Status)) {\r
	197	VarList->Key = AllocateCopyPool(StrSize(VariableName), VariableName);\r
	198	if (VarList->Key == NULL) {\r
	199	SHELL_FREE_NON_NULL(VarList->Val);\r
	200	SHELL_FREE_NON_NULL(VarList);\r
	201	Status = EFI_OUT_OF_RESOURCES;\r
	202	} else {\r
	203	InsertTailList(ListHead, &VarList->Link);\r
	204	}\r
	205	} else {\r
	206	SHELL_FREE_NON_NULL(VarList->Val);\r
	207	SHELL_FREE_NON_NULL(VarList);\r
	208	}\r
	209	} // if (VarList == NULL) ... else ...\r
	210	} // compare guid\r
	211	} // while\r
	212	SHELL_FREE_NON_NULL (VariableName);\r
	213	\r
	214	if (EFI_ERROR(Status)) {\r
	215	FreeEnvironmentVariableList(ListHead);\r
	216	}\r
	217	\r
	218	return (Status);\r
	219	}\r
	220	\r
	221	/**\r
	222	Sets a list of all Shell-Guid-based environment variables. this will\r
	223	also eliminate all existing shell environment variables (even if they\r
	224	are not on the list).\r
	225	\r
	226	This function will also deallocate the memory from List.\r
	227	\r
	228	@param[in] ListHead The pointer to LIST_ENTRY from\r
	229	GetShellEnvVarList().\r
	230	\r
	231	@retval EFI_SUCCESS the list was Set sucessfully.\r
	232	**/\r
	233	EFI_STATUS\r
	234	EFIAPI\r
	235	SetEnvironmentVariableList(\r
	236	IN LIST_ENTRY *ListHead\r
	237	)\r
	238	{\r
	239	ENV_VAR_LIST VarList;\r
	240	ENV_VAR_LIST *Node;\r
	241	EFI_STATUS Status;\r
	242	UINTN Size;\r
	243	\r
	244	InitializeListHead(&VarList.Link);\r
	245	\r
	246	//\r
	247	// Delete all the current environment variables\r
	248	//\r
	249	Status = GetEnvironmentVariableList(&VarList.Link);\r
	250	ASSERT_EFI_ERROR(Status);\r
	251	\r
	252	for ( Node = (ENV_VAR_LIST*)GetFirstNode(&VarList.Link)\r
	253	; !IsNull(&VarList.Link, &Node->Link)\r
	254	; Node = (ENV_VAR_LIST*)GetNextNode(&VarList.Link, &Node->Link)\r
	255	){\r
	256	if (Node->Key != NULL) {\r
	257	Status = SHELL_DELETE_ENVIRONMENT_VARIABLE(Node->Key);\r
	258	}\r
	259	ASSERT_EFI_ERROR(Status);\r
	260	}\r
	261	\r
	262	FreeEnvironmentVariableList(&VarList.Link);\r
	263	\r
	264	//\r
	265	// set all the variables fron the list\r
	266	//\r
	267	for ( Node = (ENV_VAR_LIST*)GetFirstNode(ListHead)\r
	268	; !IsNull(ListHead, &Node->Link)\r
	269	; Node = (ENV_VAR_LIST*)GetNextNode(ListHead, &Node->Link)\r
	270	){\r
	271	Size = StrSize(Node->Val);\r
	272	if (Node->Atts & EFI_VARIABLE_NON_VOLATILE) {\r
	273	Status = SHELL_SET_ENVIRONMENT_VARIABLE_NV(Node->Key, Size, Node->Val);\r
	274	} else {\r
	275	Status = SHELL_SET_ENVIRONMENT_VARIABLE_V (Node->Key, Size, Node->Val);\r
	276	}\r
	277	ASSERT_EFI_ERROR(Status);\r
	278	}\r
	279	FreeEnvironmentVariableList(ListHead);\r
	280	\r
	281	return (Status);\r
	282	}\r
	283	\r
	284	/**\r
	285	sets a list of all Shell-Guid-based environment variables.\r
	286	\r
	287	@param Environment Points to a NULL-terminated array of environment\r
	288	variables with the format 'x=y', where x is the\r
	289	environment variable name and y is the value.\r
	290	\r
	291	@retval EFI_SUCCESS The command executed successfully.\r
	292	@retval EFI_INVALID_PARAMETER The parameter is invalid.\r
	293	@retval EFI_OUT_OF_RESOURCES Out of resources.\r
	294	\r
	295	@sa SetEnvironmentVariableList\r
	296	**/\r
	297	EFI_STATUS\r
	298	EFIAPI\r
	299	SetEnvironmentVariables(\r
	300	IN CONST CHAR16 **Environment\r
	301	)\r
	302	{\r
	303	CONST CHAR16 *CurrentString;\r
	304	UINTN CurrentCount;\r
	305	ENV_VAR_LIST *VarList;\r
	306	ENV_VAR_LIST *Node;\r
	307	\r
	308	VarList = NULL;\r
	309	\r
	310	if (Environment == NULL) {\r
	311	return (EFI_INVALID_PARAMETER);\r
	312	}\r
	313	\r
	314	//\r
	315	// Build a list identical to the ones used for get/set list functions above\r
	316	//\r
	317	for ( CurrentCount = 0\r
	318	;\r
	319	; CurrentCount++\r
	320	){\r
	321	CurrentString = Environment[CurrentCount];\r
	322	if (CurrentString == NULL) {\r
	323	break;\r
	324	}\r
	325	ASSERT(StrStr(CurrentString, L"=") != NULL);\r
	326	Node = AllocateZeroPool(sizeof(ENV_VAR_LIST));\r
	327	if (Node == NULL) {\r
	328	SetEnvironmentVariableList(&VarList->Link);\r
	329	return (EFI_OUT_OF_RESOURCES);\r
	330	}\r
	331	\r
	332	Node->Key = AllocateZeroPool((StrStr(CurrentString, L"=") - CurrentString + 1) * sizeof(CHAR16));\r
	333	if (Node->Key == NULL) {\r
	334	SHELL_FREE_NON_NULL(Node);\r
	335	SetEnvironmentVariableList(&VarList->Link);\r
	336	return (EFI_OUT_OF_RESOURCES);\r
	337	}\r
	338	\r
	339	//\r
	340	// Copy the string into the Key, leaving the last character allocated as NULL to terminate\r
	341	//\r
	342	StrnCpy(Node->Key, CurrentString, StrStr(CurrentString, L"=") - CurrentString);\r
	343	\r
	344	//\r
	345	// ValueSize = TotalSize - already removed size - size for '=' + size for terminator (the last 2 items cancel each other)\r
	346	//\r
	347	Node->Val = AllocateCopyPool(StrSize(CurrentString) - StrSize(Node->Key), CurrentString + StrLen(Node->Key) + 1);\r
	348	if (Node->Val == NULL) {\r
	349	SHELL_FREE_NON_NULL(Node->Key);\r
	350	SHELL_FREE_NON_NULL(Node);\r
	351	SetEnvironmentVariableList(&VarList->Link);\r
	352	return (EFI_OUT_OF_RESOURCES);\r
	353	}\r
	354	\r
	355	Node->Atts = EFI_VARIABLE_BOOTSERVICE_ACCESS;\r
	356	\r
	357	if (VarList == NULL) {\r
	358	VarList = AllocateZeroPool(sizeof(ENV_VAR_LIST));\r
	359	if (VarList == NULL) {\r
	360	SHELL_FREE_NON_NULL(Node->Key);\r
	361	SHELL_FREE_NON_NULL(Node->Val);\r
	362	SHELL_FREE_NON_NULL(Node);\r
	363	return (EFI_OUT_OF_RESOURCES);\r
	364	}\r
	365	InitializeListHead(&VarList->Link);\r
	366	}\r
	367	InsertTailList(&VarList->Link, &Node->Link);\r
	368	\r
	369	} // for loop\r
	370	\r
	371	//\r
	372	// set this new list as the set of all environment variables.\r
	373	// this function also frees the memory and deletes all pre-existing\r
	374	// shell-guid based environment variables.\r
	375	//\r
	376	return (SetEnvironmentVariableList(&VarList->Link));\r
	377	}\r