MdeModulePkg BrotliCustomDecompressLib: Remove the duplicated functions

[mirror_edk2.git] / MdeModulePkg / Library / BrotliCustomDecompressLib / BrotliDecompress.c

/** @file\r
  Brotli Decompress interfaces\r
\r
  Copyright (c) 2017 - 2018, Intel Corporation. All rights reserved.<BR>\r
  SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
#include <BrotliDecompressLibInternal.h>\r
\r
/**\r
  Allocation routine used by BROTLI decompression.\r
\r
  @param Ptr              Pointer to the BROTLI_BUFF instance.\r
  @param Size             The size in bytes to be allocated.\r
\r
  @return The allocated pointer address, or NULL on failure\r
**/\r
VOID *\r
BrAlloc (\r
  IN VOID *    Ptr,\r
  IN size_t    Size\r
  )\r
{\r
  VOID          *Addr;\r
  BROTLI_BUFF   *Private;\r
\r
  Private = (BROTLI_BUFF *)Ptr;\r
\r
  if (Private->BuffSize >= Size) {\r
    Addr = Private->Buff;\r
    Private->Buff = (VOID *) ((UINT8 *)Addr + Size);\r
    Private->BuffSize -= Size;\r
    return Addr;\r
  } else {\r
    ASSERT (FALSE);\r
    return NULL;\r
  }\r
}\r
\r
/**\r
  Free routine used by BROTLI decompression.\r
\r
  @param Ptr              Pointer to the BROTLI_BUFF instance\r
  @param Address          The address to be freed\r
**/\r
VOID\r
BrFree (\r
  IN VOID *    Ptr,\r
  IN VOID *    Address\r
  )\r
{\r
  //\r
  // We use the 'scratch buffer' for allocations, so there is no free\r
  // operation required.  The scratch buffer will be freed by the caller\r
  // of the decompression code.\r
  //\r
}\r
\r
/**\r
  Decompresses a Brotli compressed source buffer.\r
\r
  Extracts decompressed data to its original form.\r
  If the compressed source data specified by Source is successfully decompressed\r
  into Destination, then EFI_SUCCESS is returned. If the compressed source data\r
  specified by Source is not in a valid compressed data format,\r
  then EFI_INVALID_PARAMETER is returned.\r
\r
  @param  Source      The source buffer containing the compressed data.\r
  @param  SourceSize  The size of source buffer.\r
  @param  Destination The destination buffer to store the decompressed data.\r
  @param  DestSize    The destination buffer size.\r
  @param  BuffInfo    The pointer to the BROTLI_BUFF instance.\r
\r
  @retval EFI_SUCCESS Decompression completed successfully, and\r
                      the uncompressed buffer is returned in Destination.\r
  @retval EFI_INVALID_PARAMETER\r
                      The source buffer specified by Source is corrupted\r
                      (not in a valid compressed format).\r
**/\r
EFI_STATUS\r
BrotliDecompress (\r
  IN CONST VOID*  Source,\r
  IN UINTN        SourceSize,\r
  IN OUT VOID*    Destination,\r
  IN OUT UINTN    DestSize,\r
  IN VOID *       BuffInfo\r
  )\r
{\r
  UINT8 *        Input;\r
  UINT8 *        Output;\r
  const UINT8 *  NextIn;\r
  UINT8 *        NextOut;\r
  size_t         TotalOut;\r
  size_t         AvailableIn;\r
  size_t         AvailableOut;\r
  VOID *         Temp;\r
  BrotliDecoderResult   Result;\r
  BrotliDecoderState *  BroState;\r
\r
  TotalOut = 0;\r
  AvailableOut = FILE_BUFFER_SIZE;\r
  Result = BROTLI_DECODER_RESULT_ERROR;\r
  BroState = BrotliDecoderCreateInstance(BrAlloc, BrFree, BuffInfo);\r
  Temp = Destination;\r
\r
  if (BroState == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
  Input = (UINT8 *)BrAlloc(BuffInfo, FILE_BUFFER_SIZE);\r
  Output = (UINT8 *)BrAlloc(BuffInfo, FILE_BUFFER_SIZE);\r
  if ((Input==NULL) || (Output==NULL)) {\r
    BrFree(BuffInfo, Input);\r
    BrFree(BuffInfo, Output);\r
    BrotliDecoderDestroyInstance(BroState);\r
    return EFI_INVALID_PARAMETER;\r
  }\r
  NextOut = Output;\r
  Result = BROTLI_DECODER_RESULT_NEEDS_MORE_INPUT;\r
  while (1) {\r
    if (Result == BROTLI_DECODER_RESULT_NEEDS_MORE_INPUT) {\r
      if (SourceSize == 0) {\r
        break;\r
      }\r
      if (SourceSize >= FILE_BUFFER_SIZE) {\r
        AvailableIn = FILE_BUFFER_SIZE;\r
      }else{\r
        AvailableIn = SourceSize;\r
      }\r
      CopyMem(Input, Source, AvailableIn);\r
      Source = (VOID *)((UINT8 *)Source + AvailableIn);\r
      SourceSize -= AvailableIn;\r
      NextIn = Input;\r
    } else if (Result == BROTLI_DECODER_RESULT_NEEDS_MORE_OUTPUT) {\r
      CopyMem(Temp, Output, FILE_BUFFER_SIZE);\r
      AvailableOut = FILE_BUFFER_SIZE;\r
      Temp = (VOID *)((UINT8 *)Temp +FILE_BUFFER_SIZE);\r
      NextOut = Output;\r
    } else {\r
      break; /* Error or success. */\r
    }\r
    Result = BrotliDecoderDecompressStream(\r
                          BroState,\r
                          &AvailableIn,\r
                          &NextIn,\r
                          &AvailableOut,\r
                          &NextOut,\r
                          &TotalOut\r
                          );\r
  }\r
  if (NextOut != Output) {\r
    CopyMem(Temp, Output, (size_t)(NextOut - Output));\r
  }\r
\r
  DestSize = TotalOut;\r
\r
  BrFree(BuffInfo, Input);\r
  BrFree(BuffInfo, Output);\r
  BrotliDecoderDestroyInstance(BroState);\r
  return (Result == BROTLI_DECODER_RESULT_SUCCESS) ? EFI_SUCCESS : EFI_INVALID_PARAMETER;\r
}\r
\r
/**\r
  Get the size of the uncompressed buffer by parsing EncodeData header.\r
\r
  @param EncodedData  Pointer to the compressed data.\r
  @param StartOffset  Start offset of the compressed data.\r
  @param EndOffset    End offset of the compressed data.\r
\r
  @return The size of the uncompressed buffer.\r
**/\r
UINT64\r
BrGetDecodedSizeOfBuf(\r
  IN UINT8 *  EncodedData,\r
  IN UINT8    StartOffset,\r
  IN UINT8    EndOffset\r
  )\r
{\r
  UINT64 DecodedSize;\r
  INTN   Index;\r
\r
  /* Parse header */\r
  DecodedSize = 0;\r
  for (Index = EndOffset - 1; Index >= StartOffset; Index--)\r
    DecodedSize = LShiftU64(DecodedSize, 8) + EncodedData[Index];\r
\r
  return DecodedSize;\r
}\r
\r
/**\r
  Given a Brotli compressed source buffer, this function retrieves the size of\r
  the uncompressed buffer and the size of the scratch buffer required\r
  to decompress the compressed source buffer.\r
\r
  Retrieves the size of the uncompressed buffer and the temporary scratch buffer\r
  required to decompress the buffer specified by Source and SourceSize.\r
  The size of the uncompressed buffer is returned in DestinationSize,\r
  the size of the scratch buffer is returned in ScratchSize, and EFI_SUCCESS is returned.\r
  This function does not have scratch buffer available to perform a thorough\r
  checking of the validity of the source data. It just retrieves the "Original Size"\r
  field from the BROTLI_SCRATCH_MAX beginning bytes of the source data and output it as DestinationSize.\r
  And ScratchSize is specific to the decompression implementation.\r
\r
  If SourceSize is less than BROTLI_SCRATCH_MAX, then ASSERT().\r
\r
  @param  Source          The source buffer containing the compressed data.\r
  @param  SourceSize      The size, in bytes, of the source buffer.\r
  @param  DestinationSize A pointer to the size, in bytes, of the uncompressed buffer\r
                          that will be generated when the compressed buffer specified\r
                          by Source and SourceSize is decompressed.\r
  @param  ScratchSize     A pointer to the size, in bytes, of the scratch buffer that\r
                          is required to decompress the compressed buffer specified\r
                          by Source and SourceSize.\r
\r
  @retval EFI_SUCCESS     The size of the uncompressed data was returned\r
                          in DestinationSize and the size of the scratch\r
                          buffer was returned in ScratchSize.\r
**/\r
EFI_STATUS\r
EFIAPI\r
BrotliUefiDecompressGetInfo (\r
  IN  CONST VOID *  Source,\r
  IN  UINT32        SourceSize,\r
  OUT UINT32 *      DestinationSize,\r
  OUT UINT32 *      ScratchSize\r
  )\r
{\r
  UINT64  GetSize;\r
  UINT8   MaxOffset;\r
\r
  ASSERT(SourceSize >= BROTLI_SCRATCH_MAX);\r
\r
  MaxOffset = BROTLI_DECODE_MAX;\r
  GetSize = BrGetDecodedSizeOfBuf((UINT8 *)Source, MaxOffset - BROTLI_INFO_SIZE, MaxOffset);\r
  *DestinationSize = (UINT32)GetSize;\r
  MaxOffset = BROTLI_SCRATCH_MAX;\r
  GetSize = BrGetDecodedSizeOfBuf((UINT8 *)Source, MaxOffset - BROTLI_INFO_SIZE, MaxOffset);\r
  *ScratchSize = (UINT32)GetSize;\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Decompresses a Brotli compressed source buffer.\r
\r
  Extracts decompressed data to its original form.\r
  If the compressed source data specified by Source is successfully decompressed\r
  into Destination, then RETURN_SUCCESS is returned.  If the compressed source data\r
  specified by Source is not in a valid compressed data format,\r
  then RETURN_INVALID_PARAMETER is returned.\r
\r
  @param  Source      The source buffer containing the compressed data.\r
  @param  SourceSize  The size of source buffer.\r
  @param  Destination The destination buffer to store the decompressed data\r
  @param  Scratch     A temporary scratch buffer that is used to perform the decompression.\r
                      This is an optional parameter that may be NULL if the\r
                      required scratch buffer size is 0.\r
\r
  @retval EFI_SUCCESS Decompression completed successfully, and\r
                      the uncompressed buffer is returned in Destination.\r
  @retval EFI_INVALID_PARAMETER\r
                      The source buffer specified by Source is corrupted\r
                      (not in a valid compressed format).\r
**/\r
EFI_STATUS\r
EFIAPI\r
BrotliUefiDecompress (\r
  IN CONST VOID *   Source,\r
  IN UINTN          SourceSize,\r
  IN OUT VOID *     Destination,\r
  IN OUT VOID *     Scratch\r
  )\r
{\r
  UINTN          DestSize = 0;\r
  EFI_STATUS     Status;\r
  BROTLI_BUFF    BroBuff;\r
  UINT64         GetSize;\r
  UINT8          MaxOffset;\r
\r
  MaxOffset = BROTLI_SCRATCH_MAX;\r
  GetSize = BrGetDecodedSizeOfBuf((UINT8 *)Source, MaxOffset - BROTLI_INFO_SIZE, MaxOffset);\r
\r
  BroBuff.Buff     = Scratch;\r
  BroBuff.BuffSize = (UINTN)GetSize;\r
\r
  Status = BrotliDecompress(\r
            (VOID *)((UINT8 *)Source + BROTLI_SCRATCH_MAX),\r
            SourceSize - BROTLI_SCRATCH_MAX,\r
            Destination,\r
            DestSize,\r
            (VOID *)(&BroBuff)\r
            );\r
\r
  return Status;\r
}\r