[mirror_edk2.git] / BaseTools / Source / C / VolInfo / VolInfo.c

/** @file\r
The tool dumps the contents of a firmware volume\r
\r
Copyright (c) 1999 - 2016, Intel Corporation. All rights reserved.<BR>\r
This program and the accompanying materials\r
are licensed and made available under the terms and conditions of the BSD License\r
which accompanies this distribution.  The full text of the license may be found at\r
http://opensource.org/licenses/bsd-license.php\r
\r
THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
\r
**/\r
\r
#include <stdio.h>\r
#include <stdlib.h>\r
#include <string.h>\r
#include <ctype.h>\r
#include <assert.h>\r
#ifdef __GNUC__\r
#include <unistd.h>\r
#else\r
#include <direct.h>\r
#endif\r
\r
#include <FvLib.h>\r
#include <Common/UefiBaseTypes.h>\r
#include <Common/UefiCapsule.h>\r
#include <Common/PiFirmwareFile.h>\r
#include <Common/PiFirmwareVolume.h>\r
#include <Guid/PiFirmwareFileSystem.h>\r
#include <IndustryStandard/PeImage.h>\r
#include <Protocol/GuidedSectionExtraction.h>\r
\r
#include "Compress.h"\r
#include "Decompress.h"\r
#include "VolInfo.h"\r
#include "CommonLib.h"\r
#include "EfiUtilityMsgs.h"\r
#include "FirmwareVolumeBufferLib.h"\r
#include "OsPath.h"\r
#include "ParseGuidedSectionTools.h"\r
#include "StringFuncs.h"\r
#include "ParseInf.h"\r
#include "PeCoffLib.h"\r
\r
//\r
// Utility global variables\r
//\r
\r
EFI_GUID  gEfiCrc32GuidedSectionExtractionProtocolGuid = EFI_CRC32_GUIDED_SECTION_EXTRACTION_PROTOCOL_GUID;\r
\r
#define UTILITY_MAJOR_VERSION      1\r
#define UTILITY_MINOR_VERSION      0\r
\r
#define UTILITY_NAME         "VolInfo"\r
\r
#define EFI_SECTION_ERROR EFIERR (100)\r
\r
#define MAX_BASENAME_LEN  60  // not good to hardcode, but let's be reasonable\r
\r
//\r
// Structure to keep a list of guid-to-basenames\r
//\r
typedef struct _GUID_TO_BASENAME {\r
  struct _GUID_TO_BASENAME  *Next;\r
  INT8                      Guid[PRINTED_GUID_BUFFER_SIZE];\r
  INT8                      BaseName[MAX_BASENAME_LEN];\r
} GUID_TO_BASENAME;\r
\r
static GUID_TO_BASENAME *mGuidBaseNameList = NULL;\r
\r
//\r
// Store GUIDed Section guid->tool mapping\r
//\r
EFI_HANDLE mParsedGuidedSectionTools = NULL;\r
\r
CHAR8* mUtilityFilename = NULL;\r
\r
BOOLEAN EnableHash = FALSE;\r
CHAR8 *OpenSslPath = NULL;\r
\r
EFI_STATUS\r
ParseGuidBaseNameFile (\r
  CHAR8    *FileName\r
  );\r
\r
EFI_STATUS\r
FreeGuidBaseNameList (\r
  VOID\r
  );\r
\r
EFI_STATUS\r
PrintGuidName (\r
  IN UINT8    *GuidStr\r
  );\r
\r
EFI_STATUS\r
ParseSection (\r
  IN UINT8  *SectionBuffer,\r
  IN UINT32 BufferLength\r
  );\r
\r
EFI_STATUS\r
DumpDepexSection (\r
  IN UINT8    *Ptr,\r
  IN UINT32   SectionLength\r
  );\r
\r
STATIC\r
EFI_STATUS\r
ReadHeader (\r
  IN FILE       *InputFile,\r
  OUT UINT32    *FvSize,\r
  OUT BOOLEAN   *ErasePolarity\r
  );\r
\r
STATIC\r
EFI_STATUS\r
PrintFileInfo (\r
  EFI_FIRMWARE_VOLUME_HEADER  *FvImage,\r
  EFI_FFS_FILE_HEADER         *FileHeader,\r
  BOOLEAN                     ErasePolarity\r
  );\r
\r
static\r
EFI_STATUS\r
PrintFvInfo (\r
  IN VOID                         *Fv,\r
  IN BOOLEAN                      IsChildFv\r
  );\r
\r
static\r
VOID\r
LoadGuidedSectionToolsTxt (\r
  IN CHAR8* FirmwareVolumeFilename\r
  );\r
\r
EFI_STATUS\r
CombinePath (\r
  IN  CHAR8* DefaultPath,\r
  IN  CHAR8* AppendPath,\r
  OUT CHAR8* NewPath\r
);\r
\r
void\r
Usage (\r
  VOID\r
  );\r
\r
int\r
main (\r
  int       argc,\r
  char      *argv[]\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  GC_TODO: Add function description\r
\r
Arguments:\r
\r
  argc  - GC_TODO: add argument description\r
  ]     - GC_TODO: add argument description\r
\r
Returns:\r
\r
  GC_TODO: add return values\r
