]> git.proxmox.com Git - mirror_edk2.git/blobdiff - MdePkg/Library/PeiExtractGuidedSectionLib/PeiExtractGuidedSectionLib.c
MdePkg: Clean up source files
[mirror_edk2.git] / MdePkg / Library / PeiExtractGuidedSectionLib / PeiExtractGuidedSectionLib.c
index ce42a5dcd21cfcffe155cdf8624a6587e1630128..793c6220e2b00c00cc91e07e6ed2c90e70996828 100644 (file)
@@ -1,23 +1,16 @@
-/*++\r
+/** @file\r
+  Provide generic extract guided section functions for PEI phase.\r
 \r
-Copyright (c) 2007, Intel Corporation\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
+  Copyright (c) 2007 - 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
+  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
-Module Name:\r
-\r
-  PeiExtractGuidedSectionLib.c\r
-\r
-Abstract:\r
-\r
-  Provide generic extract guided section functions. \r
-\r
---*/\r
+**/\r
 \r
 #include <PiPei.h>\r
 \r
@@ -27,7 +20,7 @@ Abstract:
 #include <Library/HobLib.h>\r
 #include <Library/ExtractGuidedSectionLib.h>\r
 \r
-#define PEI_EXTRACT_HANDLER_INFO_SIGNATURE EFI_SIGNATURE_32 ('P', 'E', 'H', 'I')\r
+#define PEI_EXTRACT_HANDLER_INFO_SIGNATURE SIGNATURE_32 ('P', 'E', 'H', 'I')\r
 \r
 typedef struct {\r
   UINT32                                  Signature;\r
@@ -37,32 +30,49 @@ typedef struct {
   EXTRACT_GUIDED_SECTION_GET_INFO_HANDLER *ExtractGetInfoHandlerTable;\r
 } PEI_EXTRACT_GUIDED_SECTION_HANDLER_INFO;\r
 \r
-/**
+/**\r
   Build guid hob for the global memory to store the registered guid and Handler list.\r
   If GuidHob exists, HandlerInfo will be directly got from Guid hob data.\r
 \r
-  @param[in, out]  InfoPointer   Pointer to pei handler info structure.
+  @param[in, out]  InfoPointer   The pointer to pei handler information structure.\r
 \r
-  @retval  RETURN_SUCCESS            Build Guid hob for the global memory space to store guid and funciton tables.\r
-  @retval  RETURN_OUT_OF_RESOURCES   No enough memory to allocated.
+  @retval  RETURN_SUCCESS            Build Guid hob for the global memory space to store guid and function tables.\r
+  @retval  RETURN_OUT_OF_RESOURCES   No enough memory to allocated.\r
 **/\r
 RETURN_STATUS\r
-EFIAPI\r
 PeiGetExtractGuidedSectionHandlerInfo (\r
   IN OUT PEI_EXTRACT_GUIDED_SECTION_HANDLER_INFO **InfoPointer\r
   )\r
 {\r
   PEI_EXTRACT_GUIDED_SECTION_HANDLER_INFO *HandlerInfo;\r
   EFI_PEI_HOB_POINTERS                    Hob;\r
-  \r
+\r
   //\r
-  // First try to get handler info from guid hob specified by CallerId.\r
+  // First try to get handler information from guid hob specified by CallerId.\r
   //\r
   Hob.Raw = GetNextHob (EFI_HOB_TYPE_GUID_EXTENSION, GetHobList ());\r
   while (Hob.Raw != NULL) {\r
     if (CompareGuid (&(Hob.Guid->Name), &gEfiCallerIdGuid)) {\r
       HandlerInfo = (PEI_EXTRACT_GUIDED_SECTION_HANDLER_INFO *) GET_GUID_HOB_DATA (Hob.Guid);\r
       if (HandlerInfo->Signature == PEI_EXTRACT_HANDLER_INFO_SIGNATURE) {\r
+        //\r
+        // Update Table Pointer when hob start address is changed.\r
+        //\r
+        if (HandlerInfo->ExtractHandlerGuidTable != (GUID *) (HandlerInfo + 1)) {\r
+          HandlerInfo->ExtractHandlerGuidTable    = (GUID *) (HandlerInfo + 1);\r
+          HandlerInfo->ExtractDecodeHandlerTable  = (EXTRACT_GUIDED_SECTION_DECODE_HANDLER *) (\r
+                                                      (UINT8 *)HandlerInfo->ExtractHandlerGuidTable +\r
+                                                      PcdGet32 (PcdMaximumGuidedExtractHandler) * sizeof (GUID)\r
+                                                     );\r
+          HandlerInfo->ExtractGetInfoHandlerTable = (EXTRACT_GUIDED_SECTION_GET_INFO_HANDLER *) (\r
+                                                      (UINT8 *)HandlerInfo->ExtractDecodeHandlerTable +\r
+                                                      PcdGet32 (PcdMaximumGuidedExtractHandler) *\r
+                                                      sizeof (EXTRACT_GUIDED_SECTION_DECODE_HANDLER)\r
+                                                     );\r
+        }\r
+        //\r
+        // Return HandlerInfo pointer.\r
+        //\r
         *InfoPointer = HandlerInfo;\r
         return EFI_SUCCESS;\r
       }\r
@@ -70,51 +80,63 @@ PeiGetExtractGuidedSectionHandlerInfo (
     Hob.Raw = GET_NEXT_HOB (Hob);\r
     Hob.Raw = GetNextHob (EFI_HOB_TYPE_GUID_EXTENSION, Hob.Raw);\r
   }\r
-  \r
+\r
   //\r
   // If Guid Hob is not found, Build CallerId Guid hob to store Handler Info\r
   //\r
   HandlerInfo = BuildGuidHob (\r
-                 &gEfiCallerIdGuid, \r
+                 &gEfiCallerIdGuid,\r
                  sizeof (PEI_EXTRACT_GUIDED_SECTION_HANDLER_INFO) +\r
-                 PcdGet32 (PcdMaximumGuidedExtractHandler) * \r
+                 PcdGet32 (PcdMaximumGuidedExtractHandler) *\r
                  (sizeof (GUID) + sizeof (EXTRACT_GUIDED_SECTION_DECODE_HANDLER) + sizeof (EXTRACT_GUIDED_SECTION_GET_INFO_HANDLER))\r
                 );\r
   if (HandlerInfo == NULL) {\r
     //\r
     // No enough resource to build guid hob.\r
     //\r
+    *InfoPointer = NULL;\r
     return EFI_OUT_OF_RESOURCES;\r
   }\r
+  //\r
+  // Init HandlerInfo structure\r
+  //\r
   HandlerInfo->Signature = PEI_EXTRACT_HANDLER_INFO_SIGNATURE;\r
   HandlerInfo->NumberOfExtractHandler     = 0;\r
   HandlerInfo->ExtractHandlerGuidTable    = (GUID *) (HandlerInfo + 1);\r
   HandlerInfo->ExtractDecodeHandlerTable  = (EXTRACT_GUIDED_SECTION_DECODE_HANDLER *) (\r
-                                              (UINT8 *)HandlerInfo->ExtractHandlerGuidTable + \r
+                                              (UINT8 *)HandlerInfo->ExtractHandlerGuidTable +\r
                                               PcdGet32 (PcdMaximumGuidedExtractHandler) * sizeof (GUID)\r
                                              );\r
   HandlerInfo->ExtractGetInfoHandlerTable = (EXTRACT_GUIDED_SECTION_GET_INFO_HANDLER *) (\r
-                                              (UINT8 *)HandlerInfo->ExtractDecodeHandlerTable + \r
-                                              PcdGet32 (PcdMaximumGuidedExtractHandler) * \r
-                                              sizeof (EXTRACT_GUIDED_SECTION_GET_INFO_HANDLER)\r
+                                              (UINT8 *)HandlerInfo->ExtractDecodeHandlerTable +\r
+                                              PcdGet32 (PcdMaximumGuidedExtractHandler) *\r
+                                              sizeof (EXTRACT_GUIDED_SECTION_DECODE_HANDLER)\r
                                              );\r
-  \r
+  //\r
+  // return the created HandlerInfo.\r
+  //\r
   *InfoPointer = HandlerInfo;\r
   return EFI_SUCCESS;\r
 }\r
 \r
-/**
-  Get the supported exract guided section Handler guid list.\r
-  If ExtractHandlerGuidTable = NULL, then ASSERT.\r
+/**\r
+  Retrieve the list GUIDs that have been registered through ExtractGuidedSectionRegisterHandlers().\r
+\r
+  Sets ExtractHandlerGuidTable so it points at a callee allocated array of registered GUIDs.\r
+  The total number of GUIDs in the array are returned. Since the array of GUIDs is callee allocated\r
+  and caller must treat this array of GUIDs as read-only data.\r
+  If ExtractHandlerGuidTable is NULL, then ASSERT().\r
+\r
+  @param[out]  ExtractHandlerGuidTable  A pointer to the array of GUIDs that have been registered through\r
+                                        ExtractGuidedSectionRegisterHandlers().\r
 \r
-  @param[in, out]  ExtractHandlerGuidTable   The extract Handler guid pointer list.
+  @return the number of the supported extract guided Handler.\r
 \r
-  @retval  return the number of the supported extract guided Handler.
 **/\r
 UINTN\r
 EFIAPI\r
 ExtractGuidedSectionGetGuidList (\r
-  IN OUT  GUID  **ExtractHandlerGuidTable\r
+  OUT  GUID  **ExtractHandlerGuidTable\r
   )\r
 {\r
   EFI_STATUS Status;\r
@@ -122,25 +144,46 @@ ExtractGuidedSectionGetGuidList (
 \r
   ASSERT (ExtractHandlerGuidTable != NULL);\r
 \r
+  //\r
+  // Get all registered handler information\r
+  //\r
   Status = PeiGetExtractGuidedSectionHandlerInfo (&HandlerInfo);\r
   if (EFI_ERROR (Status)) {\r
-    return Status;\r
+    *ExtractHandlerGuidTable = NULL;\r
+    return 0;\r
   }\r
 \r
+  //\r
+  // Get GuidTable and Table Number\r
+  //\r
+  ASSERT (HandlerInfo != NULL);\r
   *ExtractHandlerGuidTable = HandlerInfo->ExtractHandlerGuidTable;\r
   return HandlerInfo->NumberOfExtractHandler;\r
 }\r
 \r
-/**
-  Register Guided Section Extract and GetInfo handler.\r
+/**\r
+  Registers handlers of type EXTRACT_GUIDED_SECTION_GET_INFO_HANDLER and EXTRACT_GUIDED_SECTION_DECODE_HANDLER\r
+  for a specific GUID section type.\r
+\r
+  Registers the handlers specified by GetInfoHandler and DecodeHandler with the GUID specified by SectionGuid.\r
+  If the GUID value specified by SectionGuid has already been registered, then return RETURN_ALREADY_STARTED.\r
+  If there are not enough resources available to register the handlers  then RETURN_OUT_OF_RESOURCES is returned.\r
+\r
+  If SectionGuid is NULL, then ASSERT().\r
+  If GetInfoHandler is NULL, then ASSERT().\r
+  If DecodeHandler is NULL, then ASSERT().\r
+\r
+  @param[in]  SectionGuid    A pointer to the GUID associated with the the handlers\r
+                             of the GUIDed section type being registered.\r
+  @param[in]  GetInfoHandler The pointer to a function that examines a GUIDed section and returns the\r
+                             size of the decoded buffer and the size of an optional scratch buffer\r
+                             required to actually decode the data in a GUIDed section.\r
+  @param[in]  DecodeHandler  The pointer to a function that decodes a GUIDed section into a caller\r
+                             allocated output buffer.\r
+\r
+  @retval  RETURN_SUCCESS           The handlers were registered.\r
+  @retval  RETURN_OUT_OF_RESOURCES  There are not enough resources available to register the handlers.\r
 \r
-  @param[in]     SectionGuid    The guid matches this Extraction function.
-  @param[in]     GetInfoHandler Function to get info from guided section.\r
-  @param[in]     DecodeHandler  Function to extract guided section.
-
-  @retval  RETURN_SUCCESS           Register Guided Section Extract function successfully.
-  @retval  RETURN_OUT_OF_RESOURCES  Resource is not enough to register new function. \r
-  @retval  RETURN_INVALID_PARAMETER Input pointer to Guid value is not valid.\r
 **/\r
 RETURN_STATUS\r
 EFIAPI\r
@@ -155,11 +198,13 @@ ExtractGuidedSectionRegisterHandlers (
   PEI_EXTRACT_GUIDED_SECTION_HANDLER_INFO *HandlerInfo;\r
 \r
   //\r
-  // Check input paramter.\r
+  // Check input parameter\r
   //\r
-  if (SectionGuid == NULL) {\r
-    return RETURN_INVALID_PARAMETER;\r
-  }\r
+  ASSERT (SectionGuid != NULL);\r
+  ASSERT (GetInfoHandler != NULL);\r
+  ASSERT (DecodeHandler != NULL);\r
+\r
+\r
 \r
   //\r
   // Get the registered handler information\r
@@ -172,57 +217,77 @@ ExtractGuidedSectionRegisterHandlers (
   //\r
   // Search the match registered GetInfo handler for the input guided section.\r
   //\r
+  ASSERT (HandlerInfo != NULL);\r
   for (Index = 0; Index < HandlerInfo->NumberOfExtractHandler; Index ++) {\r
-    if (CompareGuid (&(HandlerInfo->ExtractHandlerGuidTable[Index]), SectionGuid)) {\r
-      break;\r
+    if (CompareGuid (HandlerInfo->ExtractHandlerGuidTable + Index, SectionGuid)) {\r
+      //\r
+      // If the guided handler has been registered before, only update its handler.\r
+      //\r
+      HandlerInfo->ExtractDecodeHandlerTable [Index] = DecodeHandler;\r
+      HandlerInfo->ExtractGetInfoHandlerTable [Index] = GetInfoHandler;\r
+      return RETURN_SUCCESS;\r
     }\r
   }\r
 \r
-  //\r
-  // If the guided handler has been registered before, only update its handler.\r
-  //\r
-  if (Index < HandlerInfo->NumberOfExtractHandler) {\r
-    HandlerInfo->ExtractDecodeHandlerTable [Index] = DecodeHandler;\r
-    HandlerInfo->ExtractGetInfoHandlerTable [Index] = GetInfoHandler;\r
-    return RETURN_SUCCESS;\r
-  }\r
-\r
   //\r
   // Check the global table is enough to contain new Handler.\r
   //\r
   if (HandlerInfo->NumberOfExtractHandler >= PcdGet32 (PcdMaximumGuidedExtractHandler)) {\r
     return RETURN_OUT_OF_RESOURCES;\r
   }\r
-  \r
+\r
   //\r
   // Register new Handler and guid value.\r
   //\r
-  CopyGuid (&(HandlerInfo->ExtractHandlerGuidTable [HandlerInfo->NumberOfExtractHandler]), SectionGuid);\r
+  CopyGuid (HandlerInfo->ExtractHandlerGuidTable + HandlerInfo->NumberOfExtractHandler, SectionGuid);\r
   HandlerInfo->ExtractDecodeHandlerTable [HandlerInfo->NumberOfExtractHandler] = DecodeHandler;\r
   HandlerInfo->ExtractGetInfoHandlerTable [HandlerInfo->NumberOfExtractHandler++] = GetInfoHandler;\r
-  \r
+\r
+  //\r
+  // Build the Guided Section GUID HOB to record the GUID itself.\r
+  // Then the content of the GUIDed HOB will be the same as the GUID value itself.\r
+  //\r
+  BuildGuidDataHob (\r
+    (EFI_GUID *) SectionGuid,\r
+    (VOID *) SectionGuid,\r
+    sizeof (GUID)\r
+    );\r
+\r
   return RETURN_SUCCESS;\r
 }\r
 \r
-/**
-  Get information from the guided section. This function first gets the guid value\r
-  from guided section header, then match this guid in the registered extract Handler list\r
-  to its corresponding getinfo Handler. \r
-  If not found, RETURN_INVALID_PARAMETER will be return. \r
-  If found, it will call the getinfo Handler to get the required size and attribute.\r
-\r
-  It will ASSERT () if the pointer to OutputBufferSize is NULL.\r
-  It will ASSERT () if the pointer to ScratchBufferSize is NULL.
-  It will ASSERT () if the pointer to SectionAttribute is NULL.\r
-\r
-  @param[in]  InputSection          Buffer containing the input GUIDed section to be processed. \r
-  @param[out] OutputBufferSize      The size of OutputBuffer.\r
-  @param[out] ScratchBufferSize     The size of ScratchBuffer.  \r
-  @param[out] SectionAttribute      The attribute of the input guided section.\r
-
-  @retval  RETURN_SUCCESS           Get the required information successfully.\r
-  @retval  RETURN_INVALID_PARAMETER The input data can't be parsed correctly. \r
-                                    The GUID in InputSection does not match any registered guid list.\r
+/**\r
+  Retrieves a GUID from a GUIDed section and uses that GUID to select an associated handler of type\r
+  EXTRACT_GUIDED_SECTION_GET_INFO_HANDLER that was registered with ExtractGuidedSectionRegisterHandlers().\r
+  The selected handler is used to retrieve and return the size of the decoded buffer and the size of an\r
+  optional scratch buffer required to actually decode the data in a GUIDed section.\r
+\r
+  Examines a GUIDed section specified by InputSection.\r
+  If GUID for InputSection does not match any of the GUIDs registered through ExtractGuidedSectionRegisterHandlers(),\r
+  then RETURN_UNSUPPORTED is returned.\r
+  If the GUID of InputSection does match the GUID that this handler supports, then the the associated handler\r
+  of type EXTRACT_GUIDED_SECTION_GET_INFO_HANDLER that was registered with ExtractGuidedSectionRegisterHandlers()\r
+  is used to retrieve the OututBufferSize, ScratchSize, and Attributes values. The return status from the handler of\r
+  type EXTRACT_GUIDED_SECTION_GET_INFO_HANDLER is returned.\r
+\r
+  If InputSection is NULL, then ASSERT().\r
+  If OutputBufferSize is NULL, then ASSERT().\r
+  If ScratchBufferSize is NULL, then ASSERT().\r
+  If SectionAttribute is NULL, then ASSERT().\r
+\r
+  @param[in]  InputSection       A pointer to a GUIDed section of an FFS formatted file.\r
+  @param[out] OutputBufferSize   A pointer to the size, in bytes, of an output buffer required if the buffer\r
+                                 specified by InputSection were decoded.\r
+  @param[out] ScratchBufferSize  A pointer to the size, in bytes, required as scratch space if the buffer specified by\r
+                                 InputSection were decoded.\r
+  @param[out] SectionAttribute   A pointer to the attributes of the GUIDed section.  See the Attributes field of\r
+                                 EFI_GUID_DEFINED_SECTION in the PI Specification.\r
+\r
+  @retval  RETURN_SUCCESS      Get the required information successfully.\r
+  @retval  RETURN_UNSUPPORTED  The GUID from the section specified by InputSection does not match any of\r
+                               the GUIDs registered with ExtractGuidedSectionRegisterHandlers().\r
+  @retval  Others              The return status from the handler associated with the GUID retrieved from\r
+                               the section specified by InputSection.\r
 \r
 **/\r
 RETURN_STATUS\r
@@ -231,132 +296,224 @@ ExtractGuidedSectionGetInfo (
   IN  CONST VOID    *InputSection,\r
   OUT       UINT32  *OutputBufferSize,\r
   OUT       UINT32  *ScratchBufferSize,\r
-  OUT       UINT16  *SectionAttribute   \r
+  OUT       UINT16  *SectionAttribute\r
   )\r
 {\r
   UINT32 Index;\r
   EFI_STATUS Status;\r
   PEI_EXTRACT_GUIDED_SECTION_HANDLER_INFO *HandlerInfo;\r
-  \r
+  EFI_GUID *SectionDefinitionGuid;\r
+\r
   //\r
-  // Check input paramter\r
+  // Check input parameter\r
   //\r
-  if (InputSection == NULL) {\r
-    return RETURN_INVALID_PARAMETER;\r
-  }\r
-  \r
+  ASSERT (InputSection != NULL);\r
   ASSERT (OutputBufferSize != NULL);\r
   ASSERT (ScratchBufferSize != NULL);\r
   ASSERT (SectionAttribute != NULL);\r
 \r
   //\r
-  // Get the registered handler information.\r
+  // Get all registered handler information.\r
   //\r
   Status = PeiGetExtractGuidedSectionHandlerInfo (&HandlerInfo);\r
   if (EFI_ERROR (Status)) {\r
     return Status;\r
   }\r
-  \r
+\r
+  if (IS_SECTION2 (InputSection)) {\r
+    SectionDefinitionGuid = &(((EFI_GUID_DEFINED_SECTION2 *) InputSection)->SectionDefinitionGuid);\r
+  } else {\r
+    SectionDefinitionGuid = &(((EFI_GUID_DEFINED_SECTION *) InputSection)->SectionDefinitionGuid);\r
+  }\r
+\r
   //\r
   // Search the match registered GetInfo handler for the input guided section.\r
   //\r
+  ASSERT (HandlerInfo != NULL);\r
   for (Index = 0; Index < HandlerInfo->NumberOfExtractHandler; Index ++) {\r
-    if (CompareGuid (&(HandlerInfo->ExtractHandlerGuidTable[Index]), &(((EFI_GUID_DEFINED_SECTION *) InputSection)->SectionDefinitionGuid))) {\r
-      break;\r
+    if (CompareGuid (HandlerInfo->ExtractHandlerGuidTable + Index, SectionDefinitionGuid)) {\r
+      //\r
+      // Call the match handler to get information for the input section data.\r
+      //\r
+      return HandlerInfo->ExtractGetInfoHandlerTable [Index] (\r
+                InputSection,\r
+                OutputBufferSize,\r
+                ScratchBufferSize,\r
+                SectionAttribute\r
+              );\r
     }\r
   }\r
 \r
   //\r
-  // Not found, the input guided section is not supported. \r
-  //\r
-  if (Index == HandlerInfo->NumberOfExtractHandler) {\r
-    return RETURN_INVALID_PARAMETER;\r
-  }\r
-\r
+  // Not found, the input guided section is not supported.\r
   //\r
-  // Call the match handler to getinfo for the input section data.\r
-  //\r
-  return HandlerInfo->ExtractGetInfoHandlerTable [Index] (\r
-            InputSection,\r
-            OutputBufferSize,\r
-            ScratchBufferSize,\r
-            SectionAttribute\r
-          );\r
+  return RETURN_UNSUPPORTED;\r
 }\r
 \r
-/**
-  Extract data from the guided section. This function first gets the guid value\r
-  from guided section header, then match this guid in the registered extract Handler list\r
-  to its corresponding extract Handler. \r
-  If not found, RETURN_INVALID_PARAMETER will be return. \r
-  If found, it will call this extract Handler to get output data and AuthenticationStatus.
-\r
-  It will ASSERT () if the pointer to OutputBuffer is NULL.\r
-  It will ASSERT () if the pointer to AuthenticationStatus is NULL.\r
-\r
-  @param[in]  InputSection  Buffer containing the input GUIDed section to be processed. \r
-  @param[out] OutputBuffer  OutputBuffer to point the start of the section's contents \r
-                            if guided data is not required prcessing. Otherwise,\r
-                            OutputBuffer to contain the output data, which is \r
-                            allocated by the caller.\r
-  @param[out] ScratchBuffer A pointer to a caller-allocated buffer for function internal use. \r
-  @param[out] AuthenticationStatus \r
-                            A pointer to a caller-allocated UINT32 that indicates the\r
-                            authentication status of the output buffer. 
-
-  @retval  RETURN_SUCCESS           Get the output data, size and AuthenticationStatus successfully.\r
-  @retval  RETURN_INVALID_PARAMETER The input data can't be parsed correctly. \r
-                                    The GUID in InputSection does not match any registered guid list.\r
-
+/**\r
+  Retrieves the GUID from a GUIDed section and uses that GUID to select an associated handler of type\r
+  EXTRACT_GUIDED_SECTION_DECODE_HANDLER that was registered with ExtractGuidedSectionRegisterHandlers().\r
+  The selected handler is used to decode the data in a GUIDed section and return the result in a caller\r
+  allocated output buffer.\r
+\r
+  Decodes the GUIDed section specified by InputSection.\r
+  If GUID for InputSection does not match any of the GUIDs registered through ExtractGuidedSectionRegisterHandlers(),\r
+  then RETURN_UNSUPPORTED is returned.\r
+  If the GUID of InputSection does match the GUID that this handler supports, then the the associated handler\r
+  of type EXTRACT_GUIDED_SECTION_DECODE_HANDLER that was registered with ExtractGuidedSectionRegisterHandlers()\r
+  is used to decode InputSection into the buffer specified by OutputBuffer and the authentication status of this\r
+  decode operation is returned in AuthenticationStatus.  If the decoded buffer is identical to the data in InputSection,\r
+  then OutputBuffer is set to point at the data in InputSection.  Otherwise, the decoded data will be placed in caller\r
+  allocated buffer specified by OutputBuffer.    This function is responsible for computing the  EFI_AUTH_STATUS_PLATFORM_OVERRIDE\r
+  bit of in AuthenticationStatus.  The return status from the handler of type EXTRACT_GUIDED_SECTION_DECODE_HANDLER is returned.\r
+\r
+  If InputSection is NULL, then ASSERT().\r
+  If OutputBuffer is NULL, then ASSERT().\r
+  If ScratchBuffer is NULL and this decode operation requires a scratch buffer, then ASSERT().\r
+  If AuthenticationStatus is NULL, then ASSERT().\r
+\r
+  @param[in]  InputSection   A pointer to a GUIDed section of an FFS formatted file.\r
+  @param[out] OutputBuffer   A pointer to a buffer that contains the result of a decode operation.\r
+  @param[in]  ScratchBuffer  A caller allocated buffer that may be required by this function as a scratch buffer to perform the decode operation.\r
+  @param[out] AuthenticationStatus\r
+                             A pointer to the authentication status of the decoded output buffer. See the definition\r
+                             of authentication status in the EFI_PEI_GUIDED_SECTION_EXTRACTION_PPI section of the PI\r
+                             Specification.\r
+\r
+  @retval  RETURN_SUCCESS           The buffer specified by InputSection was decoded.\r
+  @retval  RETURN_UNSUPPORTED       The section specified by InputSection does not match the GUID this handler supports.\r
+  @retval  RETURN_INVALID_PARAMETER The section specified by InputSection can not be decoded.\r
+\r
 **/\r
 RETURN_STATUS\r
 EFIAPI\r
 ExtractGuidedSectionDecode (\r
   IN  CONST VOID    *InputSection,\r
   OUT       VOID    **OutputBuffer,\r
-  OUT       VOID    *ScratchBuffer,        OPTIONAL\r
-  OUT       UINT32  *AuthenticationStatus  \r
+  IN        VOID    *ScratchBuffer,        OPTIONAL\r
+  OUT       UINT32  *AuthenticationStatus\r
   )\r
 {\r
   UINT32     Index;\r
   EFI_STATUS Status;\r
   PEI_EXTRACT_GUIDED_SECTION_HANDLER_INFO *HandlerInfo;\r
-  \r
-  if (InputSection == NULL) {\r
-    return RETURN_INVALID_PARAMETER;\r
-  }\r
-  \r
+  EFI_GUID *SectionDefinitionGuid;\r
+\r
+  //\r
+  // Check input parameter\r
+  //\r
+  ASSERT (InputSection != NULL);\r
   ASSERT (OutputBuffer != NULL);\r
   ASSERT (AuthenticationStatus != NULL);\r
-  \r
+\r
+  //\r
+  // Get all registered handler information.\r
+  //\r
   Status = PeiGetExtractGuidedSectionHandlerInfo (&HandlerInfo);\r
   if (EFI_ERROR (Status)) {\r
     return Status;\r
   }\r
 \r
+  if (IS_SECTION2 (InputSection)) {\r
+    SectionDefinitionGuid = &(((EFI_GUID_DEFINED_SECTION2 *) InputSection)->SectionDefinitionGuid);\r
+  } else {\r
+    SectionDefinitionGuid = &(((EFI_GUID_DEFINED_SECTION *) InputSection)->SectionDefinitionGuid);\r
+  }\r
+\r
   //\r
-  // Search the match registered GetInfo handler for the input guided section.\r
+  // Search the match registered Extract handler for the input guided section.\r
   //\r
+  ASSERT (HandlerInfo != NULL);\r
   for (Index = 0; Index < HandlerInfo->NumberOfExtractHandler; Index ++) {\r
-    if (CompareGuid (&(HandlerInfo->ExtractHandlerGuidTable[Index]), &(((EFI_GUID_DEFINED_SECTION *) InputSection)->SectionDefinitionGuid))) {\r
-      break;\r
+    if (CompareGuid (HandlerInfo->ExtractHandlerGuidTable + Index, SectionDefinitionGuid)) {\r
+      //\r
+      // Call the match handler to extract raw data for the input guided section.\r
+      //\r
+      return HandlerInfo->ExtractDecodeHandlerTable [Index] (\r
+                InputSection,\r
+                OutputBuffer,\r
+                ScratchBuffer,\r
+                AuthenticationStatus\r
+              );\r
     }\r
   }\r
 \r
   //\r
-  // Not found, the input guided section is not supported. \r
+  // Not found, the input guided section is not supported.\r
+  //\r
+  return RETURN_UNSUPPORTED;\r
+}\r
+\r
+/**\r
+  Retrieves handlers of type EXTRACT_GUIDED_SECTION_GET_INFO_HANDLER and\r
+  EXTRACT_GUIDED_SECTION_DECODE_HANDLER for a specific GUID section type.\r
+\r
+  Retrieves the handlers associated with SectionGuid and returns them in\r
+  GetInfoHandler and DecodeHandler.\r
+\r
+  If the GUID value specified by SectionGuid has not been registered, then\r
+  return RETURN_NOT_FOUND.\r
+\r
+  If SectionGuid is NULL, then ASSERT().\r
+\r
+  @param[in]  SectionGuid    A pointer to the GUID associated with the handlersof the GUIDed\r
+                             section type being retrieved.\r
+  @param[out] GetInfoHandler Pointer to a function that examines a GUIDed section and returns\r
+                             the size of the decoded buffer and the size of an optional scratch\r
+                             buffer required to actually decode the data in a GUIDed section.\r
+                             This is an optional parameter that may be NULL. If it is NULL, then\r
+                             the previously registered handler is not returned.\r
+  @param[out] DecodeHandler  Pointer to a function that decodes a GUIDed section into a caller\r
+                             allocated output buffer. This is an optional parameter that may be NULL.\r
+                             If it is NULL, then the previously registered handler is not returned.\r
+\r
+  @retval  RETURN_SUCCESS     The handlers were retrieved.\r
+  @retval  RETURN_NOT_FOUND   No handlers have been registered with the specified GUID.\r
+\r
+**/\r
+RETURN_STATUS\r
+EFIAPI\r
+ExtractGuidedSectionGetHandlers (\r
+  IN CONST   GUID                                     *SectionGuid,\r
+  OUT        EXTRACT_GUIDED_SECTION_GET_INFO_HANDLER  *GetInfoHandler,  OPTIONAL\r
+  OUT        EXTRACT_GUIDED_SECTION_DECODE_HANDLER    *DecodeHandler    OPTIONAL\r
+  )\r
+{\r
+  EFI_STATUS                               Status;\r
+  UINT32                                   Index;\r
+  PEI_EXTRACT_GUIDED_SECTION_HANDLER_INFO  *HandlerInfo;\r
+\r
+  //\r
+  // Check input parameter\r
+  //\r
+  ASSERT (SectionGuid != NULL);\r
+\r
+  //\r
+  // Get the registered handler information\r
   //\r
-  if (Index == HandlerInfo->NumberOfExtractHandler) {\r
-    return RETURN_INVALID_PARAMETER;\r
+  Status = PeiGetExtractGuidedSectionHandlerInfo (&HandlerInfo);\r
+  if (EFI_ERROR (Status)) {\r
+    return Status;\r
   }\r
 \r
   //\r
-  // Call the match handler to getinfo for the input section data.\r
+  // Search the match registered GetInfo handler for the input guided section.\r
   //\r
-  return HandlerInfo->ExtractDecodeHandlerTable [Index] (\r
-            InputSection,\r
-            OutputBuffer,\r
-            ScratchBuffer,\r
-            AuthenticationStatus\r
-          );\r
+  ASSERT (HandlerInfo != NULL);\r
+  for (Index = 0; Index < HandlerInfo->NumberOfExtractHandler; Index ++) {\r
+    if (CompareGuid (HandlerInfo->ExtractHandlerGuidTable + Index, SectionGuid)) {\r
+\r
+      //\r
+      // If the guided handler has been registered before, then return the registered handlers.\r
+      //\r
+      if (GetInfoHandler != NULL) {\r
+        *GetInfoHandler = HandlerInfo->ExtractGetInfoHandlerTable[Index];\r
+      }\r
+      if (DecodeHandler != NULL) {\r
+        *DecodeHandler = HandlerInfo->ExtractDecodeHandlerTable[Index];\r
+      }\r
+      return RETURN_SUCCESS;\r
+    }\r
+  }\r
+  return RETURN_NOT_FOUND;\r
 }\r