--- /dev/null
+/** @file\r
+\r
+ Copyright (c) 2008-2009, Apple Inc. All rights reserved.\r
+ \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
+\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 <PiPei.h>\r
+#include <Library/BaseMemoryLib.h>\r
+#include <Library/DebugLib.h>\r
+#include <Library/ExtractGuidedSectionLib.h>\r
+#include <Library/PcdLib.h>\r
+#include <Library/PrePiLib.h>\r
+\r
+#define PRE_PI_EXTRACT_GUIDED_SECTION_DATA_GUID { 0x385A982C, 0x2F49, 0x4043, { 0xA5, 0x1E, 0x49, 0x01, 0x02, 0x5C, 0x8B, 0x6B }}\r
+\r
+typedef struct {\r
+ UINT32 NumberOfExtractHandler;\r
+ GUID *ExtractHandlerGuidTable;\r
+ EXTRACT_GUIDED_SECTION_DECODE_HANDLER *ExtractDecodeHandlerTable;\r
+ EXTRACT_GUIDED_SECTION_GET_INFO_HANDLER *ExtractGetInfoHandlerTable;\r
+} PRE_PI_EXTRACT_GUIDED_SECTION_DATA;\r
+\r
+PRE_PI_EXTRACT_GUIDED_SECTION_DATA *\r
+GetSavedData (\r
+ VOID\r
+ )\r
+{\r
+ EFI_HOB_GUID_TYPE *GuidHob;\r
+ GUID SavedDataGuid = PRE_PI_EXTRACT_GUIDED_SECTION_DATA_GUID;\r
+\r
+ GuidHob = GetFirstGuidHob(&SavedDataGuid);\r
+ GuidHob++;\r
+ \r
+ return (PRE_PI_EXTRACT_GUIDED_SECTION_DATA *)GuidHob;\r
+}\r
+\r
+RETURN_STATUS\r
+EFIAPI\r
+ExtractGuidedSectionRegisterHandlers (\r
+ IN CONST GUID *SectionGuid,\r
+ IN EXTRACT_GUIDED_SECTION_GET_INFO_HANDLER GetInfoHandler,\r
+ IN EXTRACT_GUIDED_SECTION_DECODE_HANDLER DecodeHandler\r
+ )\r
+{\r
+ PRE_PI_EXTRACT_GUIDED_SECTION_DATA *SavedData;\r
+ UINT32 Index;\r
+ //\r
+ // Check input paramter.\r
+ //\r
+ if (SectionGuid == NULL) {\r
+ return RETURN_INVALID_PARAMETER;\r
+ }\r
+\r
+ SavedData = GetSavedData();\r
+\r
+ //\r
+ // Search the match registered GetInfo handler for the input guided section.\r
+ //\r
+ for (Index = 0; Index < SavedData->NumberOfExtractHandler; Index ++) {\r
+ if (CompareGuid (&SavedData->ExtractHandlerGuidTable[Index], SectionGuid)) {\r
+ break;\r
+ }\r
+ }\r
+\r
+ //\r
+ // If the guided handler has been registered before, only update its handler.\r
+ //\r
+ if (Index < SavedData->NumberOfExtractHandler) {\r
+ SavedData->ExtractDecodeHandlerTable [Index] = DecodeHandler;\r
+ SavedData->ExtractGetInfoHandlerTable [Index] = GetInfoHandler;\r
+ return RETURN_SUCCESS;\r
+ }\r
+ \r
+ //\r
+ // Check the global table is enough to contain new Handler.\r
+ //\r
+ if (SavedData->NumberOfExtractHandler >= PcdGet32 (PcdMaximumGuidedExtractHandler)) {\r
+ return RETURN_OUT_OF_RESOURCES;\r
+ }\r
+ \r
+ //\r
+ // Register new Handler and guid value.\r
+ //\r
+ CopyGuid (&SavedData->ExtractHandlerGuidTable [SavedData->NumberOfExtractHandler], SectionGuid);\r
+ SavedData->ExtractDecodeHandlerTable [SavedData->NumberOfExtractHandler] = DecodeHandler;\r
+ SavedData->ExtractGetInfoHandlerTable [SavedData->NumberOfExtractHandler++] = GetInfoHandler;\r
+ \r
+ return RETURN_SUCCESS;\r
+}\r
+\r
+UINTN\r
+EFIAPI\r
+ExtractGuidedSectionGetGuidList (\r
+ IN OUT GUID **ExtractHandlerGuidTable\r
+ )\r
+{\r
+ PRE_PI_EXTRACT_GUIDED_SECTION_DATA *SavedData;\r
+\r
+ ASSERT(ExtractHandlerGuidTable != NULL);\r
+ \r
+ SavedData = GetSavedData();\r
+ \r
+ *ExtractHandlerGuidTable = SavedData->ExtractHandlerGuidTable;\r
+ return SavedData->NumberOfExtractHandler;\r
+}\r
+\r
+RETURN_STATUS\r
+EFIAPI\r
+ExtractGuidedSectionGetInfo (\r
+ IN CONST VOID *InputSection,\r
+ OUT UINT32 *OutputBufferSize,\r
+ OUT UINT32 *ScratchBufferSize,\r
+ OUT UINT16 *SectionAttribute \r
+ )\r
+{\r
+ PRE_PI_EXTRACT_GUIDED_SECTION_DATA *SavedData;\r
+ UINT32 Index;\r
+ \r
+ if (InputSection == NULL) {\r
+ return RETURN_INVALID_PARAMETER;\r
+ }\r
+ \r
+ ASSERT (OutputBufferSize != NULL);\r
+ ASSERT (ScratchBufferSize != NULL);\r
+ ASSERT (SectionAttribute != NULL);\r
+ \r
+ SavedData = GetSavedData();\r
+\r
+ //\r
+ // Search the match registered GetInfo handler for the input guided section.\r
+ //\r
+ for (Index = 0; Index < SavedData->NumberOfExtractHandler; Index ++) {\r
+ if (CompareGuid (&SavedData->ExtractHandlerGuidTable[Index], &(((EFI_GUID_DEFINED_SECTION *) InputSection)->SectionDefinitionGuid))) {\r
+ break;\r
+ }\r
+ }\r
+\r
+ //\r
+ // Not found, the input guided section is not supported. \r
+ //\r
+ if (Index == SavedData->NumberOfExtractHandler) {\r
+ return RETURN_INVALID_PARAMETER;\r
+ }\r
+\r
+ //\r
+ // Call the match handler to getinfo for the input section data.\r
+ //\r
+ return SavedData->ExtractGetInfoHandlerTable [Index] (\r
+ InputSection,\r
+ OutputBufferSize,\r
+ ScratchBufferSize,\r
+ SectionAttribute\r
+ );\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
+ )\r
+{\r
+ PRE_PI_EXTRACT_GUIDED_SECTION_DATA *SavedData;\r
+ UINT32 Index;\r
+ \r
+ if (InputSection == NULL) {\r
+ return RETURN_INVALID_PARAMETER;\r
+ }\r
+ \r
+ ASSERT (OutputBuffer != NULL);\r
+ ASSERT (AuthenticationStatus != NULL);\r
+\r
+ SavedData = GetSavedData();\r
+\r
+ //\r
+ // Search the match registered GetInfo handler for the input guided section.\r
+ //\r
+ for (Index = 0; Index < SavedData->NumberOfExtractHandler; Index ++) {\r
+ if (CompareGuid (&SavedData->ExtractHandlerGuidTable[Index], &(((EFI_GUID_DEFINED_SECTION *) InputSection)->SectionDefinitionGuid))) {\r
+ break;\r
+ }\r
+ }\r
+\r
+ //\r
+ // Not found, the input guided section is not supported. \r
+ //\r
+ if (Index == SavedData->NumberOfExtractHandler) {\r
+ return RETURN_INVALID_PARAMETER;\r
+ }\r
+\r
+ //\r
+ // Call the match handler to getinfo for the input section data.\r
+ //\r
+ return SavedData->ExtractDecodeHandlerTable [Index] (\r
+ InputSection,\r
+ OutputBuffer,\r
+ ScratchBuffer,\r
+ AuthenticationStatus\r
+ );\r
+}\r
+\r
+RETURN_STATUS\r
+EFIAPI\r
+ExtractGuidedSectionLibConstructor (\r
+ VOID\r
+ )\r
+{\r
+ PRE_PI_EXTRACT_GUIDED_SECTION_DATA SavedData;\r
+ GUID HobGuid = PRE_PI_EXTRACT_GUIDED_SECTION_DATA_GUID;\r
+\r
+ //\r
+ // Allocate global pool space to store the registered handler and its guid value.\r
+ //\r
+ SavedData.ExtractHandlerGuidTable = (GUID *)AllocatePool(PcdGet32(PcdMaximumGuidedExtractHandler) * sizeof(GUID));\r
+ if (SavedData.ExtractHandlerGuidTable == NULL) {\r
+ return RETURN_OUT_OF_RESOURCES;\r
+ }\r
+ \r
+ SavedData.ExtractDecodeHandlerTable = (EXTRACT_GUIDED_SECTION_DECODE_HANDLER *)AllocatePool(PcdGet32(PcdMaximumGuidedExtractHandler) * sizeof(EXTRACT_GUIDED_SECTION_DECODE_HANDLER));\r
+ if (SavedData.ExtractDecodeHandlerTable == NULL) {\r
+ return RETURN_OUT_OF_RESOURCES;\r
+ }\r
+\r
+ SavedData.ExtractGetInfoHandlerTable = (EXTRACT_GUIDED_SECTION_GET_INFO_HANDLER *)AllocatePool(PcdGet32(PcdMaximumGuidedExtractHandler) * sizeof(EXTRACT_GUIDED_SECTION_GET_INFO_HANDLER));\r
+ if (SavedData.ExtractGetInfoHandlerTable == NULL) {\r
+ return RETURN_OUT_OF_RESOURCES;\r
+ }\r
+ \r
+ //\r
+ // the initialized number is Zero.\r
+ //\r
+ SavedData.NumberOfExtractHandler = 0;\r
+ \r
+ BuildGuidDataHob(&HobGuid, &SavedData, sizeof(SavedData));\r
+ \r
+ return RETURN_SUCCESS;\r
+}\r