[mirror_edk2.git] / Tools / CCode / Source / PeiRebase / PeiRebaseExe.c

/*++\r
\r
Copyright (c)  1999-2006 Intel Corporation. All rights reserved\r
This program and the accompanying materials are licensed and made available\r
under the terms and conditions of the BSD License which accompanies this\r
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
Module Name:\r
\r
  PeiRebaseExe.c\r
\r
Abstract:\r
\r
  This contains all code necessary to build the PeiRebase.exe utility.\r
  This utility relies heavily on the PeiRebase DLL.  Definitions for both\r
  can be found in the PEI Rebase Utility Specification, review draft.\r
\r
--*/\r
\r
#include <stdlib.h>\r
#include <stdio.h>\r
#include <string.h>\r
\r
#include <Common/UefiBaseTypes.h>\r
#include <Common/FirmwareVolumeImageFormat.h>\r
#include <Common/FirmwareFileSystem.h>\r
#include <Library/PeCoffLib.h>\r
\r
#include "CommonLib.h"\r
#include "ParseInf.h"\r
#include "FvLib.h"\r
#include "EfiUtilityMsgs.h"\r
#include "PeiRebaseExe.h"\r
\r
EFI_STATUS\r
ReadHeader (\r
  IN FILE       *InputFile,\r
  OUT UINT32    *FvSize,\r
  OUT BOOLEAN   *ErasePolarity\r
  );\r