\r
--*/\r
{\r
  FILE                        *InputFile;\r
  int                         BytesRead;\r
  EFI_FIRMWARE_VOLUME_HEADER  *FvImage;\r
  UINT32                      FvSize;\r
  EFI_STATUS                  Status;\r
  int                         Offset;\r
  BOOLEAN                     ErasePolarity;\r
  UINT64                      LogLevel;\r
  CHAR8                       *OpenSslEnv;\r
  CHAR8                       *OpenSslCommand;\r
\r
  SetUtilityName (UTILITY_NAME);\r
  //\r
  // Print utility header\r
  //\r
  printf ("%s Version %d.%d Build %s\n",\r
    UTILITY_NAME,\r
    UTILITY_MAJOR_VERSION,\r
    UTILITY_MINOR_VERSION,\r
    __BUILD_VERSION\r
    );\r
\r
  if (argc == 1) {\r
    Usage ();\r
    return -1;\r
  }\r
\r
  argc--;\r
  argv++;\r
  LogLevel = 0;\r
  Offset = 0;\r
\r
  //\r
  // Look for help options\r
  //\r
  if ((strcmp(argv[0], "-h") == 0) || (strcmp(argv[0], "--help") == 0) ||\r
      (strcmp(argv[0], "-?") == 0) || (strcmp(argv[0], "/?") == 0)) {\r
    Usage();\r
    return  STATUS_SUCCESS;\r
  }\r
  //\r
  // Version has already be printed, so just return success\r
  //\r
  if (strcmp(argv[0], "--version") == 0) {\r
    return  STATUS_SUCCESS;\r
  }\r
\r
  //\r
  // If they specified -x xref guid/basename cross-reference files, process it.\r
  // This will print the basename beside each file guid. To use it, specify\r
  // -x xref_filename to processdsc, then use xref_filename as a parameter\r
  // here.\r
  //\r
  while (argc > 0) {\r
    if ((strcmp(argv[0], "-x") == 0) || (strcmp(argv[0], "--xref") == 0)) {\r
      ParseGuidBaseNameFile (argv[1]);\r
      printf("ParseGuidBaseNameFile: %s\n", argv[1]);\r
      argc -= 2;\r
      argv += 2;\r
      continue;\r
    }\r
    if (strcmp(argv[0], "--offset") == 0) {\r
      //\r
      // Hex or decimal?\r
      //\r
      if ((argv[1][0] == '0') && (tolower ((int)argv[1][1]) == 'x')) {\r
        if (sscanf (argv[1], "%x", &Offset) != 1) {\r
          Error (NULL, 0, 1003, "Invalid option value", "Offset = %s", argv[1]);\r
          return GetUtilityStatus ();\r
        }\r
      } else {\r
        if (sscanf (argv[1], "%d", &Offset) != 1) {\r
          Error (NULL, 0, 1003, "Invalid option value", "Offset = %s", argv[1]);\r
          return GetUtilityStatus ();\r
        }\r
        //\r
        // See if they said something like "64K"\r
        //\r
        if (tolower ((int)argv[1][strlen (argv[1]) - 1]) == 'k') {\r
          Offset *= 1024;\r
        }\r
      }\r
\r
      argc -= 2;\r
      argv += 2;\r
      continue;\r
    }\r
    if ((stricmp (argv[0], "--hash") == 0)) {\r
      if (EnableHash == TRUE) {\r
        //\r
        // --hash already given in the option, ignore this one\r
        //\r
        argc --;\r
        argv ++;\r
        continue;\r
      }\r
      EnableHash = TRUE;\r
      OpenSslCommand = "openssl";\r
      OpenSslEnv = getenv("OPENSSL_PATH");\r
      if (OpenSslEnv == NULL) {\r
        OpenSslPath = OpenSslCommand;\r
      } else {\r
        OpenSslPath = malloc(strlen(OpenSslEnv)+strlen(OpenSslCommand)+1);\r
        if (OpenSslPath == NULL) {\r
          Error (NULL, 0, 4001, "Resource", "memory cannot be allocated!");\r
          return GetUtilityStatus ();\r
        }\r
        CombinePath(OpenSslEnv, OpenSslCommand, OpenSslPath);\r
      }\r
      if (OpenSslPath == NULL){\r
        Error (NULL, 0, 3000, "Open SSL command not available.  Please verify PATH or set OPENSSL_PATH.", NULL);\r
        return GetUtilityStatus ();\r
      }\r
      argc --;\r
      argv ++;\r
      continue;\r
    }\r
\r
    if ((stricmp (argv[0], "-v") == 0) || (stricmp (argv[0], "--verbose") == 0)) {\r
      SetPrintLevel (VERBOSE_LOG_LEVEL);\r
      argc --;\r
      argv ++;\r
      continue;\r
    }\r
\r
    if ((stricmp (argv[0], "-q") == 0) || (stricmp (argv[0], "--quiet") == 0)) {\r
      SetPrintLevel (KEY_LOG_LEVEL);\r
      argc --;\r
      argv ++;\r
      continue;\r
    }\r
\r
    if ((stricmp (argv[0], "-d") == 0) || (stricmp (argv[0], "--debug") == 0)) {\r
      Status = AsciiStringToUint64 (argv[1], FALSE, &LogLevel);\r
      if (EFI_ERROR (Status)) {\r
        Error (NULL, 0, 1003, "Invalid option value", "%s = %s", argv[0], argv[1]);\r
        return -1;\r
      }\r
      if (LogLevel > 9) {\r
        Error (NULL, 0, 1003, "Invalid option value", "Debug Level range is 0-9, current input level is %d", (int) LogLevel);\r
        return -1;\r
      }\r
      SetPrintLevel (LogLevel);\r
      DebugMsg (NULL, 0, 9, "Debug Mode Set", "Debug Output Mode Level %s is set!", argv[1]);\r
      argc -= 2;\r
      argv += 2;\r
      continue;\r
    }\r
\r
    mUtilityFilename = argv[0];\r
    argc --;\r
    argv ++;\r
  }\r
\r
  //\r
  // Open the file containing the FV\r
  //\r
  if (mUtilityFilename == NULL) {\r
    Error (NULL, 0, 1001, "Missing option", "Input files are not specified");\r
    return GetUtilityStatus ();\r
  }\r
  InputFile = fopen (LongFilePath (mUtilityFilename), "rb");\r
  if (InputFile == NULL) {\r
    Error (NULL, 0, 0001, "Error opening the input file", mUtilityFilename);\r
    return GetUtilityStatus ();\r
  }\r
  //\r
  // Skip over pad bytes if specified. This is used if they prepend 0xff\r
  // data to the FV image binary.\r
  //\r
  if (Offset != 0) {\r
    fseek (InputFile, Offset, SEEK_SET);\r
  }\r
  //\r
  // Determine size of FV\r
  //\r
  Status = ReadHeader (InputFile, &FvSize, &ErasePolarity);\r
  if (EFI_ERROR (Status)) {\r
    Error (NULL, 0, 0003, "error parsing FV image", "%s Header is invalid", mUtilityFilename);\r
    fclose (InputFile);\r
    return GetUtilityStatus ();\r
  }\r
  //\r
  // Allocate a buffer for the FV image\r
  //\r
  FvImage = malloc (FvSize);\r
  if (FvImage == NULL) {\r
    Error (NULL, 0, 4001, "Resource: Memory can't be allocated", NULL);\r
    fclose (InputFile);\r
    return GetUtilityStatus ();\r
  }\r
  //\r
  // Seek to the start of the image, then read the entire FV to the buffer\r
  //\r
  fseek (InputFile, Offset, SEEK_SET);\r
  BytesRead = fread (FvImage, 1, FvSize, InputFile);\r
  fclose (InputFile);\r
  if ((unsigned int) BytesRead != FvSize) {\r
    Error (NULL, 0, 0004, "error reading FvImage from", mUtilityFilename);\r
    free (FvImage);\r
    return GetUtilityStatus ();\r
  }\r
\r
  LoadGuidedSectionToolsTxt (mUtilityFilename);\r
\r
  PrintFvInfo (FvImage, FALSE);\r
\r
  //\r
  // Clean up\r
  //\r
  free (FvImage);\r
  FreeGuidBaseNameList ();\r
  return GetUtilityStatus ();\r
}\r
\r
\r
static\r
EFI_STATUS\r
PrintFvInfo (\r
  IN VOID                         *Fv,\r
  IN BOOLEAN                      IsChildFv\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  GC_TODO: Add function description\r
\r
Arguments:\r
\r
  Fv            - Firmware Volume to print information about\r
  IsChildFv     - Flag specifies whether the input FV is a child FV.\r
\r
Returns:\r
\r
  EFI_STATUS\r
\r
--*/\r
{\r
  EFI_STATUS                  Status;\r
  UINTN                       NumberOfFiles;\r
  BOOLEAN                     ErasePolarity;\r
  UINTN                       FvSize;\r
  EFI_FFS_FILE_HEADER         *CurrentFile;\r
  UINTN                       Key;\r
\r
  Status = FvBufGetSize (Fv, &FvSize);\r
\r
  NumberOfFiles = 0;\r
  ErasePolarity =\r
    (((EFI_FIRMWARE_VOLUME_HEADER*)Fv)->Attributes & EFI_FVB2_ERASE_POLARITY) ?\r
      TRUE : FALSE;\r
\r
  //\r
  // Get the first file\r
  //\r
  Key = 0;\r
  Status = FvBufFindNextFile (Fv, &Key, (VOID **) &CurrentFile);\r
  if (EFI_ERROR (Status)) {\r
    Error (NULL, 0, 0003, "error parsing FV image", "cannot find the first file in the FV image");\r
    return GetUtilityStatus ();\r
  }\r
  //\r
  // Display information about files found\r
  //\r
  while (CurrentFile != NULL) {\r
    //\r
    // Increment the number of files counter\r
    //\r
    NumberOfFiles++;\r
\r
    //\r
    // Display info about this file\r
    //\r
    Status = PrintFileInfo (Fv, CurrentFile, ErasePolarity);\r
    if (EFI_ERROR (Status)) {\r
      Error (NULL, 0, 0003, "error parsing FV image", "failed to parse a file in the FV");\r
      return GetUtilityStatus ();\r
    }\r
    //\r
    // Get the next file\r
    //\r
    Status = FvBufFindNextFile (Fv, &Key, (VOID **) &CurrentFile);\r
    if (Status == EFI_NOT_FOUND) {\r
      CurrentFile = NULL;\r
    } else if (EFI_ERROR (Status)) {\r
      Error (NULL, 0, 0003, "error parsing FV image", "cannot find the next file in the FV image");\r
      return GetUtilityStatus ();\r
    }\r
  }\r
\r
  if (IsChildFv) {\r
    printf ("There are a total of %d files in the child FV\n", (int) NumberOfFiles);\r
  } else {\r
    printf ("There are a total of %d files in this FV\n", (int) NumberOfFiles);\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
UINT32\r
GetOccupiedSize (\r
  IN UINT32  ActualSize,\r
  IN UINT32  Alignment\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  This function returns the next larger size that meets the alignment \r
  requirement specified.\r
\r
Arguments:\r
\r
  ActualSize      The size.\r
  Alignment       The desired alignment.\r
    \r
Returns:\r
 \r
  EFI_SUCCESS             Function completed successfully.\r
  EFI_ABORTED             The function encountered an error.\r
\r
--*/\r
{\r
  UINT32  OccupiedSize;\r
\r
  OccupiedSize = ActualSize;\r
  while ((OccupiedSize & (Alignment - 1)) != 0) {\r
    OccupiedSize++;\r
  }\r
\r
  return OccupiedSize;\r
}\r
\r
static\r
CHAR8 *\r
SectionNameToStr (\r
  IN EFI_SECTION_TYPE   Type\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Converts EFI Section names to Strings\r
\r
Arguments:\r
\r
  Type  - The EFI Section type\r
\r
Returns:\r
\r
  CHAR8* - Pointer to the String containing the section name.\r
\r
--*/\r
{\r
  CHAR8 *SectionStr;\r
  CHAR8 *SectionTypeStringTable[] = {\r
    //\r
    // 0X00\r
    //\r
    "EFI_SECTION_ALL",\r
    //\r
    // 0x01\r
    //\r
    "EFI_SECTION_COMPRESSION",\r
    //\r
    // 0x02\r
    //\r
    "EFI_SECTION_GUID_DEFINED",    \r
    //\r
    // 0x03\r
    //\r
    "Unknown section type - Reserved 0x03",\r
    //\r
    // 0x04\r
    //\r
    "Unknown section type - Reserved 0x04",\r
    //\r
    // 0x05\r
    //\r
    "Unknown section type - Reserved 0x05",\r
    //\r
    // 0x06\r
    //\r
    "Unknown section type - Reserved 0x06",\r
    //\r
    // 0x07\r
    //\r
    "Unknown section type - Reserved 0x07",\r
    //\r
    // 0x08\r
    //\r
    "Unknown section type - Reserved 0x08",\r
    //\r
    // 0x09\r
    //\r
    "Unknown section type - Reserved 0x09",\r
    //\r
    // 0x0A\r
    //\r
    "Unknown section type - Reserved 0x0A",\r
    //\r
    // 0x0B\r
    //\r
    "Unknown section type - Reserved 0x0B",\r
    //\r
    // 0x0C\r
    //\r
    "Unknown section type - Reserved 0x0C",\r
    //\r
    // 0x0D\r
    //\r
    "Unknown section type - Reserved 0x0D",\r
    //\r
    // 0x0E\r
    //\r
    "Unknown section type - Reserved 0x0E",\r
    //\r
    // 0x0F\r
    //\r
    "Unknown section type - Reserved 0x0E",\r
    //\r
    // 0x10\r
    //\r
    "EFI_SECTION_PE32",\r
    //\r
    // 0x11\r
    //\r
    "EFI_SECTION_PIC",\r
    //\r
    // 0x12\r
    //\r
    "EFI_SECTION_TE",    \r
    //\r
    // 0x13\r
    //\r
    "EFI_SECTION_DXE_DEPEX", \r
    //\r
    // 0x14\r
    //\r
    "EFI_SECTION_VERSION",\r
    //\r
    // 0x15\r
    //\r
    "EFI_SECTION_USER_INTERFACE",\r
    //\r
    // 0x16\r
    //\r
    "EFI_SECTION_COMPATIBILITY16",\r
    //\r
    // 0x17\r
    //\r
    "EFI_SECTION_FIRMWARE_VOLUME_IMAGE ",\r
    //\r
    // 0x18\r
    //\r
    "EFI_SECTION_FREEFORM_SUBTYPE_GUID ",\r
    //\r
    // 0x19\r
    //\r
    "EFI_SECTION_RAW",\r
    //\r
    // 0x1A\r
    //\r
    "Unknown section type - 0x1A",\r
    //\r
    // 0x1B\r
    //\r
    "EFI_SECTION_PEI_DEPEX",\r
    //\r
    // 0x1C\r
    //\r
    "EFI_SECTION_SMM_DEPEX",\r
    //\r
    // 0x1C+\r
    //\r
    "Unknown section type - Reserved - beyond last defined section"\r
  };\r
\r
  if (Type > EFI_SECTION_LAST_SECTION_TYPE) {\r
    Type = EFI_SECTION_LAST_SECTION_TYPE + 1;\r
  }\r
\r
  SectionStr = malloc (100);\r
  if (SectionStr == NULL) {\r
    printf ("Error: Out of memory resources.\n");\r
    return SectionStr;\r
  }\r
  strcpy (SectionStr, SectionTypeStringTable[Type]);\r
  return SectionStr;\r
}\r
\r
STATIC\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 (__FILE__, __LINE__, 0, "application error", "invalid parameter to function");\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
  // Print FV header information\r
  //\r
  printf ("Signature:        %s (%X)\n", (char *) Signature, (unsigned) VolumeHeader.Signature);\r
  printf ("Attributes:       %X\n", (unsigned) VolumeHeader.Attributes);\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_READ_DISABLED_CAP) {\r
    printf ("       EFI_FVB2_READ_DISABLED_CAP\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_READ_ENABLED_CAP) {\r
    printf ("       EFI_FVB2_READ_ENABLED_CAP\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_READ_STATUS) {\r
    printf ("       EFI_FVB2_READ_STATUS\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_WRITE_DISABLED_CAP) {\r
    printf ("       EFI_FVB2_WRITE_DISABLED_CAP\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_WRITE_ENABLED_CAP) {\r
    printf ("       EFI_FVB2_WRITE_ENABLED_CAP\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_WRITE_STATUS) {\r
    printf ("       EFI_FVB2_WRITE_STATUS\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_LOCK_CAP) {\r
    printf ("       EFI_FVB2_LOCK_CAP\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_LOCK_STATUS) {\r
    printf ("       EFI_FVB2_LOCK_STATUS\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_STICKY_WRITE) {\r
    printf ("       EFI_FVB2_STICKY_WRITE\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_MEMORY_MAPPED) {\r
    printf ("       EFI_FVB2_MEMORY_MAPPED\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ERASE_POLARITY) {\r
    printf ("       EFI_FVB2_ERASE_POLARITY\n");\r
    *ErasePolarity = TRUE;\r
  }\r
\r
#if (PI_SPECIFICATION_VERSION < 0x00010000)\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT) {\r
    printf ("       EFI_FVB2_ALIGNMENT\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_2) {\r
    printf ("       EFI_FVB2_ALIGNMENT_2\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_4) {\r
    printf ("       EFI_FVB2_ALIGNMENT_4\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_8) {\r
    printf ("       EFI_FVB2_ALIGNMENT_8\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_16) {\r
    printf ("       EFI_FVB2_ALIGNMENT_16\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_32) {\r
    printf ("       EFI_FVB2_ALIGNMENT_32\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_64) {\r
    printf ("        EFI_FVB2_ALIGNMENT_64\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_128) {\r
    printf ("        EFI_FVB2_ALIGNMENT_128\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_256) {\r
    printf ("        EFI_FVB2_ALIGNMENT_256\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_512) {\r
    printf ("        EFI_FVB2_ALIGNMENT_512\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_1K) {\r
    printf ("        EFI_FVB2_ALIGNMENT_1K\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_2K) {\r
    printf ("        EFI_FVB2_ALIGNMENT_2K\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_4K) {\r
    printf ("        EFI_FVB2_ALIGNMENT_4K\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_8K) {\r
    printf ("        EFI_FVB2_ALIGNMENT_8K\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_16K) {\r
    printf ("        EFI_FVB2_ALIGNMENT_16K\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_32K) {\r
    printf ("        EFI_FVB2_ALIGNMENT_32K\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_64K) {\r
    printf ("        EFI_FVB2_ALIGNMENT_64K\n");\r
  }\r
  \r
#else\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_READ_LOCK_CAP) {\r
    printf ("       EFI_FVB2_READ_LOCK_CAP\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_READ_LOCK_STATUS) {\r
    printf ("       EFI_FVB2_READ_LOCK_STATUS\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_WRITE_LOCK_CAP) {\r
    printf ("       EFI_FVB2_WRITE_LOCK_CAP\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_WRITE_LOCK_STATUS) {\r
    printf ("       EFI_FVB2_WRITE_LOCK_STATUS\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_1) {\r
    printf ("       EFI_FVB2_ALIGNMENT_1\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_2) {\r
    printf ("        EFI_FVB2_ALIGNMENT_2\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_4) {\r
    printf ("        EFI_FVB2_ALIGNMENT_4\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_8) {\r
    printf ("        EFI_FVB2_ALIGNMENT_8\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_16) {\r
    printf ("        EFI_FVB2_ALIGNMENT_16\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_32) {\r
    printf ("        EFI_FVB2_ALIGNMENT_32\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_64) {\r
    printf ("        EFI_FVB2_ALIGNMENT_64\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_128) {\r
    printf ("        EFI_FVB2_ALIGNMENT_128\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_256) {\r
    printf ("        EFI_FVB2_ALIGNMENT_256\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_512) {\r
    printf ("        EFI_FVB2_ALIGNMENT_512\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_1K) {\r
    printf ("        EFI_FVB2_ALIGNMENT_1K\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_2K) {\r
    printf ("        EFI_FVB2_ALIGNMENT_2K\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_4K) {\r
    printf ("        EFI_FVB2_ALIGNMENT_4K\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_8K) {\r
    printf ("        EFI_FVB2_ALIGNMENT_8K\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_16K) {\r
    printf ("        EFI_FVB2_ALIGNMENT_16K\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_32K) {\r
    printf ("        EFI_FVB2_ALIGNMENT_32K\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_64K) {\r
    printf ("        EFI_FVB2_ALIGNMENT_64K\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_128K) {\r
    printf ("        EFI_FVB2_ALIGNMENT_128K\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_256K) {\r
    printf ("        EFI_FVB2_ALIGNMENT_256K\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_512K) {\r
    printf ("        EFI_FVB2_ALIGNMENT_512K\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_1M) {\r
    printf ("        EFI_FVB2_ALIGNMENT_1M\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_2M) {\r
    printf ("        EFI_FVB2_ALIGNMENT_2M\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_4M) {\r
    printf ("        EFI_FVB2_ALIGNMENT_4M\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_8M) {\r
    printf ("        EFI_FVB2_ALIGNMENT_8M\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_16M) {\r
    printf ("        EFI_FVB2_ALIGNMENT_16M\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_32M) {\r
    printf ("        EFI_FVB2_ALIGNMENT_32M\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_64M) {\r
    printf ("        EFI_FVB2_ALIGNMENT_64M\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_128M) {\r
    printf ("        EFI_FVB2_ALIGNMENT_128M\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_64M) {\r
    printf ("        EFI_FVB2_ALIGNMENT_64M\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_128M) {\r
    printf ("        EFI_FVB2_ALIGNMENT_128M\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_256M) {\r
    printf ("        EFI_FVB2_ALIGNMENT_256M\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_512M) {\r
    printf ("        EFI_FVB2_ALIGNMENT_512M\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_1G) {\r
    printf ("        EFI_FVB2_ALIGNMENT_1G\n");\r
  }\r
\r
  if (VolumeHeader.Attributes & EFI_FVB2_ALIGNMENT_2G) {\r
    printf ("        EFI_FVB2_ALIGNMENT_2G\n");\r
  }\r
\r
#endif\r
  printf ("Header Length:         0x%08X\n", VolumeHeader.HeaderLength);\r
  printf ("File System ID:        ");\r
  PrintGuid (&VolumeHeader.FileSystemGuid);\r
  //\r
  // printf ("\n");\r
  //\r
  printf ("Revision:              0x%04X\n", VolumeHeader.Revision);\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
      printf ("Number of Blocks:      0x%08X\n", (unsigned) BlockMap.NumBlocks);\r
      printf ("Block Length:          0x%08X\n", (unsigned) BlockMap.Length);\r
      Size += BlockMap.NumBlocks * BlockMap.Length;\r
    }\r
\r
  } while (!(BlockMap.NumBlocks == 0 && BlockMap.Length == 0));\r
\r
  if (BytesRead != VolumeHeader.HeaderLength) {\r
    printf ("ERROR: Header length not consistent with Block Maps!\n");\r
    return EFI_ABORTED;\r
  }\r
\r
  if (VolumeHeader.FvLength != Size) {\r
    printf ("ERROR: Volume Size not consistant with Block Maps!\n");\r
    return EFI_ABORTED;\r
  }\r
\r
  printf ("Total Volume Size:     0x%08X\n", (unsigned) Size);\r
\r
  *FvSize = Size;\r
\r
  //\r
  // rewind (InputFile);\r
  //\r
  return EFI_SUCCESS;\r
}\r
\r
STATIC\r
EFI_STATUS\r
PrintFileInfo (\r
  EFI_FIRMWARE_VOLUME_HEADER  *FvImage,\r
  EFI_FFS_FILE_HEADER         *FileHeader,\r
  BOOLEAN                     ErasePolarity\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  GC_TODO: Add function description\r
\r
Arguments:\r
\r
  FvImage       - GC_TODO: add argument description\r
  FileHeader    - GC_TODO: add argument description\r
  ErasePolarity - GC_TODO: add argument description\r
\r
Returns:\r
\r
  EFI_SUCCESS - GC_TODO: Add description for return value\r
  EFI_ABORTED - GC_TODO: Add description for return value\r
\r
--*/\r
{\r
  UINT32              FileLength;\r
  UINT8               FileState;\r
  UINT8               Checksum;\r
  EFI_FFS_FILE_HEADER2 BlankHeader;\r
  EFI_STATUS          Status;\r
  UINT8               GuidBuffer[PRINTED_GUID_BUFFER_SIZE];\r
  UINT32              HeaderSize;\r
#if (PI_SPECIFICATION_VERSION < 0x00010000) \r
  UINT16              *Tail;\r
#endif\r
  //\r
  // Check if we have free space\r
  //\r
  HeaderSize = FvBufGetFfsHeaderSize(FileHeader);\r
  if (ErasePolarity) {\r
    memset (&BlankHeader, -1, HeaderSize);\r
  } else {\r
    memset (&BlankHeader, 0, HeaderSize);\r
  }\r
\r
  if (memcmp (&BlankHeader, FileHeader, HeaderSize) == 0) {\r
    return EFI_SUCCESS;\r
  }\r
  //\r
  // Print file information.\r
  //\r
  printf ("============================================================\n");\r
\r
  printf ("File Name:        ");\r
  PrintGuidToBuffer (&FileHeader->Name, GuidBuffer, sizeof (GuidBuffer), TRUE);\r
  printf ("%s  ", GuidBuffer);\r
  PrintGuidName (GuidBuffer);\r
  printf ("\n");\r
\r
  //\r
  //  PrintGuid (&FileHeader->Name);\r
  //  printf ("\n");\r
  //\r
  FileLength = FvBufGetFfsFileSize (FileHeader);\r
  printf ("File Offset:      0x%08X\n", (unsigned) ((UINTN) FileHeader - (UINTN) FvImage));\r
  printf ("File Length:      0x%08X\n", (unsigned) FileLength);\r
  printf ("File Attributes:  0x%02X\n", FileHeader->Attributes);\r
  printf ("File State:       0x%02X\n", FileHeader->State);\r
\r
  //\r
  // Print file state\r
  //\r
  FileState = GetFileState (ErasePolarity, FileHeader);\r
\r
  switch (FileState) {\r
\r
  case EFI_FILE_HEADER_CONSTRUCTION:\r
    printf ("        EFI_FILE_HEADER_CONSTRUCTION\n");\r
    return EFI_SUCCESS;\r
\r
  case EFI_FILE_HEADER_INVALID:\r
    printf ("        EFI_FILE_HEADER_INVALID\n");\r
    return EFI_SUCCESS;\r
\r
  case EFI_FILE_HEADER_VALID:\r
    printf ("        EFI_FILE_HEADER_VALID\n");\r
    Checksum  = CalculateSum8 ((UINT8 *) FileHeader, HeaderSize);\r
    Checksum  = (UINT8) (Checksum - FileHeader->IntegrityCheck.Checksum.File);\r
    Checksum  = (UINT8) (Checksum - FileHeader->State);\r
    if (Checksum != 0) {\r
      printf ("ERROR: Header checksum invalid.\n");\r
      return EFI_ABORTED;\r
    }\r
\r
    return EFI_SUCCESS;\r
\r
  case EFI_FILE_DELETED:\r
    printf ("        EFI_FILE_DELETED\n");\r
\r
  case EFI_FILE_MARKED_FOR_UPDATE:\r
    printf ("        EFI_FILE_MARKED_FOR_UPDATE\n");\r
\r
  case EFI_FILE_DATA_VALID:\r
    printf ("        EFI_FILE_DATA_VALID\n");\r
\r
    //\r
    // Calculate header checksum\r
    //\r
    Checksum  = CalculateSum8 ((UINT8 *) FileHeader, HeaderSize);\r
    Checksum  = (UINT8) (Checksum - FileHeader->IntegrityCheck.Checksum.File);\r
    Checksum  = (UINT8) (Checksum - FileHeader->State);\r
    if (Checksum != 0) {\r
      Error (NULL, 0, 0003, "error parsing FFS file", "FFS file with Guid %s has invalid header checksum", GuidBuffer);\r
      return EFI_ABORTED;\r
    }\r
\r
    FileLength = FvBufGetFfsFileSize (FileHeader);\r
\r
    if (FileHeader->Attributes & FFS_ATTRIB_CHECKSUM) {\r
      //\r
      // Calculate file checksum\r
      //\r
      Checksum  = CalculateSum8 ((UINT8 *)FileHeader + HeaderSize, FileLength - HeaderSize);\r
      Checksum  = Checksum + FileHeader->IntegrityCheck.Checksum.File;\r
      if (Checksum != 0) {\r
        Error (NULL, 0, 0003, "error parsing FFS file", "FFS file with Guid %s has invalid file checksum", GuidBuffer);\r
        return EFI_ABORTED;\r
      }\r
    } else {\r
      if (FileHeader->IntegrityCheck.Checksum.File != FFS_FIXED_CHECKSUM) {\r
        Error (NULL, 0, 0003, "error parsing FFS file", "FFS file with Guid %s has invalid header checksum -- not set to fixed value of 0xAA", GuidBuffer);\r
        return EFI_ABORTED;\r
      }\r
    }\r
#if (PI_SPECIFICATION_VERSION < 0x00010000)    \r
    //\r
    // Verify tail if present\r
    //\r
    if (FileHeader->Attributes & FFS_ATTRIB_TAIL_PRESENT) {\r
      //\r
      // Verify tail is complement of integrity check field in the header.\r
      //\r
      Tail = (UINT16 *) ((UINTN) FileHeader + GetLength (FileHeader->Size) - sizeof (EFI_FFS_INTEGRITY_CHECK));\r
      if (FileHeader->IntegrityCheck.TailReference != (UINT16)~(*Tail)) {\r
        Error (NULL, 0, 0003, "error parsing FFS file", \\r
        "FFS file with Guid %s failed in the integrity check, tail is not the complement of the header field", GuidBuffer);\r
        return EFI_ABORTED;\r
      }\r
    }\r
 #endif   \r
    break;\r
\r
  default:\r
    Error (NULL, 0, 0003, "error parsing FFS file", "FFS file with Guid %s has the invalid/unrecognized file state bits", GuidBuffer);\r
    return EFI_ABORTED;\r
  }\r
\r
  printf ("File Type:        0x%02X  ", FileHeader->Type);\r
\r
  switch (FileHeader->Type) {\r
\r
  case EFI_FV_FILETYPE_RAW:\r
    printf ("EFI_FV_FILETYPE_RAW\n");\r
    break;\r
\r
  case EFI_FV_FILETYPE_FREEFORM:\r
    printf ("EFI_FV_FILETYPE_FREEFORM\n");\r
    break;\r
\r
  case EFI_FV_FILETYPE_SECURITY_CORE:\r
    printf ("EFI_FV_FILETYPE_SECURITY_CORE\n");\r
    break;\r
\r
  case EFI_FV_FILETYPE_PEI_CORE:\r
    printf ("EFI_FV_FILETYPE_PEI_CORE\n");\r
    break;\r
\r
  case EFI_FV_FILETYPE_DXE_CORE:\r
    printf ("EFI_FV_FILETYPE_DXE_CORE\n");\r
    break;\r
\r
  case EFI_FV_FILETYPE_PEIM:\r
    printf ("EFI_FV_FILETYPE_PEIM\n");\r
    break;\r
\r
  case EFI_FV_FILETYPE_DRIVER:\r
    printf ("EFI_FV_FILETYPE_DRIVER\n");\r
    break;\r
\r
  case EFI_FV_FILETYPE_COMBINED_PEIM_DRIVER:\r
    printf ("EFI_FV_FILETYPE_COMBINED_PEIM_DRIVER\n");\r
    break;\r
\r
  case EFI_FV_FILETYPE_APPLICATION:\r
    printf ("EFI_FV_FILETYPE_APPLICATION\n");\r
    break;\r
\r
  case EFI_FV_FILETYPE_SMM:\r
    printf ("EFI_FV_FILETYPE_SMM\n");\r
    break;\r
\r
  case EFI_FV_FILETYPE_FIRMWARE_VOLUME_IMAGE:\r
    printf ("EFI_FV_FILETYPE_FIRMWARE_VOLUME_IMAGE\n");\r
    break;\r
\r
  case EFI_FV_FILETYPE_COMBINED_SMM_DXE:\r
    printf ("EFI_FV_FILETYPE_COMBINED_SMM_DXE\n");\r
    break;\r
\r
  case EFI_FV_FILETYPE_SMM_CORE:\r
    printf ("EFI_FV_FILETYPE_SMM_CORE\n");\r
    break;\r
\r
  case EFI_FV_FILETYPE_FFS_PAD:\r
    printf ("EFI_FV_FILETYPE_FFS_PAD\n");\r
    break;\r
\r
  default:\r
    printf ("\nERROR: Unrecognized file type %X.\n", FileHeader->Type);\r
    return EFI_ABORTED;\r
    break;\r
  }\r
\r
  switch (FileHeader->Type) {\r
\r
  case EFI_FV_FILETYPE_ALL:\r
  case EFI_FV_FILETYPE_RAW:\r
  case EFI_FV_FILETYPE_FFS_PAD:\r
    break;\r
\r
  default:\r
    //\r
    // All other files have sections\r
    //\r
    Status = ParseSection (\r
              (UINT8 *) ((UINTN) FileHeader + HeaderSize),\r
              FvBufGetFfsFileSize (FileHeader) - HeaderSize\r
              );\r
    if (EFI_ERROR (Status)) {\r
      //\r
      // printf ("ERROR: Parsing the FFS file.\n");\r
      //\r
      return EFI_ABORTED;\r
    }\r
    break;\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
EFI_STATUS\r
RebaseImageRead (\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
\r
EFI_STATUS\r
SetAddressToSectionHeader (\r
  IN     CHAR8   *FileName,\r
  IN OUT UINT8   *FileBuffer,\r
  IN     UINT64  NewPe32BaseAddress\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Set new base address into the section header of PeImage\r
\r
Arguments:\r
\r
  FileName           - Name of file\r
  FileBuffer         - Pointer to PeImage.\r
  NewPe32BaseAddress - New Base Address for PE image.\r
\r
Returns:\r
\r
  EFI_SUCCESS          Set new base address into this image successfully.\r
\r
--*/\r
{\r
  EFI_STATUS                            Status;\r
  PE_COFF_LOADER_IMAGE_CONTEXT          ImageContext;\r
  UINTN                                 Index;\r
  EFI_IMAGE_OPTIONAL_HEADER_UNION       *ImgHdr;\r
  EFI_IMAGE_SECTION_HEADER              *SectionHeader;\r
\r
  //\r
  // Initialize context\r
  //\r
  memset (&ImageContext, 0, sizeof (ImageContext));\r
  ImageContext.Handle     = (VOID *) FileBuffer;\r
  ImageContext.ImageRead  = (PE_COFF_LOADER_READ_FILE) RebaseImageRead;\r
  Status                  = PeCoffLoaderGetImageInfo (&ImageContext);\r
  if (EFI_ERROR (Status)) {\r
    Error (NULL, 0, 3000, "Invalid", "The input PeImage %s is not valid", FileName);\r
    return Status;\r
  }\r
\r
  if (ImageContext.RelocationsStripped) {\r
    Error (NULL, 0, 3000, "Invalid", "The input PeImage %s has no relocation to be fixed up", FileName);\r
    return Status;\r
  }\r
\r
  //\r
  // Get PeHeader pointer\r
  //\r
  ImgHdr = (EFI_IMAGE_OPTIONAL_HEADER_UNION *)(FileBuffer + ImageContext.PeCoffHeaderOffset);\r
\r
  //\r
  // Get section header list\r
  //\r
  SectionHeader = (EFI_IMAGE_SECTION_HEADER *) (\r
    (UINTN) ImgHdr +\r
    sizeof (UINT32) +\r
    sizeof (EFI_IMAGE_FILE_HEADER) +\r
    ImgHdr->Pe32.FileHeader.SizeOfOptionalHeader\r
    );\r
\r
  //\r
  // Set base address into the first section header that doesn't point to code section.\r
  //\r
  for (Index = 0; Index < ImgHdr->Pe32.FileHeader.NumberOfSections; Index ++, SectionHeader ++) {\r
    if ((SectionHeader->Characteristics & EFI_IMAGE_SCN_CNT_CODE) == 0) {\r
      *(UINT64 *) &SectionHeader->PointerToRelocations = NewPe32BaseAddress;\r
      break;\r
    }\r
  }\r
\r
  //\r
  // BaseAddress is set to section header.\r
  //\r
  return EFI_SUCCESS;\r
}\r
\r
EFI_STATUS\r
RebaseImage (\r
  IN     CHAR8   *FileName,\r
  IN OUT UINT8   *FileBuffer,\r
  IN     UINT64  NewPe32BaseAddress\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Set new base address into PeImage, and fix up PeImage based on new address.\r
\r
Arguments:\r
\r
  FileName           - Name of file\r
  FileBuffer         - Pointer to PeImage.\r
  NewPe32BaseAddress - New Base Address for PE image.\r
\r
Returns:\r
\r
  EFI_INVALID_PARAMETER   - BaseAddress is not valid.\r
  EFI_SUCCESS             - Update PeImage is correctly.\r
\r
--*/\r
{\r
  EFI_STATUS                            Status;\r
  PE_COFF_LOADER_IMAGE_CONTEXT          ImageContext;\r
  UINTN                                 Index;\r
  EFI_IMAGE_OPTIONAL_HEADER_UNION       *ImgHdr;\r
  UINT8                                 *MemoryImagePointer;\r
  EFI_IMAGE_SECTION_HEADER              *SectionHeader;\r
\r
  //\r
  // Initialize context\r
  //\r
  memset (&ImageContext, 0, sizeof (ImageContext));\r
  ImageContext.Handle     = (VOID *) FileBuffer;\r
  ImageContext.ImageRead  = (PE_COFF_LOADER_READ_FILE) RebaseImageRead;\r
  Status                  = PeCoffLoaderGetImageInfo (&ImageContext);\r
  if (EFI_ERROR (Status)) {\r
    Error (NULL, 0, 3000, "Invalid", "The input PeImage %s is not valid", FileName);\r
    return Status;\r
  }\r
\r
  if (ImageContext.RelocationsStripped) {\r
    Error (NULL, 0, 3000, "Invalid", "The input PeImage %s has no relocation to be fixed up", FileName);\r
    return Status;\r
  }\r
\r
  //\r
  // Get PeHeader pointer\r
  //\r
  ImgHdr = (EFI_IMAGE_OPTIONAL_HEADER_UNION *)(FileBuffer + ImageContext.PeCoffHeaderOffset);\r
\r
  //\r
  // Load and Relocate Image Data\r
  //\r
  MemoryImagePointer = (UINT8 *) malloc ((UINTN) ImageContext.ImageSize + ImageContext.SectionAlignment);\r
  if (MemoryImagePointer == NULL) {\r
    Error (NULL, 0, 4001, "Resource", "memory cannot be allocated on rebase of %s", FileName);\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
  memset ((VOID *) MemoryImagePointer, 0, (UINTN) ImageContext.ImageSize + ImageContext.SectionAlignment);\r
  ImageContext.ImageAddress = ((UINTN) MemoryImagePointer + ImageContext.SectionAlignment - 1) & (~((INT64)ImageContext.SectionAlignment - 1));\r
\r
  Status =  PeCoffLoaderLoadImage (&ImageContext);\r
  if (EFI_ERROR (Status)) {\r
    Error (NULL, 0, 3000, "Invalid", "LocateImage() call failed on rebase of %s", FileName);\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, 3000, "Invalid", "RelocateImage() call failed on rebase of %s", FileName);\r
    free ((VOID *) MemoryImagePointer);\r
    return Status;\r
  }\r
\r
  //\r
  // Copy Relocated data to raw image file.\r
  //\r
  SectionHeader = (EFI_IMAGE_SECTION_HEADER *) (\r
    (UINTN) ImgHdr +\r
    sizeof (UINT32) +\r
    sizeof (EFI_IMAGE_FILE_HEADER) +\r
    ImgHdr->Pe32.FileHeader.SizeOfOptionalHeader\r
    );\r
\r
  for (Index = 0; Index < ImgHdr->Pe32.FileHeader.NumberOfSections; Index ++, SectionHeader ++) {\r
    CopyMem (\r
      FileBuffer + SectionHeader->PointerToRawData,\r
      (VOID*) (UINTN) (ImageContext.ImageAddress + SectionHeader->VirtualAddress),\r
      SectionHeader->SizeOfRawData\r
      );\r
  }\r
\r
  free ((VOID *) MemoryImagePointer);\r
\r
  //\r
  // Update Image Base Address\r
  //\r
  if ((ImgHdr->Pe32.OptionalHeader.Magic == EFI_IMAGE_NT_OPTIONAL_HDR32_MAGIC) && (ImgHdr->Pe32.FileHeader.Machine != IMAGE_FILE_MACHINE_IA64)) {\r
    ImgHdr->Pe32.OptionalHeader.ImageBase = (UINT32) NewPe32BaseAddress;\r
  } else if (ImgHdr->Pe32Plus.OptionalHeader.Magic == EFI_IMAGE_NT_OPTIONAL_HDR64_MAGIC) {\r
    ImgHdr->Pe32Plus.OptionalHeader.ImageBase = NewPe32BaseAddress;\r
  } else {\r
    Error (NULL, 0, 3000, "Invalid", "unknown PE magic signature %X in PE32 image %s",\r
      ImgHdr->Pe32.OptionalHeader.Magic,\r
      FileName\r
      );\r
    return EFI_ABORTED;\r
  }\r
\r
  //\r
  // Set new base address into section header\r
  //\r
  Status = SetAddressToSectionHeader (FileName, FileBuffer, NewPe32BaseAddress);\r
\r
  return Status;\r
}\r
\r
EFI_STATUS\r
CombinePath (\r
  IN  CHAR8* DefaultPath,\r
  IN  CHAR8* AppendPath,\r
  OUT CHAR8* NewPath\r
)\r
{\r
  UINT32 DefaultPathLen;\r
  DefaultPathLen = strlen(DefaultPath);\r
  strcpy(NewPath, DefaultPath);\r
  UINT64 Index = 0;\r
  for (; Index < DefaultPathLen; Index ++) {\r
    if (NewPath[Index] == '\\' || NewPath[Index] == '/') {\r
      if (NewPath[Index + 1] != '\0') {\r
        NewPath[Index] = '/';\r
      }\r
    }\r
  }\r
  if (NewPath[Index -1] != '/') {\r
    NewPath[Index] = '/';\r
    NewPath[Index + 1] = '\0';\r
  }\r
  strcat(NewPath, AppendPath);\r
  return EFI_SUCCESS;\r
}\r
\r
EFI_STATUS\r
ParseSection (\r
  IN UINT8  *SectionBuffer,\r
  IN UINT32 BufferLength\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Parses EFI Sections\r
\r
Arguments:\r
\r
  SectionBuffer - Buffer containing the section to parse.\r
  BufferLength  - Length of SectionBuffer\r
\r
Returns:\r
\r
  EFI_SECTION_ERROR - Problem with section parsing.\r
                      (a) compression errors\r
                      (b) unrecognized section \r
  EFI_UNSUPPORTED - Do not know how to parse the section.\r
  EFI_SUCCESS - Section successfully parsed.\r
  EFI_OUT_OF_RESOURCES - Memory allocation failed.\r
\r
--*/\r
{\r
  EFI_SECTION_TYPE    Type;\r
  UINT8               *Ptr;\r
  UINT32              SectionLength;\r
  UINT32              SectionHeaderLen;\r
  CHAR8               *SectionName;\r
  EFI_STATUS          Status;\r
  UINT32              ParsedLength;\r
  UINT8               *CompressedBuffer;\r
  UINT32              CompressedLength;\r
  UINT8               *UncompressedBuffer;\r
  UINT32              UncompressedLength;\r
  UINT8               *ToolOutputBuffer;\r
  UINT32              ToolOutputLength;\r
  UINT8               CompressionType;\r
  UINT32              DstSize;\r
  UINT32              ScratchSize;\r
  UINT8               *ScratchBuffer;\r
  DECOMPRESS_FUNCTION DecompressFunction;\r
  GETINFO_FUNCTION    GetInfoFunction;\r
  // CHAR16              *name;\r
  CHAR8               *ExtractionTool;\r
  CHAR8               *ToolInputFile;\r
  CHAR8               *ToolOutputFile;\r
  CHAR8               *SystemCommandFormatString;\r
  CHAR8               *SystemCommand;\r
  EFI_GUID            *EfiGuid;\r
  UINT16              DataOffset;\r
  UINT16              Attributes;\r
  UINT32              RealHdrLen;\r
  CHAR8               *ToolInputFileName;\r
  CHAR8               *ToolOutputFileName;\r
\r
  ParsedLength = 0;\r
  ToolInputFileName = NULL;\r
  ToolOutputFileName = NULL;\r
\r
  while (ParsedLength < BufferLength) {\r
    Ptr           = SectionBuffer + ParsedLength;\r
\r
    SectionLength = GetLength (((EFI_COMMON_SECTION_HEADER *) Ptr)->Size);\r
    Type          = ((EFI_COMMON_SECTION_HEADER *) Ptr)->Type;\r
\r
    //\r
    // This is sort of an odd check, but is necessary because FFS files are\r
    // padded to a QWORD boundary, meaning there is potentially a whole section\r
    // header worth of 0xFF bytes.\r
    //\r
    if (SectionLength == 0xffffff && Type == 0xff) {\r
      ParsedLength += 4;\r
      continue;\r
    }\r
\r
    //\r
    // Get real section file size\r
    //\r
    SectionLength = GetSectionFileLength ((EFI_COMMON_SECTION_HEADER *) Ptr);\r
    SectionHeaderLen = GetSectionHeaderLength((EFI_COMMON_SECTION_HEADER *)Ptr);\r
\r
    SectionName = SectionNameToStr (Type);\r
    if (SectionName != NULL) {\r
      printf ("------------------------------------------------------------\n");\r
      printf ("  Type:  %s\n  Size:  0x%08X\n", SectionName, (unsigned) SectionLength);\r
      free (SectionName);\r
    }\r
\r
    switch (Type) {\r
    case EFI_SECTION_RAW:\r
    case EFI_SECTION_PIC:\r
    case EFI_SECTION_TE:\r
      // default is no more information\r
      break;\r
\r
    case EFI_SECTION_PE32:\r
      if (EnableHash) {\r
        ToolInputFileName  = "edk2Temp_InputEfi.tmp";\r
        ToolOutputFileName = "edk2Temp_OutputHash.tmp";\r
        RebaseImage(ToolInputFileName, (UINT8*)Ptr + SectionHeaderLen, 0);\r
        PutFileImage (\r
          ToolInputFileName,\r
          (CHAR8*)Ptr + SectionHeaderLen,\r
          SectionLength - SectionHeaderLen\r
          );\r
\r
        SystemCommandFormatString = "%s sha1 -out %s %s";\r
        SystemCommand = malloc (\r
          strlen (SystemCommandFormatString) +\r
          strlen (OpenSslPath) +\r
          strlen (ToolInputFileName) +\r
          strlen (ToolOutputFileName) +\r
          1\r
          );\r
        if (SystemCommand == NULL) {\r
          Error (NULL, 0, 4001, "Resource", "memory cannot be allocated!");\r
          return EFI_OUT_OF_RESOURCES;\r
        }\r
        sprintf (\r
          SystemCommand,\r
          SystemCommandFormatString,\r
          OpenSslPath,\r
          ToolOutputFileName,\r
          ToolInputFileName\r
          );\r
\r
        if (system (SystemCommand) != EFI_SUCCESS) {\r
          Error (NULL, 0, 3000, "Open SSL command not available.  Please verify PATH or set OPENSSL_PATH.", NULL);\r
        }\r
        else {\r
          FILE *fp;\r
          CHAR8 *StrLine;\r
          CHAR8 *NewStr;\r
          UINT32 nFileLen;\r
          if((fp = fopen(ToolOutputFileName,"r")) == NULL) {\r
            Error (NULL, 0, 0004, "Hash the PE32 image failed.", NULL);\r
          }\r
          else {\r
            fseek(fp,0,SEEK_SET);\r
            fseek(fp,0,SEEK_END);\r
            nFileLen = ftell(fp);\r
            fseek(fp,0,SEEK_SET);\r
            StrLine = malloc(nFileLen);\r
            if (StrLine == NULL) {\r
              fclose(fp);\r
              free (SystemCommand);\r
              Error (NULL, 0, 4001, "Resource", "memory cannot be allocated!");\r
              return EFI_OUT_OF_RESOURCES;\r
            }\r
            fgets(StrLine, nFileLen, fp);\r
            NewStr = strrchr (StrLine, '=');\r
            printf ("  SHA1: %s\n", NewStr + 1);\r
            free (StrLine);\r
            fclose(fp);\r
          }\r
        }\r
        remove(ToolInputFileName);\r
        remove(ToolOutputFileName);\r
        free (SystemCommand);\r
      }\r
      break;\r
\r
    case EFI_SECTION_USER_INTERFACE:\r
      printf ("  String: %ls\n", (wchar_t *) &((EFI_USER_INTERFACE_SECTION *) Ptr)->FileNameString);\r
      break;\r
\r
    case EFI_SECTION_FIRMWARE_VOLUME_IMAGE:\r
      Status = PrintFvInfo (Ptr + SectionHeaderLen, TRUE);\r
      if (EFI_ERROR (Status)) {\r
        Error (NULL, 0, 0003, "printing of FV section contents failed", NULL);\r
        return EFI_SECTION_ERROR;\r
      }\r
      break;\r
\r
    case EFI_SECTION_COMPATIBILITY16:\r
    case EFI_SECTION_FREEFORM_SUBTYPE_GUID:\r
      //\r
      // Section does not contain any further header information.\r
      //\r
      break;\r
\r
    case EFI_SECTION_PEI_DEPEX:\r
    case EFI_SECTION_DXE_DEPEX:\r
    case EFI_SECTION_SMM_DEPEX:\r
      DumpDepexSection (Ptr, SectionLength);\r
      break;\r
\r
    case EFI_SECTION_VERSION:\r
      printf ("  Build Number:  0x%02X\n", *(UINT16 *)(Ptr + SectionHeaderLen));\r
      printf ("  Version Strg:  %s\n", (char*) (Ptr + SectionHeaderLen + sizeof (UINT16)));\r
      break;\r
\r
    case EFI_SECTION_COMPRESSION:\r
      UncompressedBuffer  = NULL;\r
      if (SectionHeaderLen == sizeof (EFI_COMMON_SECTION_HEADER)) {\r
        RealHdrLen = sizeof(EFI_COMPRESSION_SECTION);\r
        UncompressedLength  = ((EFI_COMPRESSION_SECTION *)Ptr)->UncompressedLength;\r
        CompressionType     = ((EFI_COMPRESSION_SECTION *)Ptr)->CompressionType;\r
      } else {\r
        RealHdrLen = sizeof(EFI_COMPRESSION_SECTION2);\r
        UncompressedLength  = ((EFI_COMPRESSION_SECTION2 *)Ptr)->UncompressedLength;\r
        CompressionType     = ((EFI_COMPRESSION_SECTION2 *)Ptr)->CompressionType;\r
      }\r
      CompressedLength    = SectionLength - RealHdrLen;\r
      printf ("  Uncompressed Length:  0x%08X\n", (unsigned) UncompressedLength);\r
\r
      if (CompressionType == EFI_NOT_COMPRESSED) {\r
        printf ("  Compression Type:  EFI_NOT_COMPRESSED\n");\r
        if (CompressedLength != UncompressedLength) {\r
          Error (\r
            NULL,\r
            0,\r
            0,\r
            "file is not compressed, but the compressed length does not match the uncompressed length",\r
            NULL\r
            );\r
          return EFI_SECTION_ERROR;\r
        }\r
\r
        UncompressedBuffer = Ptr + RealHdrLen;\r
      } else if (CompressionType == EFI_STANDARD_COMPRESSION) {\r
        GetInfoFunction     = EfiGetInfo;\r
        DecompressFunction  = EfiDecompress;\r
        printf ("  Compression Type:  EFI_STANDARD_COMPRESSION\n");\r
\r
        CompressedBuffer  = Ptr + RealHdrLen;\r
\r
        Status            = GetInfoFunction (CompressedBuffer, CompressedLength, &DstSize, &ScratchSize);\r
        if (EFI_ERROR (Status)) {\r
          Error (NULL, 0, 0003, "error getting compression info from compression section", NULL);\r
          return EFI_SECTION_ERROR;\r
        }\r
\r
        if (DstSize != UncompressedLength) {\r
          Error (NULL, 0, 0003, "compression error in the compression section", NULL);\r
          return EFI_SECTION_ERROR;\r
        }\r
\r
        ScratchBuffer       = malloc (ScratchSize);\r
        if (ScratchBuffer == NULL) {\r
          Error (NULL, 0, 4001, "Resource", "memory cannot be allocated!");\r
          return EFI_OUT_OF_RESOURCES;\r
        }\r
        UncompressedBuffer  = malloc (UncompressedLength);\r
        if (UncompressedBuffer == NULL) {\r
          free (ScratchBuffer);\r
          Error (NULL, 0, 4001, "Resource", "memory cannot be allocated!");\r
          return EFI_OUT_OF_RESOURCES;\r
        }\r
        Status = DecompressFunction (\r
                  CompressedBuffer,\r
                  CompressedLength,\r
                  UncompressedBuffer,\r
                  UncompressedLength,\r
                  ScratchBuffer,\r
                  ScratchSize\r
                  );\r
        free (ScratchBuffer);\r
        if (EFI_ERROR (Status)) {\r
          Error (NULL, 0, 0003, "decompress failed", NULL);\r
          free (UncompressedBuffer);\r
          return EFI_SECTION_ERROR;\r
        }\r
      } else {\r
        Error (NULL, 0, 0003, "unrecognized compression type", "type 0x%X", CompressionType);\r
        return EFI_SECTION_ERROR;\r
      }\r
\r
      Status = ParseSection (UncompressedBuffer, UncompressedLength);\r
\r
      if (CompressionType == EFI_STANDARD_COMPRESSION) {\r
        //\r
        // We need to deallocate Buffer\r
        //\r
        free (UncompressedBuffer);\r
      }\r
\r
      if (EFI_ERROR (Status)) {\r
        Error (NULL, 0, 0003, "failed to parse section", NULL);\r
        return EFI_SECTION_ERROR;\r
      }\r
      break;\r
\r
    case EFI_SECTION_GUID_DEFINED:\r
      if (SectionHeaderLen == sizeof(EFI_COMMON_SECTION_HEADER)) {\r
        EfiGuid = &((EFI_GUID_DEFINED_SECTION *) Ptr)->SectionDefinitionGuid;\r
        DataOffset = ((EFI_GUID_DEFINED_SECTION *) Ptr)->DataOffset;\r
        Attributes = ((EFI_GUID_DEFINED_SECTION *) Ptr)->Attributes;\r
      } else {\r
        EfiGuid = &((EFI_GUID_DEFINED_SECTION2 *) Ptr)->SectionDefinitionGuid;\r
        DataOffset = ((EFI_GUID_DEFINED_SECTION2 *) Ptr)->DataOffset;\r
        Attributes = ((EFI_GUID_DEFINED_SECTION2 *) Ptr)->Attributes;\r
      }\r
      printf ("  SectionDefinitionGuid:  ");\r
      PrintGuid (EfiGuid);\r
      printf ("\n");\r
      printf ("  DataOffset:             0x%04X\n", (unsigned) DataOffset);\r
      printf ("  Attributes:             0x%04X\n", (unsigned) Attributes);\r
\r
      ExtractionTool =\r
        LookupGuidedSectionToolPath (\r
          mParsedGuidedSectionTools,\r
          EfiGuid\r
          );\r
\r
      if (ExtractionTool != NULL) {\r
       #ifndef __GNUC__\r
        ToolInputFile = CloneString (tmpnam (NULL));\r
        ToolOutputFile = CloneString (tmpnam (NULL));\r
       #else\r
        char tmp1[] = "/tmp/fileXXXXXX";\r
        char tmp2[] = "/tmp/fileXXXXXX";\r
        int fd1;\r
        int fd2;\r
        fd1 = mkstemp(tmp1);\r
        fd2 = mkstemp(tmp2);\r
        ToolInputFile = CloneString(tmp1);\r
        ToolOutputFile = CloneString(tmp2);\r
        close(fd1);\r
        close(fd2);\r
       #endif\r
\r
        if ((ToolInputFile == NULL) || (ToolOutputFile == NULL)) {\r
          if (ToolInputFile != NULL) {\r
            free (ToolInputFile);\r
          }\r
          if (ToolOutputFile != NULL) {\r
            free (ToolOutputFile);\r
          }\r
          free (ExtractionTool);\r
\r
          Error (NULL, 0, 4001, "Resource", "memory cannot be allocated!");\r
          return EFI_OUT_OF_RESOURCES;\r
        }\r
\r
        //\r
        // Construction 'system' command string\r
        //\r
        SystemCommandFormatString = "%s -d -o %s %s";\r
        SystemCommand = malloc (\r
          strlen (SystemCommandFormatString) +\r
          strlen (ExtractionTool) +\r
          strlen (ToolInputFile) +\r
          strlen (ToolOutputFile) +\r
          1\r
          );\r
        if (SystemCommand == NULL) {\r
          free (ToolInputFile);\r
          free (ToolOutputFile);\r
          free (ExtractionTool);\r
\r
          Error (NULL, 0, 4001, "Resource", "memory cannot be allocated!");\r
          return EFI_OUT_OF_RESOURCES;\r
        }\r
        sprintf (\r
          SystemCommand,\r
          SystemCommandFormatString,\r
          ExtractionTool,\r
          ToolOutputFile,\r
          ToolInputFile\r
          );\r
        free (ExtractionTool);\r
\r
        Status =\r
          PutFileImage (\r
            ToolInputFile,\r
            (CHAR8*) SectionBuffer + DataOffset,\r
            BufferLength - DataOffset\r
            );\r
\r
        system (SystemCommand);\r
        remove (ToolInputFile);\r
        free (ToolInputFile);\r
\r
        Status =\r
          GetFileImage (\r
            ToolOutputFile,\r
            (CHAR8 **)&ToolOutputBuffer,\r
            &ToolOutputLength\r
            );\r
        remove (ToolOutputFile);\r
        free (ToolOutputFile);\r
        free (SystemCommand);\r
        if (EFI_ERROR (Status)) {\r
          Error (NULL, 0, 0004, "unable to read decoded GUIDED section", NULL);\r
          return EFI_SECTION_ERROR;\r
        }\r
\r
        Status = ParseSection (\r
                  ToolOutputBuffer,\r
                  ToolOutputLength\r
                  );\r
        if (EFI_ERROR (Status)) {\r
          Error (NULL, 0, 0003, "parse of decoded GUIDED section failed", NULL);\r
          return EFI_SECTION_ERROR;\r
        }\r
\r
      //\r
      // Check for CRC32 sections which we can handle internally if needed.\r
      //\r
      } else if (!CompareGuid (\r
                   EfiGuid,\r
                   &gEfiCrc32GuidedSectionExtractionProtocolGuid\r
                   )\r
          ) {\r
        //\r
        // CRC32 guided section\r
        //\r
        Status = ParseSection (\r
                  SectionBuffer + DataOffset,\r
                  BufferLength - DataOffset\r
                  );\r
        if (EFI_ERROR (Status)) {\r
          Error (NULL, 0, 0003, "parse of CRC32 GUIDED section failed", NULL);\r
          return EFI_SECTION_ERROR;\r
        }\r
      } else {\r
        //\r
        // We don't know how to parse it now.\r
        //\r
        Error (NULL, 0, 0003, "Error parsing section", \\r
        "EFI_SECTION_GUID_DEFINED cannot be parsed at this time. Tool to decode this section should have been defined in GuidedSectionTools.txt (built in the FV directory).");\r
        return EFI_UNSUPPORTED;\r
      }\r
      break;\r
\r
    default:\r
      //\r
      // Unknown section, return error\r
      //\r
      Error (NULL, 0, 0003, "unrecognized section type found", "section type = 0x%X", Type);\r
      return EFI_SECTION_ERROR;\r
    }\r
\r
    ParsedLength += SectionLength;\r
    //\r
    // We make then next section begin on a 4-byte boundary\r
    //\r
    ParsedLength = GetOccupiedSize (ParsedLength, 4);\r
  }\r
\r
  if (ParsedLength < BufferLength) {\r
    Error (NULL, 0, 0003, "sections do not completely fill the sectioned buffer being parsed", NULL);\r
    return EFI_SECTION_ERROR;\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
EFI_STATUS\r
DumpDepexSection (\r
  IN UINT8    *Ptr,\r
  IN UINT32   SectionLength\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  GC_TODO: Add function description\r
\r
Arguments:\r
\r
  Ptr           - GC_TODO: add argument description\r
  SectionLength - GC_TODO: add argument description\r
\r
Returns:\r
\r
  EFI_SUCCESS - GC_TODO: Add description for return value\r
\r
--*/\r
{\r
  UINT8 GuidBuffer[PRINTED_GUID_BUFFER_SIZE];\r
\r
  //\r
  // Need at least a section header + data\r
  //\r
  if (SectionLength <= sizeof (EFI_COMMON_SECTION_HEADER)) {\r
    return EFI_SUCCESS;\r
  }\r
\r
  Ptr += GetSectionHeaderLength((EFI_COMMON_SECTION_HEADER *)Ptr);\r
  SectionLength -= GetSectionHeaderLength((EFI_COMMON_SECTION_HEADER *)Ptr);\r
  while (SectionLength > 0) {\r
    printf ("        ");\r
    switch (*Ptr) {\r
    case EFI_DEP_BEFORE:\r
      printf ("BEFORE\n");\r
      Ptr++;\r
      SectionLength--;\r
      break;\r
\r
    case EFI_DEP_AFTER:\r
      printf ("AFTER\n");\r
      Ptr++;\r
      SectionLength--;\r
      break;\r
\r
    case EFI_DEP_PUSH:\r
      printf ("PUSH\n        ");\r
      PrintGuidToBuffer ((EFI_GUID *) (Ptr + 1), GuidBuffer, sizeof (GuidBuffer), TRUE);\r
      printf ("%s  ", GuidBuffer);\r
      PrintGuidName (GuidBuffer);\r
      printf ("\n");\r
      //\r
      // PrintGuid ((EFI_GUID *)(Ptr + 1));\r
      //\r
      Ptr += 17;\r
      SectionLength -= 17;\r
      break;\r
\r
    case EFI_DEP_AND:\r
      printf ("AND\n");\r
      Ptr++;\r
      SectionLength--;\r
      break;\r
\r
    case EFI_DEP_OR:\r
      printf ("OR\n");\r
      Ptr++;\r
      SectionLength--;\r
      break;\r
\r
    case EFI_DEP_NOT:\r
      printf ("NOT\n");\r
      Ptr++;\r
      SectionLength--;\r
      break;\r
\r
    case EFI_DEP_TRUE:\r
      printf ("TRUE\n");\r
      Ptr++;\r
      SectionLength--;\r
      break;\r
\r
    case EFI_DEP_FALSE:\r
      printf ("FALSE\n");\r
      Ptr++;\r
      SectionLength--;\r
      break;\r
\r
    case EFI_DEP_END:\r
      printf ("END DEPEX\n");\r
      Ptr++;\r
      SectionLength--;\r
      break;\r
\r
    case EFI_DEP_SOR:\r
      printf ("SOR\n");\r
      Ptr++;\r
      SectionLength--;\r
      break;\r
\r
    default:\r
      printf ("Unrecognized byte in depex: 0x%X\n", *Ptr);\r
      return EFI_SUCCESS;\r
    }\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
EFI_STATUS\r
PrintGuidName (\r
  IN UINT8    *GuidStr\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  GC_TODO: Add function description\r
\r
Arguments:\r
\r
  GuidStr - GC_TODO: add argument description\r
\r
Returns:\r
\r
  EFI_SUCCESS - GC_TODO: Add description for return value\r
  EFI_INVALID_PARAMETER - GC_TODO: Add description for return value\r
\r
--*/\r
{\r
  GUID_TO_BASENAME  *GPtr;\r
  //\r
  // If we have a list of guid-to-basenames, then go through the list to\r
  // look for a guid string match. If found, print the basename to stdout,\r
  // otherwise return a failure.\r
  //\r
  GPtr = mGuidBaseNameList;\r
  while (GPtr != NULL) {\r
    if (_stricmp ((CHAR8*) GuidStr, (CHAR8*) GPtr->Guid) == 0) {\r
      printf ("%s", GPtr->BaseName);\r
      return EFI_SUCCESS;\r
    }\r
\r
    GPtr = GPtr->Next;\r
  }\r
\r
  return EFI_INVALID_PARAMETER;\r
}\r
\r
EFI_STATUS\r
ParseGuidBaseNameFile (\r
  CHAR8    *FileName\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  GC_TODO: Add function description\r
\r
Arguments:\r
\r
  FileName  - GC_TODO: add argument description\r
\r
Returns:\r
\r
  EFI_DEVICE_ERROR - GC_TODO: Add description for return value\r
  EFI_OUT_OF_RESOURCES - GC_TODO: Add description for return value\r
  EFI_SUCCESS - GC_TODO: Add description for return value\r
\r
--*/\r
{\r
  FILE              *Fptr;\r
  CHAR8             Line[MAX_LINE_LEN];\r
  CHAR8             *FormatString;\r
  INTN              FormatLength;\r
  GUID_TO_BASENAME  *GPtr;\r
\r
  if ((Fptr = fopen (LongFilePath (FileName), "r")) == NULL) {\r
    printf ("ERROR: Failed to open input cross-reference file '%s'\n", FileName);\r
    return EFI_DEVICE_ERROR;\r
  }\r
\r
  //\r
  // Generate the format string for fscanf\r
  //\r
  FormatLength = snprintf (\r
                   NULL,\r
                   0,\r
                   "%%%us %%%us",\r
                   (unsigned) sizeof (GPtr->Guid) - 1,\r
                   (unsigned) sizeof (GPtr->BaseName) - 1\r
                   ) + 1;\r
\r
  FormatString = (CHAR8 *) malloc (FormatLength);\r
  if (FormatString == NULL) {\r
    fclose (Fptr);\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  snprintf (\r
    FormatString,\r
    FormatLength,\r
    "%%%us %%%us",\r
    (unsigned) sizeof (GPtr->Guid) - 1,\r
    (unsigned) sizeof (GPtr->BaseName) - 1\r
    );\r
\r
  while (fgets (Line, sizeof (Line), Fptr) != NULL) {\r
    //\r
    // Allocate space for another guid/basename element\r
    //\r
    GPtr = malloc (sizeof (GUID_TO_BASENAME));\r
    if (GPtr == NULL) {\r
      free (FormatString);\r
      fclose (Fptr);\r
      return EFI_OUT_OF_RESOURCES;\r
    }\r
\r
    memset ((char *) GPtr, 0, sizeof (GUID_TO_BASENAME));\r
    if (sscanf (Line, FormatString, GPtr->Guid, GPtr->BaseName) == 2) {\r
      GPtr->Next        = mGuidBaseNameList;\r
      mGuidBaseNameList = GPtr;\r
    } else {\r
      //\r
      // Some sort of error. Just continue.\r
      //\r
      free (GPtr);\r
    }\r
  }\r
\r
  free (FormatString);\r
  fclose (Fptr);\r
  return EFI_SUCCESS;\r
}\r
\r
EFI_STATUS\r
FreeGuidBaseNameList (\r
  VOID\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  GC_TODO: Add function description\r
\r
Arguments:\r
\r
  None\r
\r
Returns:\r
\r
  EFI_SUCCESS - GC_TODO: Add description for return value\r
\r
--*/\r
{\r
  GUID_TO_BASENAME  *Next;\r
\r
  while (mGuidBaseNameList != NULL) {\r
    Next = mGuidBaseNameList->Next;\r
    free (mGuidBaseNameList);\r
    mGuidBaseNameList = Next;\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
\r
static\r
VOID\r
LoadGuidedSectionToolsTxt (\r
  IN CHAR8* FirmwareVolumeFilename\r
  )\r
{\r
  CHAR8* PeerFilename;\r
  CHAR8* Places[] = {\r
    NULL,\r
    //NULL,\r
    };\r
  UINTN Index;\r
\r
  Places[0] = FirmwareVolumeFilename;\r
  //Places[1] = mUtilityFilename;\r
\r
  mParsedGuidedSectionTools = NULL;\r
\r
  for (Index = 0; Index < (sizeof(Places)/sizeof(Places[0])); Index++) {\r
    PeerFilename = OsPathPeerFilePath (Places[Index], "GuidedSectionTools.txt");\r
    //printf("Loading %s...\n", PeerFilename);\r
    if (OsPathExists (PeerFilename)) {\r
      mParsedGuidedSectionTools = ParseGuidedSectionToolsFile (PeerFilename);\r
    }\r
    free (PeerFilename);\r
    if (mParsedGuidedSectionTools != NULL) {\r
      return;\r
    }\r
  }\r
}\r
\r
\r
void\r
Usage (\r
  VOID\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  GC_TODO: Add function description\r
\r
Arguments:\r
\r
  None\r
\r
Returns:\r
\r
  GC_TODO: add return values\r
\r
--*/\r
{\r
  //\r
  // Summary usage\r
  //\r
  fprintf (stdout, "Usage: %s [options] <input_file>\n\n", UTILITY_NAME);\r
\r
  //\r
  // Copyright declaration\r
  // \r
  fprintf (stdout, "Copyright (c) 2007 - 2016, Intel Corporation. All rights reserved.\n\n");\r
  fprintf (stdout, "  Display Tiano Firmware Volume FFS image information\n\n");\r
\r
  //\r
  // Details Option\r
  //\r
  fprintf (stdout, "optional arguments:\n");\r
  fprintf (stdout, "  -h, --help\n\\r
            Show this help message and exit\n");\r
  fprintf (stdout, "  --version\n\\r
           Show program's version number and exit\n");\r
  fprintf (stdout, "  -d [DEBUG], --debug [DEBUG]\n\\r
            Output DEBUG statements, where DEBUG_LEVEL is 0 (min) - 9 (max)\n");\r
  fprintf (stdout, "  -v, --verbose\n\\r
            Print informational statements\n");\r
  fprintf (stdout, "  -q, --quiet\n\\r
            Returns the exit code, error messages will be displayed\n");\r
  fprintf (stdout, "  -s, --silent\n\\r
            Returns only the exit code; informational and error\n\\r
            messages are not displayed\n");\r
  fprintf (stdout, "  -x XREF_FILENAME, --xref XREF_FILENAME\n\\r
            Parse the basename to file-guid cross reference file(s)\n");\r
  fprintf (stdout, "  -f OFFSET, --offset OFFSET\n\\r
            The offset from the start of the input file to start \n\\r
            processing an FV\n");\r
  fprintf (stdout, "  --hash\n\\r
            Generate HASH value of the entire PE image\n");\r
  fprintf (stdout, "  --sfo\n\\r
            Reserved for future use\n");\r
}\r
\r