ShellPkg: acpiview: Add GT Frame Number validation to GTDT parser

[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
  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
\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