Signed-off-by: lzeng14 Reviewed-by: lgao4 git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@12582 6f19259b-4bc3-4df7-8a09-765794883524
		
			
				
	
	
		
			237 lines
		
	
	
		
			8.6 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			237 lines
		
	
	
		
			8.6 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/** @file
 | 
						|
 | 
						|
  This library registers CRC32 guided section handler 
 | 
						|
  to parse CRC32 encapsulation section and extract raw data.
 | 
						|
  It uses UEFI boot service CalculateCrc32 to authenticate 32 bit CRC value.
 | 
						|
 | 
						|
Copyright (c) 2007 - 2011, Intel Corporation. All rights reserved.<BR>
 | 
						|
This program and the accompanying materials                          
 | 
						|
are licensed and made available under the terms and conditions of the BSD License         
 | 
						|
which accompanies this distribution.  The full text of the license may be found at        
 | 
						|
http://opensource.org/licenses/bsd-license.php                                            
 | 
						|
                                                                                          
 | 
						|
THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,                     
 | 
						|
WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.             
 | 
						|
 | 
						|
**/
 | 
						|
 | 
						|
#include <PiDxe.h>
 | 
						|
#include <Guid/Crc32GuidedSectionExtraction.h>
 | 
						|
#include <Protocol/SecurityPolicy.h>
 | 
						|
#include <Library/ExtractGuidedSectionLib.h>
 | 
						|
#include <Library/DebugLib.h>
 | 
						|
#include <Library/BaseMemoryLib.h>
 | 
						|
#include <Library/UefiBootServicesTableLib.h>
 | 
						|
 | 
						|
///
 | 
						|
/// CRC32 Guided Section header
 | 
						|
///
 | 
						|
typedef struct {
 | 
						|
  EFI_GUID_DEFINED_SECTION  GuidedSectionHeader; ///< EFI guided section header
 | 
						|
  UINT32                    CRC32Checksum;       ///< 32bit CRC check sum
 | 
						|
} CRC32_SECTION_HEADER;
 | 
						|
 | 
						|
typedef struct {
 | 
						|
  EFI_GUID_DEFINED_SECTION2 GuidedSectionHeader; ///< EFI guided section header
 | 
						|
  UINT32                    CRC32Checksum;       ///< 32bit CRC check sum
 | 
						|
} CRC32_SECTION2_HEADER;
 | 
						|
 | 
						|
