MdeModulePkg: Clean up source files

[mirror_edk2.git] / MdeModulePkg / Universal / SectionExtractionDxe / SectionExtractionDxe.c

/** @file\r
 Section Extraction DXE Driver\r
\r
Copyright (c) 2013 - 2018, 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
\r
**/\r
\r
#include <PiDxe.h>\r
#include <Protocol/GuidedSectionExtraction.h>\r
#include <Library/DebugLib.h>\r
#include <Library/ExtractGuidedSectionLib.h>\r
#include <Library/MemoryAllocationLib.h>\r
#include <Library/BaseMemoryLib.h>\r
#include <Library/UefiBootServicesTableLib.h>\r
\r
/**\r
  The ExtractSection() function processes the input section and\r
  allocates a buffer from the pool in which it returns the section\r
  contents. If the section being extracted contains\r
  authentication information (the section's\r
  GuidedSectionHeader.Attributes field has the\r
  EFI_GUIDED_SECTION_AUTH_STATUS_VALID bit set), the values\r
  returned in AuthenticationStatus must reflect the results of\r
  the authentication operation. Depending on the algorithm and\r
  size of the encapsulated data, the time that is required to do\r
  a full authentication may be prohibitively long for some\r
  classes of systems. To indicate this, use\r
  EFI_SECURITY_POLICY_PROTOCOL_GUID, which may be published by\r
  the security policy driver (see the Platform Initialization\r
  Driver Execution Environment Core Interface Specification for\r
  more details and the GUID definition). If the\r
  EFI_SECURITY_POLICY_PROTOCOL_GUID exists in the handle\r
  database, then, if possible, full authentication should be\r
  skipped and the section contents simply returned in the\r
  OutputBuffer. In this case, the\r
  EFI_AUTH_STATUS_PLATFORM_OVERRIDE bit AuthenticationStatus\r
  must be set on return. ExtractSection() is callable only from\r
  TPL_NOTIFY and below. Behavior of ExtractSection() at any\r
  EFI_TPL above TPL_NOTIFY is undefined. Type EFI_TPL is\r
  defined in RaiseTPL() in the UEFI 2.0 specification.\r
\r
\r
  @param This         Indicates the\r
                      EFI_GUIDED_SECTION_EXTRACTION_PROTOCOL instance.\r
  @param InputSection Buffer containing the input GUIDed section\r
                      to be processed. OutputBuffer OutputBuffer\r
                      is allocated from boot services pool\r
                      memory and contains the new section\r
                      stream. The caller is responsible for\r
                      freeing this buffer.\r
  @param OutputBuffer *OutputBuffer is allocated from boot services\r
                      pool memory and contains the new section stream.\r
                      The caller is responsible for freeing this buffer.\r
  @param OutputSize   A pointer to a caller-allocated UINTN in\r
                      which the size of OutputBuffer allocation\r
                      is stored. If the function returns\r
                      anything other than EFI_SUCCESS, the value\r
                      of OutputSize is undefined.\r
\r
  @param AuthenticationStatus A pointer to a caller-allocated\r
                              UINT32 that indicates the\r
                              authentication status of the\r
                              output buffer. If the input\r
                              section's\r
                              GuidedSectionHeader.Attributes\r
                              field has the\r
                              EFI_GUIDED_SECTION_AUTH_STATUS_VAL\r
                              bit as clear, AuthenticationStatus\r
                              must return zero. Both local bits\r
                              (19:16) and aggregate bits (3:0)\r
                              in AuthenticationStatus are\r
                              returned by ExtractSection().\r
                              These bits reflect the status of\r
                              the extraction operation. The bit\r
                              pattern in both regions must be\r
                              the same, as the local and\r
                              aggregate authentication statuses\r
                              have equivalent meaning at this\r
                              level. If the function returns\r
                              anything other than EFI_SUCCESS,\r
                              the value of AuthenticationStatus\r
                              is undefined.\r
\r
\r
  @retval EFI_SUCCESS          The InputSection was successfully\r
                               processed and the section contents were\r
                               returned.\r
\r
  @retval EFI_OUT_OF_RESOURCES The system has insufficient\r
                               resources to process the\r
                               request.\r
\r
  @retval EFI_INVALID_PARAMETER The GUID in InputSection does\r
                                not match this instance of the\r
                                GUIDed Section Extraction\r
                                Protocol.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
CustomGuidedSectionExtract (\r
  IN CONST  EFI_GUIDED_SECTION_EXTRACTION_PROTOCOL *This,\r
  IN CONST  VOID                                   *InputSection,\r
  OUT       VOID                                   **OutputBuffer,\r
  OUT       UINTN                                  *OutputSize,\r
  OUT       UINT32                                 *AuthenticationStatus\r
  );\r
\r
//\r
// Module global for the Section Extraction Protocol handle\r
//\r
EFI_HANDLE mSectionExtractionHandle = NULL;\r
\r
//\r
// Module global for the Section Extraction Protocol instance\r
//\r
EFI_GUIDED_SECTION_EXTRACTION_PROTOCOL mCustomGuidedSectionExtractionProtocol = {\r
  CustomGuidedSectionExtract\r
};\r
\r
/**\r
  The ExtractSection() function processes the input section and\r
  allocates a buffer from the pool in which it returns the section\r
  contents. If the section being extracted contains\r
  authentication information (the section's\r
  GuidedSectionHeader.Attributes field has the\r
  EFI_GUIDED_SECTION_AUTH_STATUS_VALID bit set), the values\r
  returned in AuthenticationStatus must reflect the results of\r
  the authentication operation. Depending on the algorithm and\r
  size of the encapsulated data, the time that is required to do\r
  a full authentication may be prohibitively long for some\r
  classes of systems. To indicate this, use\r
  EFI_SECURITY_POLICY_PROTOCOL_GUID, which may be published by\r
  the security policy driver (see the Platform Initialization\r
  Driver Execution Environment Core Interface Specification for\r
  more details and the GUID definition). If the\r
  EFI_SECURITY_POLICY_PROTOCOL_GUID exists in the handle\r
  database, then, if possible, full authentication should be\r
  skipped and the section contents simply returned in the\r
  OutputBuffer. In this case, the\r
  EFI_AUTH_STATUS_PLATFORM_OVERRIDE bit AuthenticationStatus\r
  must be set on return. ExtractSection() is callable only from\r
  TPL_NOTIFY and below. Behavior of ExtractSection() at any\r
  EFI_TPL above TPL_NOTIFY is undefined. Type EFI_TPL is\r
  defined in RaiseTPL() in the UEFI 2.0 specification.\r
\r
\r
  @param This         Indicates the\r
                      EFI_GUIDED_SECTION_EXTRACTION_PROTOCOL instance.\r
  @param InputSection Buffer containing the input GUIDed section\r
                      to be processed. OutputBuffer OutputBuffer\r
                      is allocated from boot services pool\r
                      memory and contains the new section\r
                      stream. The caller is responsible for\r
                      freeing this buffer.\r
  @param OutputBuffer *OutputBuffer is allocated from boot services\r
                      pool memory and contains the new section stream.\r
                      The caller is responsible for freeing this buffer.\r
  @param OutputSize   A pointer to a caller-allocated UINTN in\r
                      which the size of OutputBuffer allocation\r
                      is stored. If the function returns\r
                      anything other than EFI_SUCCESS, the value\r
                      of OutputSize is undefined.\r
\r
  @param AuthenticationStatus A pointer to a caller-allocated\r
                              UINT32 that indicates the\r
                              authentication status of the\r
                              output buffer. If the input\r
                              section's\r
                              GuidedSectionHeader.Attributes\r
                              field has the\r
                              EFI_GUIDED_SECTION_AUTH_STATUS_VAL\r
                              bit as clear, AuthenticationStatus\r
                              must return zero. Both local bits\r
                              (19:16) and aggregate bits (3:0)\r
                              in AuthenticationStatus are\r
                              returned by ExtractSection().\r
                              These bits reflect the status of\r
                              the extraction operation. The bit\r
                              pattern in both regions must be\r
                              the same, as the local and\r
                              aggregate authentication statuses\r
                              have equivalent meaning at this\r
                              level. If the function returns\r
                              anything other than EFI_SUCCESS,\r
                              the value of AuthenticationStatus\r
                              is undefined.\r
\r
\r
  @retval EFI_SUCCESS          The InputSection was successfully\r
                               processed and the section contents were\r
                               returned.\r
\r
  @retval EFI_OUT_OF_RESOURCES The system has insufficient\r
                               resources to process the\r
                               request.\r
\r
  @retval EFI_INVALID_PARAMETER The GUID in InputSection does\r
                                not match this instance of the\r
                                GUIDed Section Extraction\r
                                Protocol.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
CustomGuidedSectionExtract (\r
  IN CONST  EFI_GUIDED_SECTION_EXTRACTION_PROTOCOL *This,\r
  IN CONST  VOID                                   *InputSection,\r
  OUT       VOID                                   **OutputBuffer,\r
  OUT       UINTN                                  *OutputSize,\r
  OUT       UINT32                                 *AuthenticationStatus\r
  )\r
{\r
  EFI_STATUS      Status;\r
  VOID            *ScratchBuffer;\r
  VOID            *AllocatedOutputBuffer;\r
  UINT32          OutputBufferSize;\r
  UINT32          ScratchBufferSize;\r
  UINT16          SectionAttribute;\r
\r
  //\r
  // Init local variable\r
  //\r
  ScratchBuffer         = NULL;\r
  AllocatedOutputBuffer = NULL;\r
\r
  //\r
  // Call GetInfo to get the size and attribute of input guided section data.\r
  //\r
  Status = ExtractGuidedSectionGetInfo (\r
             InputSection,\r
             &OutputBufferSize,\r
             &ScratchBufferSize,\r
             &SectionAttribute\r
             );\r
\r
  if (EFI_ERROR (Status)) {\r
    DEBUG ((DEBUG_ERROR, "GetInfo from guided section Failed - %r\n", Status));\r
    return Status;\r
  }\r
\r
  if (ScratchBufferSize > 0) {\r
    //\r
    // Allocate scratch buffer\r
    //\r
    ScratchBuffer = AllocatePool (ScratchBufferSize);\r
    if (ScratchBuffer == NULL) {\r
      return EFI_OUT_OF_RESOURCES;\r
    }\r
  }\r
\r
  if (OutputBufferSize > 0) {\r
    //\r
    // Allocate output buffer\r
    //\r
    AllocatedOutputBuffer = AllocatePool (OutputBufferSize);\r
    if (AllocatedOutputBuffer == NULL) {\r
      FreePool (ScratchBuffer);\r
      return EFI_OUT_OF_RESOURCES;\r
    }\r
    *OutputBuffer = AllocatedOutputBuffer;\r
  }\r
\r
  //\r
  // Call decode function to extract raw data from the guided section.\r
  //\r
  Status = ExtractGuidedSectionDecode (\r
             InputSection,\r
             OutputBuffer,\r
             ScratchBuffer,\r
             AuthenticationStatus\r
             );\r
  if (EFI_ERROR (Status)) {\r
    //\r
    // Decode failed\r
    //\r
    if (AllocatedOutputBuffer != NULL) {\r
      FreePool (AllocatedOutputBuffer);\r
    }\r
    if (ScratchBuffer != NULL) {\r
      FreePool (ScratchBuffer);\r
    }\r
    DEBUG ((DEBUG_ERROR, "Extract guided section Failed - %r\n", Status));\r
    return Status;\r
  }\r
\r
  if (*OutputBuffer != AllocatedOutputBuffer) {\r
    //\r
    // OutputBuffer was returned as a different value,\r
    // so copy section contents to the allocated memory buffer.\r
    //\r
    CopyMem (AllocatedOutputBuffer, *OutputBuffer, OutputBufferSize);\r
    *OutputBuffer = AllocatedOutputBuffer;\r
  }\r
\r
  //\r
  // Set real size of output buffer.\r
  //\r
  *OutputSize = (UINTN) OutputBufferSize;\r
\r
  //\r
  // Free unused scratch buffer.\r
  //\r
  if (ScratchBuffer != NULL) {\r
    FreePool (ScratchBuffer);\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Main entry for the Section Extraction DXE module.\r
\r
  This routine registers the Section Extraction Protocols that have been registered\r
  with the Section Extraction Library.\r
\r
  @param[in] ImageHandle    The firmware allocated handle for the EFI image.\r
  @param[in] SystemTable    A pointer to the EFI System Table.\r
\r
  @retval EFI_SUCCESS       The entry point is executed successfully.\r
  @retval other             Some error occurs when executing this entry point.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
SectionExtractionDxeEntry (\r
  IN EFI_HANDLE        ImageHandle,\r
  IN EFI_SYSTEM_TABLE  *SystemTable\r
  )\r
{\r
  EFI_STATUS  Status;\r
  EFI_GUID    *ExtractHandlerGuidTable;\r
  UINTN       ExtractHandlerNumber;\r
\r
  //\r
  // Get custom extract guided section method guid list\r
  //\r
  ExtractHandlerNumber = ExtractGuidedSectionGetGuidList (&ExtractHandlerGuidTable);\r
\r
  //\r
  // Install custom guided extraction protocol\r
  //\r
  while (ExtractHandlerNumber-- > 0) {\r
    Status = gBS->InstallMultipleProtocolInterfaces (\r
                    &mSectionExtractionHandle,\r
                    &ExtractHandlerGuidTable [ExtractHandlerNumber], &mCustomGuidedSectionExtractionProtocol,\r
                    NULL\r
                    );\r
    ASSERT_EFI_ERROR (Status);\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r