ShellPkg: Refactor string manipulation

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

/** @file\r
  function declarations for shell environment functions.\r
\r
  Copyright (c) 2009 - 2011, 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
/**\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
  UINT64            MaxStorSize;\r
  UINT64            RemStorSize;\r
  UINT64            MaxVarSize;\r
  EFI_STATUS        Status;\r
  EFI_GUID          Guid;\r
  UINTN             ValSize;\r
  ENV_VAR_LIST      *VarList;\r
\r
  if (ListHead == NULL) {\r
    return (EFI_INVALID_PARAMETER);\r
  }\r
\r
  if (gRT->Hdr.Revision >= EFI_2_00_SYSTEM_TABLE_REVISION) {\r
    Status = gRT->QueryVariableInfo(EFI_VARIABLE_NON_VOLATILE|EFI_VARIABLE_BOOTSERVICE_ACCESS, &MaxStorSize, &RemStorSize, &MaxVarSize);\r
    if (EFI_ERROR(Status)) {\r
      return (Status);\r
    }\r
  } else {\r
    Status = EFI_SUCCESS;\r
    MaxVarSize = 16384;\r
  }\r
\r
  NameSize = (UINTN)MaxVarSize;\r
  VariableName = AllocateZeroPool(NameSize);\r
  if (VariableName == NULL) {\r
    return (EFI_OUT_OF_RESOURCES);\r
  }\r
  *VariableName = CHAR_NULL;\r
\r
  while (!EFI_ERROR(Status)) {\r
    NameSize = (UINTN)MaxVarSize;\r
    Status = gRT->GetNextVariableName(&NameSize, VariableName, &Guid);\r
    if (Status == EFI_NOT_FOUND){\r
      Status = EFI_SUCCESS;\r
      break;\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 = 0;\r
        Status = SHELL_GET_ENVIRONMENT_VARIABLE_AND_ATTRIBUTES(VariableName, &VarList->Atts, &ValSize, VarList->Val);\r
        if (Status == EFI_BUFFER_TOO_SMALL){\r
          VarList->Val = AllocateZeroPool(ValSize);\r
          if (VarList->Val == NULL) {\r
            SHELL_FREE_NON_NULL(VarList);\r
            Status = EFI_OUT_OF_RESOURCES;\r
          } else {\r
            Status = SHELL_GET_ENVIRONMENT_VARIABLE_AND_ATTRIBUTES(VariableName, &VarList->Atts, &ValSize, VarList->Val);\r
          }\r
        }\r
        if (!EFI_ERROR(Status) && VarList != NULL) {\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
        }\r
      }\r
    } // compare guid\r
  } // while\r
  FreePool(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