ShellPkg: Apply uncrustify changes

[mirror_edk2.git] / ShellPkg / Library / UefiShellLevel2CommandsLib / Set.c

/** @file\r
  Main file for attrib shell level 2 function.\r
\r
  (C) Copyright 2015 Hewlett-Packard Development Company, L.P.<BR>\r
  Copyright (c) 2009 - 2018, Intel Corporation. All rights reserved.<BR>\r
  SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
\r
#include "UefiShellLevel2CommandsLib.h"\r
\r
/**\r
  Print out each environment variable registered with the Shell 2.0 GUID.\r
\r
  If you spawn a pre 2.0 shell from the Shell 2.0 the environment variable may not carry through.\r
\r
  @retval STATUS_SUCCESS  the printout was sucessful\r
  @return any return code from GetNextVariableName except EFI_NOT_FOUND\r
**/\r
SHELL_STATUS\r
PrintAllShellEnvVars (\r
  VOID\r
  )\r
{\r
  CONST CHAR16  *Value;\r
  CONST CHAR16  *ConstEnvNameList;\r
\r
  ConstEnvNameList = gEfiShellProtocol->GetEnv (NULL);\r
  if (ConstEnvNameList == NULL) {\r
    return (SHELL_SUCCESS);\r
  }\r
\r
  while (*ConstEnvNameList != CHAR_NULL) {\r
    Value = gEfiShellProtocol->GetEnv (ConstEnvNameList);\r
    ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_SET_DISP), gShellLevel2HiiHandle, ConstEnvNameList, Value);\r
    ConstEnvNameList += StrLen (ConstEnvNameList)+1;\r
  }\r
\r
  return (SHELL_SUCCESS);\r
}\r
\r
STATIC CONST SHELL_PARAM_ITEM  SetParamList[] = {\r
  { L"-d", TypeValue },\r
  { L"-v", TypeFlag  },\r
  { NULL,  TypeMax   }\r
};\r
\r
/**\r
  Function for 'set' 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
ShellCommandRunSet (\r
  IN EFI_HANDLE        ImageHandle,\r
  IN EFI_SYSTEM_TABLE  *SystemTable\r
  )\r
{\r
  EFI_STATUS    Status;\r
  LIST_ENTRY    *Package;\r
  CONST CHAR16  *KeyName;\r
  CONST CHAR16  *Value;\r
  CHAR16        *ProblemParam;\r
  SHELL_STATUS  ShellStatus;\r
\r
  ProblemParam = NULL;\r
  ShellStatus  = SHELL_SUCCESS;\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
  //\r
  // Make sure globals are good...\r
  //\r
  Status = CommandInit ();\r
  ASSERT_EFI_ERROR (Status);\r
\r
  //\r
  // parse the command line\r
  //\r
  Status = ShellCommandLineParse (SetParamList, &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), gShellLevel2HiiHandle, L"set", ProblemParam);\r
      FreePool (ProblemParam);\r
      return (SHELL_INVALID_PARAMETER);\r
    } else {\r
      ASSERT (FALSE);\r
    }\r
  } else {\r
    //\r
    // check for "-?"\r
    //\r
    if (ShellCommandLineGetFlag (Package, L"-?")) {\r
      ASSERT (FALSE);\r
    } else if (ShellCommandLineGetRawValue (Package, 3) != NULL) {\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellLevel2HiiHandle, L"set");\r
      ShellStatus = SHELL_INVALID_PARAMETER;\r
    } else if ((ShellCommandLineGetRawValue (Package, 1) != NULL) && ShellCommandLineGetFlag (Package, L"-d")) {\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellLevel2HiiHandle, L"set");\r
      ShellStatus = SHELL_INVALID_PARAMETER;\r
    } else if (ShellCommandLineGetFlag (Package, L"-d")) {\r
      //\r
      // delete a environment variable\r
      //\r
      KeyName = ShellCommandLineGetValue (Package, L"-d");\r
      if (KeyName == NULL) {\r
        ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_VALUE), gShellLevel2HiiHandle, L"set", L"-d");\r
        ShellStatus = SHELL_INVALID_PARAMETER;\r
      } else {\r
        Status = ShellSetEnvironmentVariable (KeyName, L"", ShellCommandLineGetFlag (Package, L"-v"));\r
        if (EFI_ERROR (Status)) {\r
          ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_SET_ND), gShellLevel2HiiHandle, L"set", KeyName);\r
          ShellStatus = SHELL_DEVICE_ERROR;\r
        }\r
      }\r
    } else if (ShellCommandLineGetRawValue (Package, 1) == NULL) {\r
      //\r
      // print out all current environment variables\r
      //\r
      return (PrintAllShellEnvVars ());\r
    } else {\r
      //\r
      // we are either printing one or assigning one\r
      //\r
      KeyName = ShellCommandLineGetRawValue (Package, 1);\r
      Value   = ShellCommandLineGetRawValue (Package, 2);\r
      if ((KeyName != NULL) && (Value != NULL)) {\r
        //\r
        // assigning one\r
        //\r
        Status = ShellSetEnvironmentVariable (KeyName, Value, ShellCommandLineGetFlag (Package, L"-v"));\r
        if (EFI_ERROR (Status)) {\r
          ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_SET_ERROR_SET), gShellLevel2HiiHandle, L"set", KeyName);\r
          ShellStatus = (SHELL_STATUS)(Status & (~MAX_BIT));\r
        }\r
      } else {\r
        if (KeyName != NULL) {\r
          //\r
          // print out value for this one only.\r
          //\r
          Value = ShellGetEnvironmentVariable (KeyName);\r
          if (Value == NULL) {\r
            ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_SET_NF), gShellLevel2HiiHandle, L"set", KeyName);\r
            ShellStatus = SHELL_SUCCESS;\r
          } else {\r
            ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_SET_DISP), gShellLevel2HiiHandle, KeyName, Value);\r
            ShellStatus = SHELL_SUCCESS;\r
          }\r
        } else {\r
          ASSERT (FALSE);\r
        }\r
      }\r
    }\r
  }\r
\r
  //\r
  // free the command line package\r
  //\r
  ShellCommandLineFreeVarList (Package);\r
\r
  return (ShellStatus);\r
}\r