IntelFsp2Pkg/SplitFspBin.py: Command crashed when FV almost full.

[mirror_edk2.git] / IntelFsp2Pkg / Tools / SplitFspBin.py

diff --git a/IntelFsp2Pkg/Tools/SplitFspBin.py b/IntelFsp2Pkg/Tools/SplitFspBin.py
index b326241fa9daf080ecd7900d22198a277a1ca680..39a7ea7460b7281fba2451235fd13c8dee61b408 100644 (file)
--- a/IntelFsp2Pkg/Tools/SplitFspBin.py
+++ b/IntelFsp2Pkg/Tools/SplitFspBin.py
@@ -428,7 +428,7 @@ class FirmwareFile:
         ffssize = len(self.FfsData)\r
         offset  = sizeof(self.FfsHdr)\r
         if self.FfsHdr.Name != '\xff' * 16:\r
-            while offset < ffssize:\r
+            while offset < (ffssize - sizeof (EFI_COMMON_SECTION_HEADER)):\r
                 sechdr = EFI_COMMON_SECTION_HEADER.from_buffer (self.FfsData, offset)\r
                 sec = Section (offset, self.FfsData[offset:offset + int(sechdr.Size)])\r
                 self.SecList.append(sec)\r
@@ -453,7 +453,7 @@ class FirmwareVolume:
         else:\r
             offset = self.FvHdr.HeaderLength\r
         offset = AlignPtr(offset)\r
-        while offset < fvsize:\r
+        while offset < (fvsize - sizeof (EFI_FFS_FILE_HEADER)):\r
             ffshdr = EFI_FFS_FILE_HEADER.from_buffer (self.FvData, offset)\r
             if (ffshdr.Name == '\xff' * 16) and (int(ffshdr.Size) == 0xFFFFFF):\r
                 offset = fvsize\r
@@ -515,7 +515,7 @@ class FirmwareDevice:
         offset = 0\r
         fdsize = len(self.FdData)\r
         self.FvList  = []\r
-        while offset < fdsize:\r
+        while offset < (fdsize - sizeof (EFI_FIRMWARE_VOLUME_HEADER)):\r
             fvh = EFI_FIRMWARE_VOLUME_HEADER.from_buffer (self.FdData, offset)\r
             if b'_FVH' != fvh.Signature:\r
                 raise Exception("ERROR: Invalid FV header !")\r
@@ -838,7 +838,7 @@ def RebaseFspBin (FspBinary, FspComponent, FspBase, OutputDir, OutputFile):
 \r
 def main ():\r
     parser     = argparse.ArgumentParser()\r
-    subparsers = parser.add_subparsers(title='commands')\r
+    subparsers = parser.add_subparsers(title='commands', dest="which")\r
 \r
     parser_rebase  = subparsers.add_parser('rebase',  help='rebase a FSP into a new base address')\r
     parser_rebase.set_defaults(which='rebase')\r
@@ -880,7 +880,7 @@ def main ():
     elif args.which == 'info':\r
         ShowFspInfo (args.FspBinary)\r
     else:\r
-        pass\r
+        parser.print_help()\r
 \r
     return 0\r
 \r