/** @file\r
Main file for EfiDecompress shell Debug1 function.\r
\r
- Copyright (c) 2005 - 2010, 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
+ (C) Copyright 2015 Hewlett-Packard Development Company, L.P.<BR>\r
+ Copyright (c) 2005 - 2018, Intel Corporation. All rights reserved.<BR>\r
+ SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
\r
#include "UefiShellDebug1CommandsLib.h"\r
#include <Protocol/Decompress.h>\r
\r
-\r
/**\r
Function for 'decompress' command.\r
\r
IN EFI_SYSTEM_TABLE *SystemTable\r
)\r
{\r
- EFI_STATUS Status;\r
- LIST_ENTRY *Package;\r
- CHAR16 *ProblemParam;\r
- SHELL_STATUS ShellStatus;\r
- SHELL_FILE_HANDLE InFileHandle;\r
- SHELL_FILE_HANDLE OutFileHandle;\r
- UINT32 OutSize;\r
- UINTN OutSizeTemp;\r
- VOID *OutBuffer;\r
- UINTN InSize;\r
- VOID *InBuffer;\r
- CHAR16 *InFileName;\r
- CONST CHAR16 *OutFileName;\r
- UINT64 Temp64Bit;\r
- UINT32 ScratchSize;\r
- VOID *ScratchBuffer;\r
- EFI_DECOMPRESS_PROTOCOL *Decompress;\r
- CONST CHAR16 *TempParam;\r
-\r
- InFileName = NULL;\r
- OutFileName = NULL;\r
- OutSize = 0;\r
- ShellStatus = SHELL_SUCCESS;\r
- Status = EFI_SUCCESS;\r
- OutBuffer = NULL;\r
- InBuffer = NULL;\r
- ScratchBuffer = NULL;\r
- InFileHandle = NULL;\r
- OutFileHandle = NULL;\r
+ EFI_STATUS Status;\r
+ LIST_ENTRY *Package;\r
+ CHAR16 *ProblemParam;\r
+ SHELL_STATUS ShellStatus;\r
+ SHELL_FILE_HANDLE InFileHandle;\r
+ SHELL_FILE_HANDLE OutFileHandle;\r
+ UINT32 OutSize;\r
+ UINTN OutSizeTemp;\r
+ VOID *OutBuffer;\r
+ UINTN InSize;\r
+ VOID *InBuffer;\r
+ CHAR16 *InFileName;\r
+ CONST CHAR16 *OutFileName;\r
+ UINT64 Temp64Bit;\r
+ UINT32 ScratchSize;\r
+ VOID *ScratchBuffer;\r
+ EFI_DECOMPRESS_PROTOCOL *Decompress;\r
+ CONST CHAR16 *TempParam;\r
+\r
+ InFileName = NULL;\r
+ OutFileName = NULL;\r
+ OutSize = 0;\r
+ ScratchSize = 0;\r
+ ShellStatus = SHELL_SUCCESS;\r
+ Status = EFI_SUCCESS;\r
+ OutBuffer = NULL;\r
+ InBuffer = NULL;\r
+ ScratchBuffer = NULL;\r
+ InFileHandle = NULL;\r
+ OutFileHandle = NULL;\r
+ Decompress = NULL;\r
\r
//\r
// initialize the shell lib (we must be in non-auto-init...)\r
//\r
- Status = ShellInitialize();\r
- ASSERT_EFI_ERROR(Status);\r
+ Status = ShellInitialize ();\r
+ ASSERT_EFI_ERROR (Status);\r
\r
- Status = CommandInit();\r
- ASSERT_EFI_ERROR(Status);\r
+ Status = CommandInit ();\r
+ ASSERT_EFI_ERROR (Status);\r
\r
//\r
// parse the command line\r
//\r
Status = ShellCommandLineParse (EmptyParamList, &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
- FreePool(ProblemParam);\r
+ if (EFI_ERROR (Status)) {\r
+ if ((Status == EFI_VOLUME_CORRUPTED) && (ProblemParam != NULL)) {\r
+ ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, L"efidecompress", ProblemParam);\r
+ FreePool (ProblemParam);\r
ShellStatus = SHELL_INVALID_PARAMETER;\r
} else {\r
- ASSERT(FALSE);\r
+ ASSERT (FALSE);\r
}\r
} else {\r
- if (ShellCommandLineGetCount(Package) > 3) {\r
- ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle);\r
+ if (ShellCommandLineGetCount (Package) > 3) {\r
+ ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle, L"efidecompress");\r
ShellStatus = SHELL_INVALID_PARAMETER;\r
- } else if (ShellCommandLineGetCount(Package) < 3) {\r
- ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle);\r
+ } else if (ShellCommandLineGetCount (Package) < 3) {\r
+ ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle, L"efidecompress");\r
ShellStatus = SHELL_INVALID_PARAMETER;\r
} else {\r
- TempParam = ShellCommandLineGetRawValue(Package, 1);\r
- ASSERT(TempParam != NULL);\r
- InFileName = ShellFindFilePath(TempParam);\r
- OutFileName = ShellCommandLineGetRawValue(Package, 2);\r
+ TempParam = ShellCommandLineGetRawValue (Package, 1);\r
+ ASSERT (TempParam != NULL);\r
+ InFileName = ShellFindFilePath (TempParam);\r
+ OutFileName = ShellCommandLineGetRawValue (Package, 2);\r
if (InFileName == NULL) {\r
- ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_FIND_FAIL), gShellDebug1HiiHandle, TempParam);\r
+ ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_FILE_FIND_FAIL), gShellDebug1HiiHandle, L"efidecompress", TempParam);\r
ShellStatus = SHELL_NOT_FOUND;\r
} else {\r
- if (ShellIsDirectory(InFileName) == EFI_SUCCESS){\r
- ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_NOT_DIR), gShellDebug1HiiHandle, InFileName);\r
+ if (ShellIsDirectory (InFileName) == EFI_SUCCESS) {\r
+ ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_FILE_NOT_DIR), gShellDebug1HiiHandle, L"efidecompress", InFileName);\r
ShellStatus = SHELL_INVALID_PARAMETER;\r
}\r
- if (ShellIsDirectory(OutFileName) == EFI_SUCCESS){\r
- ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_NOT_DIR), gShellDebug1HiiHandle, OutFileName);\r
+\r
+ if (ShellIsDirectory (OutFileName) == EFI_SUCCESS) {\r
+ ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_FILE_NOT_DIR), gShellDebug1HiiHandle, L"efidecompress", OutFileName);\r
ShellStatus = SHELL_INVALID_PARAMETER;\r
}\r
+\r
if (ShellStatus == SHELL_SUCCESS) {\r
- Status = ShellOpenFileByName(InFileName, &InFileHandle, EFI_FILE_MODE_READ, 0);\r
- if (EFI_ERROR(Status)) {\r
- ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_OPEN_FAIL), gShellDebug1HiiHandle, ShellCommandLineGetRawValue(Package, 1), Status);\r
- ShellStatus = SHELL_NOT_FOUND;\r
- }\r
- Status = ShellOpenFileByName(OutFileName, &OutFileHandle, EFI_FILE_MODE_READ|EFI_FILE_MODE_WRITE|EFI_FILE_MODE_CREATE, 0);\r
- if (EFI_ERROR(Status)) {\r
- ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_OPEN_FAIL), gShellDebug1HiiHandle, ShellCommandLineGetRawValue(Package, 2), Status);\r
+ Status = ShellOpenFileByName (InFileName, &InFileHandle, EFI_FILE_MODE_READ, 0);\r
+ if (EFI_ERROR (Status)) {\r
+ ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_OPEN_FAIL), gShellDebug1HiiHandle, L"efidecompress", ShellCommandLineGetRawValue (Package, 1));\r
ShellStatus = SHELL_NOT_FOUND;\r
}\r
}\r
\r
if (ShellStatus == SHELL_SUCCESS) {\r
- Status = FileHandleGetSize(InFileHandle, &Temp64Bit);\r
- ASSERT(Temp64Bit <= (UINT32)(-1));\r
+ Status = FileHandleGetSize (InFileHandle, &Temp64Bit);\r
+ ASSERT (Temp64Bit <= (UINT32)(-1));\r
InSize = (UINTN)Temp64Bit;\r
- ASSERT_EFI_ERROR(Status);\r
- InBuffer = AllocateZeroPool(InSize);\r
- ASSERT(InBuffer != NULL);\r
- Status = gEfiShellProtocol->ReadFile(InFileHandle, &InSize, InBuffer);\r
- ASSERT_EFI_ERROR(Status);\r
-\r
- Status = gBS->LocateProtocol(&gEfiDecompressProtocolGuid, NULL, (VOID**)&Decompress);\r
- ASSERT_EFI_ERROR(Status);\r
+ ASSERT_EFI_ERROR (Status);\r
+ InBuffer = AllocateZeroPool (InSize);\r
+ if (InBuffer == NULL) {\r
+ Status = EFI_OUT_OF_RESOURCES;\r
+ } else {\r
+ Status = gEfiShellProtocol->ReadFile (InFileHandle, &InSize, InBuffer);\r
+ ASSERT_EFI_ERROR (Status);\r
\r
- Status = Decompress->GetInfo(Decompress, InBuffer, (UINT32)InSize, &OutSize, &ScratchSize);\r
- ASSERT_EFI_ERROR(Status);\r
+ Status = gBS->LocateProtocol (&gEfiDecompressProtocolGuid, NULL, (VOID **)&Decompress);\r
+ ASSERT_EFI_ERROR (Status);\r
\r
- OutBuffer = AllocateZeroPool(OutSize);\r
- ScratchBuffer = AllocateZeroPool(ScratchSize);\r
- ASSERT(OutBuffer != NULL);\r
- ASSERT(ScratchBuffer != NULL);\r
+ Status = Decompress->GetInfo (Decompress, InBuffer, (UINT32)InSize, &OutSize, &ScratchSize);\r
+ }\r
\r
- Status = Decompress->Decompress(Decompress, InBuffer, (UINT32)InSize, OutBuffer, OutSize, ScratchBuffer, ScratchSize);\r
- ASSERT_EFI_ERROR(Status);\r
+ if (EFI_ERROR (Status) || (OutSize == 0)) {\r
+ ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_EFI_DECOMPRESS_NOPE), gShellDebug1HiiHandle, InFileName);\r
+ ShellStatus = SHELL_NOT_FOUND;\r
+ } else {\r
+ Status = ShellOpenFileByName (OutFileName, &OutFileHandle, EFI_FILE_MODE_READ|EFI_FILE_MODE_WRITE|EFI_FILE_MODE_CREATE, 0);\r
+ if (EFI_ERROR (Status)) {\r
+ ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_FILE_OPEN_FAIL), gShellDebug1HiiHandle, ShellCommandLineGetRawValue (Package, 2), Status);\r
+ ShellStatus = SHELL_NOT_FOUND;\r
+ } else {\r
+ OutBuffer = AllocateZeroPool (OutSize);\r
+ ScratchBuffer = AllocateZeroPool (ScratchSize);\r
+ if ((OutBuffer == NULL) || (ScratchBuffer == NULL)) {\r
+ Status = EFI_OUT_OF_RESOURCES;\r
+ } else {\r
+ Status = Decompress->Decompress (Decompress, InBuffer, (UINT32)InSize, OutBuffer, OutSize, ScratchBuffer, ScratchSize);\r
+ }\r
+ }\r
+ }\r
\r
- if (EFI_ERROR(Status)) {\r
- ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_EFI_DECOMPRESS_FAIL), gShellDebug1HiiHandle, Status);\r
- ShellStatus = SHELL_DEVICE_ERROR;\r
+ if (EFI_ERROR (Status)) {\r
+ ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_EFI_DECOMPRESS_FAIL), gShellDebug1HiiHandle, Status);\r
+ ShellStatus = ((Status == EFI_OUT_OF_RESOURCES) ? SHELL_OUT_OF_RESOURCES : SHELL_DEVICE_ERROR);\r
} else {\r
OutSizeTemp = OutSize;\r
- Status = gEfiShellProtocol->WriteFile(OutFileHandle, &OutSizeTemp, OutBuffer);\r
- OutSize = (UINT32)OutSizeTemp;\r
- if (EFI_ERROR(Status)) {\r
- ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_WRITE_FAIL), gShellDebug1HiiHandle, OutFileName, Status);\r
+ Status = gEfiShellProtocol->WriteFile (OutFileHandle, &OutSizeTemp, OutBuffer);\r
+ OutSize = (UINT32)OutSizeTemp;\r
+ if (EFI_ERROR (Status)) {\r
+ ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_FILE_WRITE_FAIL), gShellDebug1HiiHandle, L"efidecompress", OutFileName, Status);\r
ShellStatus = SHELL_DEVICE_ERROR;\r
}\r
}\r
\r
ShellCommandLineFreeVarList (Package);\r
}\r
+\r
if (InFileHandle != NULL) {\r
- gEfiShellProtocol->CloseFile(InFileHandle);\r
+ gEfiShellProtocol->CloseFile (InFileHandle);\r
}\r
+\r
if (OutFileHandle != NULL) {\r
- gEfiShellProtocol->CloseFile(OutFileHandle);\r
+ gEfiShellProtocol->CloseFile (OutFileHandle);\r
}\r
\r
- SHELL_FREE_NON_NULL(InFileName);\r
- SHELL_FREE_NON_NULL(InBuffer);\r
- SHELL_FREE_NON_NULL(OutBuffer);\r
- SHELL_FREE_NON_NULL(ScratchBuffer);\r
+ SHELL_FREE_NON_NULL (InFileName);\r
+ SHELL_FREE_NON_NULL (InBuffer);\r
+ SHELL_FREE_NON_NULL (OutBuffer);\r
+ SHELL_FREE_NON_NULL (ScratchBuffer);\r
\r
return (ShellStatus);\r
}\r
projects / mirror_edk2.git / blobdiff