/**
 | 
						|
 | 
						|
  GetInfo gets raw data size and attribute of the input guided section.
 | 
						|
  It first checks whether the input guid section is supported. 
 | 
						|
  If not, EFI_INVALID_PARAMETER will return.
 | 
						|
 | 
						|
  @param InputSection       Buffer containing the input GUIDed section to be processed.
 | 
						|
  @param OutputBufferSize   The size of OutputBuffer.
 | 
						|
  @param ScratchBufferSize  The size of ScratchBuffer.
 | 
						|
  @param SectionAttribute   The attribute of the input guided section.
 | 
						|
 | 
						|
  @retval EFI_SUCCESS            The size of destination buffer, the size of scratch buffer and 
 | 
						|
                                 the attribute of the input section are successull retrieved.
 | 
						|
  @retval EFI_INVALID_PARAMETER  The GUID in InputSection does not match this instance guid.
 | 
						|
 | 
						|
**/
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
Crc32GuidedSectionGetInfo (
 | 
						|
  IN  CONST VOID  *InputSection,
 | 
						|
  OUT UINT32      *OutputBufferSize,
 | 
						|
  OUT UINT32      *ScratchBufferSize,
 | 
						|
  OUT UINT16      *SectionAttribute
 | 
						|
  )
 | 
						|
{
 | 
						|
  if (IS_SECTION2 (InputSection)) {
 | 
						|
    //
 | 
						|
    // Check whether the input guid section is recognized.
 | 
						|
    //
 | 
						|
    if (!CompareGuid (
 | 
						|
        &gEfiCrc32GuidedSectionExtractionGuid,
 | 
						|
        &(((EFI_GUID_DEFINED_SECTION2 *) InputSection)->SectionDefinitionGuid))) {
 | 
						|
      return EFI_INVALID_PARAMETER;
 | 
						|
    }
 | 
						|
    //
 | 
						|
    // Retrieve the size and attribute of the input section data.
 | 
						|
    //
 | 
						|
    *SectionAttribute  = ((EFI_GUID_DEFINED_SECTION2 *) InputSection)->Attributes;
 | 
						|
    *ScratchBufferSize = 0;
 | 
						|
    *OutputBufferSize  = SECTION2_SIZE (InputSection) - ((EFI_GUID_DEFINED_SECTION2 *) InputSection)->DataOffset;
 | 
						|
  } else {
 | 
						|
    //
 | 
						|
    // Check whether the input guid section is recognized.
 | 
						|
    //
 | 
						|
    if (!CompareGuid (
 | 
						|
        &gEfiCrc32GuidedSectionExtractionGuid,
 | 
						|
        &(((EFI_GUID_DEFINED_SECTION *) InputSection)->SectionDefinitionGuid))) {
 | 
						|
      return EFI_INVALID_PARAMETER;
 | 
						|
    }
 | 
						|
    //
 | 
						|
    // Retrieve the size and attribute of the input section data.
 | 
						|
    //
 | 
						|
    *SectionAttribute  = ((EFI_GUID_DEFINED_SECTION *) InputSection)->Attributes;
 | 
						|
    *ScratchBufferSize = 0;
 | 
						|
    *OutputBufferSize  = SECTION_SIZE (InputSection) - ((EFI_GUID_DEFINED_SECTION *) InputSection)->DataOffset;
 | 
						|
  }
 | 
						|
 | 
						|
  return EFI_SUCCESS;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 | 
						|
  Extraction handler tries to extract raw data from the input guided section.
 | 
						|
  It also does authentication check for 32bit CRC value in the input guided section.
 | 
						|
  It first checks whether the input guid section is supported. 
 | 
						|
  If not, EFI_INVALID_PARAMETER will return.
 | 
						|
 | 
						|
  @param InputSection    Buffer containing the input GUIDed section to be processed.
 | 
						|
  @param OutputBuffer    Buffer to contain the output raw data allocated by the caller.
 | 
						|
  @param ScratchBuffer   A pointer to a caller-allocated buffer for function internal use.
 | 
						|
  @param AuthenticationStatus A pointer to a caller-allocated UINT32 that indicates the
 | 
						|
                              authentication status of the output buffer.
 | 
						|
 | 
						|
  @retval EFI_SUCCESS            Section Data and Auth Status is extracted successfully.
 | 
						|
  @retval EFI_INVALID_PARAMETER  The GUID in InputSection does not match this instance guid.
 | 
						|
 | 
						|
**/
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
Crc32GuidedSectionHandler (
 | 
						|
  IN CONST  VOID    *InputSection,
 | 
						|
  OUT       VOID    **OutputBuffer,
 | 
						|
  IN        VOID    *ScratchBuffer,        OPTIONAL
 | 
						|
  OUT       UINT32  *AuthenticationStatus
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS                Status;
 | 
						|
  UINT32                    SectionCrc32Checksum;
 | 
						|
  UINT32                    Crc32Checksum;
 | 
						|
  UINT32                    OutputBufferSize;
 | 
						|
  VOID                      *DummyInterface;
 | 
						|
 | 
						|
  if (IS_SECTION2 (InputSection)) {
 | 
						|
    //
 | 
						|
    // Check whether the input guid section is recognized.
 | 
						|
    //
 | 
						|
    if (!CompareGuid (
 | 
						|
        &gEfiCrc32GuidedSectionExtractionGuid,
 | 
						|
        &(((EFI_GUID_DEFINED_SECTION2 *) InputSection)->SectionDefinitionGuid))) {
 | 
						|
      return EFI_INVALID_PARAMETER;
 | 
						|
    }
 | 
						|
  
 | 
						|
    //
 | 
						|
    // Get section Crc32 checksum.
 | 
						|
    //
 | 
						|
    SectionCrc32Checksum = ((CRC32_SECTION2_HEADER *) InputSection)->CRC32Checksum;
 | 
						|
    *OutputBuffer      = (UINT8 *) InputSection + ((EFI_GUID_DEFINED_SECTION2 *) InputSection)->DataOffset;
 | 
						|
    OutputBufferSize   = SECTION2_SIZE (InputSection) - ((EFI_GUID_DEFINED_SECTION2 *) InputSection)->DataOffset;
 | 
						|
 | 
						|
    //
 | 
						|
    // Implicitly CRC32 GUIDed section should have STATUS_VALID bit set
 | 
						|
    //
 | 
						|
    ASSERT (((EFI_GUID_DEFINED_SECTION2 *) InputSection)->Attributes & EFI_GUIDED_SECTION_AUTH_STATUS_VALID);
 | 
						|
    *AuthenticationStatus = EFI_AUTH_STATUS_IMAGE_SIGNED;
 | 
						|
  } else {
 | 
						|
    //
 | 
						|
    // Check whether the input guid section is recognized.
 | 
						|
    //
 | 
						|
    if (!CompareGuid (
 | 
						|
        &gEfiCrc32GuidedSectionExtractionGuid,
 | 
						|
        &(((EFI_GUID_DEFINED_SECTION *) InputSection)->SectionDefinitionGuid))) {
 | 
						|
      return EFI_INVALID_PARAMETER;
 | 
						|
    }
 | 
						|
  
 | 
						|
    //
 | 
						|
    // Get section Crc32 checksum.
 | 
						|
    //
 | 
						|
    SectionCrc32Checksum = ((CRC32_SECTION_HEADER *) InputSection)->CRC32Checksum;
 | 
						|
    *OutputBuffer      = (UINT8 *) InputSection + ((EFI_GUID_DEFINED_SECTION *) InputSection)->DataOffset;
 | 
						|
    OutputBufferSize   = SECTION_SIZE (InputSection) - ((EFI_GUID_DEFINED_SECTION *) InputSection)->DataOffset;
 | 
						|
 | 
						|
    //
 | 
						|
    // Implicitly CRC32 GUIDed section should have STATUS_VALID bit set
 | 
						|
    //
 | 
						|
    ASSERT (((EFI_GUID_DEFINED_SECTION *) InputSection)->Attributes & EFI_GUIDED_SECTION_AUTH_STATUS_VALID);
 | 
						|
    *AuthenticationStatus = EFI_AUTH_STATUS_IMAGE_SIGNED;
 | 
						|
  }
 | 
						|
 | 
						|
  //
 | 
						|
  // Init Checksum value to Zero.
 | 
						|
  //
 | 
						|
  Crc32Checksum = 0;
 | 
						|
 | 
						|
  //
 | 
						|
  // Check whether there exists EFI_SECURITY_POLICY_PROTOCOL_GUID.
 | 
						|
  //
 | 
						|
  Status = gBS->LocateProtocol (&gEfiSecurityPolicyProtocolGuid, NULL, &DummyInterface);
 | 
						|
  if (!EFI_ERROR (Status)) {
 | 
						|
    //
 | 
						|
    // If SecurityPolicy Protocol exist, AUTH platform override bit is set.
 | 
						|
    //
 | 
						|
    *AuthenticationStatus |= EFI_AUTH_STATUS_PLATFORM_OVERRIDE;
 | 
						|
  } else {
 | 
						|
    //
 | 
						|
    // Calculate CRC32 Checksum of Image
 | 
						|
    //
 | 
						|
    Status = gBS->CalculateCrc32 (*OutputBuffer, OutputBufferSize, &Crc32Checksum);
 | 
						|
    if (Status == EFI_SUCCESS) {
 | 
						|
      if (Crc32Checksum != SectionCrc32Checksum) {
 | 
						|
        //
 | 
						|
        // If Crc32 checksum is not matched, AUTH tested failed bit is set.
 | 
						|
        //
 | 
						|
        *AuthenticationStatus |= EFI_AUTH_STATUS_TEST_FAILED;
 | 
						|
      }
 | 
						|
    } else {
 | 
						|
      //
 | 
						|
      // If Crc32 checksum is not calculated, AUTH not tested bit is set.
 | 
						|
      //
 | 
						|
      *AuthenticationStatus |= EFI_AUTH_STATUS_NOT_TESTED;
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  return EFI_SUCCESS;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
  Register the handler to extract CRC32 guided section.
 | 
						|
 | 
						|
  @param  ImageHandle  ImageHandle of the loaded driver.
 | 
						|
  @param  SystemTable  Pointer to the EFI System Table.
 | 
						|
 | 
						|
  @retval  EFI_SUCCESS            Register successfully.
 | 
						|
  @retval  EFI_OUT_OF_RESOURCES   No enough memory to register this handler.
 | 
						|
**/
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
DxeCrc32GuidedSectionExtractLibConstructor (
 | 
						|
  IN EFI_HANDLE        ImageHandle,
 | 
						|
  IN EFI_SYSTEM_TABLE  *SystemTable
 | 
						|
  )
 | 
						|
{
 | 
						|
  return ExtractGuidedSectionRegisterHandlers (
 | 
						|
          &gEfiCrc32GuidedSectionExtractionGuid,
 | 
						|
          Crc32GuidedSectionGetInfo,
 | 
						|
          Crc32GuidedSectionHandler
 | 
						|
          );
 | 
						|
}
 | 
						|
 |