Fix a bug in UefiHiiLib. The size for allocating a buffer is StrSize instead of StrLen.

[mirror_edk2.git] / IntelFrameworkModulePkg / Library / LzmaCustomDecompressLib / LzmaDecompress.c

/** @file\r
  LZMA Decompress routines for edk2\r
\r
  Portions based on LZMA SDK 4.65:\r
    LzmaUtil.c -- Test application for LZMA compression\r
    2008-11-23 : Igor Pavlov : Public domain\r
\r
  Copyright (c) 2006 - 2009, Intel Corporation<BR>\r
  All rights reserved. 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 <Uefi.h>\r
#include <Library/BaseLib.h>\r
#include <Library/BaseMemoryLib.h>\r
#include <Library/DebugLib.h>\r
#include <Library/MemoryAllocationLib.h>\r
#include <Library/UefiDecompressLib.h>\r
#include <Library/ExtractGuidedSectionLib.h>\r
#include <Guid/LzmaDecompress.h>\r
\r
#include "Sdk/C/Types.h"\r
#include "Sdk/C/7zVersion.h"\r
#include "Sdk/C/LzmaDec.h"\r
\r
extern ISzAlloc g_Alloc;\r
\r
#define LZMA_HEADER_SIZE (LZMA_PROPS_SIZE + 8)\r
\r
STATIC\r
UINT64\r
GetDecodedSizeOfBuf(\r
  UINT8 *encodedData\r
  )\r
{\r
  UINT64 DecodedSize;\r
  INTN   Index;\r
\r
  /* Parse header */\r
  DecodedSize = 0;\r
  for (Index = LZMA_PROPS_SIZE + 7; Index >= LZMA_PROPS_SIZE; Index--)\r
    DecodedSize = LShiftU64(DecodedSize, 8) + encodedData[Index];\r
\r
  return DecodedSize;\r
}\r
\r
//\r
// LZMA functions and data as defined in local LzmaDecompress.h\r
//\r
\r
STATIC CONST VOID  *mSourceLastUsedWithGetInfo;\r
STATIC UINT32      mSizeOfLastSource;\r
STATIC UINT32      mDecompressedSizeForLastSource;\r
\r
/**\r
  The internal implementation of *_DECOMPRESS_PROTOCOL.GetInfo().\r
  \r
  @param Source           The source buffer containing the compressed data.\r
  @param SourceSize       The size of source buffer\r
  @param DestinationSize  The size of destination buffer.\r
  @param ScratchSize      The size of scratch buffer.\r
\r
  @retval RETURN_SUCCESS           - The size of destination buffer and the size of scratch buffer are successull retrieved.\r
  @retval RETURN_INVALID_PARAMETER - The source data is corrupted\r
**/\r
RETURN_STATUS\r
EFIAPI\r
LzmaUefiDecompressGetInfo (\r
  IN  CONST VOID  *Source,\r
  IN  UINT32      SourceSize,\r
  OUT UINT32      *DestinationSize,\r
  OUT UINT32      *ScratchSize\r
  )\r
{\r
  UInt64  DecodedSize;\r
\r
  ASSERT(SourceSize >= LZMA_HEADER_SIZE);\r
\r
  DecodedSize = GetDecodedSizeOfBuf((UINT8*)Source);\r
\r
  mSourceLastUsedWithGetInfo = Source;\r
  mSizeOfLastSource = SourceSize;\r
  mDecompressedSizeForLastSource = (UInt32)DecodedSize;\r
  *DestinationSize = mDecompressedSizeForLastSource;\r
  *ScratchSize = 0x10;\r
  return RETURN_SUCCESS;\r
}\r
\r
\r
/**\r
  The internal implementation of *_DECOMPRESS_PROTOCOL.Decompress().\r
  \r
  @param Source          - The source buffer containing the compressed data.\r
  @param Destination     - The destination buffer to store the decompressed data\r
  @param Scratch         - The buffer used internally by the decompress routine. This  buffer is needed to store intermediate data.\r
\r
  @retval RETURN_SUCCESS           - Decompression is successfull\r
  @retval RETURN_INVALID_PARAMETER - The source data is corrupted  \r
**/\r
RETURN_STATUS\r
EFIAPI\r
LzmaUefiDecompress (\r
  IN CONST VOID  *Source,\r
  IN OUT VOID    *Destination,\r
  IN OUT VOID    *Scratch\r
  )\r
{\r
  SRes        lzmaResult;\r
  ELzmaStatus status;\r
  SizeT       decodedBufSize;\r
  SizeT       encodedDataSize;\r
\r
  if (Source != mSourceLastUsedWithGetInfo) {\r
    return RETURN_INVALID_PARAMETER;\r
  }\r
\r
  decodedBufSize = (SizeT)mDecompressedSizeForLastSource;\r
  encodedDataSize = (SizeT)(mSizeOfLastSource - LZMA_HEADER_SIZE);\r
\r
  lzmaResult = LzmaDecode(\r
    Destination,\r
    &decodedBufSize,\r
    (Byte*)((UINT8*)Source + LZMA_HEADER_SIZE),\r
    &encodedDataSize,\r
    Source,\r
    LZMA_PROPS_SIZE,\r
    LZMA_FINISH_END,\r
    &status,\r
    &g_Alloc\r
    );\r
\r
  if (lzmaResult == SZ_OK) {\r
    return RETURN_SUCCESS;\r
  } else {\r
    return RETURN_INVALID_PARAMETER;\r
  }\r
}\r
\r