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

/** @file\r
Utility program to create an EFI option ROM image from binary and EFI PE32 files.\r
\r
Copyright (c) 1999 - 2018, Intel Corporation. All rights reserved.<BR>\r
SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
\r
#include "EfiUtilityMsgs.h"\r
#include "ParseInf.h"\r
#include "EfiRom.h"\r
\r
UINT64  DebugLevel = 0;\r
\r
int\r
main (\r
  int   Argc,\r
  char  *Argv[]\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Given an EFI image filename, create a ROM-able image by creating an option\r
  ROM header and PCI data structure, filling them in, and then writing the\r
  option ROM header + PCI data structure + EFI image out to the output file.\r
\r
Arguments:\r
\r
  Argc            - standard C main() argument count\r
\r
  Argv            - standard C main() argument list\r
\r
Returns:\r
\r
  0             success\r
  non-zero      otherwise\r
\r
--*/\r
{\r
  CHAR8     *Ext;\r
  FILE      *FptrOut;\r
  UINT32    Status;\r
  FILE_LIST *FList;\r
  UINT32    TotalSize;\r
  UINT32    Size;\r
  CHAR8     *Ptr0;\r
\r
  SetUtilityName(UTILITY_NAME);\r
\r
  Status  = STATUS_SUCCESS;\r
  FptrOut = NULL;\r
\r
  //\r
  // Parse the command line arguments\r
  //\r
  if (ParseCommandLine (Argc, Argv, &mOptions)) {\r
    return STATUS_ERROR;\r
  }\r
\r
  if (mOptions.Quiet) {\r
    SetPrintLevel(40);\r
  } else if (mOptions.Verbose) {\r
    SetPrintLevel(15);\r
  } else if (mOptions.Debug) {\r
    SetPrintLevel(DebugLevel);\r
  }\r
\r
  if (mOptions.Verbose) {\r
    VerboseMsg("%s tool start.\n", UTILITY_NAME);\r
  }\r
\r
  //\r
  // If dumping an image, then do that and quit\r
  //\r
  if (mOptions.DumpOption == 1) {\r
    if (mOptions.FileList != NULL) {\r
      if ((Ptr0 = strstr ((CONST CHAR8 *) mOptions.FileList->FileName, DEFAULT_OUTPUT_EXTENSION)) != NULL) {\r
        DumpImage (mOptions.FileList);\r
        goto BailOut;\r
      } else {\r
        Error (NULL, 0, 1002, "No PciRom input file", "No *.rom input file");\r
        goto BailOut;\r
      }\r
    }\r
  }\r
  //\r
  // Determine the output filename. Either what they specified on\r
  // the command line, or the first input filename with a different extension.\r
  //\r
  if (!mOptions.OutFileName[0]) {\r
    if (mOptions.FileList != NULL) {\r
      if (strlen (mOptions.FileList->FileName) >= MAX_PATH) {\r
        Status = STATUS_ERROR;\r
        Error (NULL, 0, 2000, "Invalid parameter", "Input file name is too long - %s.", mOptions.FileList->FileName);\r
        goto BailOut;\r
      }\r
      strncpy (mOptions.OutFileName, mOptions.FileList->FileName, MAX_PATH - 1);\r
      mOptions.OutFileName[MAX_PATH - 1] = 0;\r
      //\r
      // Find the last . on the line and replace the filename extension with\r
      // the default\r
      //\r
      Ext = mOptions.OutFileName + strlen (mOptions.OutFileName) - 1;\r
      while (Ext >= mOptions.OutFileName) {\r
        if ((*Ext == '.') || (*Ext == '\\')) {\r
          break;\r
        }\r
        Ext--;\r
      }\r
      //\r
      // If dot here, then insert extension here, otherwise append\r
      //\r
      if (*Ext != '.') {\r
        Ext = mOptions.OutFileName + strlen (mOptions.OutFileName);\r
      }\r
\r
      strcpy (Ext, DEFAULT_OUTPUT_EXTENSION);\r
    }\r
  }\r
  //\r
  // Make sure we don't have the same filename for input and output files\r
  //\r
  for (FList = mOptions.FileList; FList != NULL; FList = FList->Next) {\r
    if (stricmp (mOptions.OutFileName, FList->FileName) == 0) {\r
      Status = STATUS_ERROR;\r
      Error (NULL, 0, 1002, "Invalid input parameter", "Input and output file names must be different - %s = %s.", FList->FileName, mOptions.OutFileName);\r
      goto BailOut;\r
    }\r
  }\r
  //\r
  // Now open our output file\r
  //\r
  if ((FptrOut = fopen (LongFilePath (mOptions.OutFileName), "wb")) == NULL) {\r
    Error (NULL, 0, 0001, "Error opening file", "Error opening file %s", mOptions.OutFileName);\r
    goto BailOut;\r
  }\r
  //\r
  // Process all our files\r
  //\r
  TotalSize = 0;\r
  for (FList = mOptions.FileList; FList != NULL; FList = FList->Next) {\r
    Size = 0;\r
    if ((FList->FileFlags & FILE_FLAG_EFI) != 0) {\r
      if (mOptions.Verbose) {\r
        VerboseMsg("Processing EFI file    %s\n", FList->FileName);\r
      }\r
\r
      Status = ProcessEfiFile (FptrOut, FList, mOptions.VendId, mOptions.DevIdList[0], &Size);\r
    } else if ((FList->FileFlags & FILE_FLAG_BINARY) !=0 ) {\r
      if (mOptions.Verbose) {\r
        VerboseMsg("Processing binary file %s\n", FList->FileName);\r
      }\r
\r
      Status = ProcessBinFile (FptrOut, FList, &Size);\r
    } else {\r
      Error (NULL, 0, 2000, "Invalid parameter", "File type not specified, it must be either an EFI or binary file: %s.", FList->FileName);\r
      Status = STATUS_ERROR;\r
    }\r
\r
    if (mOptions.Verbose) {\r
      VerboseMsg("  Output size = 0x%X\n", (unsigned) Size);\r
    }\r
\r
    if (Status != STATUS_SUCCESS) {\r
      break;\r
    }\r
\r
    TotalSize += Size;\r
  }\r
  //\r
  // Check total size\r
  //\r
  if (TotalSize > MAX_OPTION_ROM_SIZE) {\r
    Error (NULL, 0, 2000, "Invalid parameter", "Option ROM image size exceeds limit of 0x%X bytes.", MAX_OPTION_ROM_SIZE);\r
    Status = STATUS_ERROR;\r
  }\r
\r
BailOut:\r
  if (Status == STATUS_SUCCESS) {\r
    //\r
    // Clean up our file list\r
    //\r
    while (mOptions.FileList != NULL) {\r
      FList = mOptions.FileList->Next;\r
      free (mOptions.FileList);\r
      mOptions.FileList = FList;\r
    }\r
\r
    //\r
    // Clean up device ID list\r
    //\r
    if (mOptions.DevIdList != NULL) {\r
      free (mOptions.DevIdList);\r
    }\r
  }\r
  if (FptrOut != NULL) {\r
    fclose (FptrOut);\r
  }\r
\r
  if (mOptions.Verbose) {\r
    VerboseMsg("%s tool done with return code is 0x%x.\n", UTILITY_NAME, GetUtilityStatus ());\r
  }\r
\r
  return GetUtilityStatus ();\r
}\r
\r
static\r
int\r
ProcessBinFile (\r
  FILE      *OutFptr,\r
  FILE_LIST *InFile,\r
  UINT32    *Size\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Process a binary input file.\r
\r
Arguments:\r
\r
  OutFptr     - file pointer to output binary ROM image file we're creating\r
  InFile      - structure contains information on the binary file to process\r
  Size        - pointer to where to return the size added to the output file\r
\r
Returns:\r
\r
  0 - successful\r
\r
--*/\r
{\r
  FILE                      *InFptr;\r
  UINT32                    TotalSize;\r
  UINT32                    FileSize;\r
  UINT8                     *Buffer;\r
  UINT32                    Status;\r
  PCI_EXPANSION_ROM_HEADER  *RomHdr;\r
  PCI_DATA_STRUCTURE        *PciDs23;\r
  PCI_3_0_DATA_STRUCTURE    *PciDs30;\r
  UINT32                    Index;\r
  UINT8                     ByteCheckSum;\r
  UINT16                    CodeType;\r
\r
  PciDs23 = NULL;\r
  PciDs30 = NULL;\r
  Status = STATUS_SUCCESS;\r
\r
  //\r
  // Try to open the input file\r
  //\r
  if ((InFptr = fopen (LongFilePath (InFile->FileName), "rb")) == NULL) {\r
    Error (NULL, 0, 0001, "Error opening file", "%s", InFile->FileName);\r
    return STATUS_ERROR;\r
  }\r
  //\r
  // Seek to the end of the input file and get the file size. Then allocate\r
  // a buffer to read it in to.\r
  //\r
  fseek (InFptr, 0, SEEK_END);\r
  FileSize = ftell (InFptr);\r
  if (mOptions.Verbose) {\r
    VerboseMsg("  File size   = 0x%X\n", (unsigned) FileSize);\r
  }\r
\r
  fseek (InFptr, 0, SEEK_SET);\r
  Buffer = (UINT8 *) malloc (FileSize);\r
  if (Buffer == NULL) {\r
    Error (NULL, 0, 4003, "Resource", "memory cannot be allocated!");\r
    Status = STATUS_ERROR;\r
    goto BailOut;\r
  }\r
\r
  if (fread (Buffer, FileSize, 1, InFptr) != 1) {\r
    Error (NULL, 0, 2000, "Invalid", "Failed to read all bytes from input file.");\r
    Status = STATUS_ERROR;\r
    goto BailOut;\r
  }\r
  //\r
  // Total size must be an even multiple of 512 bytes, and can't exceed\r
  // the option ROM image size.\r
  //\r
  TotalSize = FileSize;\r
  if (TotalSize & 0x1FF) {\r
    TotalSize = (TotalSize + 0x200) &~0x1ff;\r
  }\r
\r
  if (TotalSize > MAX_OPTION_ROM_SIZE) {\r
    Error (NULL, 0, 3001, "Invalid", "Option ROM image %s size exceeds limit of 0x%X bytes.", InFile->FileName, MAX_OPTION_ROM_SIZE);\r
    Status = STATUS_ERROR;\r
    goto BailOut;\r
  }\r
  //\r
  // Return the size to the caller so they can keep track of the running total.\r
  //\r
  *Size = TotalSize;\r
\r
  //\r
  // Crude check to make sure it's a legitimate ROM image\r
  //\r
  RomHdr = (PCI_EXPANSION_ROM_HEADER *) Buffer;\r
  if (RomHdr->Signature != PCI_EXPANSION_ROM_HEADER_SIGNATURE) {\r
    Error (NULL, 0, 2000, "Invalid parameter", "ROM image file has an invalid ROM signature.");\r
    Status = STATUS_ERROR;\r
    goto BailOut;\r
  }\r
  //\r
  // Make sure the pointer to the PCI data structure is within the size of the image.\r
  // Then check it for valid signature.\r
  //\r
  if ((RomHdr->PcirOffset > FileSize) || (RomHdr->PcirOffset == 0)) {\r
    Error (NULL, 0, 2000, "Invalid parameter", "Invalid PCI data structure offset.");\r
    Status = STATUS_ERROR;\r
    goto BailOut;\r
  }\r
\r
  //\r
  // Check the header is conform to PCI2.3 or PCI3.0\r
  //\r
  if (mOptions.Pci23 == 1) {\r
    PciDs23 = (PCI_DATA_STRUCTURE *) (Buffer + RomHdr->PcirOffset);\r
    if (PciDs23->Signature != PCI_DATA_STRUCTURE_SIGNATURE) {\r
      Error (NULL, 0, 2000, "Invalid parameter", "PCI data structure has an invalid signature.");\r
      Status = STATUS_ERROR;\r
      goto BailOut;\r
    }\r
  } else {\r
    //\r
    // Default setting is PCI3.0 header\r
    //\r
    PciDs30 = (PCI_3_0_DATA_STRUCTURE *)(Buffer + RomHdr->PcirOffset);\r
    if (PciDs30->Signature != PCI_DATA_STRUCTURE_SIGNATURE) {\r
      Error (NULL, 0, 2000, "Invalid parameter", "PCI data structure has an invalid signature.");\r
      Status = STATUS_ERROR;\r
      goto BailOut;\r
    }\r
  }\r
\r
  //\r
  // ReSet Option Rom size\r
  //\r
  if (mOptions.Pci23 == 1) {\r
    PciDs23->ImageLength = (UINT16) (TotalSize / 512);\r
    CodeType = PciDs23->CodeType;\r
  } else {\r
    PciDs30->ImageLength = (UINT16) (TotalSize / 512);\r
    CodeType = PciDs30->CodeType;\r
  }\r
\r
  //\r
  // If this is the last image, then set the LAST bit unless requested not\r
  // to via the command-line -n argument. Otherwise, make sure you clear it.\r
  //\r
  if ((InFile->Next == NULL) && (mOptions.NoLast == 0)) {\r
    if (mOptions.Pci23 == 1) {\r
      PciDs23->Indicator = INDICATOR_LAST;\r
    } else {\r
      PciDs30->Indicator = INDICATOR_LAST;\r
    }\r
  } else {\r
    if (mOptions.Pci23 == 1) {\r
      PciDs23->Indicator = 0;\r
    } else {\r
      PciDs30->Indicator = 0;\r
    }\r
  }\r
\r
  if (CodeType != PCI_CODE_TYPE_EFI_IMAGE) {\r
    ByteCheckSum = 0;\r
    for (Index = 0; Index < FileSize - 1; Index++) {\r
      ByteCheckSum = (UINT8) (ByteCheckSum + Buffer[Index]);\r
    }\r
\r
    Buffer[FileSize - 1] = (UINT8) ((~ByteCheckSum) + 1);\r
    if (mOptions.Verbose) {\r
      VerboseMsg("  Checksum = %02x\n\n", Buffer[FileSize - 1]);\r
    }\r
  }\r
\r
  //\r
  // Now copy the input file contents out to the output file\r
  //\r
  if (fwrite (Buffer, FileSize, 1, OutFptr) != 1) {\r
    Error (NULL, 0, 0005, "Failed to write all file bytes to output file.", NULL);\r
    Status = STATUS_ERROR;\r
    goto BailOut;\r
  }\r
\r
  TotalSize -= FileSize;\r
  //\r
  // Pad the rest of the image to make it a multiple of 512 bytes\r
  //\r
  while (TotalSize > 0) {\r
    putc (~0, OutFptr);\r
    TotalSize--;\r
  }\r
\r
BailOut:\r
  if (InFptr != NULL) {\r
    fclose (InFptr);\r
  }\r
\r
  if (Buffer != NULL) {\r
    free (Buffer);\r
  }\r
  //\r
  // Print the file name if errors occurred\r
  //\r
  if (Status != STATUS_SUCCESS) {\r
    Error (NULL, 0, 0003, "Error", "Error parsing file: %s", InFile->FileName);\r
  }\r
\r
  return Status;\r
}\r
\r
static\r
int\r
ProcessEfiFile (\r
  FILE      *OutFptr,\r
  FILE_LIST *InFile,\r
  UINT16    VendId,\r
  UINT16    DevId,\r
  UINT32    *Size\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Process a PE32 EFI file.\r
\r
Arguments:\r
\r
  OutFptr     - file pointer to output binary ROM image file we're creating\r
  InFile      - structure contains information on the PE32 file to process\r
  VendId      - vendor ID as required in the option ROM header\r
  DevId       - device ID as required in the option ROM header\r
  Size        - pointer to where to return the size added to the output file\r
\r
Returns:\r
\r
  0 - successful\r
\r
--*/\r
{\r
  UINT32                        Status;\r
  FILE                          *InFptr;\r
  EFI_PCI_EXPANSION_ROM_HEADER  RomHdr;\r
  PCI_DATA_STRUCTURE            PciDs23;\r
  PCI_3_0_DATA_STRUCTURE        PciDs30;\r
  UINT32                        FileSize;\r
  UINT32                        CompressedFileSize;\r
  UINT8                         *Buffer;\r
  UINT8                         *CompressedBuffer;\r
  UINT8                         *TempBufferPtr;\r
  UINT32                        TotalSize;\r
  UINT32                        HeaderSize;\r
  UINT16                        MachineType;\r
  UINT16                        SubSystem;\r
  UINT32                        HeaderPadBytes;\r
  UINT32                        PadBytesBeforeImage;\r
  UINT32                        PadBytesAfterImage;\r
  UINT32                        DevIdListSize;\r
\r
  //\r
  // Try to open the input file\r
  //\r
  if ((InFptr = fopen (LongFilePath (InFile->FileName), "rb")) == NULL) {\r
    Error (NULL, 0, 0001, "Open file error", "Error opening file: %s", InFile->FileName);\r
    return STATUS_ERROR;\r
  }\r
  //\r
  // Initialize our buffer pointers to null.\r
  //\r
  Buffer            = NULL;\r
  CompressedBuffer  = NULL;\r
\r
  //\r
  // Double-check the file to make sure it's what we expect it to be\r
  //\r
  Status = CheckPE32File (InFptr, &MachineType, &SubSystem);\r
  if (Status != STATUS_SUCCESS) {\r
    goto BailOut;\r
  }\r
  //\r
  // Seek to the end of the input file and get the file size\r
  //\r
  fseek (InFptr, 0, SEEK_END);\r
  FileSize = ftell (InFptr);\r
\r
  //\r
  // Get the size of the headers we're going to put in front of the image. The\r
  // EFI header must be aligned on a 4-byte boundary, so pad accordingly.\r
  //\r
  if (sizeof (RomHdr) & 0x03) {\r
    HeaderPadBytes = 4 - (sizeof (RomHdr) & 0x03);\r
  } else {\r
    HeaderPadBytes = 0;\r
  }\r
\r
  //\r
  // For Pci3.0 to use the different data structure.\r
  //\r
  if (mOptions.Pci23 == 1) {\r
    HeaderSize = sizeof (PCI_DATA_STRUCTURE) + HeaderPadBytes + sizeof (EFI_PCI_EXPANSION_ROM_HEADER);\r
  } else {\r
    if (mOptions.DevIdCount > 1) {\r
      //\r
      // Write device ID list when more than one device ID is specified.\r
      // Leave space for list plus terminator.\r
      //\r
      DevIdListSize = (mOptions.DevIdCount + 1) * sizeof (UINT16);\r
    } else {\r
      DevIdListSize = 0;\r
    }\r
    HeaderSize = sizeof (PCI_3_0_DATA_STRUCTURE) + HeaderPadBytes + DevIdListSize + sizeof (EFI_PCI_EXPANSION_ROM_HEADER);\r
  }\r
\r
  if (mOptions.Verbose) {\r
    VerboseMsg("  File size   = 0x%X\n", (unsigned) FileSize);\r
  }\r
  //\r
  // Allocate memory for the entire file (in case we have to compress), then\r
  // seek back to the beginning of the file and read it into our buffer.\r
  //\r
  Buffer = (UINT8 *) malloc (FileSize);\r
  if (Buffer == NULL) {\r
    Error (NULL, 0, 4001, "Resource", "memory cannot be allocated!");\r
    Status = STATUS_ERROR;\r
    goto BailOut;\r
  }\r
\r
  fseek (InFptr, 0, SEEK_SET);\r
  if (fread (Buffer, FileSize, 1, InFptr) != 1) {\r
    Error (NULL, 0, 0004, "Error reading file", "File %s", InFile->FileName);\r
    Status = STATUS_ERROR;\r
    goto BailOut;\r
  }\r
  //\r
  // Now determine the size of the final output file. It's either the header size\r
  // plus the file's size, or the header size plus the compressed file size.\r
  //\r
  if ((InFile->FileFlags & FILE_FLAG_COMPRESS) != 0) {\r
    //\r
    // Allocate a buffer into which we can compress the image, compress it,\r
    // and use that size as the new size.\r
    //\r
    CompressedBuffer = (UINT8 *) malloc (FileSize);\r
    if (CompressedBuffer == NULL) {\r
      Error (NULL, 0, 4001, "Resource", "memory cannot be allocated!");\r
      Status = STATUS_ERROR;\r
      goto BailOut;\r
    }\r
\r
    CompressedFileSize  = FileSize;\r
    Status              = EfiCompress (Buffer, FileSize, CompressedBuffer, &CompressedFileSize);\r
    if (Status != STATUS_SUCCESS) {\r
      Error (NULL, 0, 0007, "Error compressing file!", NULL);\r
      goto BailOut;\r
    }\r
    //\r
    // Now compute the size, then swap buffer pointers.\r
    //\r
    if (mOptions.Verbose) {\r
      VerboseMsg("  Comp size   = 0x%X\n", (unsigned) CompressedFileSize);\r
    }\r
\r
    TotalSize         = CompressedFileSize + HeaderSize;\r
    FileSize          = CompressedFileSize;\r
    TempBufferPtr     = Buffer;\r
    Buffer            = CompressedBuffer;\r
    CompressedBuffer  = TempBufferPtr;\r
  } else {\r
    TotalSize = FileSize + HeaderSize;\r
  }\r
  //\r
  // Total size must be an even multiple of 512 bytes\r
  //\r
  if (TotalSize & 0x1FF) {\r
    TotalSize = (TotalSize + 0x200) &~0x1ff;\r
  }\r
  //\r
  // Workaround:\r
  //   If compressed, put the pad bytes after the image,\r
  //   else put the pad bytes before the image.\r
  //\r
  if ((InFile->FileFlags & FILE_FLAG_COMPRESS) != 0) {\r
    PadBytesBeforeImage = 0;\r
    PadBytesAfterImage = TotalSize - (FileSize + HeaderSize);\r
  } else {\r
    PadBytesBeforeImage = TotalSize - (FileSize + HeaderSize);\r
    PadBytesAfterImage = 0;\r
  }\r
  //\r
  // Check size\r
  //\r
  if (TotalSize > MAX_OPTION_ROM_SIZE) {\r
    Error (NULL, 0, 2000, "Invalid", "Option ROM image %s size exceeds limit of 0x%X bytes.", InFile->FileName, MAX_OPTION_ROM_SIZE);\r
    Status = STATUS_ERROR;\r
    goto BailOut;\r
  }\r
  //\r
  // Return the size to the caller so they can keep track of the running total.\r
  //\r
  *Size = TotalSize;\r
\r
  //\r
  // Now fill in the ROM header. These values come from chapter 18 of the\r
  // EFI 1.02 specification.\r
  //\r
  memset (&RomHdr, 0, sizeof (RomHdr));\r
  RomHdr.Signature            = PCI_EXPANSION_ROM_HEADER_SIGNATURE;\r
  RomHdr.InitializationSize   = (UINT16) (TotalSize / 512);\r
  RomHdr.EfiSignature         = EFI_PCI_EXPANSION_ROM_HEADER_EFISIGNATURE;\r
  RomHdr.EfiSubsystem         = SubSystem;\r
  RomHdr.EfiMachineType       = MachineType;\r
  RomHdr.EfiImageHeaderOffset = (UINT16) (HeaderSize + PadBytesBeforeImage);\r
  RomHdr.PcirOffset           = (UINT16) (sizeof (RomHdr) + HeaderPadBytes);\r
  //\r
  // Set image as compressed or not\r
  //\r
  if (InFile->FileFlags & FILE_FLAG_COMPRESS) {\r
    RomHdr.CompressionType = EFI_PCI_EXPANSION_ROM_HEADER_COMPRESSED;\r
  }\r
  //\r
  // Fill in the PCI data structure\r
  //\r
  if (mOptions.Pci23 == 1) {\r
    memset (&PciDs23, 0, sizeof (PCI_DATA_STRUCTURE));\r
  } else {\r
    memset (&PciDs30, 0, sizeof (PCI_3_0_DATA_STRUCTURE));\r
  }\r
\r
  if (mOptions.Pci23 == 1) {\r
    PciDs23.Signature = PCI_DATA_STRUCTURE_SIGNATURE;\r
    PciDs23.VendorId  = VendId;\r
    PciDs23.DeviceId  = DevId;\r
    PciDs23.Length    = (UINT16) sizeof (PCI_DATA_STRUCTURE);\r
    PciDs23.Revision  = 0;\r
    //\r
    // Class code and code revision from the command line (optional)\r
    //\r
    PciDs23.ClassCode[0]  = (UINT8) InFile->ClassCode;\r
    PciDs23.ClassCode[1]  = (UINT8) (InFile->ClassCode >> 8);\r
    PciDs23.ClassCode[2]  = (UINT8) (InFile->ClassCode >> 16);\r
    PciDs23.ImageLength   = RomHdr.InitializationSize;\r
    PciDs23.CodeRevision  = InFile->CodeRevision;\r
    PciDs23.CodeType      = PCI_CODE_TYPE_EFI_IMAGE;\r
  } else {\r
    PciDs30.Signature = PCI_DATA_STRUCTURE_SIGNATURE;\r
    PciDs30.VendorId  = VendId;\r
    PciDs30.DeviceId  = DevId;\r
    if (mOptions.DevIdCount > 1) {\r
      //\r
      // Place device list immediately after PCI structure\r
      //\r
      PciDs30.DeviceListOffset = (UINT16) sizeof (PCI_3_0_DATA_STRUCTURE);\r
    } else {\r
      PciDs30.DeviceListOffset = 0;\r
    }\r
    PciDs30.Length    = (UINT16) sizeof (PCI_3_0_DATA_STRUCTURE);\r
    PciDs30.Revision  = 0x3;\r
    //\r
    // Class code and code revision from the command line (optional)\r
    //\r
    PciDs30.ClassCode[0]  = (UINT8) InFile->ClassCode;\r
    PciDs30.ClassCode[1]  = (UINT8) (InFile->ClassCode >> 8);\r
    PciDs30.ClassCode[2]  = (UINT8) (InFile->ClassCode >> 16);\r
    PciDs30.ImageLength   = RomHdr.InitializationSize;\r
    PciDs30.CodeRevision  = InFile->CodeRevision;\r
    PciDs30.CodeType      = PCI_CODE_TYPE_EFI_IMAGE;\r
    PciDs30.MaxRuntimeImageLength = 0; // to be fixed\r
    PciDs30.ConfigUtilityCodeHeaderOffset = 0; // to be fixed\r
    PciDs30.DMTFCLPEntryPointOffset = 0; // to be fixed\r
  }\r
  //\r
  // If this is the last image, then set the LAST bit unless requested not\r
  // to via the command-line -n argument.\r
  //\r
  if ((InFile->Next == NULL) && (mOptions.NoLast == 0)) {\r
    if (mOptions.Pci23 == 1) {\r
      PciDs23.Indicator = INDICATOR_LAST;\r
    } else {\r
    PciDs30.Indicator = INDICATOR_LAST;}\r
  } else {\r
    if (mOptions.Pci23 == 1) {\r
      PciDs23.Indicator = 0;\r
  } else {\r
      PciDs30.Indicator = 0;\r
    }\r
  }\r
  //\r
  // Write the ROM header to the output file\r
  //\r
  if (fwrite (&RomHdr, sizeof (RomHdr), 1, OutFptr) != 1) {\r
    Error (NULL, 0, 0002, "Failed to write ROM header to output file!", NULL);\r
    Status = STATUS_ERROR;\r
    goto BailOut;\r
  }\r
\r
  //\r
  // Write pad bytes to align the PciDs\r
  //\r
  while (HeaderPadBytes > 0) {\r
    if (putc (0, OutFptr) == EOF) {\r
      Error (NULL, 0, 0002, "Failed to write ROM header pad bytes to output file!", NULL);\r
      Status = STATUS_ERROR;\r
      goto BailOut;\r
    }\r
\r
    HeaderPadBytes--;\r
  }\r
  //\r
  // Write the PCI data structure header to the output file\r
  //\r
  if (mOptions.Pci23 == 1) {\r
    if (fwrite (&PciDs23, sizeof (PciDs23), 1, OutFptr) != 1) {\r
      Error (NULL, 0, 0002, "Failed to write PCI ROM header to output file!", NULL);\r
      Status = STATUS_ERROR;\r
      goto BailOut;\r
    }\r
  } else {\r
    if (fwrite (&PciDs30, sizeof (PciDs30), 1, OutFptr) != 1) {\r
      Error (NULL, 0, 0002, "Failed to write PCI ROM header to output file!", NULL);\r
      Status = STATUS_ERROR;\r
      goto BailOut;\r
    }\r
  }\r
\r
  //\r
  // Write the Device ID list to the output file\r
  //\r
  if (mOptions.DevIdCount > 1) {\r
    if (fwrite (mOptions.DevIdList, sizeof (UINT16), mOptions.DevIdCount, OutFptr) != mOptions.DevIdCount) {\r
      Error (NULL, 0, 0002, "Failed to write PCI device list to output file!", NULL);\r
      Status = STATUS_ERROR;\r
      goto BailOut;\r
    }\r
    //\r
    // Write two-byte terminating 0 at the end of the device list\r
    //\r
    if (putc (0, OutFptr) == EOF || putc (0, OutFptr) == EOF) {\r
      Error (NULL, 0, 0002, "Failed to write PCI device list to output file!", NULL);\r
      Status = STATUS_ERROR;\r
      goto BailOut;\r
    }\r
  }\r
\r
\r
  //\r
  // Pad head to make it a multiple of 512 bytes\r
  //\r
  while (PadBytesBeforeImage > 0) {\r
    if (putc (~0, OutFptr) == EOF) {\r
      Error (NULL, 0, 2000, "Failed to write trailing pad bytes output file!", NULL);\r
      Status = STATUS_ERROR;\r
      goto BailOut;\r
    }\r
    PadBytesBeforeImage--;\r
  }\r
  //\r
  // Now dump the input file's contents to the output file\r
  //\r
  if (fwrite (Buffer, FileSize, 1, OutFptr) != 1) {\r
    Error (NULL, 0, 0002, "Failed to write all file bytes to output file!", NULL);\r
    Status = STATUS_ERROR;\r
    goto BailOut;\r
  }\r
\r
  //\r
  // Pad the rest of the image to make it a multiple of 512 bytes\r
  //\r
  while (PadBytesAfterImage > 0) {\r
    if (putc (~0, OutFptr) == EOF) {\r
      Error (NULL, 0, 2000, "Failed to write trailing pad bytes output file!", NULL);\r
      Status = STATUS_ERROR;\r
      goto BailOut;\r
    }\r
\r
    PadBytesAfterImage--;\r
  }\r
\r
BailOut:\r
  if (InFptr != NULL) {\r
    fclose (InFptr);\r
  }\r
  //\r
  // Free up our buffers\r
  //\r
  if (Buffer != NULL) {\r
    free (Buffer);\r
  }\r
\r
  if (CompressedBuffer != NULL) {\r
    free (CompressedBuffer);\r
  }\r
  //\r
  // Print the file name if errors occurred\r
  //\r
  if (Status != STATUS_SUCCESS) {\r
    Error (NULL, 0, 0003, "Error parsing", "Error parsing file: %s", InFile->FileName);\r
  }\r
\r
  return Status;\r
}\r
\r
static\r
int\r
CheckPE32File (\r
  FILE      *Fptr,\r
  UINT16    *MachineType,\r
  UINT16    *SubSystem\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Given a file pointer to a supposed PE32 image file, verify that it is indeed a\r
  PE32 image file, and then return the machine type in the supplied pointer.\r
\r
Arguments:\r
\r
  Fptr          File pointer to the already-opened PE32 file\r
  MachineType   Location to stuff the machine type of the PE32 file. This is needed\r
                because the image may be Itanium-based, IA32, or EBC.\r
\r
Returns:\r
\r
  0             success\r
  non-zero      otherwise\r
\r
--*/\r
{\r
  EFI_IMAGE_DOS_HEADER            DosHeader;\r
  EFI_IMAGE_OPTIONAL_HEADER_UNION PeHdr;\r
\r
  //\r
  // Position to the start of the file\r
  //\r
  fseek (Fptr, 0, SEEK_SET);\r
\r
  //\r
  // Read the DOS header\r
  //\r
  if (fread (&DosHeader, sizeof (DosHeader), 1, Fptr) != 1) {\r
    Error (NULL, 0, 0004, "Failed to read the DOS stub from the input file!", NULL);\r
    return STATUS_ERROR;\r
  }\r
  //\r
  // Check the magic number (0x5A4D)\r
  //\r
  if (DosHeader.e_magic != EFI_IMAGE_DOS_SIGNATURE) {\r
    Error (NULL, 0, 2000, "Invalid parameter", "Input file does not appear to be a PE32 image (magic number)!");\r
    return STATUS_ERROR;\r
  }\r
  //\r
  // Position into the file and check the PE signature\r
  //\r
  fseek (Fptr, (long) DosHeader.e_lfanew, SEEK_SET);\r
\r
  //\r
  // Read PE headers\r
  //\r
  if (fread (&PeHdr, sizeof (PeHdr), 1, Fptr) != 1) {\r
    Error (NULL, 0, 0004, "Failed to read PE/COFF headers from input file!", NULL);\r
    return STATUS_ERROR;\r
  }\r
\r
\r
  //\r
  // Check the PE signature in the header "PE\0\0"\r
  //\r
  if (PeHdr.Pe32.Signature != EFI_IMAGE_NT_SIGNATURE) {\r
    Error (NULL, 0, 2000, "Invalid parameter", "Input file does not appear to be a PE32 image (signature)!");\r
    return STATUS_ERROR;\r
  }\r
\r
  memcpy ((char *) MachineType, &PeHdr.Pe32.FileHeader.Machine, 2);\r
\r
  if (PeHdr.Pe32.OptionalHeader.Magic == EFI_IMAGE_NT_OPTIONAL_HDR32_MAGIC) {\r
    *SubSystem = PeHdr.Pe32.OptionalHeader.Subsystem;\r
  } else if (PeHdr.Pe32Plus.OptionalHeader.Magic == EFI_IMAGE_NT_OPTIONAL_HDR64_MAGIC) {\r
    *SubSystem = PeHdr.Pe32Plus.OptionalHeader.Subsystem;\r
  } else {\r
    Error (NULL, 0, 2000, "Invalid parameter", "Unable to find subsystem type!");\r
    return STATUS_ERROR;\r
  }\r
\r
  if (mOptions.Verbose) {\r
    VerboseMsg("  Got subsystem = 0x%X from image\n", *SubSystem);\r
  }\r
\r
  //\r
  // File was successfully identified as a PE32\r
  //\r
  return STATUS_SUCCESS;\r
}\r
\r
static\r
int\r
ParseCommandLine (\r
  int         Argc,\r
  char        *Argv[],\r
  OPTIONS     *Options\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Given the Argc/Argv program arguments, and a pointer to an options structure,\r
  parse the command-line options and check their validity.\r
\r
\r
Arguments:\r
\r
  Argc            - standard C main() argument count\r
  Argv[]          - standard C main() argument list\r
  Options         - pointer to a structure to store the options in\r
\r
Returns:\r
\r
  STATUS_SUCCESS    success\r
  non-zero          otherwise\r
\r
--*/\r
{\r
  FILE_LIST *FileList;\r
  FILE_LIST *PrevFileList;\r
  UINT32    FileFlags;\r
  UINT32    ClassCode;\r
  UINT32    CodeRevision;\r
  EFI_STATUS Status;\r
  INTN       ReturnStatus;\r
  BOOLEAN    EfiRomFlag;\r
  UINT64     TempValue;\r
  char       *OptionName;\r
  UINT16     *DevIdList;\r
\r
  ReturnStatus = 0;\r
  FileFlags = 0;\r
  EfiRomFlag = FALSE;\r
\r
  //\r
  // Clear out the options\r
  //\r
  memset ((char *) Options, 0, sizeof (OPTIONS));\r
\r
  //\r
  // To avoid compile warnings\r
  //\r
  FileList                = PrevFileList = NULL;\r
\r
  Options->DevIdList      = NULL;\r
  Options->DevIdCount     = 0;\r
\r
  ClassCode               = 0;\r
  CodeRevision            = 0;\r
  //\r
  // Skip over the program name\r
  //\r
  Argc--;\r
  Argv++;\r
\r
  //\r
  // If no arguments, assume they want usage info\r
  //\r
  if (Argc == 0) {\r
    Usage ();\r
    return STATUS_ERROR;\r
  }\r
\r
  if ((stricmp(Argv[0], "-h") == 0) || (stricmp(Argv[0], "--help") == 0)) {\r
    Usage();\r
    return STATUS_ERROR;\r
  }\r
\r
  if ((stricmp(Argv[0], "--version") == 0)) {\r
    Version();\r
    return STATUS_ERROR;\r
  }\r
\r
  //\r
  // Process until no more arguments\r
  //\r
  while (Argc > 0) {\r
    if (Argv[0][0] == '-') {\r
      //\r
      // Vendor ID specified with -f\r
      //\r
      if (stricmp (Argv[0], "-f") == 0) {\r
        //\r
        // Make sure there's another parameter\r
        //\r
        Status = AsciiStringToUint64(Argv[1], FALSE, &TempValue);\r
        if (EFI_ERROR (Status)) {\r
          Error (NULL, 0, 2000, "Invalid option value", "%s = %s", Argv[0], Argv[1]);\r
          ReturnStatus = 1;\r
          goto Done;\r
        }\r
        if (TempValue >= 0x10000) {\r
          Error (NULL, 0, 2000, "Invalid option value", "Vendor Id %s out of range!", Argv[1]);\r
          ReturnStatus = 1;\r
          goto Done;\r
        }\r
        Options->VendId       = (UINT16) TempValue;\r
        Options->VendIdValid  = 1;\r
\r
        Argv++;\r
        Argc--;\r
      } else if (stricmp (Argv[0], "-i") == 0) {\r
\r
        OptionName = Argv[0];\r
\r
        //\r
        // Device IDs specified with -i\r
        // Make sure there's at least one more parameter\r
        //\r
        if (Argc < 1) {\r
          Error (NULL, 0, 2000, "Invalid parameter", "Missing Device Id with %s option!", OptionName);\r
          ReturnStatus = 1;\r
          goto Done;\r
        }\r
\r
        //\r
        // Process until another dash-argument parameter or the end of the list\r
        //\r
        while (Argc > 1 && Argv[1][0] != '-') {\r
          Status = AsciiStringToUint64(Argv[1], FALSE, &TempValue);\r
          if (EFI_ERROR (Status)) {\r
            Error (NULL, 0, 2000, "Invalid option value", "%s = %s", OptionName, Argv[1]);\r
            ReturnStatus = 1;\r
            goto Done;\r
          }\r
          //\r
          // Don't allow device IDs greater than 16 bits\r
          // Don't allow 0, since it is used as a list terminator\r
          //\r
          if (TempValue >= 0x10000 || TempValue == 0) {\r
            Error (NULL, 0, 2000, "Invalid option value", "Device Id %s out of range!", Argv[1]);\r
            ReturnStatus = 1;\r
            goto Done;\r
          }\r
\r
          DevIdList = (UINT16*) realloc (Options->DevIdList, (Options->DevIdCount + 1) * sizeof (UINT16));\r
          if (DevIdList == NULL) {\r
            Error (NULL, 0, 4001, "Resource", "memory cannot be allocated!", NULL);\r
            ReturnStatus = 1;\r
            goto Done;\r
          }\r
          Options->DevIdList = DevIdList;\r
\r
          Options->DevIdList[Options->DevIdCount++] = (UINT16) TempValue;\r
\r
          Argv++;\r
          Argc--;\r
        }\r
\r
      } else if ((stricmp (Argv[0], "-o") == 0) || (stricmp (Argv[0], "--output") == 0)) {\r
        //\r
        // Output filename specified with -o\r
        // Make sure there's another parameter\r
        //\r
        if (Argv[1] == NULL || Argv[1][0] == '-') {\r
          Error (NULL, 0, 2000, "Invalid parameter", "Missing output file name with %s option!", Argv[0]);\r
          ReturnStatus = STATUS_ERROR;\r
          goto Done;\r
        }\r
        if (strlen (Argv[1]) > MAX_PATH - 1) {\r
          Error (NULL, 0, 2000, "Invalid parameter", "Output file name %s is too long!", Argv[1]);\r
          ReturnStatus = STATUS_ERROR;\r
          goto Done;\r
        }\r
        strncpy (Options->OutFileName, Argv[1], MAX_PATH - 1);\r
        Options->OutFileName[MAX_PATH - 1] = 0;\r
\r
        Argv++;\r
        Argc--;\r
      } else if ((stricmp (Argv[0], "-h") == 0) || (stricmp (Argv[0], "--help") == 0)) {\r
        //\r
        // Help option\r
        //\r
        Usage ();\r
        ReturnStatus = STATUS_ERROR;\r
        goto Done;\r
      } else if (stricmp (Argv[0], "-b") == 0) {\r
        //\r
        // Specify binary files with -b\r
        //\r
        FileFlags = FILE_FLAG_BINARY;\r
      } else if ((stricmp (Argv[0], "-e") == 0) || (stricmp (Argv[0], "-ec") == 0)) {\r
        //\r
        // Specify EFI files with -e. Specify EFI-compressed with -c.\r
        //\r
        FileFlags = FILE_FLAG_EFI;\r
        if ((Argv[0][2] == 'c') || (Argv[0][2] == 'C')) {\r
          FileFlags |= FILE_FLAG_COMPRESS;\r
        }\r
        //\r
        // Specify not to set the LAST bit in the last file with -n\r
        //\r
      } else if (stricmp (Argv[0], "-n") == 0) {\r
        Options->NoLast = 1;\r
      } else if (((stricmp (Argv[0], "-v") == 0)) || ((stricmp (Argv[0], "--verbose") == 0))) {\r
        //\r
        // -v for verbose\r
        //\r
        Options->Verbose = 1;\r
      } else if (stricmp (Argv[0], "--debug") == 0) {\r
        Status = AsciiStringToUint64(Argv[1], FALSE, &DebugLevel);\r
        if (EFI_ERROR (Status)) {\r
          Error (NULL, 0, 2000, "Invalid option value", "%s = %s", Argv[0], Argv[1]);\r
          ReturnStatus = 1;\r
          goto Done;\r
        }\r
        if (DebugLevel > 9)  {\r
          Error (NULL, 0, 2000, "Invalid option value", "Debug Level range is 0-9, current input level is %d", Argv[1]);\r
          ReturnStatus = 1;\r
          goto Done;\r
        }\r
        if (DebugLevel>=5 && DebugLevel<=9) {\r
          Options->Debug = TRUE;\r
        } else {\r
          Options->Debug = FALSE;\r
        }\r
        Argv++;\r
        Argc--;\r
      } else if ((stricmp (Argv[0], "--quiet") == 0) || (stricmp (Argv[0], "-q") == 0)) {\r
        Options->Quiet = TRUE;\r
      } else if ((stricmp (Argv[0], "--dump") == 0) || (stricmp (Argv[0], "-d") == 0)) {\r
        //\r
        // -dump for dumping a ROM image. In this case, say that the device id\r
        // and vendor id are valid so we don't have to specify bogus ones on the\r
        // command line.\r
        //\r
        Options->DumpOption   = 1;\r
\r
        Options->VendIdValid  = 1;\r
        Options->DevIdCount   = 1;\r
        FileFlags             = FILE_FLAG_BINARY;\r
      } else if ((stricmp (Argv[0], "-l") == 0) || (stricmp (Argv[0], "--class-code") == 0)) {\r
        //\r
        // Class code value for the next file in the list.\r
        // Make sure there's another parameter\r
        //\r
        Status = AsciiStringToUint64(Argv[1], FALSE, &TempValue);\r
        if (EFI_ERROR (Status)) {\r
          Error (NULL, 0, 2000, "Invalid option value", "%s = %s", Argv[0], Argv[1]);\r
          ReturnStatus = 1;\r
          goto Done;\r
        }\r
        ClassCode = (UINT32) TempValue;\r
        if (ClassCode & 0xFF000000) {\r
          Error (NULL, 0, 2000, "Invalid parameter", "Class code %s out of range!", Argv[1]);\r
          ReturnStatus = STATUS_ERROR;\r
          goto Done;\r
        }\r
        if (FileList != NULL && FileList->ClassCode == 0) {\r
          FileList->ClassCode = ClassCode;\r
        }\r
        Argv++;\r
        Argc--;\r
      } else if ((stricmp (Argv[0], "-r") == 0) || (stricmp (Argv[0], "--Revision") == 0)) {\r
        //\r
        // Code revision in the PCI data structure. The value is for the next\r
        // file in the list.\r
        // Make sure there's another parameter\r
        //\r
        Status = AsciiStringToUint64(Argv[1], FALSE, &TempValue);\r
        if (EFI_ERROR (Status)) {\r
          Error (NULL, 0, 2000, "Invalid option value", "%s = %s", Argv[0], Argv[1]);\r
          ReturnStatus = 1;\r
          goto Done;\r
        }\r
        CodeRevision = (UINT32) TempValue;\r
        if (CodeRevision & 0xFFFF0000) {\r
          Error (NULL, 0, 2000, "Invalid parameter", "Code revision %s out of range!", Argv[1]);\r
          ReturnStatus = STATUS_ERROR;\r
          goto Done;\r
        }\r
        if (FileList != NULL && FileList->CodeRevision == 0) {\r
          FileList->CodeRevision = (UINT16) CodeRevision;\r
        }\r
        Argv++;\r
        Argc--;\r
      } else if ((stricmp (Argv[0], "-p") == 0) || (stricmp (Argv[0], "--pci23") == 0)) {\r
        //\r
        // Default layout meets PCI 3.0 specifications, specifying this flag will for a PCI 2.3 layout.\r
        //\r
        mOptions.Pci23 = 1;\r
      } else {\r
        Error (NULL, 0, 2000, "Invalid parameter", "Invalid option specified: %s", Argv[0]);\r
        ReturnStatus = STATUS_ERROR;\r
        goto Done;\r
      }\r
    } else {\r
      //\r
      // Not a slash-option argument. Must be a file name. Make sure they've specified\r
      // -e or -b already.\r
      //\r
      if ((FileFlags & (FILE_FLAG_BINARY | FILE_FLAG_EFI)) == 0) {\r
        Error (NULL, 0, 2000, "Invalid parameter", "Missing -e or -b with input file %s!", Argv[0]);\r
        ReturnStatus = STATUS_ERROR;\r
        goto Done;\r
      }\r
      //\r
      // Check Efi Option RomImage\r
      //\r
      if ((FileFlags & FILE_FLAG_EFI) == FILE_FLAG_EFI) {\r
        EfiRomFlag = TRUE;\r
      }\r
      //\r
      // Create a new file structure\r
      //\r
      FileList = (FILE_LIST *) malloc (sizeof (FILE_LIST));\r
      if (FileList == NULL) {\r
        Error (NULL, 0, 4001, "Resource", "memory cannot be allocated!", NULL);\r
        ReturnStatus = STATUS_ERROR;\r
        goto Done;\r
      }\r
\r
      //\r
      // set flag and class code for this image.\r
      //\r
      memset ((char *) FileList, 0, sizeof (FILE_LIST));\r
      FileList->FileName      = Argv[0];\r
      FileList->FileFlags     = FileFlags;\r
      FileList->ClassCode     = ClassCode;\r
      FileList->CodeRevision  = (UINT16) CodeRevision;\r
      ClassCode               = 0;\r
      CodeRevision            = 0;\r
\r
      if (Options->FileList == NULL) {\r
        Options->FileList = FileList;\r
      } else {\r
        if (PrevFileList == NULL) {\r
          PrevFileList = FileList;\r
        } else {\r
          PrevFileList->Next = FileList;\r
        }\r
      }\r
\r
      PrevFileList = FileList;\r
    }\r
    //\r
    // Next argument\r
    //\r
    Argv++;\r
    Argc--;\r
  }\r
\r
  //\r
  // Must have specified some files\r
  //\r
  if (Options->FileList == NULL) {\r
    Error (NULL, 0, 2000, "Invalid parameter", "Missing input file name!");\r
    //\r
    // No memory allocation, return directly.\r
    //\r
    return STATUS_ERROR;\r
  }\r
\r
  //\r
  // For EFI OptionRom image, Make sure a device ID and vendor ID are both specified.\r
  //\r
  if (EfiRomFlag) {\r
    if (!Options->VendIdValid) {\r
      Error (NULL, 0, 2000, "Missing Vendor ID in command line", NULL);\r
      ReturnStatus = STATUS_ERROR;\r
      goto Done;\r
    }\r
\r
    if (!Options->DevIdCount) {\r
      Error (NULL, 0, 2000, "Missing Device ID in command line", NULL);\r
      ReturnStatus = STATUS_ERROR;\r
      goto Done;\r
    }\r
  }\r
\r
   if (Options->DevIdCount > 1 && Options->Pci23) {\r
     Error (NULL, 0, 2000, "Invalid parameter", "PCI 3.0 is required when specifying multiple Device IDs");\r
     ReturnStatus = STATUS_ERROR;\r
     goto Done;\r
   }\r
\r
Done:\r
  if (ReturnStatus != 0) {\r
    while (Options->FileList != NULL) {\r
      FileList = Options->FileList->Next;\r
      free (Options->FileList);\r
      Options->FileList = FileList;\r
    }\r
  }\r
\r
  return ReturnStatus;\r
}\r
\r
static\r
void\r
Version (\r
  VOID\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Print version information for this utility.\r
\r
Arguments:\r
\r
  None.\r
\r
Returns:\r
\r
  Nothing.\r
--*/\r
{\r
 fprintf (stdout, "%s Version %d.%d %s \n", UTILITY_NAME, UTILITY_MAJOR_VERSION, UTILITY_MINOR_VERSION, __BUILD_VERSION);\r
}\r
\r
static\r
void\r
Usage (\r
  VOID\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Print usage information for this utility.\r
\r
Arguments:\r
\r
  None.\r
\r
Returns:\r
\r
  Nothing.\r
\r
--*/\r
{\r
  //\r
  // Summary usage\r
  //\r
  fprintf (stdout, "Usage: %s -f VendorId -i DeviceId [options] [file name<s>] \n\n", UTILITY_NAME);\r
\r
  //\r
  // Copyright declaration\r
  //\r
  fprintf (stdout, "Copyright (c) 2007 - 2018, Intel Corporation. All rights reserved.\n\n");\r
\r
  //\r
  // Details Option\r
  //\r
  fprintf (stdout, "Options:\n");\r
  fprintf (stdout, "  -o FileName, --output FileName\n\\r
            File will be created to store the output content.\n");\r
  fprintf (stdout, "  -e EfiFileName\n\\r
            EFI PE32 image files.\n");\r
  fprintf (stdout, "  -ec EfiFileName\n\\r
            EFI PE32 image files and will be compressed.\n");\r
  fprintf (stdout, "  -b BinFileName\n\\r
            Legacy binary files.\n");\r
  fprintf (stdout, "  -l ClassCode\n\\r
            Hex ClassCode in the PCI data structure header.\n");\r
  fprintf (stdout, "  -r Rev    Hex Revision in the PCI data structure header.\n");\r
  fprintf (stdout, "  -n        Not to automatically set the LAST bit in the last file.\n");\r
  fprintf (stdout, "  -f VendorId\n\\r
            Hex PCI Vendor ID for the device OpROM, must be specified\n");\r
  fprintf (stdout, "  -i DeviceId\n\\r
            One or more hex PCI Device IDs for the device OpROM, must be specified\n");\r
  fprintf (stdout, "  -p, --pci23\n\\r
            Default layout meets PCI 3.0 specifications\n\\r
            specifying this flag will for a PCI 2.3 layout.\n");\r
  fprintf (stdout, "  -d, --dump\n\\r
            Dump the headers of an existing option ROM image.\n");\r
  fprintf (stdout, "  -v, --verbose\n\\r
            Turn on verbose output with informational messages.\n");\r
  fprintf (stdout, "  --version Show program's version number and exit.\n");\r
  fprintf (stdout, "  -h, --help\n\\r
            Show this help message and exit.\n");\r
  fprintf (stdout, "  -q, --quiet\n\\r
            Disable all messages except FATAL ERRORS.\n");\r
  fprintf (stdout, "  --debug [#,0-9]\n\\r
            Enable debug messages at level #.\n");\r
}\r
\r
static\r
void\r
DumpImage (\r
  FILE_LIST *InFile\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Dump the headers of an existing option ROM image\r
\r
Arguments:\r
\r
  InFile  - the file name of an existing option ROM image\r
\r
Returns:\r
\r
  none\r
\r
--*/\r
{\r
  PCI_EXPANSION_ROM_HEADER      PciRomHdr;\r
  FILE                          *InFptr;\r
  UINT32                        ImageStart;\r
  UINT32                        ImageCount;\r
  EFI_PCI_EXPANSION_ROM_HEADER  EfiRomHdr;\r
  PCI_DATA_STRUCTURE            PciDs23;\r
  PCI_3_0_DATA_STRUCTURE        PciDs30;\r
  UINT16                        DevId;\r
\r
  //\r
  // Open the input file\r
  //\r
  if ((InFptr = fopen (LongFilePath (InFile->FileName), "rb")) == NULL) {\r
    Error (NULL, 0, 0001, "Error opening file", InFile->FileName);\r
    return ;\r
  }\r
  //\r
  // Go through the image and dump the header stuff for each\r
  //\r
  ImageCount = 0;\r
  for (;;) {\r
    //\r
    // Save our position in the file, since offsets in the headers\r
    // are relative to the particular image.\r
    //\r
    ImageStart = ftell (InFptr);\r
    ImageCount++;\r
\r
    //\r
    // Read the option ROM header. Have to assume a raw binary image for now.\r
    //\r
    if (fread (&PciRomHdr, sizeof (PciRomHdr), 1, InFptr) != 1) {\r
      Error (NULL, 0, 3001, "Not supported", "Failed to read PCI ROM header from file!");\r
      goto BailOut;\r
    }\r
\r
    //\r
    // Dump the contents of the header\r
    //\r
    fprintf (stdout, "Image %u -- Offset 0x%X\n", (unsigned) ImageCount, (unsigned) ImageStart);\r
    fprintf (stdout, "  ROM header contents\n");\r
    fprintf (stdout, "    Signature              0x%04X\n", PciRomHdr.Signature);\r
    fprintf (stdout, "    PCIR offset            0x%04X\n", PciRomHdr.PcirOffset);\r
    //\r
    // Find PCI data structure\r
    //\r
    if (fseek (InFptr, ImageStart + PciRomHdr.PcirOffset, SEEK_SET)) {\r
      Error (NULL, 0, 3001, "Not supported", "Failed to seek to PCI data structure!");\r
      goto BailOut;\r
    }\r
    //\r
    // Read and dump the PCI data structure\r
    //\r
    memset (&PciDs23, 0, sizeof (PciDs23));\r
    memset (&PciDs30, 0, sizeof (PciDs30));\r
    if (mOptions.Pci23 == 1) {\r
      if (fread (&PciDs23, sizeof (PciDs23), 1, InFptr) != 1) {\r
        Error (NULL, 0, 3001, "Not supported", "Failed to read PCI data structure from file %s!", InFile->FileName);\r
        goto BailOut;\r
      }\r
    } else {\r
      if (fread (&PciDs30, sizeof (PciDs30), 1, InFptr) != 1) {\r
        Error (NULL, 0, 3001, "Not supported", "Failed to read PCI data structure from file %s!", InFile->FileName);\r
        goto BailOut;\r
      }\r
    }\r
    if (mOptions.Verbose) {\r
      VerboseMsg("Read PCI data structure from file %s", InFile->FileName);\r
    }\r
\r
    //fprintf (stdout, "  PCI Data Structure\n");\r
    if (mOptions.Pci23 == 1) {\r
    fprintf (\r
      stdout,\r
      "    Signature              %c%c%c%c\n",\r
      (char) PciDs23.Signature,\r
      (char) (PciDs23.Signature >> 8),\r
      (char) (PciDs23.Signature >> 16),\r
      (char) (PciDs23.Signature >> 24)\r
      );\r
    fprintf (stdout, "    Vendor ID              0x%04X\n", PciDs23.VendorId);\r
    fprintf (stdout, "    Device ID              0x%04X\n", PciDs23.DeviceId);\r
    fprintf (stdout, "    Length                 0x%04X\n", PciDs23.Length);\r
    fprintf (stdout, "    Revision               0x%04X\n", PciDs23.Revision);\r
    fprintf (\r
      stdout,\r
      "    Class Code             0x%06X\n",\r
      (unsigned) (PciDs23.ClassCode[0] | (PciDs23.ClassCode[1] << 8) | (PciDs23.ClassCode[2] << 16))\r
      );\r
    fprintf (stdout, "    Image size             0x%X\n", (unsigned) PciDs23.ImageLength * 512);\r
    fprintf (stdout, "    Code revision:         0x%04X\n", PciDs23.CodeRevision);\r
    fprintf (stdout, "    Indicator              0x%02X", PciDs23.Indicator);\r
    } else {\r
    fprintf (\r
      stdout,\r
      "    Signature               %c%c%c%c\n",\r
      (char) PciDs30.Signature,\r
      (char) (PciDs30.Signature >> 8),\r
      (char) (PciDs30.Signature >> 16),\r
      (char) (PciDs30.Signature >> 24)\r
      );\r
    fprintf (stdout, "    Vendor ID               0x%04X\n", PciDs30.VendorId);\r
    fprintf (stdout, "    Device ID               0x%04X\n", PciDs30.DeviceId);\r
    fprintf (stdout, "    Length                  0x%04X\n", PciDs30.Length);\r
    fprintf (stdout, "    Revision                0x%04X\n", PciDs30.Revision);\r
    fprintf (stdout, "    DeviceListOffset        0x%02X\n", PciDs30.DeviceListOffset);\r
    if (PciDs30.DeviceListOffset) {\r
      //\r
      // Print device ID list\r
      //\r
      fprintf (stdout, "    Device list contents\n");\r
      if (fseek (InFptr, ImageStart + PciRomHdr.PcirOffset + PciDs30.DeviceListOffset, SEEK_SET)) {\r
        Error (NULL, 0, 3001, "Not supported", "Failed to seek to PCI device ID list!");\r
        goto BailOut;\r
      }\r
\r
      //\r
      // Loop until terminating 0\r
      //\r
      do {\r
        if (fread (&DevId, sizeof (DevId), 1, InFptr) != 1) {\r
          Error (NULL, 0, 3001, "Not supported", "Failed to read PCI device ID list from file %s!", InFile->FileName);\r
          goto BailOut;\r
        }\r
        if (DevId) {\r
          fprintf (stdout, "      0x%04X\n", DevId);\r
        }\r
      } while (DevId);\r
\r
    }\r
    fprintf (\r
      stdout,\r
      "    Class Code              0x%06X\n",\r
      (unsigned) (PciDs30.ClassCode[0] | (PciDs30.ClassCode[1] << 8) | (PciDs30.ClassCode[2] << 16))\r
      );\r
    fprintf (stdout, "    Image size              0x%X\n", (unsigned) PciDs30.ImageLength * 512);\r
    fprintf (stdout, "    Code revision:          0x%04X\n", PciDs30.CodeRevision);\r
    fprintf (stdout, "    MaxRuntimeImageLength   0x%02X\n", PciDs30.MaxRuntimeImageLength);\r
    fprintf (stdout, "    ConfigUtilityCodeHeaderOffset 0x%02X\n", PciDs30.ConfigUtilityCodeHeaderOffset);\r
    fprintf (stdout, "    DMTFCLPEntryPointOffset 0x%02X\n", PciDs30.DMTFCLPEntryPointOffset);\r
    fprintf (stdout, "    Indicator               0x%02X", PciDs30.Indicator);\r
    }\r
    //\r
    // Print the indicator, used to flag the last image\r
    //\r
    if (PciDs23.Indicator == INDICATOR_LAST || PciDs30.Indicator == INDICATOR_LAST) {\r
      fprintf (stdout, "   (last image)\n");\r
    } else {\r
      fprintf (stdout, "\n");\r
    }\r
    //\r
    // Print the code type. If EFI code, then we can provide more info.\r
    //\r
    if (mOptions.Pci23 == 1) {\r
      fprintf (stdout, "    Code type              0x%02X", PciDs23.CodeType);\r
    } else {\r
      fprintf (stdout, "    Code type               0x%02X", PciDs30.CodeType);\r
    }\r
    if (PciDs23.CodeType == PCI_CODE_TYPE_EFI_IMAGE || PciDs30.CodeType == PCI_CODE_TYPE_EFI_IMAGE) {\r
      fprintf (stdout, "   (EFI image)\n");\r
      //\r
      // Re-read the header as an EFI ROM header, then dump more info\r
      //\r
      fprintf (stdout, "  EFI ROM header contents\n");\r
      if (fseek (InFptr, ImageStart, SEEK_SET)) {\r
        Error (NULL, 0, 5001, "Failed to re-seek to ROM header structure!", NULL);\r
        goto BailOut;\r
      }\r
\r
      if (fread (&EfiRomHdr, sizeof (EfiRomHdr), 1, InFptr) != 1) {\r
        Error (NULL, 0, 5001, "Failed to read EFI PCI ROM header from file!", NULL);\r
        goto BailOut;\r
      }\r
      //\r
      // Now dump more info\r
      //\r
      fprintf (stdout, "    EFI Signature          0x%04X\n", (unsigned) EfiRomHdr.EfiSignature);\r
      fprintf (\r
        stdout,\r
        "    Compression Type       0x%04X ",\r
        EfiRomHdr.CompressionType\r
        );\r
      if (EfiRomHdr.CompressionType == EFI_PCI_EXPANSION_ROM_HEADER_COMPRESSED) {\r
        fprintf (stdout, "(compressed)\n");\r
      } else {\r
        fprintf (stdout, "(not compressed)\n");\r
      }\r
\r
      fprintf (\r
        stdout,\r
        "    Machine type           0x%04X (%s)\n",\r
        EfiRomHdr.EfiMachineType,\r
        GetMachineTypeStr (EfiRomHdr.EfiMachineType)\r
        );\r
      fprintf (\r
        stdout,\r
        "    Subsystem              0x%04X (%s)\n",\r
        EfiRomHdr.EfiSubsystem,\r
        GetSubsystemTypeStr (EfiRomHdr.EfiSubsystem)\r
        );\r
      fprintf (\r
        stdout,\r
        "    EFI image offset       0x%04X (@0x%X)\n",\r
        EfiRomHdr.EfiImageHeaderOffset,\r
        EfiRomHdr.EfiImageHeaderOffset + (unsigned) ImageStart\r
        );\r
\r
    } else {\r
      //\r
      // Not an EFI image\r
      //\r
      fprintf (stdout, "\n");\r
    }\r
    //\r
    // If code type is EFI image, then dump it as well?\r
    //\r
    // if (PciDs.CodeType == PCI_CODE_TYPE_EFI_IMAGE) {\r
    // }\r
    //\r
    // If last image, then we're done\r
    //\r
    if (PciDs23.Indicator == INDICATOR_LAST || PciDs30.Indicator == INDICATOR_LAST) {\r
      goto BailOut;\r
    }\r
    //\r
    // Seek to the start of the next image\r
    //\r
    if (mOptions.Pci23 == 1) {\r
      if (fseek (InFptr, ImageStart + (PciDs23.ImageLength * 512), SEEK_SET)) {\r
        Error (NULL, 0, 3001, "Not supported", "Failed to seek to next image!");\r
        goto BailOut;\r
      }\r
    } else {\r
      if (fseek (InFptr, ImageStart + (PciDs30.ImageLength * 512), SEEK_SET)) {\r
        Error (NULL, 0, 3001, "Not supported", "Failed to seek to next image!");\r
        goto BailOut;\r
      }\r
    }\r
  }\r
\r
BailOut:\r
  fclose (InFptr);\r
}\r
\r
char *\r
GetMachineTypeStr (\r
  UINT16    MachineType\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  GC_TODO: Add function description\r
\r
Arguments:\r
\r
  MachineType - GC_TODO: add argument description\r
\r
Returns:\r
\r
  GC_TODO: add return values\r
\r
--*/\r
{\r
  int Index;\r
\r
  for (Index = 0; mMachineTypes[Index].Name != NULL; Index++) {\r
    if (mMachineTypes[Index].Value == MachineType) {\r
      return mMachineTypes[Index].Name;\r
    }\r
  }\r
\r
  return "unknown";\r
}\r
\r
static\r
char *\r
GetSubsystemTypeStr (\r
  UINT16  SubsystemType\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  GC_TODO: Add function description\r
\r
Arguments:\r
\r
  SubsystemType - GC_TODO: add argument description\r
\r
Returns:\r
\r
  GC_TODO: add return values\r
\r
--*/\r
{\r
  int Index;\r
\r
  for (Index = 0; mSubsystemTypes[Index].Name != NULL; Index++) {\r
    if (mSubsystemTypes[Index].Value == SubsystemType) {\r
      return mSubsystemTypes[Index].Name;\r
    }\r
  }\r
\r
  return "unknown";\r
}\r