/** @file\r
Main file for SetVar shell Debug1 function.\r
\r
- Copyright (c) 2010, Intel Corporation. All rights reserved.<BR>\r
+ (C) Copyright 2015 Hewlett-Packard Development Company, L.P.<BR>\r
+ Copyright (c) 2010 - 2014, 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
{NULL, TypeMax}\r
};\r
\r
+/**\r
+ Function for 'setvar' 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
ShellCommandRunSetVar (\r
UINTN Size;\r
UINTN LoopVar;\r
EFI_DEVICE_PATH_PROTOCOL *DevPath;\r
- EFI_DEVICE_PATH_FROM_TEXT_PROTOCOL *DevPathFromText;\r
\r
ShellStatus = SHELL_SUCCESS;\r
Status = EFI_SUCCESS;\r
Status = ShellCommandLineParse (ParamList, &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), gShellDebug1HiiHandle, ProblemParam);\r
+ ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, L"setvar", ProblemParam); \r
FreePool(ProblemParam);\r
ShellStatus = SHELL_INVALID_PARAMETER;\r
} else {\r
}\r
} else {\r
if (ShellCommandLineGetCount(Package) < 2) {\r
- ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle);\r
+ ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle, L"setvar"); \r
ShellStatus = SHELL_INVALID_PARAMETER;\r
} else if (ShellCommandLineGetCount(Package) > 3) {\r
- ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle);\r
+ ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle, L"setvar"); \r
ShellStatus = SHELL_INVALID_PARAMETER;\r
} else {\r
VariableName = ShellCommandLineGetRawValue(Package, 1);\r
StringGuid = ShellCommandLineGetValue(Package, L"-guid");\r
Status = ConvertStringToGuid(StringGuid, &Guid);\r
if (EFI_ERROR(Status)) {\r
- ShellStatus = SHELL_NOT_FOUND;\r
+ ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellDebug1HiiHandle, L"setvar", StringGuid); \r
+ ShellStatus = SHELL_INVALID_PARAMETER;\r
}\r
}\r
- if (Data == NULL) {\r
+ if (Data == NULL || Data[0] != L'=') {\r
//\r
// Display what's there\r
//\r
Status = gRT->GetVariable((CHAR16*)VariableName, &Guid, &Attributes, &Size, Buffer);\r
if (Status == EFI_BUFFER_TOO_SMALL) {\r
- Buffer = AllocatePool(Size);\r
+ Buffer = AllocateZeroPool(Size);\r
Status = gRT->GetVariable((CHAR16*)VariableName, &Guid, &Attributes, &Size, Buffer);\r
}\r
if (!EFI_ERROR(Status)&& Buffer != NULL) {\r
}\r
ShellPrintEx(-1, -1, L"\r\n");\r
} else {\r
- ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_GET), gShellDebug1HiiHandle, &Guid, VariableName, Status);\r
+ ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_GET), gShellDebug1HiiHandle, L"setvar", &Guid, VariableName); \r
ShellStatus = SHELL_ACCESS_DENIED;\r
}\r
} else if (StrCmp(Data, L"=") == 0) {\r
//\r
Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, 0, NULL);\r
if (EFI_ERROR(Status)) {\r
- ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, &Guid, VariableName, Status);\r
+ ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, L"setvar", &Guid, VariableName); \r
ShellStatus = SHELL_ACCESS_DENIED;\r
} else {\r
ASSERT(ShellStatus == SHELL_SUCCESS);\r
}\r
} else {\r
- if (Data[0] == L'=') {\r
- Data++;\r
- }\r
//\r
- // Change what's there\r
+ // Change what's there or create a new one.\r
//\r
- if (ShellCommandLineGetFlag(Package, L"-bs")) {\r
- Attributes |= EFI_VARIABLE_BOOTSERVICE_ACCESS;\r
- }\r
- if (ShellCommandLineGetFlag(Package, L"-rt")) {\r
- Attributes |= EFI_VARIABLE_RUNTIME_ACCESS;\r
- }\r
- if (ShellCommandLineGetFlag(Package, L"-nv")) {\r
- Attributes |= EFI_VARIABLE_NON_VOLATILE;\r
+\r
+ ASSERT(Data[0] == L'=');\r
+ Data++;\r
+\r
+ //\r
+ // Determine if the variable exists and get the attributes\r
+ //\r
+ Status = gRT->GetVariable((CHAR16*)VariableName, &Guid, &Attributes, &Size, Buffer);\r
+ if (Status == EFI_BUFFER_TOO_SMALL) {\r
+ Buffer = AllocateZeroPool(Size);\r
+ Status = gRT->GetVariable((CHAR16*)VariableName, &Guid, &Attributes, &Size, Buffer);\r
}\r
- if (ShellIsHexOrDecimalNumber(Data, TRUE, FALSE)) {\r
+\r
+ if (EFI_ERROR(Status) || Buffer == NULL) {\r
//\r
- // arbitrary buffer\r
+ // Creating a new variable. determine attributes from command line.\r
//\r
- Buffer = AllocateZeroPool((StrLen(Data) / 2));\r
- for (LoopVar = 0 ; LoopVar < (StrLen(Data) / 2) ; LoopVar++) {\r
- ((UINT8*)Buffer)[LoopVar] = (UINT8)(HexCharToUintn(Data[LoopVar*2]) * 16);\r
- ((UINT8*)Buffer)[LoopVar] = (UINT8)(((UINT8*)Buffer)[LoopVar] + HexCharToUintn(Data[LoopVar*2+1]));\r
+ Attributes = 0;\r
+ if (ShellCommandLineGetFlag(Package, L"-bs")) {\r
+ Attributes |= EFI_VARIABLE_BOOTSERVICE_ACCESS;\r
}\r
- Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, StrLen(Data) / 2, Buffer);\r
- if (EFI_ERROR(Status)) {\r
- ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, &Guid, VariableName, Status);\r
- ShellStatus = SHELL_ACCESS_DENIED;\r
+ if (ShellCommandLineGetFlag(Package, L"-rt")) {\r
+ Attributes |= EFI_VARIABLE_RUNTIME_ACCESS |\r
+ EFI_VARIABLE_BOOTSERVICE_ACCESS;\r
+ }\r
+ if (ShellCommandLineGetFlag(Package, L"-nv")) {\r
+ Attributes |= EFI_VARIABLE_NON_VOLATILE;\r
+ }\r
+ }\r
+ SHELL_FREE_NON_NULL(Buffer);\r
+\r
+ //\r
+ // What type is the new data.\r
+ //\r
+ if (ShellIsHexOrDecimalNumber(Data, TRUE, FALSE)) {\r
+ if (StrLen(Data) % 2 != 0) {\r
+ ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellDebug1HiiHandle, L"setvar", Data); \r
+ ShellStatus = SHELL_INVALID_PARAMETER;\r
} else {\r
- ASSERT(ShellStatus == SHELL_SUCCESS);\r
+ //\r
+ // arbitrary buffer\r
+ //\r
+ Buffer = AllocateZeroPool((StrLen(Data) / 2));\r
+ if (Buffer == NULL) {\r
+ Status = EFI_OUT_OF_RESOURCES;\r
+ } else {\r
+ for (LoopVar = 0 ; LoopVar < (StrLen(Data) / 2) ; LoopVar++) {\r
+ ((UINT8*)Buffer)[LoopVar] = (UINT8)(HexCharToUintn(Data[LoopVar*2]) * 16);\r
+ ((UINT8*)Buffer)[LoopVar] = (UINT8)(((UINT8*)Buffer)[LoopVar] + HexCharToUintn(Data[LoopVar*2+1]));\r
+ }\r
+ Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, StrLen(Data) / 2, Buffer);\r
+ }\r
+ if (EFI_ERROR(Status)) {\r
+ ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, L"setvar", &Guid, VariableName); \r
+ ShellStatus = SHELL_ACCESS_DENIED;\r
+ } else {\r
+ ASSERT(ShellStatus == SHELL_SUCCESS);\r
+ }\r
}\r
} else if (StrnCmp(Data, L"\"", 1) == 0) {\r
//\r
//\r
Data++;\r
Buffer = AllocateZeroPool(StrSize(Data) / 2);\r
- AsciiSPrint(Buffer, StrSize(Data) / 2, "%s", Data);\r
- ((CHAR8*)Buffer)[AsciiStrLen(Buffer)-1] = CHAR_NULL;\r
-\r
- Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, AsciiStrSize(Buffer)-sizeof(CHAR8), Buffer);\r
+ if (Buffer == NULL) {\r
+ Status = EFI_OUT_OF_RESOURCES;\r
+ } else {\r
+ AsciiSPrint(Buffer, StrSize(Data) / 2, "%s", Data);\r
+ ((CHAR8*)Buffer)[AsciiStrLen(Buffer)-1] = CHAR_NULL;\r
+ Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, AsciiStrSize(Buffer)-sizeof(CHAR8), Buffer);\r
+ }\r
if (EFI_ERROR(Status)) {\r
- ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, &Guid, VariableName, Status);\r
+ ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, L"setvar", &Guid, VariableName); \r
ShellStatus = SHELL_ACCESS_DENIED;\r
} else {\r
ASSERT(ShellStatus == SHELL_SUCCESS);\r
Data++;\r
Data++;\r
Buffer = AllocateZeroPool(StrSize(Data));\r
- UnicodeSPrint(Buffer, StrSize(Data), L"%s", Data);\r
- ((CHAR16*)Buffer)[StrLen(Buffer)-1] = CHAR_NULL;\r
-\r
- Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, StrSize(Buffer)-sizeof(CHAR16), Buffer);\r
- if (EFI_ERROR(Status)) {\r
- ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, &Guid, VariableName, Status);\r
- ShellStatus = SHELL_ACCESS_DENIED;\r
+ if (Buffer == NULL) {\r
+ ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_OUT_MEM), gShellDebug1HiiHandle, L"setvar"); \r
+ ShellStatus = SHELL_OUT_OF_RESOURCES;\r
} else {\r
- ASSERT(ShellStatus == SHELL_SUCCESS);\r
+ UnicodeSPrint(Buffer, StrSize(Data), L"%s", Data);\r
+ ((CHAR16*)Buffer)[StrLen(Buffer)-1] = CHAR_NULL;\r
+\r
+ Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, StrSize(Buffer)-sizeof(CHAR16), Buffer);\r
+ if (EFI_ERROR(Status)) {\r
+ ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, L"setvar", &Guid, VariableName); \r
+ ShellStatus = SHELL_ACCESS_DENIED;\r
+ } else {\r
+ ASSERT(ShellStatus == SHELL_SUCCESS);\r
+ }\r
}\r
} else if (StrnCmp(Data, L"--", 2) == 0) {\r
//\r
//\r
Data++;\r
Data++;\r
- Status = gBS->LocateProtocol(&gEfiDevicePathFromTextProtocolGuid, NULL, (VOID**)&DevPathFromText);\r
- ASSERT_EFI_ERROR(Status);\r
- DevPath = DevPathFromText->ConvertTextToDevicePath(Data);\r
+ DevPath = ConvertTextToDevicePath(Data);\r
if (DevPath == NULL) {\r
- ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_DPFT), gShellDebug1HiiHandle, Status);\r
+ ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_DPFT), gShellDebug1HiiHandle, L"setvar"); \r
ShellStatus = SHELL_INVALID_PARAMETER;\r
} else {\r
Status = gRT->SetVariable((CHAR16*)VariableName, &Guid, Attributes, GetDevicePathSize(DevPath), DevPath);\r
if (EFI_ERROR(Status)) {\r
- ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, &Guid, VariableName, Status);\r
+ ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SETVAR_ERROR_SET), gShellDebug1HiiHandle, L"setvar", &Guid, VariableName); \r
ShellStatus = SHELL_ACCESS_DENIED;\r
} else {\r
ASSERT(ShellStatus == SHELL_SUCCESS);\r
}\r
}\r
} else {\r
- ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, Data);\r
+ ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellDebug1HiiHandle, L"setvar", Data); \r
ShellStatus = SHELL_INVALID_PARAMETER;\r
}\r
}\r