BaseUefiTianoCustomDecompressLib.c

Go to the documentation of this file.

 
#include <PiPei.h>
#include <Library/ExtractGuidedSectionLib.h>
#include "BaseUefiDecompressLibInternals.h"
 
RETURN_STATUS
EFIAPI
TianoDecompressGetInfo (
  IN  CONST VOID  *InputSection,
  OUT UINT32      *OutputBufferSize,
  OUT UINT32      *ScratchBufferSize,
  OUT UINT16      *SectionAttribute
  )
 
{
  ASSERT (SectionAttribute != NULL);
 
  if (InputSection == NULL) {
    return RETURN_INVALID_PARAMETER;
  }
 
  if (IS_SECTION2 (InputSection)) {
    if (!CompareGuid (
           &gTianoCustomDecompressGuid,
           &(((EFI_GUID_DEFINED_SECTION2 *)InputSection)->SectionDefinitionGuid)
           ))
    {
      return RETURN_INVALID_PARAMETER;
    }
 
    //
    // Get guid attribute of guid section.
    //
    *SectionAttribute = ((EFI_GUID_DEFINED_SECTION2 *)InputSection)->Attributes;
 
    //
    // Call Tiano GetInfo to get the required size info.
    //
    return UefiDecompressGetInfo (
             (UINT8 *)InputSection + ((EFI_GUID_DEFINED_SECTION2 *)InputSection)->DataOffset,
             SECTION2_SIZE (InputSection) - ((EFI_GUID_DEFINED_SECTION2 *)InputSection)->DataOffset,
             OutputBufferSize,
             ScratchBufferSize
             );
  } else {
    if (!CompareGuid (
           &gTianoCustomDecompressGuid,
           &(((EFI_GUID_DEFINED_SECTION *)InputSection)->SectionDefinitionGuid)
           ))
    {
      return RETURN_INVALID_PARAMETER;
    }
 
    //
    // Get guid attribute of guid section.
    //
    *SectionAttribute = ((EFI_GUID_DEFINED_SECTION *)InputSection)->Attributes;
 
    //
    // Call Tiano GetInfo to get the required size info.
    //
    return UefiDecompressGetInfo (
             (UINT8 *)InputSection + ((EFI_GUID_DEFINED_SECTION *)InputSection)->DataOffset,
             SECTION_SIZE (InputSection) - ((EFI_GUID_DEFINED_SECTION *)InputSection)->DataOffset,
             OutputBufferSize,
             ScratchBufferSize
             );
  }
}
 
RETURN_STATUS
EFIAPI
TianoDecompress (
  IN CONST  VOID    *InputSection,
  OUT       VOID    **OutputBuffer,
  IN        VOID    *ScratchBuffer         OPTIONAL,
  OUT       UINT32  *AuthenticationStatus
  )
{
  ASSERT (OutputBuffer != NULL);
  ASSERT (InputSection != NULL);
 
  if (IS_SECTION2 (InputSection)) {
    if (!CompareGuid (
           &gTianoCustomDecompressGuid,
           &(((EFI_GUID_DEFINED_SECTION2 *)InputSection)->SectionDefinitionGuid)
           ))
    {
      return RETURN_INVALID_PARAMETER;
    }
 
    //
    // Set Authentication to Zero.
    //
    *AuthenticationStatus = 0;
 
    //
    // Call Tiano Decompress to get the raw data
    //
    return UefiTianoDecompress (
             (UINT8 *)InputSection + ((EFI_GUID_DEFINED_SECTION2 *)InputSection)->DataOffset,
             *OutputBuffer,
             ScratchBuffer,
             2
             );
  } else {
    if (!CompareGuid (
           &gTianoCustomDecompressGuid,
           &(((EFI_GUID_DEFINED_SECTION *)InputSection)->SectionDefinitionGuid)
           ))
    {
      return RETURN_INVALID_PARAMETER;
    }
 
    //
    // Set Authentication to Zero.
    //
    *AuthenticationStatus = 0;
 
    //
    // Call Tiano Decompress to get the raw data
    //
    return UefiTianoDecompress (
             (UINT8 *)InputSection + ((EFI_GUID_DEFINED_SECTION *)InputSection)->DataOffset,
             *OutputBuffer,
             ScratchBuffer,
             2
             );
  }
}
 
RETURN_STATUS
EFIAPI
TianoDecompressLibConstructor (
  VOID
  )
{
  return ExtractGuidedSectionRegisterHandlers (
           &gTianoCustomDecompressGuid,
           TianoDecompressGetInfo,
           TianoDecompress
           );
}