MdeModulePkg/Core/Pei/FwVol/FwVol.c

   1 /** @file
   2   Pei Core Firmware File System service routines.
   3
   4 Copyright (c) 2006 - 2010, Intel Corporation
   5 All rights reserved. This program and the accompanying materials
   6 are licensed and made available under the terms and conditions of the BSD License
   7 which accompanies this distribution.  The full text of the license may be found at
   8 http://opensource.org/licenses/bsd-license.php
   9
  10 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
  11 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
  12
  13 **/
  14
  15 #include "FwVol.h"
  16
  17 EFI_PEI_NOTIFY_DESCRIPTOR mNotifyOnFvInfoList = {
  18   (EFI_PEI_PPI_DESCRIPTOR_NOTIFY_CALLBACK | EFI_PEI_PPI_DESCRIPTOR_TERMINATE_LIST),
  19   &gEfiPeiFirmwareVolumeInfoPpiGuid,
  20   FirmwareVolmeInfoPpiNotifyCallback
  21 };
  22
  23 EFI_PEI_FIRMWARE_VOLUME_PPI mPeiFfs2FvPpi = {
  24   PeiFfs2FvPpiProcessVolume,
  25   PeiFfs2FvPpiFindFileByType,
  26   PeiFfs2FvPpiFindFileByName,
  27   PeiFfs2FvPpiGetFileInfo,
  28   PeiFfs2FvPpiGetVolumeInfo,
  29   PeiFfs2FvPpiFindSectionByType
  30 };
  31
  32 EFI_PEI_PPI_DESCRIPTOR  mPeiFfs2FvPpiList = {
  33   (EFI_PEI_PPI_DESCRIPTOR_PPI | EFI_PEI_PPI_DESCRIPTOR_TERMINATE_LIST),
  34   &gEfiFirmwareFileSystem2Guid,
  35   &mPeiFfs2FvPpi
  36 };
  37
  38 /**
  39   Returns the file state set by the highest zero bit in the State field
  40
  41   @param ErasePolarity   Erase Polarity  as defined by EFI_FVB2_ERASE_POLARITY
  42                          in the Attributes field.
  43   @param FfsHeader       Pointer to FFS File Header.
  44
  45   @retval EFI_FFS_FILE_STATE File state is set by the highest none zero bit
  46                              in the header State field.
  47 **/
  48 EFI_FFS_FILE_STATE
  49 GetFileState(
  50   IN UINT8                ErasePolarity,
  51   IN EFI_FFS_FILE_HEADER  *FfsHeader
  52   )
  53 {
  54   EFI_FFS_FILE_STATE  FileState;
  55   EFI_FFS_FILE_STATE  HighestBit;
  56
  57   FileState = FfsHeader->State;
  58
  59   if (ErasePolarity != 0) {
  60     FileState = (EFI_FFS_FILE_STATE)~FileState;
  61   }
  62
  63   //
  64   // Get file state set by its highest none zero bit.
  65   //
  66   HighestBit = 0x80;
  67   while (HighestBit != 0 && (HighestBit & FileState) == 0) {
  68     HighestBit >>= 1;
  69   }
  70
  71   return HighestBit;
  72 }
  73
  74 /**
  75   Calculates the checksum of the header of a file.
  76
  77   @param FileHeader      Pointer to FFS File Header.
  78
  79   @return Checksum of the header.
  80           Zero means the header is good.
  81           Non-zero means the header is bad.
  82 **/
  83 UINT8
  84 CalculateHeaderChecksum (
  85   IN EFI_FFS_FILE_HEADER  *FileHeader
  86   )
  87 {
  88   EFI_FFS_FILE_HEADER TestFileHeader;
  89
  90   CopyMem (&TestFileHeader, FileHeader, sizeof (EFI_FFS_FILE_HEADER));
  91   //
  92   // Ingore State and File field in FFS header.
  93   //
  94   TestFileHeader.State = 0;
  95   TestFileHeader.IntegrityCheck.Checksum.File = 0;
  96
  97   return CalculateSum8 ((CONST UINT8 *) &TestFileHeader, sizeof (EFI_FFS_FILE_HEADER));
  98 }
  99
 100 /**
 101   Find FV handler according to FileHandle in that FV.
 102
 103   @param FileHandle      Handle of file image
 104
 105   @return Pointer to instance of PEI_CORE_FV_HANDLE.
 106 **/
 107 PEI_CORE_FV_HANDLE*
 108 FileHandleToVolume (
 109   IN   EFI_PEI_FILE_HANDLE          FileHandle
 110   )
 111 {
 112   UINTN                       Index;
 113   PEI_CORE_INSTANCE           *PrivateData;
 114   EFI_FIRMWARE_VOLUME_HEADER  *FwVolHeader;
 115
 116   PrivateData = PEI_CORE_INSTANCE_FROM_PS_THIS (GetPeiServicesTablePointer ());
 117
 118   for (Index = 0; Index < PrivateData->FvCount; Index++) {
 119     FwVolHeader = PrivateData->Fv[Index].FvHeader;
 120     if (((UINT64) (UINTN) FileHandle > (UINT64) (UINTN) FwVolHeader ) &&   \
 121         ((UINT64) (UINTN) FileHandle <= ((UINT64) (UINTN) FwVolHeader + FwVolHeader->FvLength - 1))) {
 122       return &PrivateData->Fv[Index];
 123     }
 124   }
 125   return NULL;
 126 }
 127
 128 /**
 129   Given the input file pointer, search for the first matching file in the
 130   FFS volume as defined by SearchType. The search starts from FileHeader inside
 131   the Firmware Volume defined by FwVolHeader.
 132   If SearchType is EFI_FV_FILETYPE_ALL, the first FFS file will return without check its file type.
 133   If SearchType is PEI_CORE_INTERNAL_FFS_FILE_DISPATCH_TYPE,
 134   the first PEIM, or COMBINED PEIM or FV file type FFS file will return.
 135
 136   @param FvHandle        Pointer to the FV header of the volume to search
 137   @param FileName        File name
 138   @param SearchType      Filter to find only files of this type.
 139                          Type EFI_FV_FILETYPE_ALL causes no filtering to be done.
 140   @param FileHandle      This parameter must point to a valid FFS volume.
 141   @param AprioriFile     Pointer to AprioriFile image in this FV if has
 142
 143   @return EFI_NOT_FOUND  No files matching the search criteria were found
 144   @retval EFI_SUCCESS    Success to search given file
 145
 146 **/
 147 EFI_STATUS
 148 FindFileEx (
 149   IN  CONST EFI_PEI_FV_HANDLE        FvHandle,
 150   IN  CONST EFI_GUID                 *FileName,   OPTIONAL
 151   IN        EFI_FV_FILETYPE          SearchType,
 152   IN OUT    EFI_PEI_FILE_HANDLE      *FileHandle,
 153   IN OUT    EFI_PEI_FV_HANDLE        *AprioriFile  OPTIONAL
 154   )
 155 {
 156   EFI_FIRMWARE_VOLUME_HEADER           *FwVolHeader;
 157   EFI_FFS_FILE_HEADER                   **FileHeader;
 158   EFI_FFS_FILE_HEADER                   *FfsFileHeader;
 159   UINT32                                FileLength;
 160   UINT32                                FileOccupiedSize;
 161   UINT32                                FileOffset;
 162   UINT64                                FvLength;
 163   UINT8                                 ErasePolarity;
 164   UINT8                                 FileState;
 165
 166   //
 167   // Convert the handle of FV to FV header for memory-mapped firmware volume
 168   //
 169   FwVolHeader = (EFI_FIRMWARE_VOLUME_HEADER *) FvHandle;
 170   FileHeader  = (EFI_FFS_FILE_HEADER **)FileHandle;
 171
 172   FvLength = FwVolHeader->FvLength;
 173   if ((FwVolHeader->Attributes & EFI_FVB2_ERASE_POLARITY) != 0) {
 174     ErasePolarity = 1;
 175   } else {
 176     ErasePolarity = 0;
 177   }
 178
 179   //
 180   // If FileHeader is not specified (NULL) or FileName is not NULL,
 181   // start with the first file in the firmware volume.  Otherwise,
 182   // start from the FileHeader.
 183   //
 184   if ((*FileHeader == NULL) || (FileName != NULL)) {
 185     FfsFileHeader = (EFI_FFS_FILE_HEADER *)((UINT8 *)FwVolHeader + FwVolHeader->HeaderLength);
 186   } else {
 187     //
 188     // Length is 24 bits wide so mask upper 8 bits
 189     // FileLength is adjusted to FileOccupiedSize as it is 8 byte aligned.
 190     //
 191     FileLength = *(UINT32 *)(*FileHeader)->Size & 0x00FFFFFF;
 192     FileOccupiedSize = GET_OCCUPIED_SIZE (FileLength, 8);
 193     FfsFileHeader = (EFI_FFS_FILE_HEADER *)((UINT8 *)*FileHeader + FileOccupiedSize);
 194   }
 195
 196   FileOffset = (UINT32) ((UINT8 *)FfsFileHeader - (UINT8 *)FwVolHeader);
 197   ASSERT (FileOffset <= 0xFFFFFFFF);
 198
 199   while (FileOffset < (FvLength - sizeof (EFI_FFS_FILE_HEADER))) {
 200     //
 201     // Get FileState which is the highest bit of the State
 202     //
 203     FileState = GetFileState (ErasePolarity, FfsFileHeader);
 204     switch (FileState) {
 205
 206     case EFI_FILE_HEADER_INVALID:
 207       FileOffset    += sizeof(EFI_FFS_FILE_HEADER);
 208       FfsFileHeader =  (EFI_FFS_FILE_HEADER *)((UINT8 *)FfsFileHeader + sizeof(EFI_FFS_FILE_HEADER));
 209       break;
 210
 211     case EFI_FILE_DATA_VALID:
 212     case EFI_FILE_MARKED_FOR_UPDATE:
 213       if (CalculateHeaderChecksum (FfsFileHeader) != 0) {
 214         ASSERT (FALSE);
 215         *FileHeader = NULL;
 216         return EFI_NOT_FOUND;
 217       }
 218
 219       FileLength       = *(UINT32 *)(FfsFileHeader->Size) & 0x00FFFFFF;
 220       FileOccupiedSize = GET_OCCUPIED_SIZE(FileLength, 8);
 221
 222       if (FileName != NULL) {
 223         if (CompareGuid (&FfsFileHeader->Name, (EFI_GUID*)FileName)) {
 224           *FileHeader = FfsFileHeader;
 225           return EFI_SUCCESS;
 226         }
 227       } else if (SearchType == PEI_CORE_INTERNAL_FFS_FILE_DISPATCH_TYPE) {
 228         if ((FfsFileHeader->Type == EFI_FV_FILETYPE_PEIM) ||
 229             (FfsFileHeader->Type == EFI_FV_FILETYPE_COMBINED_PEIM_DRIVER) ||
 230             (FfsFileHeader->Type == EFI_FV_FILETYPE_FIRMWARE_VOLUME_IMAGE)) {
 231
 232           *FileHeader = FfsFileHeader;
 233           return EFI_SUCCESS;
 234         } else if (AprioriFile != NULL) {
 235           if (FfsFileHeader->Type == EFI_FV_FILETYPE_FREEFORM) {
 236             if (CompareGuid (&FfsFileHeader->Name, &gPeiAprioriFileNameGuid)) {
 237               *AprioriFile = FfsFileHeader;
 238             }
 239           }
 240         }
 241       } else if (((SearchType == FfsFileHeader->Type) || (SearchType == EFI_FV_FILETYPE_ALL)) &&
 242                  (FfsFileHeader->Type != EFI_FV_FILETYPE_FFS_PAD)) {
 243         *FileHeader = FfsFileHeader;
 244         return EFI_SUCCESS;
 245       }
 246
 247       FileOffset    += FileOccupiedSize;
 248       FfsFileHeader =  (EFI_FFS_FILE_HEADER *)((UINT8 *)FfsFileHeader + FileOccupiedSize);
 249       break;
 250
 251     case EFI_FILE_DELETED:
 252       FileLength       =  *(UINT32 *)(FfsFileHeader->Size) & 0x00FFFFFF;
 253       FileOccupiedSize =  GET_OCCUPIED_SIZE(FileLength, 8);
 254       FileOffset       += FileOccupiedSize;
 255       FfsFileHeader    =  (EFI_FFS_FILE_HEADER *)((UINT8 *)FfsFileHeader + FileOccupiedSize);
 256       break;
 257
 258     default:
 259       *FileHeader = NULL;
 260       return EFI_NOT_FOUND;
 261     }
 262   }
 263
 264   *FileHeader = NULL;
 265   return EFI_NOT_FOUND;
 266 }
 267
 268 /**
 269   Initialize PeiCore Fv List.
 270
 271   @param PrivateData     - Pointer to PEI_CORE_INSTANCE.
 272   @param SecCoreData     - Pointer to EFI_SEC_PEI_HAND_OFF.
 273 **/
 274 VOID
 275 PeiInitializeFv (
 276   IN  PEI_CORE_INSTANCE           *PrivateData,
 277   IN CONST EFI_SEC_PEI_HAND_OFF   *SecCoreData
 278   )
 279 {
 280   EFI_STATUS                    Status;
 281   EFI_PEI_FIRMWARE_VOLUME_PPI   *FvPpi;
 282   EFI_PEI_FV_HANDLE             FvHandle;
 283   EFI_FIRMWARE_VOLUME_HEADER    *BfvHeader;
 284
 285   //
 286   // Install FV_PPI for FFS2 file system.
 287   //
 288   PeiServicesInstallPpi (&mPeiFfs2FvPpiList);
 289
 290   BfvHeader = (EFI_FIRMWARE_VOLUME_HEADER *)SecCoreData->BootFirmwareVolumeBase;
 291
 292   //
 293   // The FV_PPI in BFV's format should be installed.
 294   //
 295   Status = PeiServicesLocatePpi (
 296              &BfvHeader->FileSystemGuid,
 297              0,
 298              NULL,
 299              (VOID**)&FvPpi
 300              );
 301   ASSERT_EFI_ERROR (Status);
 302
 303   //
 304   // Get handle of BFV
 305   //
 306   FvPpi->ProcessVolume (
 307            FvPpi,
 308            SecCoreData->BootFirmwareVolumeBase,
 309            (UINTN)BfvHeader->FvLength,
 310            &FvHandle
 311            );
 312
 313   //
 314   // Update internal PEI_CORE_FV array.
 315   //
 316   PrivateData->Fv[PrivateData->FvCount].FvHeader = BfvHeader;
 317   PrivateData->Fv[PrivateData->FvCount].FvPpi    = FvPpi;
 318   PrivateData->Fv[PrivateData->FvCount].FvHandle = FvHandle;
 319   DEBUG ((
 320     EFI_D_INFO,
 321     "The %dth FV start address is 0x%11p, size is 0x%08x, handle is 0x%p\n",
 322     (UINT32) PrivateData->FvCount,
 323     (VOID *) BfvHeader,
 324     BfvHeader->FvLength,
 325     FvHandle
 326     ));
 327   PrivateData->FvCount ++;
 328
 329   //
 330   // Post a call-back for the FvInfoPPI services to expose
 331   // additional Fvs to PeiCore.
 332   //
 333   Status = PeiServicesNotifyPpi (&mNotifyOnFvInfoList);
 334   ASSERT_EFI_ERROR (Status);
 335
 336 }
 337
 338 /**
 339   Process Firmware Volum Information once FvInfoPPI install.
 340   The FV Info will be registered into PeiCore private data structure.
 341   And search the inside FV image, if found, the new FV INFO PPI will be installed.
 342
 343   @param PeiServices       An indirect pointer to the EFI_PEI_SERVICES table published by the PEI Foundation
 344   @param NotifyDescriptor  Address of the notification descriptor data structure.
 345   @param Ppi               Address of the PPI that was installed.
 346
 347   @retval EFI_SUCCESS    The FV Info is registered into PeiCore private data structure.
 348   @return if not EFI_SUCESS, fail to verify FV.
 349
 350 **/
 351 EFI_STATUS
 352 EFIAPI
 353 FirmwareVolmeInfoPpiNotifyCallback (
 354   IN EFI_PEI_SERVICES              **PeiServices,
 355   IN EFI_PEI_NOTIFY_DESCRIPTOR     *NotifyDescriptor,
 356   IN VOID                          *Ppi
 357   )
 358 {
 359   EFI_PEI_FIRMWARE_VOLUME_INFO_PPI      *FvInfoPpi;
 360   EFI_PEI_FIRMWARE_VOLUME_PPI           *FvPpi;
 361   PEI_CORE_INSTANCE                     *PrivateData;
 362   EFI_STATUS                            Status;
 363   EFI_PEI_FV_HANDLE                     FvHandle;
 364   UINTN                                 FvIndex;
 365   EFI_PEI_FILE_HANDLE                   FileHandle;
 366   VOID                                  *DepexData;
 367
 368   Status       = EFI_SUCCESS;
 369   PrivateData  = PEI_CORE_INSTANCE_FROM_PS_THIS (PeiServices);
 370
 371   if (PrivateData->FvCount >= FixedPcdGet32 (PcdPeiCoreMaxFvSupported)) {
 372     DEBUG ((EFI_D_ERROR, "The number of Fv Images (%d) exceed the max supported FVs (%d) in Pei", PrivateData->FvCount + 1, FixedPcdGet32 (PcdPeiCoreMaxFvSupported)));
 373     DEBUG ((EFI_D_ERROR, "PcdPeiCoreMaxFvSupported value need be reconfigurated in DSC"));
 374     ASSERT (FALSE);
 375   }
 376
 377   FvInfoPpi = (EFI_PEI_FIRMWARE_VOLUME_INFO_PPI *)Ppi;
 378
 379   //
 380   // Locate the corresponding FV_PPI according to founded FV's format guid
 381   //
 382   Status = PeiServicesLocatePpi (
 383              &FvInfoPpi->FvFormat,
 384              0,
 385              NULL,
 386              (VOID**)&FvPpi
 387              );
 388   if (!EFI_ERROR (Status)) {
 389     //
 390     // Process new found FV and get FV handle.
 391     //
 392     Status = FvPpi->ProcessVolume (FvPpi, FvInfoPpi->FvInfo, FvInfoPpi->FvInfoSize, &FvHandle);
 393     if (EFI_ERROR (Status)) {
 394       DEBUG ((EFI_D_ERROR, "Fail to process new found FV, FV may be corrupted!\n"));
 395       return Status;
 396     }
 397
 398     //
 399     // Check whether the FV has already been processed.
 400     //
 401     for (FvIndex = 0; FvIndex < PrivateData->FvCount; FvIndex ++) {
 402       if (PrivateData->Fv[FvIndex].FvHandle == FvHandle) {
 403         DEBUG ((EFI_D_INFO, "The Fv %p has already been processed!\n", FvInfoPpi->FvInfo));
 404         return EFI_SUCCESS;
 405       }
 406     }
 407
 408     //
 409     // Update internal PEI_CORE_FV array.
 410     //
 411     PrivateData->Fv[PrivateData->FvCount].FvHeader = (EFI_FIRMWARE_VOLUME_HEADER*) FvInfoPpi->FvInfo;
 412     PrivateData->Fv[PrivateData->FvCount].FvPpi    = FvPpi;
 413     PrivateData->Fv[PrivateData->FvCount].FvHandle = FvHandle;
 414     DEBUG ((
 415       EFI_D_INFO,
 416       "The %dth FV start address is 0x%11p, size is 0x%08x, handle is 0x%p\n",
 417       (UINT32) PrivateData->FvCount,
 418       (VOID *) FvInfoPpi->FvInfo,
 419       FvInfoPpi->FvInfoSize,
 420       FvHandle
 421       ));
 422     PrivateData->FvCount ++;
 423
 424     //
 425     // Scan and process the new discoveried FV for EFI_FV_FILETYPE_FIRMWARE_VOLUME_IMAGE
 426     //
 427     FileHandle = NULL;
 428     do {
 429       Status = FvPpi->FindFileByType (
 430                         FvPpi,
 431                         EFI_FV_FILETYPE_FIRMWARE_VOLUME_IMAGE,
 432                         FvHandle,
 433                         &FileHandle
 434                        );
 435       if (!EFI_ERROR (Status)) {
 436         Status = FvPpi->FindSectionByType (
 437                           FvPpi,
 438                           EFI_SECTION_PEI_DEPEX,
 439                           FileHandle,
 440                           (VOID**)&DepexData
 441                           );
 442         if (!EFI_ERROR (Status)) {
 443           if (!PeimDispatchReadiness (PeiServices, DepexData)) {
 444             //
 445             // Dependency is not satisfied.
 446             //
 447             continue;
 448           }
 449         }
 450
 451         DEBUG ((EFI_D_INFO, "Found firmware volume Image File %p in FV[%d] %p\n", FileHandle, PrivateData->FvCount - 1, FvHandle));
 452         ProcessFvFile (&PrivateData->Fv[PrivateData->FvCount - 1], FileHandle);
 453       }
 454     } while (FileHandle != NULL);
 455   } else {
 456     DEBUG ((EFI_D_ERROR, "Fail to process FV %p because no corresponding EFI_FIRMWARE_VOLUME_PPI is found!\n", FvInfoPpi->FvInfo));
 457
 458     AddUnknownFormatFvInfo (PrivateData, &FvInfoPpi->FvFormat, FvInfoPpi->FvInfo, FvInfoPpi->FvInfoSize);
 459   }
 460
 461   return EFI_SUCCESS;
 462 }
 463
 464 /**
 465   Go through the file to search SectionType section.
 466   Search within encapsulation sections (compression and GUIDed) recursively,
 467   until the match section is found.
 468
 469   @param PeiServices       An indirect pointer to the EFI_PEI_SERVICES table published by the PEI Foundation.
 470   @param SectionType       Filter to find only section of this type.
 471   @param Section           From where to search.
 472   @param SectionSize       The file size to search.
 473   @param OutputBuffer      A pointer to the discovered section, if successful.
 474                            NULL if section not found
 475
 476   @return EFI_NOT_FOUND    The match section is not found.
 477   @return EFI_SUCCESS      The match section is found.
 478
 479 **/
 480 EFI_STATUS
 481 ProcessSection (
 482   IN CONST EFI_PEI_SERVICES     **PeiServices,
 483   IN EFI_SECTION_TYPE           SectionType,
 484   IN EFI_COMMON_SECTION_HEADER  *Section,
 485   IN UINTN                      SectionSize,
 486   OUT VOID                      **OutputBuffer
 487   )
 488 {
 489   EFI_STATUS                              Status;
 490   UINT32                                  SectionLength;
 491   UINT32                                  ParsedLength;
 492   EFI_PEI_GUIDED_SECTION_EXTRACTION_PPI   *GuidSectionPpi;
 493   EFI_PEI_DECOMPRESS_PPI                  *DecompressPpi;
 494   VOID                                    *PpiOutput;
 495   UINTN                                   PpiOutputSize;
 496   UINTN                                   Index;
 497   UINT32                                  Authentication;
 498   PEI_CORE_INSTANCE                       *PrivateData;
 499
 500   PrivateData   = PEI_CORE_INSTANCE_FROM_PS_THIS (PeiServices);
 501   *OutputBuffer = NULL;
 502   ParsedLength  = 0;
 503   Index         = 0;
 504   Status        = EFI_NOT_FOUND;
 505   PpiOutput     = NULL;
 506   PpiOutputSize = 0;
 507   while (ParsedLength < SectionSize) {
 508     if (Section->Type == SectionType) {
 509       *OutputBuffer = (VOID *)(Section + 1);
 510       return EFI_SUCCESS;
 511     } else if ((Section->Type == EFI_SECTION_GUID_DEFINED) || (Section->Type == EFI_SECTION_COMPRESSION)) {
 512       //
 513       // Check the encapsulated section is extracted into the cache data.
 514       //
 515       for (Index = 0; Index < PrivateData->CacheSection.AllSectionCount; Index ++) {
 516         if (Section == PrivateData->CacheSection.Section[Index]) {
 517           PpiOutput     = PrivateData->CacheSection.SectionData[Index];
 518           PpiOutputSize = PrivateData->CacheSection.SectionSize[Index];
 519           //
 520           // Search section directly from the cache data.
 521           //
 522           return ProcessSection (
 523                    PeiServices,
 524                    SectionType,
 525                    PpiOutput,
 526                    PpiOutputSize,
 527                    OutputBuffer
 528                    );
 529         }
 530       }
 531
 532       Status = EFI_NOT_FOUND;
 533       if (Section->Type == EFI_SECTION_GUID_DEFINED) {
 534         Status = PeiServicesLocatePpi (
 535                    &((EFI_GUID_DEFINED_SECTION *)Section)->SectionDefinitionGuid,
 536                    0,
 537                    NULL,
 538                    (VOID **) &GuidSectionPpi
 539                    );
 540         if (!EFI_ERROR (Status)) {
 541           Status = GuidSectionPpi->ExtractSection (
 542                                      GuidSectionPpi,
 543                                      Section,
 544                                      &PpiOutput,
 545                                      &PpiOutputSize,
 546                                      &Authentication
 547                                      );
 548         }
 549       } else if (Section->Type == EFI_SECTION_COMPRESSION) {
 550         Status = PeiServicesLocatePpi (&gEfiPeiDecompressPpiGuid, 0, NULL, (VOID **) &DecompressPpi);
 551         if (!EFI_ERROR (Status)) {
 552           Status = DecompressPpi->Decompress (
 553                                     DecompressPpi,
 554                                     (CONST EFI_COMPRESSION_SECTION*) Section,
 555                                     &PpiOutput,
 556                                     &PpiOutputSize
 557                                     );
 558         }
 559       }
 560
 561       if (!EFI_ERROR (Status)) {
 562         //
 563         // Update cache section data.
 564         //
 565         if (PrivateData->CacheSection.AllSectionCount < CACHE_SETION_MAX_NUMBER) {
 566           PrivateData->CacheSection.AllSectionCount ++;
 567         }
 568         PrivateData->CacheSection.Section [PrivateData->CacheSection.SectionIndex]     = Section;
 569         PrivateData->CacheSection.SectionData [PrivateData->CacheSection.SectionIndex] = PpiOutput;
 570         PrivateData->CacheSection.SectionSize [PrivateData->CacheSection.SectionIndex] = PpiOutputSize;
 571         PrivateData->CacheSection.SectionIndex = (PrivateData->CacheSection.SectionIndex + 1)%CACHE_SETION_MAX_NUMBER;
 572
 573         return ProcessSection (
 574                  PeiServices,
 575                  SectionType,
 576                  PpiOutput,
 577                  PpiOutputSize,
 578                  OutputBuffer
 579                  );
 580       }
 581     }
 582
 583     //
 584     // Size is 24 bits wide so mask upper 8 bits.
 585     // SectionLength is adjusted it is 4 byte aligned.
 586     // Go to the next section
 587     //
 588     SectionLength = *(UINT32 *)Section->Size & 0x00FFFFFF;
 589     SectionLength = GET_OCCUPIED_SIZE (SectionLength, 4);
 590     ASSERT (SectionLength != 0);
 591     ParsedLength += SectionLength;
 592     Section = (EFI_COMMON_SECTION_HEADER *)((UINT8 *)Section + SectionLength);
 593   }
 594
 595   return EFI_NOT_FOUND;
 596 }
 597
 598
 599 /**
 600   Searches for the next matching section within the specified file.
 601
 602   @param PeiServices     An indirect pointer to the EFI_PEI_SERVICES table published by the PEI Foundation
 603   @param SectionType     Filter to find only sections of this type.
 604   @param FileHandle      Pointer to the current file to search.
 605   @param SectionData     A pointer to the discovered section, if successful.
 606                          NULL if section not found
 607
 608   @retval EFI_NOT_FOUND  The section was not found.
 609   @retval EFI_SUCCESS    The section was found.
 610
 611 **/
 612 EFI_STATUS
 613 EFIAPI
 614 PeiFfsFindSectionData (
 615   IN CONST EFI_PEI_SERVICES    **PeiServices,
 616   IN     EFI_SECTION_TYPE      SectionType,
 617   IN     EFI_PEI_FILE_HANDLE   FileHandle,
 618   OUT VOID                     **SectionData
 619   )
 620 {
 621   PEI_CORE_FV_HANDLE           *CoreFvHandle;
 622
 623   CoreFvHandle = FileHandleToVolume (FileHandle);
 624   if ((CoreFvHandle == NULL) || (CoreFvHandle->FvPpi == NULL)) {
 625     return EFI_NOT_FOUND;
 626   }
 627
 628   return CoreFvHandle->FvPpi->FindSectionByType (CoreFvHandle->FvPpi, SectionType, FileHandle, SectionData);
 629 }
 630
 631 /**
 632   Searches for the next matching file in the firmware volume.
 633
 634   @param PeiServices     An indirect pointer to the EFI_PEI_SERVICES table published by the PEI Foundation.
 635   @param SearchType      Filter to find only files of this type.
 636                          Type EFI_FV_FILETYPE_ALL causes no filtering to be done.
 637   @param FvHandle        Handle of firmware volume in which to search.
 638   @param FileHandle      On entry, points to the current handle from which to begin searching or NULL to start
 639                          at the beginning of the firmware volume. On exit, points the file handle of the next file
 640                          in the volume or NULL if there are no more files.
 641
 642   @retval EFI_NOT_FOUND  The file was not found.
 643   @retval EFI_NOT_FOUND  The header checksum was not zero.
 644   @retval EFI_SUCCESS    The file was found.
 645
 646 **/
 647 EFI_STATUS
 648 EFIAPI
 649 PeiFfsFindNextFile (
 650   IN CONST EFI_PEI_SERVICES      **PeiServices,
 651   IN UINT8                       SearchType,
 652   IN EFI_PEI_FV_HANDLE           FvHandle,
 653   IN OUT EFI_PEI_FILE_HANDLE     *FileHandle
 654   )
 655 {
 656   PEI_CORE_FV_HANDLE      *CoreFvHandle;
 657
 658   CoreFvHandle = FvHandleToCoreHandle (FvHandle);
 659
 660   //
 661   // To make backward compatiblity, if can not find corresponding the handle of FV
 662   // then treat FV as build-in FFS2 format and memory mapped FV that FV handle is pointed
 663   // to the address of first byte of FV.
 664   //
 665   if ((CoreFvHandle == NULL) && FeaturePcdGet (PcdFrameworkCompatibilitySupport)) {
 666     return FindFileEx (FvHandle, NULL, SearchType, FileHandle, NULL);
 667   }
 668
 669   if ((CoreFvHandle == NULL) || CoreFvHandle->FvPpi == NULL) {
 670     return EFI_NOT_FOUND;
 671   }
 672
 673   return CoreFvHandle->FvPpi->FindFileByType (CoreFvHandle->FvPpi, SearchType, FvHandle, FileHandle);
 674 }
 675
 676
 677 /**
 678   Search the firmware volumes by index
 679
 680   @param PeiServices     An indirect pointer to the EFI_PEI_SERVICES table published by the PEI Foundation
 681   @param Instance        This instance of the firmware volume to find. The value 0 is the Boot Firmware
 682                          Volume (BFV).
 683   @param VolumeHandle    On exit, points to the next volume handle or NULL if it does not exist.
 684
 685   @retval EFI_INVALID_PARAMETER  VolumeHandle is NULL
 686   @retval EFI_NOT_FOUND          The volume was not found.
 687   @retval EFI_SUCCESS            The volume was found.
 688
 689 **/
 690 EFI_STATUS
 691 EFIAPI
 692 PeiFfsFindNextVolume (
 693   IN CONST EFI_PEI_SERVICES         **PeiServices,
 694   IN     UINTN                      Instance,
 695   IN OUT EFI_PEI_FV_HANDLE          *VolumeHandle
 696   )
 697 {
 698   PEI_CORE_INSTANCE  *Private;
 699   PEI_CORE_FV_HANDLE *CoreFvHandle;
 700
 701   if (VolumeHandle == NULL) {
 702     return EFI_INVALID_PARAMETER;
 703   }
 704
 705   Private = PEI_CORE_INSTANCE_FROM_PS_THIS (PeiServices);
 706
 707   CoreFvHandle = FindNextCoreFvHandle (Private, Instance);
 708   if (CoreFvHandle == NULL) {
 709     *VolumeHandle = NULL;
 710     return EFI_NOT_FOUND;
 711   }
 712
 713   *VolumeHandle = CoreFvHandle->FvHandle;
 714
 715   return EFI_SUCCESS;
 716 }
 717
 718
 719 /**
 720   Find a file within a volume by its name.
 721
 722   @param FileName        A pointer to the name of the file to find within the firmware volume.
 723   @param VolumeHandle    The firmware volume to search
 724   @param FileHandle      Upon exit, points to the found file's handle
 725                          or NULL if it could not be found.
 726
 727   @retval EFI_SUCCESS            File was found.
 728   @retval EFI_NOT_FOUND          File was not found.
 729   @retval EFI_INVALID_PARAMETER  VolumeHandle or FileHandle or FileName was NULL.
 730
 731 **/
 732 EFI_STATUS
 733 EFIAPI
 734 PeiFfsFindFileByName (
 735   IN  CONST EFI_GUID        *FileName,
 736   IN  EFI_PEI_FV_HANDLE     VolumeHandle,
 737   OUT EFI_PEI_FILE_HANDLE   *FileHandle
 738   )
 739 {
 740   PEI_CORE_FV_HANDLE            *CoreFvHandle;
 741
 742   if ((VolumeHandle == NULL) || (FileName == NULL) || (FileHandle == NULL)) {
 743     return EFI_INVALID_PARAMETER;
 744   }
 745
 746   CoreFvHandle = FvHandleToCoreHandle (VolumeHandle);
 747   if ((CoreFvHandle == NULL) || (CoreFvHandle->FvPpi == NULL)) {
 748     return EFI_NOT_FOUND;
 749   }
 750
 751   return CoreFvHandle->FvPpi->FindFileByName (CoreFvHandle->FvPpi, FileName, &VolumeHandle, FileHandle);
 752 }
 753
 754 /**
 755   Returns information about a specific file.
 756
 757   @param FileHandle       Handle of the file.
 758   @param FileInfo         Upon exit, points to the file’s information.
 759
 760   @retval EFI_INVALID_PARAMETER If FileInfo is NULL.
 761   @retval EFI_INVALID_PARAMETER If FileHandle does not represent a valid file.
 762   @retval EFI_SUCCESS           File information returned.
 763
 764 **/
 765 EFI_STATUS
 766 EFIAPI
 767 PeiFfsGetFileInfo (
 768   IN EFI_PEI_FILE_HANDLE  FileHandle,
 769   OUT EFI_FV_FILE_INFO    *FileInfo
 770   )
 771 {
 772   PEI_CORE_FV_HANDLE          *CoreFvHandle;
 773
 774   if ((FileHandle == NULL) || (FileInfo == NULL)) {
 775     return EFI_INVALID_PARAMETER;
 776   }
 777
 778   //
 779   // Retrieve the FirmwareVolume which the file resides in.
 780   //
 781   CoreFvHandle = FileHandleToVolume (FileHandle);
 782   if ((CoreFvHandle == NULL) || (CoreFvHandle->FvPpi == NULL)) {
 783     return EFI_INVALID_PARAMETER;
 784   }
 785
 786   return CoreFvHandle->FvPpi->GetFileInfo (CoreFvHandle->FvPpi, FileHandle, FileInfo);
 787 }
 788
 789
 790 /**
 791   Returns information about the specified volume.
 792
 793   This function returns information about a specific firmware
 794   volume, including its name, type, attributes, starting address
 795   and size.
 796
 797   @param VolumeHandle   Handle of the volume.
 798   @param VolumeInfo     Upon exit, points to the volume's information.
 799
 800   @retval EFI_SUCCESS             Volume information returned.
 801   @retval EFI_INVALID_PARAMETER   If VolumeHandle does not represent a valid volume.
 802   @retval EFI_INVALID_PARAMETER   If VolumeHandle is NULL.
 803   @retval EFI_SUCCESS             Information successfully returned.
 804   @retval EFI_INVALID_PARAMETER   The volume designated by the VolumeHandle is not available.
 805
 806 **/
 807 EFI_STATUS
 808 EFIAPI
 809 PeiFfsGetVolumeInfo (
 810   IN EFI_PEI_FV_HANDLE  VolumeHandle,
 811   OUT EFI_FV_INFO       *VolumeInfo
 812   )
 813 {
 814   PEI_CORE_FV_HANDLE                     *CoreHandle;
 815
 816   if ((VolumeInfo == NULL) || (VolumeHandle == NULL)) {
 817     return EFI_INVALID_PARAMETER;
 818   }
 819
 820   CoreHandle = FvHandleToCoreHandle (VolumeHandle);
 821
 822   if ((CoreHandle == NULL) || (CoreHandle->FvPpi == NULL)) {
 823     return EFI_INVALID_PARAMETER;
 824   }
 825
 826   return CoreHandle->FvPpi->GetVolumeInfo (CoreHandle->FvPpi, VolumeHandle, VolumeInfo);
 827 }
 828
 829 /**
 830   Get Fv image from the FV type file, then install FV INFO ppi, Build FV hob.
 831
 832   @param ParentFvCoreHandle   Pointer of EFI_CORE_FV_HANDLE to parent Fv image that contain this Fv image.
 833   @param ParentFvFileHandle   File handle of a Fv type file that contain this Fv image.
 834
 835   @retval EFI_NOT_FOUND         FV image can't be found.
 836   @retval EFI_SUCCESS           Successfully to process it.
 837   @retval EFI_OUT_OF_RESOURCES  Can not allocate page when aligning FV image
 838   @retval Others                Can not find EFI_SECTION_FIRMWARE_VOLUME_IMAGE section
 839
 840 **/
 841 EFI_STATUS
 842 ProcessFvFile (
 843   IN  PEI_CORE_FV_HANDLE          *ParentFvCoreHandle,
 844   IN  EFI_PEI_FILE_HANDLE         ParentFvFileHandle
 845   )
 846 {
 847   EFI_STATUS                    Status;
 848   EFI_FV_INFO                   ParentFvImageInfo;
 849   UINT32                        FvAlignment;
 850   VOID                          *NewFvBuffer;
 851   EFI_PEI_HOB_POINTERS          HobPtr;
 852   EFI_PEI_FIRMWARE_VOLUME_PPI   *ParentFvPpi;
 853   EFI_PEI_FV_HANDLE             ParentFvHandle;
 854   EFI_FIRMWARE_VOLUME_HEADER    *FvHeader;
 855   EFI_FV_FILE_INFO              FileInfo;
 856
 857   //
 858   // Check if this EFI_FV_FILETYPE_FIRMWARE_VOLUME_IMAGE file has already
 859   // been extracted.
 860   //
 861   HobPtr.Raw = GetHobList ();
 862   while ((HobPtr.Raw = GetNextHob (EFI_HOB_TYPE_FV2, HobPtr.Raw)) != NULL) {
 863     if (CompareGuid (&(((EFI_FFS_FILE_HEADER *)ParentFvFileHandle)->Name), &HobPtr.FirmwareVolume2->FileName)) {
 864       //
 865       // this FILE has been dispatched, it will not be dispatched again.
 866       //
 867       DEBUG ((EFI_D_INFO, "FV file %p has been dispatched!\r\n", ParentFvFileHandle));
 868       return EFI_SUCCESS;
 869     }
 870     HobPtr.Raw = GET_NEXT_HOB (HobPtr);
 871   }
 872
 873   ParentFvHandle = ParentFvCoreHandle->FvHandle;
 874   ParentFvPpi    = ParentFvCoreHandle->FvPpi;
 875
 876   //
 877   // Find FvImage in FvFile
 878   //
 879   Status = ParentFvPpi->FindSectionByType (
 880                           ParentFvPpi,
 881                           EFI_SECTION_FIRMWARE_VOLUME_IMAGE,
 882                           ParentFvFileHandle,
 883                           (VOID **)&FvHeader
 884                           );
 885
 886   if (EFI_ERROR (Status)) {
 887     return Status;
 888   }
 889
 890   //
 891   // FvAlignment must be more than 8 bytes required by FvHeader structure.
 892   //
 893   FvAlignment = 1 << ((FvHeader->Attributes & EFI_FVB2_ALIGNMENT) >> 16);
 894   if (FvAlignment < 8) {
 895     FvAlignment = 8;
 896   }
 897
 898   //
 899   // Check FvImage
 900   //
 901   if ((UINTN) FvHeader % FvAlignment != 0) {
 902     NewFvBuffer = AllocateAlignedPages (EFI_SIZE_TO_PAGES ((UINT32) FvHeader->FvLength), FvAlignment);
 903     if (NewFvBuffer == NULL) {
 904       return EFI_OUT_OF_RESOURCES;
 905     }
 906     CopyMem (NewFvBuffer, FvHeader, (UINTN) FvHeader->FvLength);
 907     FvHeader = (EFI_FIRMWARE_VOLUME_HEADER*) NewFvBuffer;
 908   }
 909
 910   Status = ParentFvPpi->GetVolumeInfo (ParentFvPpi, ParentFvHandle, &ParentFvImageInfo);
 911   ASSERT_EFI_ERROR (Status);
 912
 913   Status = ParentFvPpi->GetFileInfo (ParentFvPpi, ParentFvFileHandle, &FileInfo);
 914   ASSERT_EFI_ERROR (Status);
 915
 916   //
 917   // Install FvPpi and Build FvHob
 918   //
 919   PeiServicesInstallFvInfoPpi (
 920     &FvHeader->FileSystemGuid,
 921     (VOID**) FvHeader,
 922     (UINT32) FvHeader->FvLength,
 923     &ParentFvImageInfo.FvName,
 924     &FileInfo.FileName
 925     );
 926
 927   //
 928   // Inform the extracted FvImage to Fv HOB consumer phase, i.e. DXE phase
 929   //
 930   BuildFvHob (
 931     (EFI_PHYSICAL_ADDRESS) (UINTN) FvHeader,
 932     FvHeader->FvLength
 933     );
 934
 935   //
 936   // Makes the encapsulated volume show up in DXE phase to skip processing of
 937   // encapsulated file again.
 938   //
 939   BuildFv2Hob (
 940     (EFI_PHYSICAL_ADDRESS) (UINTN) FvHeader,
 941     FvHeader->FvLength,
 942     &ParentFvImageInfo.FvName,
 943     &FileInfo.FileName
 944     );
 945
 946   return EFI_SUCCESS;
 947 }
 948
 949 /**
 950   Process a firmware volume and create a volume handle.
 951
 952   Create a volume handle from the information in the buffer. For
 953   memory-mapped firmware volumes, Buffer and BufferSize refer to
 954   the start of the firmware volume and the firmware volume size.
 955   For non memory-mapped firmware volumes, this points to a
 956   buffer which contains the necessary information for creating
 957   the firmware volume handle. Normally, these values are derived
 958   from the EFI_FIRMWARE_VOLUME_INFO_PPI.
 959
 960
 961   @param This                   Points to this instance of the
 962                                 EFI_PEI_FIRMWARE_VOLUME_PPI.
 963   @param Buffer                 Points to the start of the buffer.
 964   @param BufferSize             Size of the buffer.
 965   @param FvHandle               Points to the returned firmware volume
 966                                 handle. The firmware volume handle must
 967                                 be unique within the system.
 968
 969   @retval EFI_SUCCESS           Firmware volume handle created.
 970   @retval EFI_VOLUME_CORRUPTED  Volume was corrupt.
 971
 972 **/
 973 EFI_STATUS
 974 EFIAPI
 975 PeiFfs2FvPpiProcessVolume (
 976   IN  CONST  EFI_PEI_FIRMWARE_VOLUME_PPI *This,
 977   IN  VOID                               *Buffer,
 978   IN  UINTN                              BufferSize,
 979   OUT EFI_PEI_FV_HANDLE                  *FvHandle
 980   )
 981 {
 982   EFI_STATUS          Status;
 983
 984   ASSERT (FvHandle != NULL);
 985
 986   if (Buffer == NULL) {
 987     return EFI_VOLUME_CORRUPTED;
 988   }
 989
 990   //
 991   // The build-in EFI_PEI_FIRMWARE_VOLUME_PPI for FFS2 support memory-mapped
 992   // FV image and the handle is pointed to Fv image's buffer.
 993   //
 994   *FvHandle = (EFI_PEI_FV_HANDLE) Buffer;
 995
 996   //
 997   // Do verify for given FV buffer.
 998   //
 999   Status = VerifyFv ((EFI_FIRMWARE_VOLUME_HEADER*) Buffer);
1000   if (EFI_ERROR(Status)) {
1001     DEBUG ((EFI_D_ERROR, "Fail to verify FV which address is 0x%11p", Buffer));
1002     return EFI_VOLUME_CORRUPTED;
1003   }
1004
1005   return EFI_SUCCESS;
1006 }
1007
1008 /**
1009   Finds the next file of the specified type.
1010
1011   This service enables PEI modules to discover additional firmware files.
1012   The FileHandle must be unique within the system.
1013
1014   @param This           Points to this instance of the
1015                         EFI_PEI_FIRMWARE_VOLUME_PPI.
1016   @param SearchType     A filter to find only files of this type. Type
1017                         EFI_FV_FILETYPE_ALL causes no filtering to be
1018                         done.
1019   @param FvHandle       Handle of firmware volume in which to
1020                         search.
1021   @param FileHandle     Points to the current handle from which to
1022                         begin searching or NULL to start at the
1023                         beginning of the firmware volume. Updated
1024                         upon return to reflect the file found.
1025
1026   @retval EFI_SUCCESS   The file was found.
1027   @retval EFI_NOT_FOUND The file was not found. FileHandle contains NULL.
1028
1029 **/
1030 EFI_STATUS
1031 EFIAPI
1032 PeiFfs2FvPpiFindFileByType (
1033   IN CONST  EFI_PEI_FIRMWARE_VOLUME_PPI *This,
1034   IN        EFI_FV_FILETYPE             SearchType,
1035   IN        EFI_PEI_FV_HANDLE           FvHandle,
1036   IN OUT    EFI_PEI_FILE_HANDLE         *FileHandle
1037   )
1038 {
1039   return FindFileEx (FvHandle, NULL, SearchType, FileHandle, NULL);
1040 }
1041
1042 /**
1043   Find a file within a volume by its name.
1044
1045   This service searches for files with a specific name, within
1046   either the specified firmware volume or all firmware volumes.
1047
1048   @param This                   Points to this instance of the
1049                                 EFI_PEI_FIRMWARE_VOLUME_PPI.
1050   @param FileName               A pointer to the name of the file to find
1051                                 within the firmware volume.
1052   @param FvHandle               Upon entry, the pointer to the firmware
1053                                 volume to search or NULL if all firmware
1054                                 volumes should be searched. Upon exit, the
1055                                 actual firmware volume in which the file was
1056                                 found.
1057   @param FileHandle             Upon exit, points to the found file's
1058                                 handle or NULL if it could not be found.
1059
1060   @retval EFI_SUCCESS           File was found.
1061   @retval EFI_NOT_FOUND         File was not found.
1062   @retval EFI_INVALID_PARAMETER FvHandle or FileHandle or
1063                                 FileName was NULL.
1064
1065
1066 **/
1067 EFI_STATUS
1068 EFIAPI
1069 PeiFfs2FvPpiFindFileByName (
1070   IN  CONST  EFI_PEI_FIRMWARE_VOLUME_PPI *This,
1071   IN  CONST  EFI_GUID                    *FileName,
1072   IN  EFI_PEI_FV_HANDLE                  *FvHandle,
1073   OUT EFI_PEI_FILE_HANDLE                *FileHandle
1074   )
1075 {
1076   EFI_STATUS        Status;
1077   PEI_CORE_INSTANCE *PrivateData;
1078   UINTN             Index;
1079
1080   if ((FvHandle == NULL) || (FileName == NULL) || (FileHandle == NULL)) {
1081     return EFI_INVALID_PARAMETER;
1082   }
1083
1084   if (*FvHandle != NULL) {
1085     Status = FindFileEx (*FvHandle, FileName, 0, FileHandle, NULL);
1086     if (Status == EFI_NOT_FOUND) {
1087       *FileHandle = NULL;
1088     }
1089   } else {
1090     //
1091     // If *FvHandle = NULL, so search all FV for given filename
1092     //
1093     Status = EFI_NOT_FOUND;
1094
1095     PrivateData = PEI_CORE_INSTANCE_FROM_PS_THIS (GetPeiServicesTablePointer());
1096     for (Index = 0; Index < PrivateData->FvCount; Index ++) {
1097       //
1098       // Only search the FV which is associated with a EFI_PEI_FIRMWARE_VOLUME_PPI instance.
1099       //
1100       if (PrivateData->Fv[Index].FvPpi != NULL) {
1101         Status = FindFileEx (PrivateData->Fv[Index].FvHandle, FileName, 0, FileHandle, NULL);
1102         if (!EFI_ERROR (Status)) {
1103           *FvHandle = PrivateData->Fv[Index].FvHandle;
1104           break;
1105         }
1106       }
1107     }
1108   }
1109
1110   return Status;
1111 }
1112
1113 /**
1114   Returns information about a specific file.
1115
1116   This function returns information about a specific
1117   file, including its file name, type, attributes, starting
1118   address and size.
1119
1120   @param This                     Points to this instance of the
1121                                   EFI_PEI_FIRMWARE_VOLUME_PPI.
1122   @param FileHandle               Handle of the file.
1123   @param FileInfo                 Upon exit, points to the file's
1124                                   information.
1125
1126   @retval EFI_SUCCESS             File information returned.
1127   @retval EFI_INVALID_PARAMETER   If FileHandle does not
1128                                   represent a valid file.
1129   @retval EFI_INVALID_PARAMETER   If FileInfo is NULL.
1130
1131 **/
1132 EFI_STATUS
1133 EFIAPI
1134 PeiFfs2FvPpiGetFileInfo (
1135   IN  CONST EFI_PEI_FIRMWARE_VOLUME_PPI   *This,
1136   IN        EFI_PEI_FILE_HANDLE           FileHandle,
1137   OUT       EFI_FV_FILE_INFO              *FileInfo
1138   )
1139 {
1140   UINT8                       FileState;
1141   UINT8                       ErasePolarity;
1142   EFI_FFS_FILE_HEADER         *FileHeader;
1143   PEI_CORE_FV_HANDLE          *CoreFvHandle;
1144
1145   if ((FileHandle == NULL) || (FileInfo == NULL)) {
1146     return EFI_INVALID_PARAMETER;
1147   }
1148
1149   //
1150   // Retrieve the FirmwareVolume which the file resides in.
1151   //
1152   CoreFvHandle = FileHandleToVolume (FileHandle);
1153   if (CoreFvHandle == NULL) {
1154     return EFI_INVALID_PARAMETER;
1155   }
1156
1157   if (CoreFvHandle->FvHeader->Attributes & EFI_FVB2_ERASE_POLARITY) {
1158     ErasePolarity = 1;
1159   } else {
1160     ErasePolarity = 0;
1161   }
1162
1163   //
1164   // Get FileState which is the highest bit of the State
1165   //
1166   FileState = GetFileState (ErasePolarity, (EFI_FFS_FILE_HEADER*)FileHandle);
1167
1168   switch (FileState) {
1169     case EFI_FILE_DATA_VALID:
1170     case EFI_FILE_MARKED_FOR_UPDATE:
1171       break;
1172     default:
1173       return EFI_INVALID_PARAMETER;
1174     }
1175
1176   FileHeader = (EFI_FFS_FILE_HEADER *)FileHandle;
1177   CopyMem (&FileInfo->FileName, &FileHeader->Name, sizeof(EFI_GUID));
1178   FileInfo->FileType = FileHeader->Type;
1179   FileInfo->FileAttributes = FileHeader->Attributes;
1180   FileInfo->BufferSize = ((*(UINT32 *)FileHeader->Size) & 0x00FFFFFF) -  sizeof (EFI_FFS_FILE_HEADER);
1181   FileInfo->Buffer = (FileHeader + 1);
1182   return EFI_SUCCESS;
1183 }
1184
1185 /**
1186   This function returns information about the firmware volume.
1187
1188   @param This                     Points to this instance of the
1189                                   EFI_PEI_FIRMWARE_VOLUME_PPI.
1190   @param FvHandle                 Handle to the firmware handle.
1191   @param VolumeInfo               Points to the returned firmware volume
1192                                   information.
1193
1194   @retval EFI_SUCCESS             Information returned successfully.
1195   @retval EFI_INVALID_PARAMETER   FvHandle does not indicate a valid
1196                                   firmware volume or VolumeInfo is NULL.
1197
1198 **/
1199 EFI_STATUS
1200 EFIAPI
1201 PeiFfs2FvPpiGetVolumeInfo (
1202   IN  CONST  EFI_PEI_FIRMWARE_VOLUME_PPI   *This,
1203   IN  EFI_PEI_FV_HANDLE                    FvHandle,
1204   OUT EFI_FV_INFO                          *VolumeInfo
1205   )
1206 {
1207   EFI_FIRMWARE_VOLUME_HEADER             FwVolHeader;
1208   EFI_FIRMWARE_VOLUME_EXT_HEADER         *FwVolExHeaderInfo;
1209
1210   if ((VolumeInfo == NULL) || (FvHandle == NULL)) {
1211     return EFI_INVALID_PARAMETER;
1212   }
1213
1214   //
1215   // VolumeHandle may not align at 8 byte,
1216   // but FvLength is UINT64 type, which requires FvHeader align at least 8 byte.
1217   // So, Copy FvHeader into the local FvHeader structure.
1218   //
1219   CopyMem (&FwVolHeader, FvHandle, sizeof (EFI_FIRMWARE_VOLUME_HEADER));
1220
1221   //
1222   // Check Fv Image Signature
1223   //
1224   if (FwVolHeader.Signature != EFI_FVH_SIGNATURE) {
1225     return EFI_INVALID_PARAMETER;
1226   }
1227
1228   ZeroMem (VolumeInfo, sizeof (EFI_FV_INFO));
1229   VolumeInfo->FvAttributes  = FwVolHeader.Attributes;
1230   VolumeInfo->FvStart       = (VOID *) FvHandle;
1231   VolumeInfo->FvSize        = FwVolHeader.FvLength;
1232   CopyMem (&VolumeInfo->FvFormat, &FwVolHeader.FileSystemGuid, sizeof(EFI_GUID));
1233
1234   if (FwVolHeader.ExtHeaderOffset != 0) {
1235     FwVolExHeaderInfo = (EFI_FIRMWARE_VOLUME_EXT_HEADER*)(((UINT8 *)FvHandle) + FwVolHeader.ExtHeaderOffset);
1236     CopyMem (&VolumeInfo->FvName, &FwVolExHeaderInfo->FvName, sizeof(EFI_GUID));
1237   }
1238
1239   return EFI_SUCCESS;
1240 }
1241
1242 /**
1243   Find the next matching section in the firmware file.
1244
1245   This service enables PEI modules to discover sections
1246   of a given type within a valid file.
1247
1248   @param This             Points to this instance of the
1249                           EFI_PEI_FIRMWARE_VOLUME_PPI.
1250   @param SearchType       A filter to find only sections of this
1251                           type.
1252   @param FileHandle       Handle of firmware file in which to
1253                           search.
1254   @param SectionData      Updated upon  return to point to the
1255                           section found.
1256
1257   @retval EFI_SUCCESS     Section was found.
1258   @retval EFI_NOT_FOUND   Section of the specified type was not
1259                           found. SectionData contains NULL.
1260 **/
1261 EFI_STATUS
1262 EFIAPI
1263 PeiFfs2FvPpiFindSectionByType (
1264   IN  CONST EFI_PEI_FIRMWARE_VOLUME_PPI    *This,
1265   IN        EFI_SECTION_TYPE               SearchType,
1266   IN        EFI_PEI_FILE_HANDLE            FileHandle,
1267   OUT VOID                                 **SectionData
1268   )
1269 {
1270   EFI_FFS_FILE_HEADER                     *FfsFileHeader;
1271   UINT32                                  FileSize;
1272   EFI_COMMON_SECTION_HEADER               *Section;
1273
1274   FfsFileHeader = (EFI_FFS_FILE_HEADER *)(FileHandle);
1275
1276   //
1277   // Size is 24 bits wide so mask upper 8 bits.
1278   // Does not include FfsFileHeader header size
1279   // FileSize is adjusted to FileOccupiedSize as it is 8 byte aligned.
1280   //
1281   Section = (EFI_COMMON_SECTION_HEADER *)(FfsFileHeader + 1);
1282   FileSize = *(UINT32 *)(FfsFileHeader->Size) & 0x00FFFFFF;
1283   FileSize -= sizeof (EFI_FFS_FILE_HEADER);
1284
1285   return ProcessSection (
1286            GetPeiServicesTablePointer (),
1287            SearchType,
1288            Section,
1289            FileSize,
1290            SectionData
1291            );
1292 }
1293
1294 /**
1295   Convert the handle of FV to pointer of corresponding PEI_CORE_FV_HANDLE.
1296
1297   @param FvHandle   The handle of a FV.
1298
1299   @retval NULL if can not find.
1300   @return Pointer of corresponding PEI_CORE_FV_HANDLE.
1301 **/
1302 PEI_CORE_FV_HANDLE *
1303 FvHandleToCoreHandle (
1304   IN EFI_PEI_FV_HANDLE  FvHandle
1305   )
1306 {
1307   UINTN             Index;
1308   PEI_CORE_INSTANCE *PrivateData;
1309
1310   PrivateData = PEI_CORE_INSTANCE_FROM_PS_THIS (GetPeiServicesTablePointer());
1311   for (Index = 0; Index < PrivateData->FvCount; Index ++) {
1312     if (FvHandle == PrivateData->Fv[Index].FvHandle) {
1313       return &PrivateData->Fv[Index];
1314     }
1315   }
1316
1317   return NULL;
1318 }
1319
1320 /**
1321   Get instance of PEI_CORE_FV_HANDLE for next volume according to given index.
1322
1323   This routine also will install FvInfo ppi for FV hob in PI ways.
1324
1325   @param Private    Pointer of PEI_CORE_INSTANCE
1326   @param Instance   The index of FV want to be searched.
1327
1328   @return Instance of PEI_CORE_FV_HANDLE.
1329 **/
1330 PEI_CORE_FV_HANDLE *
1331 FindNextCoreFvHandle (
1332   IN PEI_CORE_INSTANCE  *Private,
1333   IN UINTN              Instance
1334   )
1335 {
1336   UINTN                    Index;
1337   BOOLEAN                  Match;
1338   EFI_HOB_FIRMWARE_VOLUME  *FvHob;
1339
1340   //
1341   // Handle Framework FvHob and Install FvInfo Ppi for it.
1342   //
1343   if (FeaturePcdGet (PcdFrameworkCompatibilitySupport)) {
1344     //
1345     // Loop to search the wanted FirmwareVolume which supports FFS
1346     //
1347     FvHob = (EFI_HOB_FIRMWARE_VOLUME *)GetFirstHob (EFI_HOB_TYPE_FV);
1348     while (FvHob != NULL) {
1349       for (Index = 0, Match = FALSE; Index < Private->FvCount; Index++) {
1350         if ((EFI_PEI_FV_HANDLE)(UINTN)FvHob->BaseAddress == Private->Fv[Index].FvHeader) {
1351           Match = TRUE;
1352           break;
1353         }
1354       }
1355       //
1356       // If Not Found, Install FvInfo Ppi for it.
1357       //
1358       if (!Match) {
1359         PeiServicesInstallFvInfoPpi (
1360           &(((EFI_FIRMWARE_VOLUME_HEADER *)(UINTN)FvHob->BaseAddress)->FileSystemGuid),
1361           (VOID *)(UINTN)FvHob->BaseAddress,
1362           (UINT32)FvHob->Length,
1363           NULL,
1364           NULL
1365           );
1366       }
1367       FvHob = (EFI_HOB_FIRMWARE_VOLUME *)GetNextHob (EFI_HOB_TYPE_FV, (VOID *)((UINTN)FvHob + FvHob->Header.HobLength));
1368     }
1369   }
1370
1371   if (Instance >= Private->FvCount) {
1372     return NULL;
1373   }
1374
1375   return &Private->Fv[Instance];
1376 }
1377
1378 /**
1379   After PeiCore image is shadowed into permanent memory, all build-in FvPpi should
1380   be re-installed with the instance in permanent memory and all cached FvPpi pointers in
1381   PrivateData->Fv[] array should be fixed up to be pointed to the one in permenant
1382   memory.
1383
1384   @param PrivateData   Pointer to PEI_CORE_INSTANCE.
1385 **/
1386 VOID
1387 PeiReinitializeFv (
1388   IN  PEI_CORE_INSTANCE           *PrivateData
1389   )
1390 {
1391   VOID                    *OldFfs2FvPpi;
1392   EFI_PEI_PPI_DESCRIPTOR  *OldDescriptor;
1393   UINTN                   Index;
1394   EFI_STATUS              Status;
1395
1396   //
1397   // Locate old build-in Ffs2 EFI_PEI_FIRMWARE_VOLUME_PPI which
1398   // in flash.
1399   //
1400   Status = PeiServicesLocatePpi (
1401             &gEfiFirmwareFileSystem2Guid,
1402             0,
1403             &OldDescriptor,
1404             &OldFfs2FvPpi
1405             );
1406   ASSERT_EFI_ERROR (Status);
1407
1408   //
1409   // Re-install the EFI_PEI_FIRMWARE_VOLUME_PPI for build-in Ffs2
1410   // which is shadowed from flash to permanent memory within PeiCore image.
1411   //
1412   Status = PeiServicesReInstallPpi (OldDescriptor, &mPeiFfs2FvPpiList);
1413   ASSERT_EFI_ERROR (Status);
1414
1415   //
1416   // Fixup all FvPpi pointers for the implementation in flash to permanent memory.
1417   //
1418   for (Index = 0; Index < FixedPcdGet32 (PcdPeiCoreMaxFvSupported); Index ++) {
1419     if (PrivateData->Fv[Index].FvPpi == OldFfs2FvPpi) {
1420       PrivateData->Fv[Index].FvPpi = &mPeiFfs2FvPpi;
1421     }
1422   }
1423 }
1424
1425 /**
1426   Report the information for a new discoveried FV in unknown third-party format.
1427
1428   If the EFI_PEI_FIRMWARE_VOLUME_PPI has not been installed for third-party FV format, but
1429   the FV in this format has been discoveried, then this FV's information will be cached into
1430   PEI_CORE_INSTANCE's UnknownFvInfo array.
1431   Also a notification would be installed for unknown third-party FV format guid, if EFI_PEI_FIRMWARE_VOLUME_PPI
1432   is installed later by platform's PEIM, the original unknown third-party FV will be processed by
1433   using new installed EFI_PEI_FIRMWARE_VOLUME_PPI.
1434
1435   @param PrivateData  Point to instance of PEI_CORE_INSTANCE
1436   @param Format       Point to the unknown third-party format guid.
1437   @param FvInfo       Point to FvInfo buffer.
1438   @param FvInfoSize   The size of FvInfo buffer.
1439
1440   @retval EFI_OUT_OF_RESOURCES  The FV info array in PEI_CORE_INSTANCE has no more spaces.
1441   @retval EFI_SUCCESS           Success to add the information for unknown FV.
1442 **/
1443 EFI_STATUS
1444 AddUnknownFormatFvInfo (
1445   IN PEI_CORE_INSTANCE *PrivateData,
1446   IN EFI_GUID          *Format,
1447   IN VOID              *FvInfo,
1448   IN UINT32            FvInfoSize
1449   )
1450 {
1451   PEI_CORE_UNKNOW_FORMAT_FV_INFO    *NewUnknownFv;
1452
1453   if (PrivateData->UnknownFvInfoCount + 1 >= FixedPcdGet32 (PcdPeiCoreMaxPeimPerFv)) {
1454     return EFI_OUT_OF_RESOURCES;
1455   }
1456
1457   NewUnknownFv = &PrivateData->UnknownFvInfo[PrivateData->UnknownFvInfoCount];
1458   PrivateData->UnknownFvInfoCount ++;
1459
1460   CopyGuid (&NewUnknownFv->FvFormat, Format);
1461   NewUnknownFv->FvInfo     = FvInfo;
1462   NewUnknownFv->FvInfoSize = FvInfoSize;
1463   NewUnknownFv->NotifyDescriptor.Flags  = (EFI_PEI_PPI_DESCRIPTOR_NOTIFY_CALLBACK | EFI_PEI_PPI_DESCRIPTOR_TERMINATE_LIST);
1464   NewUnknownFv->NotifyDescriptor.Guid   = &NewUnknownFv->FvFormat;
1465   NewUnknownFv->NotifyDescriptor.Notify = ThirdPartyFvPpiNotifyCallback;
1466
1467   PeiServicesNotifyPpi (&NewUnknownFv->NotifyDescriptor);
1468   return EFI_SUCCESS;
1469 }
1470
1471 /**
1472   Find the FV information according to third-party FV format guid.
1473
1474   This routine also will remove the FV information found by given FV format guid from
1475   PrivateData->UnknownFvInfo[].
1476
1477   @param PrivateData      Point to instance of PEI_CORE_INSTANCE
1478   @param Format           Point to given FV format guid
1479   @param FvInfo           On return, the pointer of FV information buffer
1480   @param FvInfoSize       On return, the size of FV information buffer.
1481
1482   @retval EFI_NOT_FOUND  The FV is not found for new installed EFI_PEI_FIRMWARE_VOLUME_PPI
1483   @retval EFI_SUCCESS    Success to find a FV which could be processed by new installed EFI_PEI_FIRMWARE_VOLUME_PPI.
1484 **/
1485 EFI_STATUS
1486 FindUnknownFormatFvInfo (
1487   IN  PEI_CORE_INSTANCE *PrivateData,
1488   IN  EFI_GUID          *Format,
1489   OUT VOID              **FvInfo,
1490   OUT UINT32            *FvInfoSize
1491   )
1492 {
1493   UINTN Index;
1494   UINTN Index2;
1495
1496   Index = 0;
1497   for (; Index < PrivateData->UnknownFvInfoCount; Index ++) {
1498     if (CompareGuid (Format, &PrivateData->UnknownFvInfo[Index].FvFormat)) {
1499       break;
1500     }
1501   }
1502
1503   if (Index == PrivateData->UnknownFvInfoCount) {
1504     return EFI_NOT_FOUND;
1505   }
1506
1507   *FvInfo     = PrivateData->UnknownFvInfo[Index].FvInfo;
1508   *FvInfoSize = PrivateData->UnknownFvInfo[Index].FvInfoSize;
1509
1510   //
1511   // Remove an entry from UnknownFvInfo array.
1512   //
1513   Index2 = Index + 1;
1514   for (;Index2 < PrivateData->UnknownFvInfoCount; Index2 ++, Index ++) {
1515     CopyMem (&PrivateData->UnknownFvInfo[Index], &PrivateData->UnknownFvInfo[Index2], sizeof (PEI_CORE_UNKNOW_FORMAT_FV_INFO));
1516   }
1517   PrivateData->UnknownFvInfoCount --;
1518   return EFI_SUCCESS;
1519 }
1520
1521 /**
1522   Notification callback function for EFI_PEI_FIRMWARE_VOLUME_PPI.
1523
1524   When a EFI_PEI_FIRMWARE_VOLUME_PPI is installed to support new FV format, this
1525   routine is called to process all discoveried FVs in this format.
1526
1527   @param PeiServices       An indirect pointer to the EFI_PEI_SERVICES table published by the PEI Foundation
1528   @param NotifyDescriptor  Address of the notification descriptor data structure.
1529   @param Ppi               Address of the PPI that was installed.
1530
1531   @retval EFI_SUCCESS  The notification callback is processed correctly.
1532 **/
1533 EFI_STATUS
1534 EFIAPI
1535 ThirdPartyFvPpiNotifyCallback (
1536   IN EFI_PEI_SERVICES              **PeiServices,
1537   IN EFI_PEI_NOTIFY_DESCRIPTOR     *NotifyDescriptor,
1538   IN VOID                          *Ppi
1539   )
1540 {
1541   PEI_CORE_INSTANCE            *PrivateData;
1542   EFI_PEI_FIRMWARE_VOLUME_PPI  *FvPpi;
1543   VOID                         *FvInfo;
1544   UINT32                       FvInfoSize;
1545   EFI_STATUS                   Status;
1546   EFI_PEI_FV_HANDLE            FvHandle;
1547   BOOLEAN                      IsProcessed;
1548   UINTN                        FvIndex;
1549   EFI_PEI_FILE_HANDLE          FileHandle;
1550   VOID                         *DepexData;
1551
1552   PrivateData  = PEI_CORE_INSTANCE_FROM_PS_THIS (PeiServices);
1553   FvPpi = (EFI_PEI_FIRMWARE_VOLUME_PPI*) Ppi;
1554
1555   do {
1556     Status = FindUnknownFormatFvInfo (PrivateData, NotifyDescriptor->Guid, &FvInfo, &FvInfoSize);
1557     if (EFI_ERROR (Status)) {
1558       return EFI_SUCCESS;
1559     }
1560
1561     //
1562     // Process new found FV and get FV handle.
1563     //
1564     Status = FvPpi->ProcessVolume (FvPpi, FvInfo, FvInfoSize, &FvHandle);
1565     if (EFI_ERROR (Status)) {
1566       DEBUG ((EFI_D_ERROR, "Fail to process the FV 0x%p, FV may be corrupted!\n", FvInfo));
1567       continue;
1568     }
1569
1570     //
1571     // Check whether the FV has already been processed.
1572     //
1573     IsProcessed = FALSE;
1574     for (FvIndex = 0; FvIndex < PrivateData->FvCount; FvIndex ++) {
1575       if (PrivateData->Fv[FvIndex].FvHandle == FvHandle) {
1576         DEBUG ((EFI_D_INFO, "The Fv %p has already been processed!\n", FvInfo));
1577         IsProcessed = TRUE;
1578         break;
1579       }
1580     }
1581
1582     if (IsProcessed) {
1583       continue;
1584     }
1585
1586     //
1587     // Update internal PEI_CORE_FV array.
1588     //
1589     PrivateData->Fv[PrivateData->FvCount].FvHeader = (EFI_FIRMWARE_VOLUME_HEADER*) FvInfo;
1590     PrivateData->Fv[PrivateData->FvCount].FvPpi    = FvPpi;
1591     PrivateData->Fv[PrivateData->FvCount].FvHandle = FvHandle;
1592     DEBUG ((
1593       EFI_D_INFO,
1594       "The %dth FV start address is 0x%11p, size is 0x%08x, handle is 0x%p\n",
1595       (UINT32) PrivateData->FvCount,
1596       (VOID *) FvInfo,
1597       FvInfoSize,
1598       FvHandle
1599       ));
1600     PrivateData->FvCount ++;
1601
1602     //
1603     // Scan and process the new discoveried FV for EFI_FV_FILETYPE_FIRMWARE_VOLUME_IMAGE
1604     //
1605     FileHandle = NULL;
1606     do {
1607       Status = FvPpi->FindFileByType (
1608                         FvPpi,
1609                         EFI_FV_FILETYPE_FIRMWARE_VOLUME_IMAGE,
1610                         FvHandle,
1611                         &FileHandle
1612                        );
1613       if (!EFI_ERROR (Status)) {
1614         Status = FvPpi->FindSectionByType (
1615                           FvPpi,
1616                           EFI_SECTION_PEI_DEPEX,
1617                           FileHandle,
1618                           (VOID**)&DepexData
1619                           );
1620         if (!EFI_ERROR (Status)) {
1621           if (!PeimDispatchReadiness (PeiServices, DepexData)) {
1622             //
1623             // Dependency is not satisfied.
1624             //
1625             continue;
1626           }
1627         }
1628
1629         DEBUG ((EFI_D_INFO, "Found firmware volume Image File %p in FV[%d] %p\n", FileHandle, PrivateData->FvCount - 1, FvHandle));
1630         ProcessFvFile (&PrivateData->Fv[PrivateData->FvCount - 1], FileHandle);
1631       }
1632     } while (FileHandle != NULL);
1633   } while (TRUE);
1634 }