[mirror_edk2.git] / MdeModulePkg / Core / Pei / FwVol / FwVol.c

/** @file\r
  Pei Core Firmware File System service routines.\r
  \r
Copyright (c) 2006 - 2009, 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
                                                                                          \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 "FwVol.h"\r
\r
EFI_PEI_NOTIFY_DESCRIPTOR mNotifyOnFvInfoList = {\r
  (EFI_PEI_PPI_DESCRIPTOR_NOTIFY_CALLBACK | EFI_PEI_PPI_DESCRIPTOR_TERMINATE_LIST),\r
  &gEfiPeiFirmwareVolumeInfoPpiGuid,\r
  FirmwareVolmeInfoPpiNotifyCallback \r
};\r
\r
EFI_PEI_FIRMWARE_VOLUME_PPI mPeiFfs2FvPpi = {\r
  PeiFfs2FvPpiProcessVolume,\r
  PeiFfs2FvPpiFindFileByType,\r
  PeiFfs2FvPpiFindFileByName,\r
  PeiFfs2FvPpiGetFileInfo,\r
  PeiFfs2FvPpiGetVolumeInfo,\r
  PeiFfs2FvPpiFindSectionByType\r
};\r
            \r
EFI_PEI_PPI_DESCRIPTOR  mPeiFfs2FvPpiList = {\r
  (EFI_PEI_PPI_DESCRIPTOR_PPI | EFI_PEI_PPI_DESCRIPTOR_TERMINATE_LIST),\r
  &gEfiFirmwareFileSystem2Guid,\r
  &mPeiFfs2FvPpi\r
};\r
 \r
/**\r
  Returns the file state set by the highest zero bit in the State field\r
\r
  @param ErasePolarity   Erase Polarity  as defined by EFI_FVB2_ERASE_POLARITY\r
                         in the Attributes field.\r
  @param FfsHeader       Pointer to FFS File Header.\r
\r
  @retval EFI_FFS_FILE_STATE File state is set by the highest none zero bit \r
                             in the header State field.\r
**/\r
EFI_FFS_FILE_STATE\r
GetFileState(\r
  IN UINT8                ErasePolarity,\r
  IN EFI_FFS_FILE_HEADER  *FfsHeader\r
  )\r
{\r
  EFI_FFS_FILE_STATE  FileState;\r
  EFI_FFS_FILE_STATE  HighestBit;\r
\r
  FileState = FfsHeader->State;\r
\r
  if (ErasePolarity != 0) {\r
    FileState = (EFI_FFS_FILE_STATE)~FileState;\r
  }\r
  \r
  //\r
  // Get file state set by its highest none zero bit.\r
  //\r
  HighestBit = 0x80;\r
  while (HighestBit != 0 && (HighestBit & FileState) == 0) {\r
    HighestBit >>= 1;\r
  }\r
\r
  return HighestBit;\r
} \r
\r
/**\r
  Calculates the checksum of the header of a file.\r
\r
  @param FileHeader      Pointer to FFS File Header.\r
\r
  @return Checksum of the header.\r
          Zero means the header is good.\r
          Non-zero means the header is bad.\r
**/\r
UINT8\r
CalculateHeaderChecksum (\r
  IN EFI_FFS_FILE_HEADER  *FileHeader\r
  )\r
{\r
  EFI_FFS_FILE_HEADER TestFileHeader;\r
  \r
  CopyMem (&TestFileHeader, FileHeader, sizeof (EFI_FFS_FILE_HEADER));\r
  //\r
  // Ingore State and File field in FFS header.\r
  //\r
  TestFileHeader.State = 0;\r
  TestFileHeader.IntegrityCheck.Checksum.File = 0;\r
\r
  return CalculateSum8 ((CONST UINT8 *) &TestFileHeader, sizeof (EFI_FFS_FILE_HEADER));\r
}\r
\r
/**\r
  Find FV handler according to FileHandle in that FV.\r
\r
  @param FileHandle      Handle of file image\r
  \r
  @return Pointer to instance of PEI_CORE_FV_HANDLE.\r
**/\r
PEI_CORE_FV_HANDLE*\r
FileHandleToVolume (\r
  IN   EFI_PEI_FILE_HANDLE          FileHandle\r
  )\r
{\r
  UINTN                       Index;\r
  PEI_CORE_INSTANCE           *PrivateData;\r
  EFI_FIRMWARE_VOLUME_HEADER  *FwVolHeader;\r
\r
  PrivateData = PEI_CORE_INSTANCE_FROM_PS_THIS (GetPeiServicesTablePointer ());\r
  \r
  for (Index = 0; Index < PrivateData->FvCount; Index++) {\r
    FwVolHeader = PrivateData->Fv[Index].FvHeader;\r
    if (((UINT64) (UINTN) FileHandle > (UINT64) (UINTN) FwVolHeader ) &&   \\r
        ((UINT64) (UINTN) FileHandle <= ((UINT64) (UINTN) FwVolHeader + FwVolHeader->FvLength - 1))) {\r
      return &PrivateData->Fv[Index];\r
    }\r
  }\r
  return NULL;\r
}\r
\r
/**\r
  Given the input file pointer, search for the first matching file in the\r
  FFS volume as defined by SearchType. The search starts from FileHeader inside\r
  the Firmware Volume defined by FwVolHeader.\r
  If SearchType is EFI_FV_FILETYPE_ALL, the first FFS file will return without check its file type.\r
  If SearchType is PEI_CORE_INTERNAL_FFS_FILE_DISPATCH_TYPE, \r
  the first PEIM, or COMBINED PEIM or FV file type FFS file will return.  \r
\r
  @param FvHandle        Pointer to the FV header of the volume to search\r
  @param FileName        File name\r
  @param SearchType      Filter to find only files of this type.\r
                         Type EFI_FV_FILETYPE_ALL causes no filtering to be done.\r
  @param FileHandle      This parameter must point to a valid FFS volume.\r
  @param AprioriFile     Pointer to AprioriFile image in this FV if has\r
\r
  @return EFI_NOT_FOUND  No files matching the search criteria were found\r
  @retval EFI_SUCCESS    Success to search given file\r
\r
**/\r
EFI_STATUS\r
FindFileEx (\r
  IN  CONST EFI_PEI_FV_HANDLE        FvHandle,\r
  IN  CONST EFI_GUID                 *FileName,   OPTIONAL\r
  IN        EFI_FV_FILETYPE          SearchType,\r
  IN OUT    EFI_PEI_FILE_HANDLE      *FileHandle,\r
  IN OUT    EFI_PEI_FV_HANDLE        *AprioriFile  OPTIONAL\r
  )\r
{\r
  EFI_FIRMWARE_VOLUME_HEADER           *FwVolHeader;\r
  EFI_FFS_FILE_HEADER                   **FileHeader;\r
  EFI_FFS_FILE_HEADER                   *FfsFileHeader;\r
  UINT32                                FileLength;\r
  UINT32                                FileOccupiedSize;\r
  UINT32                                FileOffset;\r
  UINT64                                FvLength;\r
  UINT8                                 ErasePolarity;\r
  UINT8                                 FileState;\r
  \r
  //\r
  // Convert the handle of FV to FV header for memory-mapped firmware volume\r
  //\r
  FwVolHeader = (EFI_FIRMWARE_VOLUME_HEADER *) FvHandle;\r
  FileHeader  = (EFI_FFS_FILE_HEADER **)FileHandle;\r
\r
  FvLength = FwVolHeader->FvLength;\r
  if ((FwVolHeader->Attributes & EFI_FVB2_ERASE_POLARITY) != 0) {\r
    ErasePolarity = 1;\r
  } else {\r
    ErasePolarity = 0;\r
  }\r
  \r
  //\r
  // If FileHeader is not specified (NULL) or FileName is not NULL,\r
  // start with the first file in the firmware volume.  Otherwise,\r
  // start from the FileHeader.\r
  //\r
  if ((*FileHeader == NULL) || (FileName != NULL)) {\r
    FfsFileHeader = (EFI_FFS_FILE_HEADER *)((UINT8 *)FwVolHeader + FwVolHeader->HeaderLength);\r
  } else {\r
    //\r
    // Length is 24 bits wide so mask upper 8 bits\r
    // FileLength is adjusted to FileOccupiedSize as it is 8 byte aligned.\r
    //\r
    FileLength = *(UINT32 *)(*FileHeader)->Size & 0x00FFFFFF;\r
    FileOccupiedSize = GET_OCCUPIED_SIZE (FileLength, 8);\r
    FfsFileHeader = (EFI_FFS_FILE_HEADER *)((UINT8 *)*FileHeader + FileOccupiedSize);\r
  }\r
  \r
  FileOffset = (UINT32) ((UINT8 *)FfsFileHeader - (UINT8 *)FwVolHeader);\r
  ASSERT (FileOffset <= 0xFFFFFFFF);\r
\r
  while (FileOffset < (FvLength - sizeof (EFI_FFS_FILE_HEADER))) {\r
    //\r
    // Get FileState which is the highest bit of the State \r
    //\r
    FileState = GetFileState (ErasePolarity, FfsFileHeader);\r
    switch (FileState) {\r
\r
    case EFI_FILE_HEADER_INVALID:\r
      FileOffset    += sizeof(EFI_FFS_FILE_HEADER);\r
      FfsFileHeader =  (EFI_FFS_FILE_HEADER *)((UINT8 *)FfsFileHeader + sizeof(EFI_FFS_FILE_HEADER));\r
      break;\r
        \r
    case EFI_FILE_DATA_VALID:\r
    case EFI_FILE_MARKED_FOR_UPDATE:\r
      if (CalculateHeaderChecksum (FfsFileHeader) != 0) {\r
        ASSERT (FALSE);\r
        *FileHeader = NULL;\r
        return EFI_NOT_FOUND;\r
      }\r
\r
      FileLength       = *(UINT32 *)(FfsFileHeader->Size) & 0x00FFFFFF;\r
      FileOccupiedSize = GET_OCCUPIED_SIZE(FileLength, 8);\r
\r
      if (FileName != NULL) {\r
        if (CompareGuid (&FfsFileHeader->Name, (EFI_GUID*)FileName)) {\r
          *FileHeader = FfsFileHeader;\r
          return EFI_SUCCESS;\r
        }\r
      } else if (SearchType == PEI_CORE_INTERNAL_FFS_FILE_DISPATCH_TYPE) {\r
        if ((FfsFileHeader->Type == EFI_FV_FILETYPE_PEIM) || \r
            (FfsFileHeader->Type == EFI_FV_FILETYPE_COMBINED_PEIM_DRIVER) ||\r
            (FfsFileHeader->Type == EFI_FV_FILETYPE_FIRMWARE_VOLUME_IMAGE)) { \r
          \r
          *FileHeader = FfsFileHeader;\r
          return EFI_SUCCESS;\r
        } else if (AprioriFile != NULL) {\r
          if (FfsFileHeader->Type == EFI_FV_FILETYPE_FREEFORM) {\r
            if (CompareGuid (&FfsFileHeader->Name, &gPeiAprioriFileNameGuid)) {\r
              *AprioriFile = FfsFileHeader;\r
            }           \r
          } \r
        }\r
      } else if (((SearchType == FfsFileHeader->Type) || (SearchType == EFI_FV_FILETYPE_ALL)) && \r
                 (FfsFileHeader->Type != EFI_FV_FILETYPE_FFS_PAD)) { \r
        *FileHeader = FfsFileHeader;\r
        return EFI_SUCCESS;\r
      }\r
\r
      FileOffset    += FileOccupiedSize; \r
      FfsFileHeader =  (EFI_FFS_FILE_HEADER *)((UINT8 *)FfsFileHeader + FileOccupiedSize);\r
      break;\r
    \r
    case EFI_FILE_DELETED:\r
      FileLength       =  *(UINT32 *)(FfsFileHeader->Size) & 0x00FFFFFF;\r
      FileOccupiedSize =  GET_OCCUPIED_SIZE(FileLength, 8);\r
      FileOffset       += FileOccupiedSize;\r
      FfsFileHeader    =  (EFI_FFS_FILE_HEADER *)((UINT8 *)FfsFileHeader + FileOccupiedSize);\r
      break;\r
\r
    default:\r
      *FileHeader = NULL;\r
      return EFI_NOT_FOUND;\r
    } \r
  }\r
  \r
  *FileHeader = NULL;\r
  return EFI_NOT_FOUND;  \r
}\r
\r
/**\r
  Initialize PeiCore Fv List.\r
\r
  @param PrivateData     - Pointer to PEI_CORE_INSTANCE.\r
  @param SecCoreData     - Pointer to EFI_SEC_PEI_HAND_OFF.\r
**/\r
VOID \r
PeiInitializeFv (\r
  IN  PEI_CORE_INSTANCE           *PrivateData,\r
  IN CONST EFI_SEC_PEI_HAND_OFF   *SecCoreData\r
  )\r
{\r
  EFI_STATUS                    Status;\r
  EFI_PEI_FIRMWARE_VOLUME_PPI   *FvPpi;\r
  EFI_PEI_FV_HANDLE             FvHandle;\r
  EFI_FIRMWARE_VOLUME_HEADER    *BfvHeader;\r
  \r
  //\r
  // Install FV_PPI for FFS2 file system.\r
  //\r
  PeiServicesInstallPpi (&mPeiFfs2FvPpiList);\r
  \r
  BfvHeader = (EFI_FIRMWARE_VOLUME_HEADER *)SecCoreData->BootFirmwareVolumeBase;\r
  \r
  //\r
  // The FV_PPI in BFV's format should be installed.\r
  //\r
  Status = PeiServicesLocatePpi (\r
             &BfvHeader->FileSystemGuid,\r
             0,\r
             NULL,\r
             (VOID**)&FvPpi\r
             );\r
  ASSERT_EFI_ERROR (Status);\r
    \r
  //\r
  // Get handle of BFV\r
  //\r
  FvPpi->ProcessVolume (\r
           FvPpi, \r
           SecCoreData->BootFirmwareVolumeBase,\r
           (UINTN)BfvHeader->FvLength,\r
           &FvHandle\r
           );\r
                        \r
  //\r
  // Post a call-back for the FvInfoPPI services to expose\r
  // additional Fvs to PeiCore.\r
  //\r
  Status = PeiServicesNotifyPpi (&mNotifyOnFvInfoList);\r
  ASSERT_EFI_ERROR (Status);\r
\r
}\r
  \r