\r
int\r
main (\r
  int  argc,\r
  char **argv\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  This utility relocates PEI XIP PE32s in a FV.\r
\r
Arguments:\r
\r
  argc          - Number of command line arguments\r
  argv[]:\r
  BaseAddress     The base address to use for rebasing the FV.  The correct\r
                  format is a hex number preceded by 0x.\r
  InputFileName   The name of the input FV file.\r
  OutputFileName  The name of the output FV file.\r
\r
  Arguments come in pair in any order.\r
    -I InputFileName\r
    -O OutputFileName\r
    -B BaseAddress\r
\r
Returns:\r
\r
  0   No error conditions detected.\r
  1   One or more of the input parameters is invalid.\r
  2   A resource required by the utility was unavailable.\r
      Most commonly this will be memory allocation or file creation.\r
  3   PeiRebase.dll could not be loaded.\r
  4   Error executing the PEI rebase.\r
\r
--*/\r
{\r
  UINT8                       Index;\r
  CHAR8                       InputFileName[_MAX_PATH];\r
  CHAR8                       *OutputFileName;\r
  EFI_PHYSICAL_ADDRESS        XipBase, BsBase, RtBase;\r
  UINT32                      BaseTypes;\r
  EFI_STATUS                  Status;\r
  FILE                        *InputFile;\r
  FILE                        *OutputFile;\r
  FILE                        *LogFile;\r
  UINT64                      FvOffset;\r
  UINT32                      FileCount;\r
  int                         BytesRead;\r
  EFI_FIRMWARE_VOLUME_HEADER  *FvImage;\r
  UINT32                      FvSize;\r
  EFI_FFS_FILE_HEADER         *CurrentFile;\r
  BOOLEAN                     ErasePolarity;\r
  MEMORY_FILE                 InfMemoryFile;\r
  CHAR8                       StringBuffer[0x100];\r
\r
  ErasePolarity = FALSE;\r
  //\r
  // Set utility name for error/warning reporting purposes.\r
  //\r
  SetUtilityName (UTILITY_NAME);\r
  \r
  if (argc == 1) {\r
    Usage();\r
    return STATUS_ERROR;\r
  }\r
    \r
  if ((strcmp(argv[1], "-h") == 0) || (strcmp(argv[1], "--help") == 0) ||\r
      (strcmp(argv[1], "-?") == 0) || (strcmp(argv[1], "/?") == 0)) {\r
    Usage();\r
    return STATUS_ERROR;\r
  }\r
  \r
  if ((strcmp(argv[1], "-V") == 0) || (strcmp(argv[1], "--version") == 0)) {\r
    Version();\r
    return STATUS_ERROR;\r
  }\r
 \r
  //\r
  // Verify the correct number of arguments\r
  //\r
  if (argc != MAX_ARGS) {\r
    Usage ();\r
    return STATUS_ERROR;\r
  }\r
\r
  //\r
  // Initialize variables\r
  //\r
  InputFileName[0]  = '\0';\r
  OutputFileName    = NULL;\r
  XipBase = BsBase = RtBase = 0;\r
  BaseTypes         = 0;\r
  FvOffset          = 0;\r
  FileCount         = 0;\r
  ErasePolarity     = FALSE;\r
  InputFile         = NULL;\r
  OutputFile        = NULL;\r
  LogFile           = NULL;\r
  FvImage           = NULL;\r
\r
  //\r
  // Parse the command line arguments\r
  //\r
  for (Index = 1; Index < MAX_ARGS; Index += 2) {\r
    //\r
    // Make sure argument pair begin with - or /\r
    //\r
    if (argv[Index][0] != '-' && argv[Index][0] != '/') {\r
      Usage ();\r
      Error (NULL, 0, 0, argv[Index], "unrecognized option");\r
      return STATUS_ERROR;\r
    }\r
    //\r
    // Make sure argument specifier is only one letter\r
    //\r
    if (argv[Index][2] != 0) {\r
      Usage ();\r
      Error (NULL, 0, 0, argv[Index], "unrecognized option");\r
      return STATUS_ERROR;\r
    }\r
    //\r
    // Determine argument to read\r
    //\r
    switch (argv[Index][1]) {\r
    case 'I':\r
    case 'i':\r
      if (strlen (InputFileName) == 0) {\r
        strcpy (InputFileName, argv[Index + 1]);\r
      } else {\r
        Usage ();\r
        Error (NULL, 0, 0, argv[Index + 1], "only one -i InputFileName may be specified");\r
        return STATUS_ERROR;\r
      }\r
      break;\r
\r
    case 'O':\r
    case 'o':\r
      if (OutputFileName == NULL) {\r
        OutputFileName = argv[Index + 1];\r
      } else {\r
        Usage ();\r
        Error (NULL, 0, 0, argv[Index + 1], "only one -o OutputFileName may be specified");\r
        return STATUS_ERROR;\r
      }\r
      break;\r
\r
    case 'F':\r
    case 'f':\r
      //\r
      // Load INF file into memory & initialize MEMORY_FILE structure\r
      //\r
      Status = GetFileImage (argv[Index + 1], &InfMemoryFile.FileImage, (UINT32*)&InfMemoryFile.Eof);\r
      InfMemoryFile.Eof = InfMemoryFile.FileImage + (UINT32)(UINTN)InfMemoryFile.Eof;\r
      InfMemoryFile.CurrentFilePointer = InfMemoryFile.FileImage;\r
      if (EFI_ERROR (Status)) {\r
        Error (NULL, 0, 0, argv[Index + 1], "Error opening FvInfFile");\r
        return STATUS_ERROR;\r
      }\r
\r
      //\r
      // Read BaseAddress from fv.inf file\r
      //\r
      FindToken (&InfMemoryFile, "[options]", "EFI_BASE_ADDRESS", 0, StringBuffer);\r
\r
      //\r
      // Free INF file image\r
      //\r
      free (InfMemoryFile.FileImage);\r
\r
      //\r
      // Point argv[Index + 1] to StringBuffer so that it could be processed as "-b"\r
      //\r
      argv[Index + 1] = StringBuffer;\r
\r
    case 'B':\r
    case 'b':\r
      if (BaseTypes & 1) {\r
        Usage ();\r
        Error (NULL, 0, 0, argv[Index + 1], "XipBaseAddress may be specified only once by either -b or -f");\r
        return STATUS_ERROR;\r
      }\r
\r
      Status = AsciiStringToUint64 (argv[Index + 1], FALSE, &XipBase);\r
      if (EFI_ERROR (Status)) {\r
        Usage ();\r
        Error (NULL, 0, 0, argv[Index + 1], "invalid hex digit given for XIP base address");\r
        return STATUS_ERROR;\r
      }\r
\r
      BaseTypes |= 1;\r
      break;\r
\r
    case 'D':\r
    case 'd':\r
      if (BaseTypes & 2) {\r
        Usage ();\r
        Error (NULL, 0, 0, argv[Index + 1], "-d BsBaseAddress may be specified only once");\r
        return STATUS_ERROR;\r
      }\r
\r
      Status = AsciiStringToUint64 (argv[Index + 1], FALSE, &BsBase);\r
      if (EFI_ERROR (Status)) {\r
        Usage ();\r
        Error (NULL, 0, 0, argv[Index + 1], "invalid hex digit given for BS_DRIVER base address");\r
        return STATUS_ERROR;\r
      }\r
\r
      BaseTypes |= 2;\r
      break;\r
\r
    case 'R':\r
    case 'r':\r
      if (BaseTypes & 4) {\r
        Usage ();\r
        Error (NULL, 0, 0, argv[Index + 1], "-r RtBaseAddress may be specified only once");\r
        return STATUS_ERROR;\r
      }\r
\r
      Status = AsciiStringToUint64 (argv[Index + 1], FALSE, &RtBase);\r
      if (EFI_ERROR (Status)) {\r
        Usage ();\r
        Error (NULL, 0, 0, argv[Index + 1], "invalid hex digit given for RT_DRIVER base address");\r
        return STATUS_ERROR;\r
      }\r
\r
      BaseTypes |= 4;\r
      break;\r
\r
    default:\r
      Usage ();\r
      Error (NULL, 0, 0, argv[Index], "unrecognized argument");\r
      return STATUS_ERROR;\r
      break;\r
    }\r
  }\r
  //\r
  // Open the file containing the FV\r
  //\r
  InputFile = fopen (InputFileName, "rb");\r
  if (InputFile == NULL) {\r
    Error (NULL, 0, 0, InputFileName, "could not open input file for reading");\r
    return STATUS_ERROR;\r
  }\r
\r
  //\r
  // Open the log file\r
  //\r
  strcat (InputFileName, ".log");\r
  LogFile = fopen (InputFileName, "a");\r
  if (LogFile == NULL) {\r
    Error (NULL, 0, 0, InputFileName, "could not append to log file");\r
  }\r
\r
  //\r
  // Determine size of FV\r
  //\r
  Status = ReadHeader (InputFile, &FvSize, &ErasePolarity);\r
  if (EFI_ERROR (Status)) {\r
    Error (NULL, 0, 0, "could not parse the FV header", NULL);\r
    goto Finish;\r
  }\r
  //\r
  // Allocate a buffer for the FV image\r
  //\r
  FvImage = malloc (FvSize);\r
  if (FvImage == NULL) {\r
    Error (NULL, 0, 0, "application error", "memory allocation failed");\r
    goto Finish;\r
  }\r
  //\r
  // Read the entire FV to the buffer\r
  //\r
  BytesRead = fread (FvImage, 1, FvSize, InputFile);\r
  fclose (InputFile);\r
  InputFile = NULL;\r
  if ((unsigned int) BytesRead != FvSize) {\r
    Error (NULL, 0, 0, InputFileName, "failed to read from file");\r
    goto Finish;\r
  }\r
  //\r
  // Prepare to walk the FV image\r
  //\r
  InitializeFvLib (FvImage, FvSize);\r
  //\r
  // Get the first file\r
  //\r
  Status = GetNextFile (NULL, &CurrentFile);\r
  if (EFI_ERROR (Status)) {\r
    Error (NULL, 0, 0, "cannot find the first file in the FV image", NULL);\r
    goto Finish;\r
  }\r
  //\r
  // Check if each file should be rebased\r
  //\r
  while (CurrentFile != NULL) {\r
    //\r
    // Rebase this file\r
    //\r
    FfsRebase (\r
      CurrentFile,\r
      BaseTypes,\r
      XipBase + (UINTN)CurrentFile - (UINTN)FvImage,\r
      &BsBase,\r
      &RtBase,\r
      LogFile\r
      );\r
\r
    if (EFI_ERROR (Status)) {\r
      switch (Status) {\r
\r
      case EFI_INVALID_PARAMETER:\r
        Error (NULL, 0, 0, "invalid parameter passed to FfsRebase", NULL);\r
        break;\r
\r
      case EFI_ABORTED:\r
        Error (NULL, 0, 0, "error detected while rebasing -- aborted", NULL);\r
        break;\r
\r
      case EFI_OUT_OF_RESOURCES:\r
        Error (NULL, 0, 0, "FfsRebase could not allocate required resources", NULL);\r
        break;\r
\r
      case EFI_NOT_FOUND:\r
        Error (NULL, 0, 0, "FfsRebase could not locate a PE32 section", NULL);\r
        break;\r
\r
      default:\r
        Error (NULL, 0, 0, "FfsRebase returned unknown status", "status=0x%08X", Status);\r
        break;\r
      }\r
\r
      goto Finish;\r
    }\r
    //\r
    // Get the next file\r
    //\r
    Status = GetNextFile (CurrentFile, &CurrentFile);\r
    if (EFI_ERROR (Status)) {\r
      Error (NULL, 0, 0, "cannot find the next file in the FV image", NULL);\r
      goto Finish;\r
    }\r
  }\r
  //\r
  // Open the output file\r
  //\r
  OutputFile = fopen (OutputFileName, "wb");\r
  if (OutputFile == NULL) {\r
    Error (NULL, 0, 0, OutputFileName, "failed to open output file");\r
    goto Finish;\r
  }\r
\r
  if (fwrite (FvImage, 1, FvSize, OutputFile) != FvSize) {\r
    Error (NULL, 0, 0, "failed to write to output file", 0);\r
    goto Finish;\r
  }\r
\r
Finish:\r
  if (InputFile != NULL) {\r
    fclose (InputFile);\r
  }\r
  //\r
  // If we created an output file, and there was an error, remove it so\r
  // subsequent builds will rebuild it.\r
  //\r
  if (OutputFile != NULL) {\r
    if (GetUtilityStatus () == STATUS_ERROR) {\r
      remove (OutputFileName);\r
    }\r
\r
    fclose (OutputFile);\r
  }\r
\r
  if (LogFile != NULL) {\r
    fclose (LogFile);\r
  }\r
\r
  if (FvImage != NULL) {\r
    free (FvImage);\r
  }\r
\r
  return GetUtilityStatus ();\r
}\r
\r
EFI_STATUS\r
ReadHeader (\r
  IN FILE       *InputFile,\r
  OUT UINT32    *FvSize,\r
  OUT BOOLEAN   *ErasePolarity\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  This function determines the size of the FV and the erase polarity.  The\r
  erase polarity is the FALSE value for file state.\r
\r
Arguments:\r
\r
  InputFile       The file that contains the FV image.\r
  FvSize          The size of the FV.\r
  ErasePolarity   The FV erase polarity.\r
\r
Returns:\r
\r
  EFI_SUCCESS             Function completed successfully.\r
  EFI_INVALID_PARAMETER   A required parameter was NULL or is out of range.\r
  EFI_ABORTED             The function encountered an error.\r
\r
--*/\r
{\r
  EFI_FIRMWARE_VOLUME_HEADER  VolumeHeader;\r
  EFI_FV_BLOCK_MAP_ENTRY      BlockMap;\r
  UINTN                       Signature[2];\r
  UINTN                       BytesRead;\r
  UINT32                      Size;\r
\r
  BytesRead = 0;\r
  Size      = 0;\r
  //\r
  // Check input parameters\r
  //\r
  if ((InputFile == NULL) || (FvSize == NULL) || (ErasePolarity == NULL)) {\r
    Error (NULL, 0, 0, "ReadHeader()", "invalid input parameter");\r
    return EFI_INVALID_PARAMETER;\r
  }\r
  //\r
  // Read the header\r
  //\r
  fread (&VolumeHeader, sizeof (EFI_FIRMWARE_VOLUME_HEADER) - sizeof (EFI_FV_BLOCK_MAP_ENTRY), 1, InputFile);\r
  BytesRead     = sizeof (EFI_FIRMWARE_VOLUME_HEADER) - sizeof (EFI_FV_BLOCK_MAP_ENTRY);\r
  Signature[0]  = VolumeHeader.Signature;\r
  Signature[1]  = 0;\r
\r
  //\r
  // Get erase polarity\r
  //\r
  if (VolumeHeader.Attributes & EFI_FVB_ERASE_POLARITY) {\r
    *ErasePolarity = TRUE;\r
  }\r
\r
  do {\r
    fread (&BlockMap, sizeof (EFI_FV_BLOCK_MAP_ENTRY), 1, InputFile);\r
    BytesRead += sizeof (EFI_FV_BLOCK_MAP_ENTRY);\r
\r
    if (BlockMap.NumBlocks != 0) {\r
      Size += BlockMap.NumBlocks * BlockMap.BlockLength;\r
    }\r
\r
  } while (!(BlockMap.NumBlocks == 0 && BlockMap.BlockLength == 0));\r
\r
  if (VolumeHeader.FvLength != Size) {\r
    Error (NULL, 0, 0, "volume size not consistant with block maps", NULL);\r
    return EFI_ABORTED;\r
  }\r
\r
  *FvSize = Size;\r
\r
  rewind (InputFile);\r
\r
  return EFI_SUCCESS;\r
}\r
\r
VOID\r
Version (\r
  VOID\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Displays the standard utility information to SDTOUT\r
\r
Arguments:\r
\r
  None\r
\r
Returns:\r
\r
  None\r
\r
--*/\r
{\r
  printf ("%s v%d.%d -PEI Rebase Utility.\n", UTILITY_NAME, UTILITY_MAJOR_VERSION, UTILITY_MINOR_VERSION);\r
  printf ("Copyright (c) 1999-2007 Intel Corporation. All rights reserved.\n");\r
}\r
\r
VOID\r
Usage (\r
  VOID\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Displays the utility usage syntax to STDOUT\r
\r
Arguments:\r
\r
  None\r
\r
Returns:\r
\r
  None\r
\r
--*/\r
{\r
  Version();\r
  \r
  printf (\r
    "\nUsage: %s -I InputFileName -O OutputFileName -B BaseAddress\n",\r
    UTILITY_NAME\r
    );\r
  printf ("  Where:\n");\r
  printf ("     InputFileName is the name of the EFI FV file to rebase.\n");\r
  printf ("     OutputFileName is the desired output file name.\n");\r
  printf ("     BaseAddress is the FV base address to rebase agains.\n");\r
  printf ("  Argument pair may be in any order.\n");\r
}\r
\r
EFI_STATUS\r
FfsRebase (\r
  IN OUT  EFI_FFS_FILE_HEADER       *FfsFile,\r
  IN      UINT32                    Flags,\r
  IN OUT  EFI_PHYSICAL_ADDRESS      XipBase,\r
  IN OUT  EFI_PHYSICAL_ADDRESS      *BsBase,\r
  IN OUT  EFI_PHYSICAL_ADDRESS      *RtBase,\r
  OUT     FILE                      *LogFile\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  This function determines if a file is XIP and should be rebased.  It will\r
  rebase any PE32 sections found in the file using the base address.\r
\r
Arguments:\r
\r
  FfsFile           A pointer to Ffs file image.\r
  BaseAddress       The base address to use for rebasing the file image.\r
\r
Returns:\r
\r
  EFI_SUCCESS             The image was properly rebased.\r
  EFI_INVALID_PARAMETER   An input parameter is invalid.\r
  EFI_ABORTED             An error occurred while rebasing the input file image.\r
  EFI_OUT_OF_RESOURCES    Could not allocate a required resource.\r
  EFI_NOT_FOUND           No compressed sections could be found.\r
\r
--*/\r
{\r
  EFI_STATUS                            Status;\r
  PE_COFF_LOADER_IMAGE_CONTEXT          ImageContext;\r
  UINTN                                 MemoryImagePointer;\r
  UINTN                                 MemoryImagePointerAligned;\r
  EFI_PHYSICAL_ADDRESS                  ImageAddress;\r
  UINT64                                ImageSize;\r
  EFI_PHYSICAL_ADDRESS                  EntryPoint;\r
  UINT32                                Pe32ImageSize;\r
  EFI_PHYSICAL_ADDRESS                  NewPe32BaseAddress;\r
  UINTN                                 Index;\r
  EFI_FILE_SECTION_POINTER              CurrentPe32Section;\r
  EFI_FFS_FILE_STATE                    SavedState;\r
  EFI_IMAGE_NT_HEADERS32                *PeHdr;\r
  EFI_IMAGE_NT_HEADERS64                *PePlusHdr;\r
  UINT32                                *PeHdrSizeOfImage;\r
  UINT32                                *PeHdrChecksum;\r
  EFI_TE_IMAGE_HEADER                   *TEImageHeader;\r
  UINT8                                 *TEBuffer;\r
  EFI_IMAGE_DOS_HEADER                  *DosHeader;\r
  UINT8                                 FileGuidString[80];\r
  UINT32                                TailSize;\r
  EFI_FFS_FILE_TAIL                     TailValue;\r
  EFI_PHYSICAL_ADDRESS                  *BaseToUpdate;\r
\r
  //\r
  // Verify input parameters\r
  //\r
  if (FfsFile == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
  //\r
  // Convert the GUID to a string so we can at least report which file\r
  // if we find an error.\r
  //\r
  PrintGuidToBuffer (&FfsFile->Name, FileGuidString, sizeof (FileGuidString), TRUE);\r
  if (FfsFile->Attributes & FFS_ATTRIB_TAIL_PRESENT) {\r
    TailSize = sizeof (EFI_FFS_FILE_TAIL);\r
  } else {\r
    TailSize = 0;\r
  }\r
  //\r
  // Do some cursory checks on the FFS file contents\r
  //\r
  Status = VerifyFfsFile (FfsFile);\r
  if (EFI_ERROR (Status)) {\r
    Error (NULL, 0, 0, "file does not appear to be a valid FFS file, cannot be rebased", FileGuidString);\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  //\r
  // We only process files potentially containing PE32 sections.\r
  //\r
  switch (FfsFile->Type) {\r
    case EFI_FV_FILETYPE_SECURITY_CORE:\r
    case EFI_FV_FILETYPE_PEI_CORE:\r
    case EFI_FV_FILETYPE_PEIM:\r
    case EFI_FV_FILETYPE_COMBINED_PEIM_DRIVER:\r
    case EFI_FV_FILETYPE_DRIVER:\r
    case EFI_FV_FILETYPE_DXE_CORE:\r
      break;\r
    default:\r
      return EFI_SUCCESS;\r
  }\r
\r
  //\r
  // Rebase each PE32 section\r
  //\r
  Status      = EFI_SUCCESS;\r
  for (Index = 1;; Index++) {\r
    Status = GetSectionByType (FfsFile, EFI_SECTION_PE32, Index, &CurrentPe32Section);\r
    if (EFI_ERROR (Status)) {\r
      break;\r
    }\r
\r
\r
    //\r
    // Initialize context\r
    //\r
    memset (&ImageContext, 0, sizeof (ImageContext));\r
    ImageContext.Handle     = (VOID *) ((UINTN) CurrentPe32Section.Pe32Section + sizeof (EFI_PE32_SECTION));\r
    ImageContext.ImageRead  = (PE_COFF_LOADER_READ_FILE) FfsRebaseImageRead;\r
    Status                  = PeCoffLoaderGetImageInfo (&ImageContext);\r
    if (EFI_ERROR (Status)) {\r
      Error (NULL, 0, 0, "GetImageInfo() call failed on rebase", FileGuidString);\r
      return Status;\r
    }\r
\r
    //\r
    // Calculate the PE32 base address, based on file type\r
    //\r
    switch (FfsFile->Type) {\r
      case EFI_FV_FILETYPE_SECURITY_CORE:\r
      case EFI_FV_FILETYPE_PEI_CORE:\r
      case EFI_FV_FILETYPE_PEIM:\r
      case EFI_FV_FILETYPE_COMBINED_PEIM_DRIVER:\r
        if ((Flags & 1) == 0) {\r
          //\r
          // We aren't relocating XIP code, so skip it.\r
          //\r
          return EFI_SUCCESS;\r
        }\r
\r
        NewPe32BaseAddress =\r
          XipBase +\r
          (UINTN)CurrentPe32Section.Pe32Section +\r
          sizeof (EFI_COMMON_SECTION_HEADER) -\r
          (UINTN)FfsFile;\r
        BaseToUpdate = &XipBase;\r
        break;\r
\r
      case EFI_FV_FILETYPE_DRIVER:\r
        PeHdr = (EFI_IMAGE_NT_HEADERS32*)(\r
          (UINTN)CurrentPe32Section.Pe32Section +\r
          sizeof (EFI_COMMON_SECTION_HEADER) +\r
          ImageContext.PeCoffHeaderOffset\r
          );\r
        switch (PeHdr->OptionalHeader.Subsystem) {\r
          case EFI_IMAGE_SUBSYSTEM_EFI_RUNTIME_DRIVER:\r
            if ((Flags & 4) == 0) {\r
              //\r
              // RT drivers aren't supposed to be relocated\r
              //\r
              continue;\r
            }\r
\r
            NewPe32BaseAddress = *RtBase;\r
            BaseToUpdate = RtBase;\r
            break;\r
\r
          default:\r
            //\r
            // We treat all other subsystems the same as BS_DRIVER\r
            //\r
            if ((Flags & 2) == 0) {\r
              //\r
              // Skip all BS_DRIVER's\r
              //\r
              continue;\r
            }\r
\r
            NewPe32BaseAddress = *BsBase;\r
            BaseToUpdate = BsBase;\r
            break;\r
        }\r
        break;\r
\r
      case EFI_FV_FILETYPE_DXE_CORE:\r
        if ((Flags & 2) == 0) {\r
          //\r
          // Skip DXE core\r
          //\r
          return EFI_SUCCESS;\r
        }\r
\r
        NewPe32BaseAddress = *BsBase;\r
        BaseToUpdate = BsBase;\r
        break;\r
\r
      default:\r
        //\r
        // Not supported file type\r
        //\r
        return EFI_SUCCESS;\r
    }\r
\r
    //\r
    // Allocate a buffer for the image to be loaded into.\r
    //\r
    Pe32ImageSize       = GetLength (CurrentPe32Section.Pe32Section->CommonHeader.Size) - sizeof (EFI_PE32_SECTION);\r
    MemoryImagePointer  = (UINTN) (malloc (Pe32ImageSize + 0x100000));\r
    if (MemoryImagePointer == 0) {\r
      Error (NULL, 0, 0, "memory allocation failure", NULL);\r
      return EFI_OUT_OF_RESOURCES;\r
    }\r
    memset ((void *) MemoryImagePointer, 0, Pe32ImageSize + 0x100000);\r
    MemoryImagePointerAligned = (MemoryImagePointer + 0x0FFFF) & (-1 << 16);\r
\r
    ImageContext.ImageAddress = MemoryImagePointerAligned;\r
\r
    Status                    = PeCoffLoaderLoadImage (&ImageContext);\r
    if (EFI_ERROR (Status)) {\r
      Error (NULL, 0, 0, "LoadImage() call failed on rebase", FileGuidString);\r
      free ((VOID *) MemoryImagePointer);\r
      return Status;\r
    }\r
    \r
    //\r
    // Check if section-alignment and file-alignment match or not\r
    //\r
    if (!(ImageContext.IsTeImage)) {\r
      PeHdr = (EFI_IMAGE_NT_HEADERS *)((UINTN)ImageContext.ImageAddress + \r
                                              ImageContext.PeCoffHeaderOffset);\r
      if (PeHdr->OptionalHeader.SectionAlignment != PeHdr->OptionalHeader.FileAlignment) {\r
        Error (NULL, 0, 0, "Section-Alignment and File-Alignment does not match", FileGuidString);\r
        free ((VOID *) MemoryImagePointer);\r
        return EFI_ABORTED;\r
      }\r
    }\r
    else {\r
      //\r
      // BUGBUG: TE Image Header lack section-alignment and file-alignment info\r
      //\r
    }\r
\r
    ImageContext.DestinationAddress = NewPe32BaseAddress;\r
    Status                          = PeCoffLoaderRelocateImage (&ImageContext);\r
    if (EFI_ERROR (Status)) {\r
      Error (NULL, 0, 0, "RelocateImage() call failed on rebase", FileGuidString);\r
      free ((VOID *) MemoryImagePointer);\r
      return Status;\r
    }\r
\r
    ImageAddress  = ImageContext.ImageAddress;\r
    ImageSize     = ImageContext.ImageSize;\r
    EntryPoint    = ImageContext.EntryPoint;\r
\r
    if (ImageSize > Pe32ImageSize) {\r
      Error (\r
        NULL,\r
        0,\r
        0,\r
        "rebased image is larger than original PE32 image",\r
        "0x%X > 0x%X, file %s",\r
        ImageSize,\r
        Pe32ImageSize,\r
        FileGuidString\r
        );\r
      free ((VOID *) MemoryImagePointer);\r
      return EFI_ABORTED;\r
    }\r
\r
    //\r
    // Update BASE address\r
    //\r
    fprintf (\r
      LogFile,\r
      "%s %016I64X %s\n",\r
      FileGuidString,\r
      ImageContext.DestinationAddress,\r
      ImageContext.PdbPointer == NULL ? "*" : ImageContext.PdbPointer\r
      );\r
    *BaseToUpdate += EFI_SIZE_TO_PAGES (ImageContext.ImageSize) * EFI_PAGE_SIZE;\r
\r
    //\r
    // Since we may have updated the Codeview RVA, we need to insure the PE\r
    // header indicates the image is large enough to contain the Codeview data\r
    // so it will be loaded properly later if the PEIM is reloaded into memory...\r
    //\r
    PeHdr = (VOID *) ((UINTN) ImageAddress + ImageContext.PeCoffHeaderOffset);\r
    PePlusHdr = (EFI_IMAGE_NT_HEADERS64*)PeHdr;\r
    if (PeHdr->FileHeader.Machine == EFI_IMAGE_MACHINE_IA32) {\r
      PeHdrSizeOfImage  = (UINT32 *) (&(PeHdr->OptionalHeader).SizeOfImage);\r
      PeHdrChecksum     = (UINT32 *) (&(PeHdr->OptionalHeader).CheckSum);\r
    } else if (PeHdr->FileHeader.Machine == EFI_IMAGE_MACHINE_IA64) {\r
      PeHdrSizeOfImage  = (UINT32 *) (&(PePlusHdr->OptionalHeader).SizeOfImage);\r
      PeHdrChecksum     = (UINT32 *) (&(PePlusHdr->OptionalHeader).CheckSum);\r
    } else if (PeHdr->FileHeader.Machine == EFI_IMAGE_MACHINE_X64) {\r
      PeHdrSizeOfImage  = (UINT32 *) (&(PePlusHdr->OptionalHeader).SizeOfImage);\r
      PeHdrChecksum     = (UINT32 *) (&(PePlusHdr->OptionalHeader).CheckSum);\r
    } else {\r
      Error (\r
        NULL,\r
        0,\r
        0,\r
        "unknown machine type in PE32 image",\r
        "machine type=0x%X, file=%s",\r
        (UINT32) PeHdr->FileHeader.Machine,\r
        FileGuidString\r
        );\r
      free ((VOID *) MemoryImagePointer);\r
      return EFI_ABORTED;\r
    }\r
\r
    if (*PeHdrSizeOfImage != ImageContext.ImageSize) {\r
      *PeHdrSizeOfImage = (UINT32) ImageContext.ImageSize;\r
      if (*PeHdrChecksum) {\r
        *PeHdrChecksum = 0;\r
      }\r
    }\r
\r
    memcpy (CurrentPe32Section.Pe32Section + 1, (VOID *) MemoryImagePointerAligned, (UINT32) ImageSize);\r
\r
    free ((VOID *) MemoryImagePointer);\r
\r
    //\r
    // Now update file checksum\r
    //\r
    if (FfsFile->Attributes & FFS_ATTRIB_TAIL_PRESENT) {\r
      TailSize = sizeof (EFI_FFS_FILE_TAIL);\r
    } else {\r
      TailSize = 0;\r
    }\r
\r
    if (FfsFile->Attributes & FFS_ATTRIB_CHECKSUM) {\r
      SavedState  = FfsFile->State;\r
      FfsFile->IntegrityCheck.Checksum.File = 0;\r
      FfsFile->State                        = 0;\r
      if (FfsFile->Attributes & FFS_ATTRIB_CHECKSUM) {\r
        FfsFile->IntegrityCheck.Checksum.File = CalculateChecksum8 (\r
                                                  (UINT8 *) FfsFile,\r
                                                  GetLength (FfsFile->Size) - TailSize\r
                                                  );\r
      } else {\r
        FfsFile->IntegrityCheck.Checksum.File = FFS_FIXED_CHECKSUM;\r
      }\r
\r
      FfsFile->State = SavedState;\r
    }\r
    //\r
    // Update tail if present\r
    //\r
    if (FfsFile->Attributes & FFS_ATTRIB_TAIL_PRESENT) {\r
      TailValue = (EFI_FFS_FILE_TAIL) (~(FfsFile->IntegrityCheck.TailReference));\r
      *(EFI_FFS_FILE_TAIL *) (((UINTN) FfsFile + GetLength (FfsFile->Size) - sizeof (EFI_FFS_FILE_TAIL))) = TailValue;\r
    }\r
  }\r
\r
  if ((Flags & 1) == 0 || (\r
      FfsFile->Type != EFI_FV_FILETYPE_SECURITY_CORE &&\r
      FfsFile->Type != EFI_FV_FILETYPE_PEI_CORE &&\r
\r
      FfsFile->Type != EFI_FV_FILETYPE_PEIM &&\r
      FfsFile->Type != EFI_FV_FILETYPE_COMBINED_PEIM_DRIVER\r
      )) {\r
    //\r
    // Only XIP code may have a TE section\r
    //\r
    return EFI_SUCCESS;\r
  }\r
\r
  //\r
  // Now process TE sections\r
  //\r
  for (Index = 1;; Index++) {\r
    Status = GetSectionByType (FfsFile, EFI_SECTION_TE, Index, &CurrentPe32Section);\r
    if (EFI_ERROR (Status)) {\r
      break;\r
    }\r
\r
    //\r
    // Calculate the TE base address, the FFS file base plus the offset of the TE section less the size stripped off\r
    // by GenTEImage\r
    //\r
    TEImageHeader = (EFI_TE_IMAGE_HEADER *) ((UINT8 *) CurrentPe32Section.Pe32Section + sizeof (EFI_COMMON_SECTION_HEADER));\r
\r
    NewPe32BaseAddress = ((UINT32) XipBase) +\r
      (\r
        (UINTN) CurrentPe32Section.Pe32Section +\r
        sizeof (EFI_COMMON_SECTION_HEADER) +\r
        sizeof (EFI_TE_IMAGE_HEADER) -\r
        TEImageHeader->StrippedSize -\r
        (UINTN) FfsFile\r
      );\r
\r
    //\r
    // Allocate a buffer to unshrink the image into.\r
    //\r
    Pe32ImageSize = GetLength (CurrentPe32Section.Pe32Section->CommonHeader.Size) - sizeof (EFI_PE32_SECTION) -\r
    sizeof (EFI_TE_IMAGE_HEADER);\r
    Pe32ImageSize += TEImageHeader->StrippedSize;\r
    TEBuffer = (UINT8 *) malloc (Pe32ImageSize);\r
    if (TEBuffer == NULL) {\r
      Error (NULL, 0, 0, "failed to allocate memory", NULL);\r
      return EFI_OUT_OF_RESOURCES;\r
    }\r
    //\r
    // Expand the image into our buffer and fill in critical fields in the DOS header\r
    // Fill in fields required by the loader.\r
    // At offset 0x3C is the offset to the PE signature. We'll put it immediately following the offset value\r
    // itself.\r
    //\r
    memset (TEBuffer, 0, Pe32ImageSize);\r
    DosHeader = (EFI_IMAGE_DOS_HEADER *) TEBuffer;\r
    DosHeader->e_magic = EFI_IMAGE_DOS_SIGNATURE;\r
    *(UINT32 *) (TEBuffer + 0x3C) = 0x40;\r
    PeHdr = (EFI_IMAGE_NT_HEADERS *) (TEBuffer + 0x40);\r
    PePlusHdr = (EFI_IMAGE_NT_HEADERS64*)PeHdr;\r
    PeHdr->Signature = EFI_IMAGE_NT_SIGNATURE;\r
    PeHdr->FileHeader.Machine = TEImageHeader->Machine;\r
    PeHdr->FileHeader.NumberOfSections = TEImageHeader->NumberOfSections;\r
\r
    //\r
    // Say the size of the optional header is the total we stripped off less the size of a PE file header and PE signature and\r
    // the 0x40 bytes for our DOS header.\r
    //\r
    PeHdr->FileHeader.SizeOfOptionalHeader = (UINT16) (TEImageHeader->StrippedSize - 0x40 - sizeof (UINT32) - sizeof (EFI_IMAGE_FILE_HEADER));\r
    if (TEImageHeader->Machine == EFI_IMAGE_MACHINE_IA32) {\r
      PeHdr->OptionalHeader.Magic = EFI_IMAGE_NT_OPTIONAL_HDR32_MAGIC;\r
    } else if (TEImageHeader->Machine == EFI_IMAGE_MACHINE_IA64) {\r
      PePlusHdr->OptionalHeader.Magic = EFI_IMAGE_NT_OPTIONAL_HDR64_MAGIC;\r
    } else if (TEImageHeader->Machine == EFI_IMAGE_MACHINE_X64) {\r
      PePlusHdr->OptionalHeader.Magic = EFI_IMAGE_NT_OPTIONAL_HDR64_MAGIC;\r
    } else {\r
      Error (\r
        NULL,\r
        0,\r
        0,\r
        "unknown machine type in TE image",\r
        "machine type=0x%X, file=%s",\r
        (UINT32) TEImageHeader->Machine,\r
        FileGuidString\r
        );\r
      free (TEBuffer);\r
      return EFI_ABORTED;\r
    }\r
\r
    if (PeHdr->OptionalHeader.Magic == EFI_IMAGE_NT_OPTIONAL_HDR32_MAGIC) {\r
      PeHdr->OptionalHeader.ImageBase     = (UINTN) (TEImageHeader->ImageBase - TEImageHeader->StrippedSize + sizeof (EFI_TE_IMAGE_HEADER));\r
      PeHdr->OptionalHeader.SizeOfImage   = Pe32ImageSize;\r
      PeHdr->OptionalHeader.Subsystem     = TEImageHeader->Subsystem;\r
      PeHdr->OptionalHeader.SizeOfHeaders = TEImageHeader->StrippedSize + TEImageHeader->NumberOfSections *\r
                                            sizeof (EFI_IMAGE_SECTION_HEADER) - 12;\r
\r
      //\r
      // Set NumberOfRvaAndSizes in the optional header to what we had available in the original image\r
      //\r
      if ((TEImageHeader->DataDirectory[EFI_TE_IMAGE_DIRECTORY_ENTRY_BASERELOC].VirtualAddress != 0) ||\r
          (TEImageHeader->DataDirectory[EFI_TE_IMAGE_DIRECTORY_ENTRY_BASERELOC].Size != 0)\r
          ) {\r
        PeHdr->OptionalHeader.NumberOfRvaAndSizes = EFI_IMAGE_DIRECTORY_ENTRY_BASERELOC + 1;\r
        PeHdr->OptionalHeader.DataDirectory[EFI_IMAGE_DIRECTORY_ENTRY_BASERELOC].VirtualAddress = TEImageHeader->DataDirectory[EFI_TE_IMAGE_DIRECTORY_ENTRY_BASERELOC].VirtualAddress;\r
        PeHdr->OptionalHeader.DataDirectory[EFI_IMAGE_DIRECTORY_ENTRY_BASERELOC].Size = TEImageHeader->DataDirectory[EFI_TE_IMAGE_DIRECTORY_ENTRY_BASERELOC].Size;\r
      }\r
\r
      if ((TEImageHeader->DataDirectory[EFI_TE_IMAGE_DIRECTORY_ENTRY_DEBUG].VirtualAddress != 0) ||\r
          (TEImageHeader->DataDirectory[EFI_TE_IMAGE_DIRECTORY_ENTRY_DEBUG].Size != 0)\r
          ) {\r
        PeHdr->OptionalHeader.DataDirectory[EFI_IMAGE_DIRECTORY_ENTRY_DEBUG].VirtualAddress = TEImageHeader->DataDirectory[EFI_TE_IMAGE_DIRECTORY_ENTRY_DEBUG].VirtualAddress;\r
        PeHdr->OptionalHeader.DataDirectory[EFI_IMAGE_DIRECTORY_ENTRY_DEBUG].Size = TEImageHeader->DataDirectory[EFI_TE_IMAGE_DIRECTORY_ENTRY_DEBUG].Size;\r
        if (PeHdr->OptionalHeader.NumberOfRvaAndSizes < EFI_IMAGE_DIRECTORY_ENTRY_DEBUG + 1) {\r
          PeHdr->OptionalHeader.NumberOfRvaAndSizes = EFI_IMAGE_DIRECTORY_ENTRY_DEBUG + 1;\r
        }\r
      }\r
      //\r
      // NOTE: These values are defaults, and should be verified to be correct in the GenTE utility\r
      //\r
      PeHdr->OptionalHeader.SectionAlignment = 0x10;\r
    } else {\r
      PePlusHdr->OptionalHeader.ImageBase     = (UINTN) (TEImageHeader->ImageBase - TEImageHeader->StrippedSize + sizeof (EFI_TE_IMAGE_HEADER));\r
      PePlusHdr->OptionalHeader.SizeOfImage   = Pe32ImageSize;\r
      PePlusHdr->OptionalHeader.Subsystem     = TEImageHeader->Subsystem;\r
      PePlusHdr->OptionalHeader.SizeOfHeaders = TEImageHeader->StrippedSize + TEImageHeader->NumberOfSections *\r
                                                sizeof (EFI_IMAGE_SECTION_HEADER) - 12;\r
\r
      //\r
      // Set NumberOfRvaAndSizes in the optional header to what we had available in the original image\r
      //\r
      if ((TEImageHeader->DataDirectory[EFI_TE_IMAGE_DIRECTORY_ENTRY_BASERELOC].VirtualAddress != 0) ||\r
          (TEImageHeader->DataDirectory[EFI_TE_IMAGE_DIRECTORY_ENTRY_BASERELOC].Size != 0)\r
          ) {\r
        PePlusHdr->OptionalHeader.NumberOfRvaAndSizes = EFI_IMAGE_DIRECTORY_ENTRY_BASERELOC + 1;\r
        PePlusHdr->OptionalHeader.DataDirectory[EFI_IMAGE_DIRECTORY_ENTRY_BASERELOC].VirtualAddress = TEImageHeader->DataDirectory[EFI_TE_IMAGE_DIRECTORY_ENTRY_BASERELOC].VirtualAddress;\r
        PePlusHdr->OptionalHeader.DataDirectory[EFI_IMAGE_DIRECTORY_ENTRY_BASERELOC].Size = TEImageHeader->DataDirectory[EFI_TE_IMAGE_DIRECTORY_ENTRY_BASERELOC].Size;\r
      }\r
\r
      if ((TEImageHeader->DataDirectory[EFI_TE_IMAGE_DIRECTORY_ENTRY_DEBUG].VirtualAddress != 0) ||\r
          (TEImageHeader->DataDirectory[EFI_TE_IMAGE_DIRECTORY_ENTRY_DEBUG].Size != 0)\r
          ) {\r
        PePlusHdr->OptionalHeader.DataDirectory[EFI_IMAGE_DIRECTORY_ENTRY_DEBUG].VirtualAddress = TEImageHeader->DataDirectory[EFI_TE_IMAGE_DIRECTORY_ENTRY_DEBUG].VirtualAddress;\r
        PePlusHdr->OptionalHeader.DataDirectory[EFI_IMAGE_DIRECTORY_ENTRY_DEBUG].Size = TEImageHeader->DataDirectory[EFI_TE_IMAGE_DIRECTORY_ENTRY_DEBUG].Size;\r
        if (PePlusHdr->OptionalHeader.NumberOfRvaAndSizes < EFI_IMAGE_DIRECTORY_ENTRY_DEBUG + 1) {\r
          PePlusHdr->OptionalHeader.NumberOfRvaAndSizes = EFI_IMAGE_DIRECTORY_ENTRY_DEBUG + 1;\r
        }\r
      }\r
      //\r
      // NOTE: These values are defaults, and should be verified to be correct in the GenTE utility\r
      //\r
      PePlusHdr->OptionalHeader.SectionAlignment = 0x10;\r
    }\r
\r
    //\r
    // Copy the rest of the image to its original offset\r
    //\r
    memcpy (\r
      TEBuffer + TEImageHeader->StrippedSize,\r
      (UINT8 *) CurrentPe32Section.Pe32Section + sizeof (EFI_PE32_SECTION) + sizeof (EFI_TE_IMAGE_HEADER),\r
      GetLength (CurrentPe32Section.Pe32Section->CommonHeader.Size) - sizeof (EFI_PE32_SECTION) -\r
      sizeof (EFI_TE_IMAGE_HEADER)\r
      );\r
\r
    //\r
    // Initialize context\r
    //\r
    memset (&ImageContext, 0, sizeof (ImageContext));\r
    ImageContext.Handle     = (VOID *) TEBuffer;\r
    ImageContext.ImageRead  = (PE_COFF_LOADER_READ_FILE) FfsRebaseImageRead;\r
\r
    Status                  = PeCoffLoaderGetImageInfo (&ImageContext);\r
\r
    if (EFI_ERROR (Status)) {\r
      Error (NULL, 0, 0, "GetImageInfo() call failed on rebase of TE image", FileGuidString);\r
      free (TEBuffer);\r
      return Status;\r
    }\r
    //\r
    // Allocate a buffer for the image to be loaded into.\r
    //\r
    MemoryImagePointer = (UINTN) (malloc (Pe32ImageSize + 0x100000));\r
    if (MemoryImagePointer == 0) {\r
      Error (NULL, 0, 0, "memory allocation error on rebase of TE image", FileGuidString);\r
      free (TEBuffer);\r
      return EFI_OUT_OF_RESOURCES;\r
    }\r
    memset ((void *) MemoryImagePointer, 0, Pe32ImageSize + 0x100000);\r
    MemoryImagePointerAligned = (MemoryImagePointer + 0x0FFFF) & (-1 << 16);\r
    \r
\r
    ImageContext.ImageAddress = MemoryImagePointerAligned;\r
    Status                    = PeCoffLoaderLoadImage (&ImageContext);\r
    if (EFI_ERROR (Status)) {\r
      Error (NULL, 0, 0, "LoadImage() call failed on rebase of TE image", FileGuidString);\r
      free (TEBuffer);\r
      free ((VOID *) MemoryImagePointer);\r
      return Status;\r
    }\r
\r
    ImageContext.DestinationAddress = NewPe32BaseAddress;\r
    Status                          = PeCoffLoaderRelocateImage (&ImageContext);\r
    if (EFI_ERROR (Status)) {\r
      Error (NULL, 0, 0, "RelocateImage() call failed on rebase of TE image", FileGuidString);\r
      free ((VOID *) MemoryImagePointer);\r
      free (TEBuffer);\r
      return Status;\r
    }\r
\r
    ImageAddress  = ImageContext.ImageAddress;\r
    ImageSize     = ImageContext.ImageSize;\r
    EntryPoint    = ImageContext.EntryPoint;\r
\r
    //\r
    // Since we may have updated the Codeview RVA, we need to insure the PE\r
    // header indicates the image is large enough to contain the Codeview data\r
    // so it will be loaded properly later if the PEIM is reloaded into memory...\r
    //\r
    PeHdr = (VOID *) ((UINTN) ImageAddress + ImageContext.PeCoffHeaderOffset);\r
    PePlusHdr = (EFI_IMAGE_NT_HEADERS64*)PeHdr;\r
    if (PeHdr->FileHeader.Machine == EFI_IMAGE_MACHINE_IA32) {\r
      PeHdrSizeOfImage  = (UINT32 *) (&(PeHdr->OptionalHeader).SizeOfImage);\r
      PeHdrChecksum     = (UINT32 *) (&(PeHdr->OptionalHeader).CheckSum);\r
    } else if (PeHdr->FileHeader.Machine == EFI_IMAGE_MACHINE_IA64) {\r
      PeHdrSizeOfImage  = (UINT32 *) (&(PePlusHdr->OptionalHeader).SizeOfImage);\r
      PeHdrChecksum     = (UINT32 *) (&(PePlusHdr->OptionalHeader).CheckSum);\r
    } else if (PeHdr->FileHeader.Machine == EFI_IMAGE_MACHINE_X64) {\r
      PeHdrSizeOfImage  = (UINT32 *) (&(PePlusHdr->OptionalHeader).SizeOfImage);\r
      PeHdrChecksum     = (UINT32 *) (&(PePlusHdr->OptionalHeader).CheckSum);\r
    } else {\r
      Error (\r
        NULL,\r
        0,\r
        0,\r
        "unknown machine type in TE image",\r
        "machine type=0x%X, file=%s",\r
        (UINT32) PeHdr->FileHeader.Machine,\r
        FileGuidString\r
        );\r
      free ((VOID *) MemoryImagePointer);\r
      free (TEBuffer);\r
      return EFI_ABORTED;\r
    }\r
\r
    if (*PeHdrSizeOfImage != ImageContext.ImageSize) {\r
      *PeHdrSizeOfImage = (UINT32) ImageContext.ImageSize;\r
      if (*PeHdrChecksum) {\r
        *PeHdrChecksum = 0;\r
      }\r
    }\r
\r
    TEImageHeader->ImageBase = (UINT64) (NewPe32BaseAddress + TEImageHeader->StrippedSize - sizeof (EFI_TE_IMAGE_HEADER));\r
    memcpy (\r
      (UINT8 *) (CurrentPe32Section.Pe32Section + 1) + sizeof (EFI_TE_IMAGE_HEADER),\r
      (VOID *) ((UINT8 *) MemoryImagePointerAligned + TEImageHeader->StrippedSize),\r
      GetLength (CurrentPe32Section.Pe32Section->CommonHeader.Size) - sizeof (EFI_PE32_SECTION) -\r
      sizeof (EFI_TE_IMAGE_HEADER)\r
      );\r
    if (FfsFile->Attributes & FFS_ATTRIB_TAIL_PRESENT) {\r
      TailSize = sizeof (EFI_FFS_FILE_TAIL);\r
    } else {\r
      TailSize = 0;\r
    }\r
    //\r
    // Now update file checksum\r
    //\r
    if (FfsFile->Attributes & FFS_ATTRIB_CHECKSUM) {\r
      SavedState  = FfsFile->State;\r
      FfsFile->IntegrityCheck.Checksum.File = 0;\r
      FfsFile->State                        = 0;\r
      if (FfsFile->Attributes & FFS_ATTRIB_CHECKSUM) {\r
        FfsFile->IntegrityCheck.Checksum.File = CalculateChecksum8 (\r
                                                  (UINT8 *) FfsFile,\r
                                                  GetLength (FfsFile->Size) - TailSize\r
                                                  );\r
      } else {\r
        FfsFile->IntegrityCheck.Checksum.File = FFS_FIXED_CHECKSUM;\r
      }\r
\r
      FfsFile->State = SavedState;\r
    }\r
    //\r
    // Update tail if present\r
    //\r
    if (FfsFile->Attributes & FFS_ATTRIB_TAIL_PRESENT) {\r
      TailValue = (EFI_FFS_FILE_TAIL) (~(FfsFile->IntegrityCheck.TailReference));\r
      *(EFI_FFS_FILE_TAIL *) (((UINTN) FfsFile + GetLength (FfsFile->Size) - sizeof (EFI_FFS_FILE_TAIL))) = TailValue;\r
    }\r
\r
    fprintf (\r
      LogFile,\r
      "%s %016I64X %s\n",\r
      FileGuidString,\r
      ImageContext.DestinationAddress,\r
      ImageContext.PdbPointer == NULL ? "*" : ImageContext.PdbPointer\r
      );\r
\r
    //\r
    // Free buffers\r
    //\r
    free ((VOID *) MemoryImagePointer);\r
    free (TEBuffer);\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
EFI_STATUS\r
FfsRebaseImageRead (\r
  IN     VOID    *FileHandle,\r
  IN     UINTN   FileOffset,\r
  IN OUT UINT32  *ReadSize,\r
  OUT    VOID    *Buffer\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Support routine for the PE/COFF Loader that reads a buffer from a PE/COFF file\r
\r
Arguments:\r
\r
  FileHandle - The handle to the PE/COFF file\r
\r
  FileOffset - The offset, in bytes, into the file to read\r
\r
  ReadSize   - The number of bytes to read from the file starting at FileOffset\r
\r
  Buffer     - A pointer to the buffer to read the data into.\r
\r
Returns:\r
\r
  EFI_SUCCESS - ReadSize bytes of data were read into Buffer from the PE/COFF file starting at FileOffset\r
\r
--*/\r
{\r
  CHAR8   *Destination8;\r
  CHAR8   *Source8;\r
  UINT32  Length;\r
\r
  Destination8  = Buffer;\r
  Source8       = (CHAR8 *) ((UINTN) FileHandle + FileOffset);\r
  Length        = *ReadSize;\r
  while (Length--) {\r
    *(Destination8++) = *(Source8++);\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r