Hot fixing for the bug that GenFw can not convert ELF to PE format correctly, the root cause is invalid size of OptionalHeader is computed.

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@8976 6f19259b-4bc3-4df7-8a09-765794883524

17 files changed, 25 insertions, 7 deletions

diff --git a/BaseTools/Bin/Win32/BootSectImage.exe b/BaseTools/Bin/Win32/BootSectImage.exe
index a0cf0fd..966832d 100755
--- a/BaseTools/Bin/Win32/BootSectImage.exe
+++ b/BaseTools/Bin/Win32/BootSectImage.exe
diff --git a/BaseTools/Bin/Win32/EfiLdrImage.exe b/BaseTools/Bin/Win32/EfiLdrImage.exe
index 57662bf..89c655d 100755
--- a/BaseTools/Bin/Win32/EfiLdrImage.exe
+++ b/BaseTools/Bin/Win32/EfiLdrImage.exe
diff --git a/BaseTools/Bin/Win32/EfiRom.exe b/BaseTools/Bin/Win32/EfiRom.exe
index 9ca2fb2..643162e 100755
--- a/BaseTools/Bin/Win32/EfiRom.exe
+++ b/BaseTools/Bin/Win32/EfiRom.exe
diff --git a/BaseTools/Bin/Win32/GenBootSector.exe b/BaseTools/Bin/Win32/GenBootSector.exe
index 08180b4..48a17af 100755
--- a/BaseTools/Bin/Win32/GenBootSector.exe
+++ b/BaseTools/Bin/Win32/GenBootSector.exe
diff --git a/BaseTools/Bin/Win32/GenCrc32.exe b/BaseTools/Bin/Win32/GenCrc32.exe
index 1b9020e..01234af 100755
--- a/BaseTools/Bin/Win32/GenCrc32.exe
+++ b/BaseTools/Bin/Win32/GenCrc32.exe
diff --git a/BaseTools/Bin/Win32/GenFfs.exe b/BaseTools/Bin/Win32/GenFfs.exe
index 3ffc0fa..3ee7925 100755
--- a/BaseTools/Bin/Win32/GenFfs.exe
+++ b/BaseTools/Bin/Win32/GenFfs.exe
diff --git a/BaseTools/Bin/Win32/GenFv.exe b/BaseTools/Bin/Win32/GenFv.exe
index a01bf7e..6011dd3 100755
--- a/BaseTools/Bin/Win32/GenFv.exe
+++ b/BaseTools/Bin/Win32/GenFv.exe
diff --git a/BaseTools/Bin/Win32/GenFw.exe b/BaseTools/Bin/Win32/GenFw.exe
index 0b6333a..f2b04ed 100755
--- a/BaseTools/Bin/Win32/GenFw.exe
+++ b/BaseTools/Bin/Win32/GenFw.exe
diff --git a/BaseTools/Bin/Win32/GenPage.exe b/BaseTools/Bin/Win32/GenPage.exe
index 9a1a566..41ac434 100755
--- a/BaseTools/Bin/Win32/GenPage.exe
+++ b/BaseTools/Bin/Win32/GenPage.exe
diff --git a/BaseTools/Bin/Win32/GenSec.exe b/BaseTools/Bin/Win32/GenSec.exe
index aafcdcc..0c828b8 100755
--- a/BaseTools/Bin/Win32/GenSec.exe
+++ b/BaseTools/Bin/Win32/GenSec.exe
diff --git a/BaseTools/Bin/Win32/GenVtf.exe b/BaseTools/Bin/Win32/GenVtf.exe
index 26e904a..32eb2c1 100755
--- a/BaseTools/Bin/Win32/GenVtf.exe
+++ b/BaseTools/Bin/Win32/GenVtf.exe
diff --git a/BaseTools/Bin/Win32/LzmaCompress.exe b/BaseTools/Bin/Win32/LzmaCompress.exe
index ce231f8..8ca0bbc 100644
--- a/BaseTools/Bin/Win32/LzmaCompress.exe
+++ b/BaseTools/Bin/Win32/LzmaCompress.exe
diff --git a/BaseTools/Bin/Win32/Split.exe b/BaseTools/Bin/Win32/Split.exe
index 222eccf..4b01eba 100755
--- a/BaseTools/Bin/Win32/Split.exe
+++ b/BaseTools/Bin/Win32/Split.exe
diff --git a/BaseTools/Bin/Win32/TianoCompress.exe b/BaseTools/Bin/Win32/TianoCompress.exe
index 04114bd..b95b67f7 100755
--- a/BaseTools/Bin/Win32/TianoCompress.exe
+++ b/BaseTools/Bin/Win32/TianoCompress.exe
diff --git a/BaseTools/Bin/Win32/VfrCompile.exe b/BaseTools/Bin/Win32/VfrCompile.exe
index 3233415..e9b9977 100755
--- a/BaseTools/Bin/Win32/VfrCompile.exe
+++ b/BaseTools/Bin/Win32/VfrCompile.exe
diff --git a/BaseTools/Bin/Win32/VolInfo.exe b/BaseTools/Bin/Win32/VolInfo.exe
index 6e50174..ebbe398 100755
--- a/BaseTools/Bin/Win32/VolInfo.exe
+++ b/BaseTools/Bin/Win32/VolInfo.exe
diff --git a/BaseTools/Source/C/GenFw/GenFw.c b/BaseTools/Source/C/GenFw/GenFw.c
index 2278948..d570f63 100644
--- a/BaseTools/Source/C/GenFw/GenFw.c
+++ b/BaseTools/Source/C/GenFw/GenFw.c
@@ -583,7 +583,21 @@ ScanSections(
   //

   CoffOffset = sizeof(EFI_IMAGE_DOS_HEADER) + 0x40;

   NtHdrOffset = CoffOffset;

-  CoffOffset += sizeof(EFI_IMAGE_OPTIONAL_HEADER_UNION);

+  switch (Ehdr->e_machine) {

+  case EM_386:

+  case EM_ARM:

+    CoffOffset += sizeof (EFI_IMAGE_NT_HEADERS32);

+	break;

+  case EM_X86_64:

+  case EM_IA_64:

+	CoffOffset += sizeof (EFI_IMAGE_NT_HEADERS64);

+	break;

+  default:

+    VerboseMsg ("%s unknown e_machine type. Assume IA-32", (UINTN)Ehdr->e_machine);

+	CoffOffset += sizeof (EFI_IMAGE_NT_HEADERS32);

+	break;

+  }

+

   TableOffset = CoffOffset;

   CoffOffset += CoffNbrSections * sizeof(EFI_IMAGE_SECTION_HEADER);

 

@@ -1144,6 +1158,7 @@ ConvertElf (
   //

   // Compute sections new address.

   //

+  

   ScanSections();

 

   VerboseMsg ("Compute sections new address.");

@@ -2660,9 +2675,12 @@ Returns:
   //

   // Init variable.

   //

-  DebugDirectoryEntryRva    = 0;

-  ExportDirectoryEntryRva   = 0;

-  ResourceDirectoryEntryRva = 0;

+  DebugDirectoryEntryRva           = 0;

+  ExportDirectoryEntryRva          = 0;

+  ResourceDirectoryEntryRva        = 0;

+  DebugDirectoryEntryFileOffset    = 0;

+  ExportDirectoryEntryFileOffset   = 0;

+  ResourceDirectoryEntryFileOffset = 0;

   DosHdr   = (EFI_IMAGE_DOS_HEADER *)  FileBuffer;

   FileHdr  = (EFI_IMAGE_FILE_HEADER *) (FileBuffer + DosHdr->e_lfanew + sizeof (UINT32));

 

@@ -2745,17 +2763,17 @@ Returns:
   //

   FileHdr->TimeDateStamp = 0;

 

-  if (ExportDirectoryEntryRva != 0) {

+  if (ExportDirectoryEntryFileOffset != 0) {

     NewTimeStamp  = (UINT32 *) (FileBuffer + ExportDirectoryEntryFileOffset + sizeof (UINT32));

     *NewTimeStamp = 0;

   }

 

-  if (ResourceDirectoryEntryRva != 0) {

+  if (ResourceDirectoryEntryFileOffset != 0) {

     NewTimeStamp  = (UINT32 *) (FileBuffer + ResourceDirectoryEntryFileOffset + sizeof (UINT32));

     *NewTimeStamp = 0;

   }

 

-  if (DebugDirectoryEntryRva != 0) {

+  if (DebugDirectoryEntryFileOffset != 0) {

     DebugEntry = (EFI_IMAGE_DEBUG_DIRECTORY_ENTRY *) (FileBuffer + DebugDirectoryEntryFileOffset);

     DebugEntry->TimeDateStamp = 0;

     if (ZeroDebugFlag) {