/**\r
  Process Firmware Volum Information once FvInfoPPI install.\r
  The FV Info will be registered into PeiCore private data structure.\r
  And search the inside FV image, if found, the new FV INFO PPI will be installed.\r
\r
  @param PeiServices       An indirect pointer to the EFI_PEI_SERVICES table published by the PEI Foundation\r
  @param NotifyDescriptor  Address of the notification descriptor data structure.\r
  @param Ppi               Address of the PPI that was installed.\r
\r
  @retval EFI_SUCCESS    The FV Info is registered into PeiCore private data structure.\r
  @return if not EFI_SUCESS, fail to verify FV.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
FirmwareVolmeInfoPpiNotifyCallback (\r
  IN EFI_PEI_SERVICES              **PeiServices,\r
  IN EFI_PEI_NOTIFY_DESCRIPTOR     *NotifyDescriptor,\r
  IN VOID                          *Ppi\r
  )\r
{\r
  EFI_PEI_FIRMWARE_VOLUME_INFO_PPI      *FvInfoPpi;\r
  EFI_PEI_FIRMWARE_VOLUME_PPI           *FvPpi;\r
  PEI_CORE_INSTANCE                     *PrivateData;\r
  EFI_STATUS                            Status;\r
  EFI_PEI_FV_HANDLE                     FvHandle;\r
  UINTN                                 FvIndex;\r
  \r
  Status       = EFI_SUCCESS;\r
  PrivateData  = PEI_CORE_INSTANCE_FROM_PS_THIS (PeiServices);\r
\r
  if (PrivateData->FvCount >= FixedPcdGet32 (PcdPeiCoreMaxFvSupported)) {\r
    DEBUG ((EFI_D_ERROR, "The number of Fv Images (%d) exceed the max supported FVs (%d) in Pei", PrivateData->FvCount + 1, FixedPcdGet32 (PcdPeiCoreMaxFvSupported)));\r
    DEBUG ((EFI_D_ERROR, "PcdPeiCoreMaxFvSupported value need be reconfigurated in DSC"));\r
    ASSERT (FALSE);\r
  }\r
\r
  FvInfoPpi = (EFI_PEI_FIRMWARE_VOLUME_INFO_PPI *)Ppi;\r
\r
  //\r
  // Locate the corresponding FV_PPI according to founded FV's format guid\r
  //\r
  Status = PeiServicesLocatePpi (\r
             &FvInfoPpi->FvFormat, \r
             0, \r
             NULL,\r
             (VOID**)&FvPpi\r
             );\r
  if (!EFI_ERROR (Status)) {\r
    //\r
    // Process new found FV and get FV handle.\r
    //\r
    Status = FvPpi->ProcessVolume (FvPpi, FvInfoPpi->FvInfo, FvInfoPpi->FvInfoSize, &FvHandle);\r
    if (EFI_ERROR (Status)) {\r
      DEBUG ((EFI_D_ERROR, "Fail to process new found FV, FV may be corrupted!"));\r
      return Status;\r
    }\r
    DEBUG ((EFI_D_INFO, "Found and process new FV %p, all fv's count is %d\n", FvHandle, PrivateData->FvCount));\r
  } else {\r
    DEBUG ((EFI_D_ERROR, "Fail to process FV %p because no corresponding EFI_FIRMWARE_VOLUME_PPI is found!\n", FvInfoPpi->FvInfo));\r
    \r
    //\r
    // If can not find EFI_FIRMWARE_VOLUME_PPI to process firmware to get FvHandle, \r
    // use the address of FV buffer as its handle.\r
    //\r
    FvHandle = FvInfoPpi->FvInfo;\r
    \r
    //\r
    // Check whether the FV has already been processed.\r
    //\r
    for (FvIndex = 0; FvIndex < PrivateData->FvCount; FvIndex ++) {\r
      if (PrivateData->Fv[FvIndex].FvHandle == FvHandle) {\r
        DEBUG ((EFI_D_INFO, "The Fv %p has already been processed!\n", FvHandle));\r
        return EFI_SUCCESS;\r
      }\r
    }    \r
    \r
    PrivateData->Fv[PrivateData->FvCount].FvHeader = (EFI_FIRMWARE_VOLUME_HEADER*) FvInfoPpi->FvInfo;\r
    PrivateData->Fv[PrivateData->FvCount].FvPpi    = NULL;\r
    PrivateData->Fv[PrivateData->FvCount].FvHandle = FvHandle;\r
    PrivateData->FvCount ++;\r
  }\r
  \r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Go through the file to search SectionType section. \r
  Search within encapsulation sections (compression and GUIDed) recursively, \r
  until the match section is found.\r
  \r
  @param PeiServices       An indirect pointer to the EFI_PEI_SERVICES table published by the PEI Foundation.\r
  @param SectionType       Filter to find only section of this type.\r
  @param Section           From where to search.\r
  @param SectionSize       The file size to search.\r
  @param OutputBuffer      A pointer to the discovered section, if successful.\r
                           NULL if section not found\r
\r
  @return EFI_NOT_FOUND    The match section is not found.\r
  @return EFI_SUCCESS      The match section is found.\r
\r
**/\r
EFI_STATUS\r
ProcessSection (\r
  IN CONST EFI_PEI_SERVICES     **PeiServices,\r
  IN EFI_SECTION_TYPE           SectionType,\r
  IN EFI_COMMON_SECTION_HEADER  *Section,\r
  IN UINTN                      SectionSize,\r
  OUT VOID                      **OutputBuffer\r
  )\r
{\r
  EFI_STATUS                              Status;\r
  UINT32                                  SectionLength;\r
  UINT32                                  ParsedLength;\r
  EFI_PEI_GUIDED_SECTION_EXTRACTION_PPI   *GuidSectionPpi;\r
  EFI_PEI_DECOMPRESS_PPI                  *DecompressPpi;\r
  VOID                                    *PpiOutput;\r
  UINTN                                   PpiOutputSize;\r
  UINTN                                   Index;\r
  UINT32                                  Authentication;\r
  PEI_CORE_INSTANCE                       *PrivateData;\r
\r
  PrivateData   = PEI_CORE_INSTANCE_FROM_PS_THIS (PeiServices);\r
  *OutputBuffer = NULL;\r
  ParsedLength  = 0;\r
  Index         = 0;\r
  Status        = EFI_NOT_FOUND;\r
  PpiOutput     = NULL;\r
  PpiOutputSize = 0;\r
  while (ParsedLength < SectionSize) {\r
    if (Section->Type == SectionType) {\r
      *OutputBuffer = (VOID *)(Section + 1);\r
      return EFI_SUCCESS;\r
    } else if ((Section->Type == EFI_SECTION_GUID_DEFINED) || (Section->Type == EFI_SECTION_COMPRESSION)) {\r
      //\r
      // Check the encapsulated section is extracted into the cache data.\r
      //\r
      for (Index = 0; Index < PrivateData->CacheSection.AllSectionCount; Index ++) {\r
        if (Section == PrivateData->CacheSection.Section[Index]) {\r
          PpiOutput     = PrivateData->CacheSection.SectionData[Index];\r
          PpiOutputSize = PrivateData->CacheSection.SectionSize[Index];\r
          //\r
          // Search section directly from the cache data.\r
          //\r
          return ProcessSection (\r
                   PeiServices,\r
                   SectionType, \r
                   PpiOutput, \r
                   PpiOutputSize, \r
                   OutputBuffer \r
                   );\r
        }\r
      }\r
      \r
      Status = EFI_NOT_FOUND;\r
      if (Section->Type == EFI_SECTION_GUID_DEFINED) {\r
        Status = PeiServicesLocatePpi (\r
                   &((EFI_GUID_DEFINED_SECTION *)Section)->SectionDefinitionGuid, \r
                   0, \r
                   NULL, \r
                   (VOID **) &GuidSectionPpi\r
                   );\r
        if (!EFI_ERROR (Status)) {\r
          Status = GuidSectionPpi->ExtractSection (\r
                                     GuidSectionPpi,\r
                                     Section,\r
                                     &PpiOutput,\r
                                     &PpiOutputSize,\r
                                     &Authentication\r
                                     );\r
        }\r
      } else if (Section->Type == EFI_SECTION_COMPRESSION) {\r
        Status = PeiServicesLocatePpi (&gEfiPeiDecompressPpiGuid, 0, NULL, (VOID **) &DecompressPpi);\r
        if (!EFI_ERROR (Status)) {\r
          Status = DecompressPpi->Decompress (\r
                                    DecompressPpi,\r
                                    (CONST EFI_COMPRESSION_SECTION*) Section,\r
                                    &PpiOutput,\r
                                    &PpiOutputSize\r
                                    );\r
        }\r
      }\r
      \r
      if (!EFI_ERROR (Status)) {\r
        //\r
        // Update cache section data.\r
        //\r
        if (PrivateData->CacheSection.AllSectionCount < CACHE_SETION_MAX_NUMBER) {\r
          PrivateData->CacheSection.AllSectionCount ++;\r
        }\r
        PrivateData->CacheSection.Section [PrivateData->CacheSection.SectionIndex]     = Section;\r
        PrivateData->CacheSection.SectionData [PrivateData->CacheSection.SectionIndex] = PpiOutput;\r
        PrivateData->CacheSection.SectionSize [PrivateData->CacheSection.SectionIndex] = PpiOutputSize;\r
        PrivateData->CacheSection.SectionIndex = (PrivateData->CacheSection.SectionIndex + 1)%CACHE_SETION_MAX_NUMBER;\r
        \r
        return ProcessSection (\r
                 PeiServices,\r
                 SectionType, \r
                 PpiOutput, \r
                 PpiOutputSize, \r
                 OutputBuffer \r
                 );\r
      }\r
    }\r
\r
    //\r
    // Size is 24 bits wide so mask upper 8 bits. \r
    // SectionLength is adjusted it is 4 byte aligned.\r
    // Go to the next section\r
    //\r
    SectionLength = *(UINT32 *)Section->Size & 0x00FFFFFF;\r
    SectionLength = GET_OCCUPIED_SIZE (SectionLength, 4);\r
    ASSERT (SectionLength != 0);\r
    ParsedLength += SectionLength;\r
    Section = (EFI_COMMON_SECTION_HEADER *)((UINT8 *)Section + SectionLength);\r
  }\r
  \r
  return EFI_NOT_FOUND;\r
}\r
\r
\r
/**\r
  Searches for the next matching section within the specified file.\r
\r
  @param PeiServices     An indirect pointer to the EFI_PEI_SERVICES table published by the PEI Foundation\r
  @param SectionType     Filter to find only sections of this type.\r
  @param FileHandle      Pointer to the current file to search.\r
  @param SectionData     A pointer to the discovered section, if successful.\r
                         NULL if section not found\r
\r
  @retval EFI_NOT_FOUND  The section was not found.\r
  @retval EFI_SUCCESS    The section was found.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
PeiFfsFindSectionData (\r
  IN CONST EFI_PEI_SERVICES    **PeiServices,\r
  IN     EFI_SECTION_TYPE      SectionType,\r
  IN     EFI_PEI_FILE_HANDLE   FileHandle,\r
  OUT VOID                     **SectionData\r
  )\r
{\r
  PEI_CORE_FV_HANDLE           *CoreFvHandle;\r
  \r
  CoreFvHandle = FileHandleToVolume (FileHandle);\r
  if ((CoreFvHandle == NULL) || (CoreFvHandle->FvPpi == NULL)) {\r
    return EFI_NOT_FOUND;\r
  }\r
  \r
  return CoreFvHandle->FvPpi->FindSectionByType (CoreFvHandle->FvPpi, SectionType, FileHandle, SectionData);\r
}\r
\r
/**\r
  Searches for the next matching file in the firmware volume.\r
\r
  @param PeiServices     An indirect pointer to the EFI_PEI_SERVICES table published by the PEI Foundation.\r
  @param SearchType      Filter to find only files of this type.\r
                         Type EFI_FV_FILETYPE_ALL causes no filtering to be done.\r
  @param VolumeHandle    Handle of firmware volume in which to search.\r
  @param FileHandle      On entry, points to the current handle from which to begin searching or NULL to start\r
                         at the beginning of the firmware volume. On exit, points the file handle of the next file\r
                         in the volume or NULL if there are no more files.\r
\r
  @retval EFI_NOT_FOUND  The file was not found.\r
  @retval EFI_NOT_FOUND  The header checksum was not zero.\r
  @retval EFI_SUCCESS    The file was found.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
PeiFfsFindNextFile (\r
  IN CONST EFI_PEI_SERVICES      **PeiServices,\r
  IN UINT8                       SearchType,\r
  IN EFI_PEI_FV_HANDLE           FvHandle,\r
  IN OUT EFI_PEI_FILE_HANDLE     *FileHandle\r
  )\r
{\r
  PEI_CORE_FV_HANDLE      *CoreFvHandle;\r
  \r
  CoreFvHandle = FvHandleToCoreHandle (FvHandle);\r
  \r
  if ((CoreFvHandle == NULL) || CoreFvHandle->FvPpi == NULL) {\r
    return EFI_NOT_FOUND;\r
  }\r
  \r
  return CoreFvHandle->FvPpi->FindFileByType (CoreFvHandle->FvPpi, SearchType, FvHandle, FileHandle);\r
}\r
\r
\r
/**\r
  Search the firmware volumes by index\r
\r
  @param PeiServices     An indirect pointer to the EFI_PEI_SERVICES table published by the PEI Foundation\r
  @param Instance        This instance of the firmware volume to find. The value 0 is the Boot Firmware\r
                         Volume (BFV).\r
  @param VolumeHandle    On exit, points to the next volume handle or NULL if it does not exist.\r
\r
  @retval EFI_INVALID_PARAMETER  VolumeHandle is NULL\r
  @retval EFI_NOT_FOUND          The volume was not found.\r
  @retval EFI_SUCCESS            The volume was found.\r
\r
**/\r
EFI_STATUS \r
EFIAPI\r
PeiFfsFindNextVolume (\r
  IN CONST EFI_PEI_SERVICES         **PeiServices,\r
  IN     UINTN                      Instance,\r
  IN OUT EFI_PEI_FV_HANDLE          *VolumeHandle\r
  )\r
{\r
  PEI_CORE_INSTANCE  *Private;\r
  PEI_CORE_FV_HANDLE *CoreFvHandle;\r
  \r
  Private = PEI_CORE_INSTANCE_FROM_PS_THIS (PeiServices);\r
  \r
  CoreFvHandle = FindNextCoreFvHandle (Private, Instance);\r
  if (CoreFvHandle == NULL) {\r
    *VolumeHandle = NULL;\r
    return EFI_NOT_FOUND;\r
  }\r
  \r
  *VolumeHandle = CoreFvHandle->FvHandle;\r
  \r
  return EFI_SUCCESS;\r
}\r
\r
\r
/**\r
  Find a file within a volume by its name.\r
\r
  @param FileName        A pointer to the name of the file to find within the firmware volume.\r
  @param VolumeHandle    The firmware volume to search\r
  @param FileHandle      Upon exit, points to the found file's handle \r
                         or NULL if it could not be found.\r
\r
  @retval EFI_SUCCESS            File was found.\r
  @retval EFI_NOT_FOUND          File was not found.\r
  @retval EFI_INVALID_PARAMETER  VolumeHandle or FileHandle or FileName was NULL.\r
\r
**/\r
EFI_STATUS\r
EFIAPI \r
PeiFfsFindFileByName (\r
  IN  CONST EFI_GUID        *FileName,\r
  IN  EFI_PEI_FV_HANDLE     VolumeHandle,\r
  OUT EFI_PEI_FILE_HANDLE   *FileHandle\r
  )\r
{\r
  PEI_CORE_FV_HANDLE            *CoreFvHandle;\r
  \r
  if ((VolumeHandle == NULL) || (FileName == NULL) || (FileHandle == NULL)) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
  \r
  CoreFvHandle = FvHandleToCoreHandle (VolumeHandle);\r
  if ((CoreFvHandle == NULL) || (CoreFvHandle->FvPpi == NULL)) {\r
    return EFI_NOT_FOUND;\r
  }\r
  \r
  return CoreFvHandle->FvPpi->FindFileByName (CoreFvHandle->FvPpi, FileName, &VolumeHandle, FileHandle);\r
}\r
\r
/**\r
  Returns information about a specific file.\r
\r
  @param FileHandle       Handle of the file.\r
  @param FileInfo         Upon exit, points to the file’s information.\r
\r
  @retval EFI_INVALID_PARAMETER If FileInfo is NULL.\r
  @retval EFI_INVALID_PARAMETER If FileHandle does not represent a valid file.\r
  @retval EFI_SUCCESS           File information returned.\r
\r
**/\r
EFI_STATUS\r
EFIAPI \r
PeiFfsGetFileInfo (\r
  IN EFI_PEI_FILE_HANDLE  FileHandle,\r
  OUT EFI_FV_FILE_INFO    *FileInfo\r
  )\r
{\r
  PEI_CORE_FV_HANDLE          *CoreFvHandle;\r
  \r
  if ((FileHandle == NULL) || (FileInfo == NULL)) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  //\r
  // Retrieve the FirmwareVolume which the file resides in.\r
  //\r
  CoreFvHandle = FileHandleToVolume (FileHandle);\r
  if ((CoreFvHandle == NULL) || (CoreFvHandle->FvPpi == NULL)) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  return CoreFvHandle->FvPpi->GetFileInfo (CoreFvHandle->FvPpi, FileHandle, FileInfo);\r
}\r
\r
\r
/**\r
  Returns information about the specified volume.\r
\r
  @param VolumeHandle    Handle of the volume.\r
  @param VolumeInfo      Upon exit, points to the volume’s information.\r
\r
  @retval EFI_INVALID_PARAMETER If VolumeHandle does not represent a valid volume.\r
  @retval EFI_INVALID_PARAMETER If VolumeInfo is NULL.\r
  @retval EFI_SUCCESS           Volume information returned.\r
**/\r
EFI_STATUS\r
EFIAPI \r
PeiFfsGetVolumeInfo (\r
  IN EFI_PEI_FV_HANDLE  VolumeHandle,\r
  OUT EFI_FV_INFO       *VolumeInfo\r
  )\r
{\r
  PEI_CORE_FV_HANDLE                     *CoreHandle;\r
  \r
  if (VolumeInfo == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
  \r
  CoreHandle = FvHandleToCoreHandle (VolumeHandle);\r
  \r
  if ((CoreHandle == NULL) || (CoreHandle->FvPpi == NULL)) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
  \r
  return CoreHandle->FvPpi->GetVolumeInfo (CoreHandle->FvPpi, VolumeHandle, VolumeInfo);\r
}\r
\r
/**\r
  Get Fv image from the FV type file, then install FV INFO ppi, Build FV hob.\r
\r
  @param ParentFvCoreHandle   Pointer of EFI_CORE_FV_HANDLE to parent Fv image that contain this Fv image.\r
  @param ParentFvFileHandle   File handle of a Fv type file that contain this Fv image.\r
\r
  @retval EFI_NOT_FOUND         FV image can't be found.\r
  @retval EFI_SUCCESS           Successfully to process it.\r
  @retval EFI_OUT_OF_RESOURCES  Can not allocate page when aligning FV image\r
  @retval Others                Can not find EFI_SECTION_FIRMWARE_VOLUME_IMAGE section\r
  \r
**/\r
EFI_STATUS\r
ProcessFvFile (\r
  IN  PEI_CORE_FV_HANDLE          *ParentFvCoreHandle,\r
  IN  EFI_PEI_FILE_HANDLE         ParentFvFileHandle\r
  )\r
{\r
  EFI_STATUS                    Status;\r
  EFI_FV_INFO                   ParentFvImageInfo;\r
  UINT32                        FvAlignment;\r
  VOID                          *NewFvBuffer;\r
  EFI_PEI_HOB_POINTERS          HobPtr;\r
  EFI_PEI_FIRMWARE_VOLUME_PPI   *ParentFvPpi;\r
  EFI_PEI_FV_HANDLE             ParentFvHandle;\r
  EFI_FIRMWARE_VOLUME_HEADER    *FvHeader;\r
  EFI_FV_FILE_INFO              FileInfo;\r
  \r
  //\r
  // Check if this EFI_FV_FILETYPE_FIRMWARE_VOLUME_IMAGE file has already\r
  // been extracted.\r
  //\r
  HobPtr.Raw = GetHobList ();\r
  while ((HobPtr.Raw = GetNextHob (EFI_HOB_TYPE_FV2, HobPtr.Raw)) != NULL) {\r
    if (CompareGuid (&(((EFI_FFS_FILE_HEADER *)ParentFvFileHandle)->Name), &HobPtr.FirmwareVolume2->FileName)) {\r
      //\r
      // this FILE has been dispatched, it will not be dispatched again.\r
      //\r
      DEBUG ((EFI_D_INFO, "FV file %p has been dispatched!\r\n", ParentFvFileHandle));\r
      return EFI_SUCCESS;\r
    }\r
    HobPtr.Raw = GET_NEXT_HOB (HobPtr);\r
  }\r
\r
  ParentFvHandle = ParentFvCoreHandle->FvHandle;\r
  ParentFvPpi    = ParentFvCoreHandle->FvPpi;\r
  \r
  //\r
  // Find FvImage in FvFile\r
  //\r
  Status = ParentFvPpi->FindSectionByType (\r
                          ParentFvPpi,\r
                          EFI_SECTION_FIRMWARE_VOLUME_IMAGE,\r
                          ParentFvFileHandle,\r
                          (VOID **)&FvHeader\r
                          );\r
             \r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  //\r
  // FvAlignment must be more than 8 bytes required by FvHeader structure.\r
  //\r
  FvAlignment = 1 << ((FvHeader->Attributes & EFI_FVB2_ALIGNMENT) >> 16);\r
  if (FvAlignment < 8) {\r
    FvAlignment = 8;\r
  }\r
  \r
  //\r
  // Check FvImage\r
  //\r
  if ((UINTN) FvHeader % FvAlignment != 0) {\r
    NewFvBuffer = AllocateAlignedPages (EFI_SIZE_TO_PAGES ((UINT32) FvHeader->FvLength), FvAlignment);\r
    if (NewFvBuffer == NULL) {\r
      return EFI_OUT_OF_RESOURCES;\r
    }\r
    CopyMem (NewFvBuffer, FvHeader, (UINTN) FvHeader->FvLength);\r
    FvHeader = (EFI_FIRMWARE_VOLUME_HEADER*) NewFvBuffer;\r
  }\r
  \r
  Status = ParentFvPpi->GetVolumeInfo (ParentFvPpi, ParentFvHandle, &ParentFvImageInfo);\r
  ASSERT_EFI_ERROR (Status);\r
  \r
  Status = ParentFvPpi->GetFileInfo (ParentFvPpi, ParentFvFileHandle, &FileInfo);\r
  ASSERT_EFI_ERROR (Status);\r
  \r
  //\r
  // Install FvPpi and Build FvHob\r
  //\r
  PeiServicesInstallFvInfoPpi (\r
    &FvHeader->FileSystemGuid,\r
    (VOID**) FvHeader,\r
    (UINT32) FvHeader->FvLength,\r
    &ParentFvImageInfo.FvName,\r
    &FileInfo.FileName\r
    );\r
\r
  //\r
  // Inform the extracted FvImage to Fv HOB consumer phase, i.e. DXE phase\r
  //\r
  BuildFvHob (\r
    (EFI_PHYSICAL_ADDRESS) (UINTN) FvHeader,\r
    FvHeader->FvLength\r
    );\r
\r
  //\r
  // Makes the encapsulated volume show up in DXE phase to skip processing of\r
  // encapsulated file again.\r
  //\r
  BuildFv2Hob (\r
    (EFI_PHYSICAL_ADDRESS) (UINTN) FvHeader,\r
    FvHeader->FvLength,\r
    &ParentFvImageInfo.FvName,\r
    &FileInfo.FileName\r
    );\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Process a firmware volume and create a volume handle.\r
\r
  Create a volume handle from the information in the buffer. For\r
  memory-mapped firmware volumes, Buffer and BufferSize refer to\r
  the start of the firmware volume and the firmware volume size.\r
  For non memory-mapped firmware volumes, this points to a\r
  buffer which contains the necessary information for creating\r
  the firmware volume handle. Normally, these values are derived\r
  from the EFI_FIRMWARE_VOLUME_INFO_PPI.\r
  \r
  \r
  @param This                   Points to this instance of the\r
                                EFI_PEI_FIRMWARE_VOLUME_PPI.\r
  @param Buffer                 Points to the start of the buffer.\r
  @param BufferSize             Size of the buffer.\r
  @param FvHandle               Points to the returned firmware volume\r
                                handle. The firmware volume handle must\r
                                be unique within the system. \r
\r
  @retval EFI_SUCCESS           Firmware volume handle created.\r
  @retval EFI_VOLUME_CORRUPTED  Volume was corrupt.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
PeiFfs2FvPpiProcessVolume (\r
  IN  CONST  EFI_PEI_FIRMWARE_VOLUME_PPI *This,\r
  IN  VOID                               *Buffer,\r
  IN  UINTN                              BufferSize,\r
  OUT EFI_PEI_FV_HANDLE                  *FvHandle\r
  )\r
{\r
  EFI_STATUS          Status;\r
  PEI_CORE_INSTANCE   *PrivateData;\r
  EFI_PEI_FILE_HANDLE FileHandle;\r
  VOID                *DepexData;\r
  EFI_PEI_SERVICES    **PeiServices;\r
  UINTN               FvIndex;\r
  \r
  PeiServices = (EFI_PEI_SERVICES**) GetPeiServicesTablePointer ();\r
  PrivateData = PEI_CORE_INSTANCE_FROM_PS_THIS (PeiServices);\r
  \r
  //\r
  // The build-in EFI_PEI_FIRMWARE_VOLUME_PPI for FFS2 support memory-mapped\r
  // FV image and the handle is pointed to Fv image's buffer.\r
  //\r
  *FvHandle = (EFI_PEI_FV_HANDLE) Buffer;\r
  \r
  //\r
  // Do verify for given FV buffer.\r
  //\r
  Status = VerifyFv ((EFI_FIRMWARE_VOLUME_HEADER*) Buffer);\r
  if (EFI_ERROR(Status)) {\r
    DEBUG ((EFI_D_ERROR, "Fail to verify FV which address is 0x%11p", Buffer));\r
    return EFI_VOLUME_CORRUPTED;\r
  }\r
\r
  //\r
  // Check whether the FV has already been processed.\r
  //\r
  for (FvIndex = 0; FvIndex < PrivateData->FvCount; FvIndex ++) {\r
    if (PrivateData->Fv[FvIndex].FvHandle == *FvHandle) {\r
      DEBUG ((EFI_D_INFO, "The Fv %p has already been processed!", Buffer));\r
      return EFI_SUCCESS;\r
    }\r
  }\r
  \r
  //\r
  // Update internal PEI_CORE_FV array.\r
  //\r
  PrivateData->Fv[PrivateData->FvCount].FvHeader = (EFI_FIRMWARE_VOLUME_HEADER*) Buffer;\r
  PrivateData->Fv[PrivateData->FvCount].FvPpi    = (EFI_PEI_FIRMWARE_VOLUME_PPI*) This;\r
  PrivateData->Fv[PrivateData->FvCount].FvHandle = *FvHandle;\r
  \r
  DEBUG ((EFI_D_INFO, \r
          "The %dth FV start address is 0x%11p and size is 0x%08x\n", \r
          (UINT32) PrivateData->FvCount, \r
          (VOID *) Buffer, \r
          BufferSize\r
          ));\r
  PrivateData->FvCount ++;\r
  \r
  do {\r
    Status = This->FindFileByType (\r
                     This,\r
                     EFI_FV_FILETYPE_FIRMWARE_VOLUME_IMAGE,\r
                     *FvHandle,\r
                     &FileHandle\r
                     );\r
    if (!EFI_ERROR (Status)) {\r
      Status = This->FindSectionByType (\r
                       This,\r
                       EFI_SECTION_PEI_DEPEX,\r
                       FileHandle,\r
                       (VOID**)&DepexData\r
                       );\r
      if (!EFI_ERROR (Status)) {\r
        if (!PeimDispatchReadiness (PeiServices, DepexData)) {\r
          //\r
          // Dependency is not satisfied.\r
          //\r
          continue;\r
        }\r
      }\r
      \r
      DEBUG ((EFI_D_INFO, "Found firmware volume Image File %p in FV[%d] %p", FileHandle, PrivateData->FvCount - 1, *FvHandle));\r
      ProcessFvFile (&PrivateData->Fv[PrivateData->FvCount - 1], FileHandle);\r
    }\r
  } while (FileHandle != NULL);\r
\r
  return EFI_SUCCESS;\r
}  \r
\r
/**\r
  Finds the next file of the specified type.\r
\r
  This service enables PEI modules to discover additional firmware files. \r
  The FileHandle must be unique within the system.\r
\r
  @param This           Points to this instance of the\r
                        EFI_PEI_FIRMWARE_VOLUME_PPI.\r
  @param SearchType     A filter to find only files of this type. Type\r
                        EFI_FV_FILETYPE_ALL causes no filtering to be\r
                        done.\r
  @param FvHandle       Handle of firmware volume in which to\r
                        search.\r
  @param FileHandle     Points to the current handle from which to\r
                        begin searching or NULL to start at the\r
                        beginning of the firmware volume. Updated\r
                        upon return to reflect the file found.\r
\r
  @retval EFI_SUCCESS   The file was found.\r
  @retval EFI_NOT_FOUND The file was not found. FileHandle contains NULL.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
PeiFfs2FvPpiFindFileByType (\r
  IN CONST  EFI_PEI_FIRMWARE_VOLUME_PPI *This,\r
  IN        EFI_FV_FILETYPE             SearchType,\r
  IN        EFI_PEI_FV_HANDLE           FvHandle,\r
  IN OUT    EFI_PEI_FILE_HANDLE         *FileHandle\r
  )\r
{ \r
  return FindFileEx (FvHandle, NULL, SearchType, FileHandle, NULL);\r
}\r
\r
/**\r
  Find a file within a volume by its name. \r
  \r
  This service searches for files with a specific name, within\r
  either the specified firmware volume or all firmware volumes.\r
\r
  @param This                   Points to this instance of the\r
                                EFI_PEI_FIRMWARE_VOLUME_PPI.\r
  @param FileName               A pointer to the name of the file to find\r
                                within the firmware volume.\r
  @param FvHandle               Upon entry, the pointer to the firmware\r
                                volume to search or NULL if all firmware\r
                                volumes should be searched. Upon exit, the\r
                                actual firmware volume in which the file was\r
                                found.\r
  @param FileHandle             Upon exit, points to the found file's\r
                                handle or NULL if it could not be found.\r
\r
  @retval EFI_SUCCESS           File was found.\r
  @retval EFI_NOT_FOUND         File was not found.\r
  @retval EFI_INVALID_PARAMETER FvHandle or FileHandle or\r
                                FileName was NULL.\r
\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
PeiFfs2FvPpiFindFileByName (\r
  IN  CONST  EFI_PEI_FIRMWARE_VOLUME_PPI *This,\r
  IN  CONST  EFI_GUID                    *FileName,\r
  IN  EFI_PEI_FV_HANDLE                  *FvHandle,\r
  OUT EFI_PEI_FILE_HANDLE                *FileHandle  \r
  )\r
{\r
  EFI_STATUS        Status;\r
  PEI_CORE_INSTANCE *PrivateData;\r
  UINTN             Index;\r
  \r
  if ((FvHandle == NULL) || (FileName == NULL) || (FileHandle == NULL)) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
  \r
  if (*FvHandle != NULL) {\r
    Status = FindFileEx (*FvHandle, FileName, 0, FileHandle, NULL);\r
    if (Status == EFI_NOT_FOUND) {\r
      *FileHandle = NULL;\r
    }\r
  } else {   \r
    //\r
    // If *FvHandle = NULL, so search all FV for given filename\r
    //\r
    Status = EFI_NOT_FOUND;\r
    \r
    PrivateData = PEI_CORE_INSTANCE_FROM_PS_THIS (GetPeiServicesTablePointer());\r
    for (Index = 0; Index < PrivateData->FvCount; Index ++) {\r
      //\r
      // Only search the FV which is associated with a EFI_PEI_FIRMWARE_VOLUME_PPI instance.\r
      //\r
      if (PrivateData->Fv[Index].FvPpi != NULL) {\r
        Status = FindFileEx (PrivateData->Fv[Index].FvHandle, FileName, 0, FileHandle, NULL);\r
        if (!EFI_ERROR (Status)) {\r
          *FvHandle = PrivateData->Fv[Index].FvHandle;\r
        }\r
      }\r
    }\r
  }\r
  \r
  return Status;  \r
}  \r
\r
/**\r
  Returns information about a specific file.\r
\r
  This function returns information about a specific\r
  file, including its file name, type, attributes, starting\r
  address and size. \r
   \r
  @param This                     Points to this instance of the\r
                                  EFI_PEI_FIRMWARE_VOLUME_PPI.\r
  @param FileHandle               Handle of the file.\r
  @param FileInfo                 Upon exit, points to the file's\r
                                  information.\r
\r
  @retval EFI_SUCCESS             File information returned.\r
  @retval EFI_INVALID_PARAMETER   If FileHandle does not\r
                                  represent a valid file.\r
  @retval EFI_INVALID_PARAMETER   If FileInfo is NULL.\r
  \r
**/ \r
EFI_STATUS\r
EFIAPI\r
PeiFfs2FvPpiGetFileInfo (\r
  IN  CONST EFI_PEI_FIRMWARE_VOLUME_PPI   *This, \r
  IN        EFI_PEI_FILE_HANDLE           FileHandle, \r
  OUT       EFI_FV_FILE_INFO              *FileInfo\r
  )\r
{\r
  UINT8                       FileState;\r
  UINT8                       ErasePolarity;\r
  EFI_FFS_FILE_HEADER         *FileHeader;\r
  PEI_CORE_FV_HANDLE          *CoreFvHandle;\r
  \r
  if ((FileHandle == NULL) || (FileInfo == NULL)) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  //\r
  // Retrieve the FirmwareVolume which the file resides in.\r
  //\r
  CoreFvHandle = FileHandleToVolume (FileHandle);\r
  if (CoreFvHandle == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  if (CoreFvHandle->FvHeader->Attributes & EFI_FVB2_ERASE_POLARITY) {\r
    ErasePolarity = 1;\r
  } else {\r
    ErasePolarity = 0;\r
  }\r
\r
  //\r
  // Get FileState which is the highest bit of the State \r
  //\r
  FileState = GetFileState (ErasePolarity, (EFI_FFS_FILE_HEADER*)FileHandle);\r
\r
  switch (FileState) {\r
    case EFI_FILE_DATA_VALID:\r
    case EFI_FILE_MARKED_FOR_UPDATE:\r
      break;  \r
    default:\r
      return EFI_INVALID_PARAMETER;\r
    }\r
\r
  FileHeader = (EFI_FFS_FILE_HEADER *)FileHandle;\r
  CopyMem (&FileInfo->FileName, &FileHeader->Name, sizeof(EFI_GUID));\r
  FileInfo->FileType = FileHeader->Type;\r
  FileInfo->FileAttributes = FileHeader->Attributes;\r
  FileInfo->BufferSize = ((*(UINT32 *)FileHeader->Size) & 0x00FFFFFF) -  sizeof (EFI_FFS_FILE_HEADER);\r
  FileInfo->Buffer = (FileHeader + 1);\r
  return EFI_SUCCESS;  \r
}  \r
  \r
/**\r
  This function returns information about the firmware volume.\r
  \r
  @param This                     Points to this instance of the\r
                                  EFI_PEI_FIRMWARE_VOLUME_PPI.\r
  @param FvHandle                 Handle to the firmware handle.\r
  @param VolumeInfo               Points to the returned firmware volume\r
                                  information.\r
\r
  @retval EFI_SUCCESS             Information returned successfully.\r
  @retval EFI_INVALID_PARAMETER   FvHandle does not indicate a valid\r
                                  firmware volume or VolumeInfo is NULL.\r
\r
**/   \r
EFI_STATUS\r
EFIAPI\r
PeiFfs2FvPpiGetVolumeInfo (\r
  IN  CONST  EFI_PEI_FIRMWARE_VOLUME_PPI   *This, \r
  IN  EFI_PEI_FV_HANDLE                    FvHandle, \r
  OUT EFI_FV_INFO                          *VolumeInfo\r
  )\r
{\r
  EFI_FIRMWARE_VOLUME_HEADER             FwVolHeader;\r
  EFI_FIRMWARE_VOLUME_EXT_HEADER         *FwVolExHeaderInfo;\r
\r
  if (VolumeInfo == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
  \r
  //\r
  // VolumeHandle may not align at 8 byte, \r
  // but FvLength is UINT64 type, which requires FvHeader align at least 8 byte. \r
  // So, Copy FvHeader into the local FvHeader structure.\r
  //\r
  CopyMem (&FwVolHeader, FvHandle, sizeof (EFI_FIRMWARE_VOLUME_HEADER));\r
\r
  //\r
  // Check Fv Image Signature\r
  //\r
  if (FwVolHeader.Signature != EFI_FVH_SIGNATURE) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  ZeroMem (VolumeInfo, sizeof (EFI_FV_INFO));\r
  VolumeInfo->FvAttributes  = FwVolHeader.Attributes;\r
  VolumeInfo->FvStart       = (VOID *) FvHandle;\r
  VolumeInfo->FvSize        = FwVolHeader.FvLength;\r
  CopyMem (&VolumeInfo->FvFormat, &FwVolHeader.FileSystemGuid, sizeof(EFI_GUID));\r
\r
  if (FwVolHeader.ExtHeaderOffset != 0) {\r
    FwVolExHeaderInfo = (EFI_FIRMWARE_VOLUME_EXT_HEADER*)(((UINT8 *)FvHandle) + FwVolHeader.ExtHeaderOffset);\r
    CopyMem (&VolumeInfo->FvName, &FwVolExHeaderInfo->FvName, sizeof(EFI_GUID));\r
  }\r
  \r
  return EFI_SUCCESS;  \r
}    \r
\r
/**\r
  Find the next matching section in the firmware file.\r
  \r
  This service enables PEI modules to discover sections\r
  of a given type within a valid file.\r
  \r
  @param This             Points to this instance of the\r
                          EFI_PEI_FIRMWARE_VOLUME_PPI.\r
  @param SearchType       A filter to find only sections of this\r
                          type.\r
  @param FileHandle       Handle of firmware file in which to\r
                          search.\r
  @param SectionData      Updated upon  return to point to the\r
                          section found.\r
  \r
  @retval EFI_SUCCESS     Section was found.\r
  @retval EFI_NOT_FOUND   Section of the specified type was not\r
                          found. SectionData contains NULL.\r
**/\r
EFI_STATUS\r
EFIAPI\r
PeiFfs2FvPpiFindSectionByType (\r
  IN  CONST EFI_PEI_FIRMWARE_VOLUME_PPI    *This,\r
  IN        EFI_SECTION_TYPE               SearchType,\r
  IN        EFI_PEI_FILE_HANDLE            FileHandle,\r
  OUT VOID                                 **SectionData\r
  )\r
{\r
  EFI_FFS_FILE_HEADER                     *FfsFileHeader;\r
  UINT32                                  FileSize;\r
  EFI_COMMON_SECTION_HEADER               *Section;\r
  \r
  FfsFileHeader = (EFI_FFS_FILE_HEADER *)(FileHandle);\r
\r
  //\r
  // Size is 24 bits wide so mask upper 8 bits. \r
  // Does not include FfsFileHeader header size\r
  // FileSize is adjusted to FileOccupiedSize as it is 8 byte aligned.\r
  //\r
  Section = (EFI_COMMON_SECTION_HEADER *)(FfsFileHeader + 1);\r
  FileSize = *(UINT32 *)(FfsFileHeader->Size) & 0x00FFFFFF;\r
  FileSize -= sizeof (EFI_FFS_FILE_HEADER);\r
\r
  return ProcessSection (\r
           GetPeiServicesTablePointer (),\r
           SearchType, \r
           Section, \r
           FileSize, \r
           SectionData\r
           );  \r
}  \r
\r
/**\r
  Convert the handle of FV to pointer of corresponding PEI_CORE_FV_HANDLE.\r
  \r
  @param FvHandle   The handle of a FV.\r
  \r
  @retval NULL if can not find.\r
  @return Pointer of corresponding PEI_CORE_FV_HANDLE. \r
**/\r
PEI_CORE_FV_HANDLE *\r
FvHandleToCoreHandle (\r
  IN EFI_PEI_FV_HANDLE  FvHandle\r
  )\r
{\r
  UINTN             Index;\r
  PEI_CORE_INSTANCE *PrivateData;\r
  \r
  PrivateData = PEI_CORE_INSTANCE_FROM_PS_THIS (GetPeiServicesTablePointer());  \r
  for (Index = 0; Index < PrivateData->FvCount; Index ++) {\r
    if (FvHandle == PrivateData->Fv[Index].FvHandle) {\r
      return &PrivateData->Fv[Index];\r
    }\r
  }\r
  \r
  return NULL;\r
}  \r
\r
/**\r
  Get instance of PEI_CORE_FV_HANDLE for next volume according to given index.\r
  \r
  This routine also will install FvInfo ppi for FV hob in PI ways.\r
  \r
  @param Private    Pointer of PEI_CORE_INSTANCE\r
  @param Instance   The index of FV want to be searched.\r
  \r
  @return Instance of PEI_CORE_FV_HANDLE.\r
**/\r
PEI_CORE_FV_HANDLE *\r
FindNextCoreFvHandle (\r
  IN PEI_CORE_INSTANCE  *Private,\r
  IN UINTN              Instance\r
  )\r
{\r
  UINTN                    Index;\r
  BOOLEAN                  Match;\r
  EFI_HOB_FIRMWARE_VOLUME  *FvHob;\r
  \r
  //\r
  // Handle Framework FvHob and Install FvInfo Ppi for it.\r
  //\r
  if (FeaturePcdGet (PcdFrameworkCompatibilitySupport)) {\r
    //\r
    // Loop to search the wanted FirmwareVolume which supports FFS\r
    //\r
    FvHob = (EFI_HOB_FIRMWARE_VOLUME *)GetFirstHob (EFI_HOB_TYPE_FV);\r
    while (FvHob != NULL) {\r
      for (Index = 0, Match = FALSE; Index < Private->FvCount; Index++) {\r
        if ((EFI_PEI_FV_HANDLE)(UINTN)FvHob->BaseAddress == Private->Fv[Index].FvHeader) {\r
          Match = TRUE;\r
          break;\r
        }\r
      }\r
      //\r
      // If Not Found, Install FvInfo Ppi for it.\r
      //\r
      if (!Match) {\r
        PeiServicesInstallFvInfoPpi (\r
          &(((EFI_FIRMWARE_VOLUME_HEADER *)(UINTN)FvHob->BaseAddress)->FileSystemGuid),\r
          (VOID *)(UINTN)FvHob->BaseAddress,\r
          (UINT32)FvHob->Length,\r
          NULL,\r
          NULL\r
          );\r
      }\r
      FvHob = (EFI_HOB_FIRMWARE_VOLUME *)GetNextHob (EFI_HOB_TYPE_FV, (VOID *)((UINTN)FvHob + FvHob->Header.HobLength)); \r
    }\r
  }\r
\r
  if (Instance >= Private->FvCount) {\r
    return NULL;\r
  }\r
\r
  return &Private->Fv[Instance];\r
}  \r