+++ /dev/null
-/*++\r
-\r
-Copyright (c) 2004 - 2010, 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
-Module Name:\r
-\r
- fwimage.c\r
-\r
-Abstract:\r
-\r
- Converts a pe32/pe32+ image to an FW image type\r
-\r
---*/\r
-\r
-#include <windows.h>\r
-#include <stdio.h>\r
-#include <stdlib.h>\r
-#include <string.h>\r
-#include <time.h>\r
-\r
-#include "TianoCommon.h"\r
-#include "EfiImage.h"\r
-#include "EfiUtilityMsgs.c"\r
-\r
-#define UTILITY_NAME "FwImage"\r
-#define UTILITY_VERSION "v1.0"\r
-\r
-typedef union {\r
- IMAGE_NT_HEADERS32 PeHeader32;\r
- IMAGE_NT_HEADERS64 PeHeader64;\r
-} PE_HEADER;\r
-\r
-VOID\r
-Usage (\r
- VOID\r
- )\r
-{\r
- int Index;\r
- const char *Str[] = {\r
- UTILITY_NAME" "UTILITY_VERSION" - Intel Firmware Image Utility",\r
- " Copyright (C), 2004 - 2008 Intel Corporation",\r
-\r
-#if ( defined(UTILITY_BUILD) && defined(UTILITY_VENDOR) )\r
- " Built from "UTILITY_BUILD", project of "UTILITY_VENDOR,\r
-#endif\r
-\r
- "",\r
- "Usage:",\r
- " "UTILITY_NAME" [OPTION]... FWTYPE SOURCE [DEST]",\r
- "Description:",\r
- " Converts a pe32/pe32+ SOURCE to DEST with FWTYPE image type.",\r
- "Options:",\r
- " FWTYPE Can be one of APPLICATION, BS_DRIVER, RT_DRIVER, SAL_RT_DRIVER,",\r
- " COMBINED_PEIM_DRIVER, SECURITY_CORE, PEI_CORE, PE32_PEIM and",\r
- " RELOCATABLE_PEIM",\r
- " -t time-date Add Time Stamp for output image",\r
- " -e Not clear ExceptionTable for output image",\r
- " -r Not strip zero pending of .reloc for output image",\r
- NULL\r
- };\r
-\r
- for (Index = 0; Str[Index] != NULL; Index++) {\r
- fprintf (stdout, "%s\n", Str[Index]);\r
- } \r
-}\r
-\r
-static\r
-STATUS\r
-FReadFile (\r
- FILE *in,\r
- VOID **Buffer,\r
- UINTN *Length\r
- )\r
-{\r
- fseek (in, 0, SEEK_END);\r
- *Length = ftell (in);\r
- *Buffer = malloc (*Length);\r
- fseek (in, 0, SEEK_SET);\r
- fread (*Buffer, *Length, 1, in);\r
- return STATUS_SUCCESS;\r
-}\r
-\r
-static\r
-STATUS\r
-FWriteFile (\r
- FILE *out,\r
- VOID *Buffer,\r
- UINTN Length\r
- )\r
-{\r
- fseek (out, 0, SEEK_SET);\r
- fwrite (Buffer, Length, 1, out);\r
- if ((ULONG) ftell (out) != Length) {\r
- Error (NULL, 0, 0, "write error", NULL);\r
- return STATUS_ERROR;\r
- }\r
- free (Buffer);\r
- return STATUS_SUCCESS;\r
-}\r
-\r
-VOID\r
-ZeroExceptionTable (\r
- IN UINT8 *FileBuffer,\r
- IN EFI_IMAGE_DOS_HEADER *DosHdr,\r
- IN PE_HEADER *PeHdr\r
- )\r
-{\r
- UINT32 PdataSize;\r
- UINT32 PdataOffset;\r
- UINT32 PdataRVASize;\r
- UINT32 PdataRVA;\r
- UINT32 SectionOffset;\r
- UINT16 SectionNumber;\r
- UINT32 SectionNameSize;\r
- EFI_IMAGE_SECTION_HEADER *Section;\r
-\r
- PdataSize = 0;\r
- PdataOffset = 0;\r
- PdataRVASize = 0;\r
- PdataRVA = 0;\r
- SectionOffset = 0;\r
-\r
- //\r
- // Search .pdata section\r
- //\r
- if (PeHdr->PeHeader32.OptionalHeader.Magic == IMAGE_NT_OPTIONAL_HDR32_MAGIC) {\r
- if ((PeHdr->PeHeader32.OptionalHeader.NumberOfRvaAndSizes > IMAGE_DIRECTORY_ENTRY_EXCEPTION) &&\r
- (PeHdr->PeHeader32.OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_EXCEPTION].VirtualAddress != 0) &&\r
- (PeHdr->PeHeader32.OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_EXCEPTION].Size != 0)) {\r
-\r
- PdataRVA = PeHdr->PeHeader32.OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_EXCEPTION].VirtualAddress;\r
- PdataRVASize = PeHdr->PeHeader32.OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_EXCEPTION].Size;\r
-\r
- PeHdr->PeHeader32.OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_EXCEPTION].VirtualAddress = 0;\r
- PeHdr->PeHeader32.OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_EXCEPTION].Size = 0;\r
-\r
- SectionOffset = sizeof(PeHdr->PeHeader32);\r
- }\r
- } else {\r
- if ((PeHdr->PeHeader64.OptionalHeader.NumberOfRvaAndSizes > IMAGE_DIRECTORY_ENTRY_EXCEPTION) &&\r
- (PeHdr->PeHeader64.OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_EXCEPTION].VirtualAddress != 0) &&\r
- (PeHdr->PeHeader64.OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_EXCEPTION].Size != 0)) {\r
-\r
- PdataRVA = PeHdr->PeHeader64.OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_EXCEPTION].VirtualAddress;\r
- PdataRVASize = PeHdr->PeHeader64.OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_EXCEPTION].Size;\r
-\r
- PeHdr->PeHeader64.OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_EXCEPTION].VirtualAddress = 0;\r
- PeHdr->PeHeader64.OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_EXCEPTION].Size = 0;\r
-\r
- SectionOffset = sizeof(PeHdr->PeHeader64);\r
- }\r
- }\r
-\r
- if ((PdataRVASize != 0) && (PdataRVA != 0)) {\r
-\r
- SectionNumber = PeHdr->PeHeader32.FileHeader.NumberOfSections;\r
- SectionNameSize = sizeof(Section->Name);\r
- while (SectionNumber > 0) {\r
- Section = (EFI_IMAGE_SECTION_HEADER *) &FileBuffer[DosHdr->e_lfanew + SectionOffset];\r
- if (strcmp (Section->Name, ".pdata") == 0) {\r
- //\r
- // Zero .pdata Section Header Name\r
- //\r
- memset (\r
- FileBuffer + DosHdr->e_lfanew + SectionOffset,\r
- 0,\r
- SectionNameSize);\r
-\r
- //\r
- // Zero .pdata Secton raw data\r
- //\r
- PdataOffset = Section->PointerToRawData;\r
- PdataSize = Section->SizeOfRawData;\r
- memset (FileBuffer + PdataOffset, 0, PdataSize);\r
- break;\r
- }\r
- SectionNumber--;\r
- SectionOffset += sizeof(EFI_IMAGE_SECTION_HEADER);\r
- }\r
- }\r
- \r
- return ;\r
-}\r
-\r
-VOID\r
-StripZeroPendingReloc (\r
- IN UINT8 *FileBuffer,\r
- IN OUT UINTN *FileLength,\r
- IN EFI_IMAGE_DOS_HEADER *DosHdr,\r
- IN PE_HEADER *PeHdr\r
- )\r
-{\r
- EFI_IMAGE_OPTIONAL_HEADER32 *Optional32;\r
- EFI_IMAGE_OPTIONAL_HEADER64 *Optional64;\r
- EFI_IMAGE_SECTION_HEADER *SectionHeader;\r
- UINTN AllignedRelocSize;\r
- UINTN Index;\r
-\r
- if (PeHdr->PeHeader32.OptionalHeader.Magic == IMAGE_NT_OPTIONAL_HDR32_MAGIC) {\r
- Optional32 = (EFI_IMAGE_OPTIONAL_HEADER32 *)&PeHdr->PeHeader32.OptionalHeader;\r
- if ((Optional32->NumberOfRvaAndSizes > EFI_IMAGE_DIRECTORY_ENTRY_BASERELOC) &&\r
- (Optional32->DataDirectory[EFI_IMAGE_DIRECTORY_ENTRY_BASERELOC].Size != 0)) {\r
- SectionHeader = (EFI_IMAGE_SECTION_HEADER *)(FileBuffer + DosHdr->e_lfanew + sizeof(UINT32) + sizeof (EFI_IMAGE_FILE_HEADER) + PeHdr->PeHeader32.FileHeader.SizeOfOptionalHeader);\r
- for (Index = 0; Index < PeHdr->PeHeader32.FileHeader.NumberOfSections; Index++, SectionHeader++) {\r
- //\r
- // Look for the Section Header that starts as the same virtual address as the Base Relocation Data Directory\r
- //\r
- if (strcmp (SectionHeader->Name, ".reloc") == 0) {\r
- SectionHeader->Misc.VirtualSize = Optional32->DataDirectory[EFI_IMAGE_DIRECTORY_ENTRY_BASERELOC].Size;\r
-\r
- AllignedRelocSize = (Optional32->DataDirectory[EFI_IMAGE_DIRECTORY_ENTRY_BASERELOC].Size +\r
- Optional32->FileAlignment - 1) & (~(Optional32->FileAlignment - 1));\r
- //\r
- // Check to see if there is zero padding at the end of the base relocations\r
- //\r
- if (AllignedRelocSize < SectionHeader->SizeOfRawData) {\r
- //\r
- // Check to see if the base relocations are at the end of the file\r
- //\r
- if (SectionHeader->PointerToRawData + SectionHeader->SizeOfRawData == Optional32->SizeOfImage) {\r
- //\r
- // All the required conditions are met to strip the zero padding of the end of the base relocations section\r
- //\r
- Optional32->SizeOfImage -= (SectionHeader->SizeOfRawData - AllignedRelocSize);\r
- Optional32->SizeOfInitializedData -= (SectionHeader->SizeOfRawData - AllignedRelocSize);\r
- SectionHeader->SizeOfRawData = AllignedRelocSize;\r
- *FileLength = Optional32->SizeOfImage;\r
- }\r
- }\r
- }\r
- }\r
- }\r
- } else {\r
- Optional64 = (EFI_IMAGE_OPTIONAL_HEADER64 *)&PeHdr->PeHeader64.OptionalHeader;\r
- if ((Optional64->NumberOfRvaAndSizes > EFI_IMAGE_DIRECTORY_ENTRY_BASERELOC) &&\r
- (Optional64->DataDirectory[EFI_IMAGE_DIRECTORY_ENTRY_BASERELOC].Size != 0)) {\r
- SectionHeader = (EFI_IMAGE_SECTION_HEADER *)(FileBuffer + DosHdr->e_lfanew + sizeof(UINT32) + sizeof (EFI_IMAGE_FILE_HEADER) + PeHdr->PeHeader64.FileHeader.SizeOfOptionalHeader);\r
- for (Index = 0; Index < PeHdr->PeHeader64.FileHeader.NumberOfSections; Index++, SectionHeader++) {\r
- //\r
- // Look for the Section Header that starts as the same virtual address as the Base Relocation Data Directory\r
- //\r
- if (strcmp (SectionHeader->Name, ".reloc") == 0) {\r
- SectionHeader->Misc.VirtualSize = Optional64->DataDirectory[EFI_IMAGE_DIRECTORY_ENTRY_BASERELOC].Size;\r
-\r
- AllignedRelocSize = (Optional64->DataDirectory[EFI_IMAGE_DIRECTORY_ENTRY_BASERELOC].Size +\r
- Optional64->FileAlignment - 1) & (~(Optional64->FileAlignment - 1));\r
- //\r
- // Check to see if there is zero padding at the end of the base relocations\r
- //\r
- if (AllignedRelocSize < SectionHeader->SizeOfRawData) {\r
- //\r
- // Check to see if the base relocations are at the end of the file\r
- //\r
- if (SectionHeader->PointerToRawData + SectionHeader->SizeOfRawData == Optional64->SizeOfImage) {\r
- //\r
- // All the required conditions are met to strip the zero padding of the end of the base relocations section\r
- //\r
- Optional64->SizeOfImage -= (SectionHeader->SizeOfRawData - AllignedRelocSize);\r
- Optional64->SizeOfInitializedData -= (SectionHeader->SizeOfRawData - AllignedRelocSize);\r
- SectionHeader->SizeOfRawData = AllignedRelocSize;\r
- *FileLength = Optional64->SizeOfImage;\r
- }\r
- }\r
- }\r
- }\r
- }\r
- }\r
-}\r
-\r
-int\r
-main (\r
- int argc,\r
- char *argv[]\r
- )\r
-/*++\r
-\r
-Routine Description:\r
-\r
- Main function.\r
-\r
-Arguments:\r
-\r
- argc - Number of command line parameters.\r
- argv - Array of pointers to command line parameter strings.\r
-\r
-Returns:\r
-\r
- STATUS_SUCCESS - Utility exits successfully.\r
- STATUS_ERROR - Some error occurred during execution.\r
-\r
---*/\r
-{\r
- ULONG Type;\r
- PUCHAR Ext;\r
- PUCHAR p;\r
- PUCHAR pe;\r
- PUCHAR OutImageName;\r
- UCHAR outname[500];\r
- FILE *fpIn;\r
- FILE *fpOut;\r
- EFI_IMAGE_DOS_HEADER *DosHdr;\r
- PE_HEADER *PeHdr;\r
- time_t TimeStamp;\r
- struct tm TimeStruct;\r
- EFI_IMAGE_DOS_HEADER BackupDosHdr;\r
- ULONG Index;\r
- BOOLEAN TimeStampPresent;\r
- BOOLEAN NeedClearExceptionTable;\r
- BOOLEAN NeedStripZeroPendingReloc;\r
- UINT8 *FileBuffer;\r
- UINTN FileLength;\r
- EFI_IMAGE_OPTIONAL_HEADER32 *Optional32;\r
- EFI_IMAGE_OPTIONAL_HEADER64 *Optional64;\r
-\r
- SetUtilityName (UTILITY_NAME);\r
- //\r
- // Assign to fix compile warning\r
- //\r
- OutImageName = NULL;\r
- Type = 0;\r
- Ext = 0;\r
- TimeStamp = 0;\r
- TimeStampPresent = FALSE;\r
-\r
- NeedClearExceptionTable = TRUE;\r
- NeedStripZeroPendingReloc = TRUE;\r
-\r
- //\r
- // Look for -t time-date option first. If the time is "0", then\r
- // skip it.\r
- //\r
- if ((argc > 2) && !strcmp (argv[1], "-t")) {\r
- TimeStampPresent = TRUE;\r
- if (strcmp (argv[2], "0") != 0) {\r
- //\r
- // Convert the string to a value\r
- //\r
- memset ((char *) &TimeStruct, 0, sizeof (TimeStruct));\r
- if (sscanf(\r
- argv[2], "%d/%d/%d,%d:%d:%d",\r
- &TimeStruct.tm_mon, /* months since January - [0,11] */\r
- &TimeStruct.tm_mday, /* day of the month - [1,31] */\r
- &TimeStruct.tm_year, /* years since 1900 */\r
- &TimeStruct.tm_hour, /* hours since midnight - [0,23] */\r
- &TimeStruct.tm_min, /* minutes after the hour - [0,59] */\r
- &TimeStruct.tm_sec /* seconds after the minute - [0,59] */\r
- ) != 6) {\r
- Error (NULL, 0, 0, argv[2], "failed to convert to mm/dd/yyyy,hh:mm:ss format");\r
- return STATUS_ERROR;\r
- }\r
- //\r
- // Now fixup some of the fields\r
- //\r
- TimeStruct.tm_mon--;\r
- TimeStruct.tm_year -= 1900;\r
- //\r
- // Sanity-check values?\r
- // Convert\r
- //\r
- TimeStamp = mktime (&TimeStruct);\r
- if (TimeStamp == (time_t) - 1) {\r
- Error (NULL, 0, 0, argv[2], "failed to convert time");\r
- return STATUS_ERROR;\r
- }\r
- }\r
- //\r
- // Skip over the args\r
- //\r
- argc -= 2;\r
- argv += 2;\r
- }\r
-\r
- //\r
- // Look for -e option.\r
- //\r
- if ((argc > 1) && !strcmp (argv[1], "-e")) {\r
- NeedClearExceptionTable = FALSE;\r
- //\r
- // Skip over the args\r
- //\r
- argc -= 1;\r
- argv += 1;\r
- }\r
-\r
- //\r
- // Look for -r option\r
- //\r
- if ((argc > 1) && !strcmp (argv[1], "-r")) {\r
- NeedStripZeroPendingReloc = FALSE;\r
- //\r
- // Skip over the args\r
- //\r
- argc -= 1;\r
- argv += 1;\r
- }\r
-\r
- //\r
- // Check for enough args\r
- //\r
- if (argc < 3) {\r
- Usage ();\r
- return STATUS_ERROR;\r
- }\r
-\r
- if (argc == 4) {\r
- OutImageName = argv[3];\r
- }\r
- //\r
- // Get new image type\r
- //\r
- p = argv[1];\r
- if (*p == '/' || *p == '\\') {\r
- p += 1;\r
- }\r
-\r
- if (_stricmp (p, "app") == 0 || _stricmp (p, "APPLICATION") == 0) {\r
- Type = EFI_IMAGE_SUBSYSTEM_EFI_APPLICATION;\r
- Ext = ".efi";\r
-\r
- } else if (_stricmp (p, "bsdrv") == 0 || _stricmp (p, "BS_DRIVER") == 0) {\r
- Type = EFI_IMAGE_SUBSYSTEM_EFI_BOOT_SERVICE_DRIVER;\r
- Ext = ".efi";\r
-\r
- } else if (_stricmp (p, "rtdrv") == 0 || _stricmp (p, "RT_DRIVER") == 0) {\r
- Type = EFI_IMAGE_SUBSYSTEM_EFI_RUNTIME_DRIVER;\r
- Ext = ".efi";\r
-\r
- } else if (_stricmp (p, "rtdrv") == 0 || _stricmp (p, "SAL_RT_DRIVER") == 0) {\r
- Type = EFI_IMAGE_SUBSYSTEM_SAL_RUNTIME_DRIVER;\r
- Ext = ".efi";\r
- } else if (_stricmp (p, "SECURITY_CORE") == 0) {\r
- Type = EFI_IMAGE_SUBSYSTEM_EFI_BOOT_SERVICE_DRIVER;\r
- Ext = ".sec";\r
- } else if (_stricmp (p, "peim") == 0 ||\r
- _stricmp (p, "PEI_CORE") == 0 ||\r
- _stricmp (p, "PE32_PEIM") == 0 ||\r
- _stricmp (p, "RELOCATABLE_PEIM") == 0 ||\r
- _stricmp (p, "combined_peim_driver") == 0\r
- ) {\r
- Type = EFI_IMAGE_SUBSYSTEM_EFI_BOOT_SERVICE_DRIVER;\r
- Ext = ".pei";\r
- } else {\r
- Usage ();\r
- return STATUS_ERROR;\r
- }\r
- //\r
- // open source file\r
- //\r
- fpIn = fopen (argv[2], "rb");\r
- if (!fpIn) {\r
- Error (NULL, 0, 0, argv[2], "failed to open input file for reading");\r
- return STATUS_ERROR;\r
- }\r
- FReadFile (fpIn, (VOID **)&FileBuffer, &FileLength);\r
- //\r
- // Read the dos & pe hdrs of the image\r
- //\r
- DosHdr = (EFI_IMAGE_DOS_HEADER *) FileBuffer;\r
- if (DosHdr->e_magic != IMAGE_DOS_SIGNATURE) {\r
- Error (NULL, 0, 0, argv[2], "DOS header signature not found in source image");\r
- fclose (fpIn);\r
- return STATUS_ERROR;\r
- }\r
-\r
- PeHdr = (PE_HEADER *)(FileBuffer + DosHdr->e_lfanew);\r
- if (PeHdr->PeHeader32.Signature != IMAGE_NT_SIGNATURE) {\r
- Error (NULL, 0, 0, argv[2], "PE header signature not found in source image");\r
- fclose (fpIn);\r
- return STATUS_ERROR;\r
- }\r
- //\r
- // open output file\r
- //\r
- strcpy (outname, argv[2]);\r
- pe = NULL;\r
- for (p = outname; *p; p++) {\r
- if (*p == '.') {\r
- pe = p;\r
- }\r
- }\r
-\r
- if (!pe) {\r
- pe = p;\r
- }\r
-\r
- strcpy (pe, Ext);\r
-\r
- if (!OutImageName) {\r
- OutImageName = outname;\r
- }\r
-\r
- fpOut = fopen (OutImageName, "w+b");\r
- if (!fpOut) {\r
- Error (NULL, 0, 0, OutImageName, "could not open output file for writing");\r
- fclose (fpIn);\r
- return STATUS_ERROR;\r
- }\r
- //\r
- // Zero all unused fields of the DOS header\r
- //\r
- memcpy (&BackupDosHdr, DosHdr, sizeof (EFI_IMAGE_DOS_HEADER));\r
- memset (DosHdr, 0, sizeof (EFI_IMAGE_DOS_HEADER));\r
- DosHdr->e_magic = BackupDosHdr.e_magic;\r
- DosHdr->e_lfanew = BackupDosHdr.e_lfanew;\r
-\r
- for (Index = sizeof (EFI_IMAGE_DOS_HEADER); Index < (ULONG) DosHdr->e_lfanew; Index++) {\r
- FileBuffer[Index] = (UINT8) DosHdr->e_cp;\r
- }\r
- \r
- //\r
- // Modify some fields in the PE header\r
- //\r
-\r
- //\r
- // TimeDateStamp's offset is fixed for PE32/32+\r
- //\r
- if (TimeStampPresent) {\r
- PeHdr->PeHeader32.FileHeader.TimeDateStamp = (UINT32) TimeStamp;\r
- }\r
-\r
- //\r
- // PE32/32+ has different optional header layout\r
- // Determine format is PE32 or PE32+ before modification\r
- //\r
- if (PeHdr->PeHeader32.OptionalHeader.Magic == IMAGE_NT_OPTIONAL_HDR32_MAGIC) {\r
- //\r
- // PE32 image\r
- //\r
- Optional32 = (EFI_IMAGE_OPTIONAL_HEADER32 *)&PeHdr->PeHeader32.OptionalHeader;\r
-\r
- Optional32->MajorLinkerVersion = 0;\r
- Optional32->MinorLinkerVersion = 0;\r
- Optional32->MajorOperatingSystemVersion = 0;\r
- Optional32->MinorOperatingSystemVersion = 0;\r
- Optional32->MajorImageVersion = 0;\r
- Optional32->MinorImageVersion = 0;\r
- Optional32->MajorSubsystemVersion = 0;\r
- Optional32->MinorSubsystemVersion = 0;\r
- Optional32->Win32VersionValue = 0;\r
- Optional32->CheckSum = 0;\r
- Optional32->SizeOfStackReserve = 0;\r
- Optional32->SizeOfStackCommit = 0;\r
- Optional32->SizeOfHeapReserve = 0;\r
- Optional32->SizeOfHeapCommit = 0;\r
- Optional32->Subsystem = (USHORT) Type;\r
-\r
- //\r
- // Strip zero padding at the end of the .reloc section \r
- //\r
- if (NeedStripZeroPendingReloc) {\r
- StripZeroPendingReloc (FileBuffer, &FileLength, DosHdr, PeHdr);\r
- }\r
- } else if (PeHdr->PeHeader32.OptionalHeader.Magic == IMAGE_NT_OPTIONAL_HDR64_MAGIC) {\r
- //\r
- // PE32+ image\r
- //\r
- Optional64 = (EFI_IMAGE_OPTIONAL_HEADER64 *)&PeHdr->PeHeader64.OptionalHeader;\r
-\r
- Optional64->MajorLinkerVersion = 0;\r
- Optional64->MinorLinkerVersion = 0;\r
- Optional64->MajorOperatingSystemVersion = 0;\r
- Optional64->MinorOperatingSystemVersion = 0;\r
- Optional64->MajorImageVersion = 0;\r
- Optional64->MinorImageVersion = 0;\r
- Optional64->MajorSubsystemVersion = 0;\r
- Optional64->MinorSubsystemVersion = 0;\r
- Optional64->Win32VersionValue = 0;\r
- Optional64->CheckSum = 0;\r
- Optional64->SizeOfStackReserve = 0;\r
- Optional64->SizeOfStackCommit = 0;\r
- Optional64->SizeOfHeapReserve = 0;\r
- Optional64->SizeOfHeapCommit = 0;\r
- Optional64->Subsystem = (USHORT) Type;\r
-\r
- //\r
- // Strip zero padding at the end of the .reloc section \r
- //\r
- if (NeedStripZeroPendingReloc) {\r
- StripZeroPendingReloc (FileBuffer, &FileLength, DosHdr, PeHdr);\r
- }\r
- } else {\r
- Error (NULL, 0, 0, argv[2], "Unsupported PE image");\r
- fclose (fpIn);\r
- fclose (fpOut);\r
- return STATUS_ERROR;\r
- }\r
-\r
- //\r
- // Zero PDATA section for smaller binary size after compression\r
- //\r
- if (NeedClearExceptionTable) {\r
- ZeroExceptionTable (FileBuffer, DosHdr, PeHdr);\r
- }\r
-\r
- FWriteFile (fpOut, FileBuffer, FileLength);\r
-\r
- //\r
- // Done\r
- //\r
- fclose (fpIn);\r
- fclose (fpOut);\r
-\r
- return STATUS_SUCCESS;\r
-}\r
projects / mirror_edk2.git / blobdiff