ArmPlatformPkg: Remove PcdStandalone from Sec module and Introduce ArmPlatformSecExtr...

[mirror_edk2.git] / EmbeddedPkg / Ebl / Variable.c

/** @file\r
*\r
*  Copyright (c) 2011, ARM Limited. All rights reserved.\r
*  \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 "Ebl.h"\r
\r
#include <Guid/GlobalVariable.h>\r
\r
EFI_STATUS\r
EblGetCmd (\r
  IN UINTN  Argc,\r
  IN CHAR8  **Argv\r
  )\r
{\r
  EFI_STATUS        Status = EFI_INVALID_PARAMETER;\r
  UINTN       Size;\r
  VOID*       Value;\r
  CHAR8*      AsciiVariableName = NULL;\r
  CHAR16*     VariableName;\r
  UINT32      Index;\r
\r
  if (Argc == 1) {\r
    AsciiPrint("Variable name is missing.\n");\r
    return Status;\r
  }\r
\r
  for (Index = 1; Index < Argc; Index++) {\r
    if (Argv[Index][0] == '-') {\r
      AsciiPrint("Warning: '%a' not recognized.\n",Argv[Index]);\r
    } else {\r
      AsciiVariableName = Argv[Index];\r
    }\r
  }\r
\r
  if (AsciiVariableName == NULL) {\r
    AsciiPrint("Variable name is missing.\n");\r
    return Status;\r
  } else {\r
    VariableName = AllocatePool((AsciiStrLen (AsciiVariableName) + 1) * sizeof (CHAR16));\r
    AsciiStrToUnicodeStr (AsciiVariableName,VariableName);\r
  }\r
\r
  // Try to get the variable size.\r
  Value = NULL;\r
  Size = 0;\r
  Status = gRT->GetVariable (VariableName, &gEfiGlobalVariableGuid, NULL, &Size, Value);\r
  if (Status == EFI_NOT_FOUND) {\r
    AsciiPrint("Variable name '%s' not found.\n",VariableName);\r
  } else if (Status == EFI_BUFFER_TOO_SMALL) {\r
    // Get the environment variable value\r
    Value = AllocatePool (Size);\r
    if (Value == NULL) {\r
      return EFI_OUT_OF_RESOURCES;\r
    }\r
\r
    Status = gRT->GetVariable ((CHAR16 *)VariableName, &gEfiGlobalVariableGuid, NULL, &Size, Value);\r
    if (EFI_ERROR (Status)) {\r
      AsciiPrint("Error: '%r'\n",Status);\r
    } else {\r
      AsciiPrint("%a=%a\n",AsciiVariableName,Value);\r
    }\r
    FreePool(Value);\r
  } else {\r
    AsciiPrint("Error: '%r'\n",Status);\r
  }\r
\r
  FreePool(VariableName);\r
  return Status;\r
}\r
\r
EFI_STATUS\r
EblSetCmd (\r
  IN UINTN  Argc,\r
  IN CHAR8  **Argv\r
  )\r
{\r
  EFI_STATUS    Status = EFI_INVALID_PARAMETER;\r
  CHAR8*        AsciiVariableSetting = NULL;\r
  CHAR8*        AsciiVariableName;\r
  CHAR8*        AsciiValue;\r
  UINT32        AsciiValueLength;\r
  CHAR16*       VariableName;\r
  UINT32        Index;\r
  UINT32        EscapedQuotes = 0;\r
  BOOLEAN       Volatile = FALSE;\r
\r
  if (Argc == 1) {\r
    AsciiPrint("Variable name is missing.\n");\r
    return Status;\r
  }\r
\r
  for (Index = 1; Index < Argc; Index++) {\r
    if (AsciiStrCmp(Argv[Index],"-v") == 0) {\r
      Volatile = 0;\r
    } else if (Argv[Index][0] == '-') {\r
      AsciiPrint("Warning: '%a' not recognized.\n",Argv[Index]);\r
    } else {\r
      AsciiVariableSetting = Argv[Index];\r
    }\r
  }\r
\r
  if (AsciiVariableSetting == NULL) {\r
    AsciiPrint("Variable name is missing.\n");\r
    return Status;\r
  }\r
  \r
  // Check if it is a valid variable setting\r
  AsciiValue = AsciiStrStr (AsciiVariableSetting,"=");\r
  if (AsciiValue == NULL) {\r
    //\r
    // There is no value. It means this variable will be deleted\r
    //\r
\r
    // Convert VariableName into Unicode\r
    VariableName = AllocatePool((AsciiStrLen (AsciiVariableSetting) + 1) * sizeof (CHAR16));\r
    AsciiStrToUnicodeStr (AsciiVariableSetting,VariableName);\r
\r
    Status = gRT->SetVariable (\r
                          VariableName,\r
                          &gEfiGlobalVariableGuid,\r
                          ( !Volatile ? EFI_VARIABLE_NON_VOLATILE : 0) |\r
                          EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_RUNTIME_ACCESS,\r
                          0,\r
                          NULL\r
                          );\r
    if (!EFI_ERROR(Status)) {\r
      AsciiPrint("Variable '%s' deleted\n",VariableName);\r
    } else {\r
      AsciiPrint("Variable setting is incorrect. It should be VariableName=Value\n");\r
    }\r
    return Status;\r
  }\r
\r
  AsciiValue[0] = '\0';\r
  AsciiVariableName = AsciiVariableSetting;\r
  AsciiValue++;\r
\r
  // Clean AsciiValue from quote\r
  if (AsciiValue[0] == '"') {\r
    AsciiValue++;\r
  }\r
  AsciiValueLength = AsciiStrLen (AsciiValue);\r
  if ((AsciiValue[AsciiValueLength-2] != '\\') && (AsciiValue[AsciiValueLength-1] == '"')) {\r
    AsciiValue[AsciiValueLength-1] = '\0';\r
  }\r
\r
  // Clean AsciiValue from escaped quotes\r
  for (Index = 0; Index < AsciiValueLength; Index++) {\r
    if ((Index > 0) && (AsciiValue[Index-1] == '\\') && (AsciiValue[Index] == '"')) {\r
      EscapedQuotes++;\r
    }\r
    AsciiValue[Index-EscapedQuotes] = AsciiValue[Index];\r
  }\r
  // Fill the end of the value with '\0'\r
  for (Index = 0; Index < EscapedQuotes; Index++) {\r
    AsciiValue[AsciiValueLength-1-Index] = '\0';\r
  }\r
\r
  // Convert VariableName into Unicode\r
  VariableName = AllocatePool((AsciiStrLen (AsciiVariableName) + 1) * sizeof (CHAR16));\r
  AsciiStrToUnicodeStr (AsciiVariableName,VariableName);\r
\r
  Status = gRT->SetVariable (\r
                      VariableName,\r
                      &gEfiGlobalVariableGuid,\r
                      ( !Volatile ? EFI_VARIABLE_NON_VOLATILE : 0) |\r
                      EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_RUNTIME_ACCESS,\r
                      AsciiStrLen (AsciiValue)+1,\r
                      AsciiValue\r
                      );\r
  if (!EFI_ERROR(Status)) {\r
    AsciiPrint("'%a'='%a'\n",AsciiVariableName,AsciiValue);\r
  }\r
\r
  return Status;\r
}\r
\r
GLOBAL_REMOVE_IF_UNREFERENCED const EBL_COMMAND_TABLE mCmdVariableTemplate[] =\r
{\r
  {\r
    "get",\r
    " ; get UEFI variable\n\r   [v]; verbose",\r
    NULL,\r
    EblGetCmd\r
  },\r
  {\r
    "set",\r
    " ; set UEFI variable\n\r   [v]; create volatile variable",\r
    NULL,\r
    EblSetCmd\r
  }\r
};\r
\r
/**\r
  Initialize the commands in this in this file\r
**/\r
VOID\r
EblInitializeVariableCmds (\r
  VOID\r
  )\r
{\r
  EblAddCommands (mCmdVariableTemplate, sizeof (mCmdVariableTemplate)/sizeof (EBL_COMMAND_TABLE));\r
}\r