Fix PeLoad function can't handle invalid PeImage to skip correct debug entry.

[mirror_edk2.git] / Tools / CCode / Source / PeCoffLoader / BasePeCoff.c

diff --git a/Tools/CCode/Source/PeCoffLoader/BasePeCoff.c b/Tools/CCode/Source/PeCoffLoader/BasePeCoff.c
index 72df90f389be21ec50ec48dfea3ad45fd645b869..06d4b83af7e6a11036d134baccaab0fc44bc31b3 100644 (file)
--- a/Tools/CCode/Source/PeCoffLoader/BasePeCoff.c
+++ b/Tools/CCode/Source/PeCoffLoader/BasePeCoff.c
@@ -346,7 +346,7 @@ Returns:
       }\r
 \r
       if (DebugDirectoryEntryFileOffset != 0) {    \r
-        for (Index = 0; Index < (DebugDirectoryEntry->Size); Index += Size) {\r
+        for (Index = 0; Index < DebugDirectoryEntry->Size; Index += sizeof (EFI_IMAGE_DEBUG_DIRECTORY_ENTRY)) {\r
           //\r
           // Read next debug directory entry\r
           //\r
@@ -438,7 +438,7 @@ Returns:
     }\r
 \r
     if (DebugDirectoryEntryFileOffset != 0) {\r
-      for (Index = 0; Index < (DebugDirectoryEntry->Size); Index += Size) {\r
+      for (Index = 0; Index < DebugDirectoryEntry->Size; Index += sizeof (EFI_IMAGE_DEBUG_DIRECTORY_ENTRY)) {\r
         //\r
         // Read next debug directory entry\r
         //\r