Support EOF data in debug data entries.

Washi1337 · Sep 22, 2024 · 4a2b36f · 4a2b36f
1 parent f70da79
commit 4a2b36f
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/src/AsmResolver.PE/Debug/SerializedDebugDataEntry.cs b/src/AsmResolver.PE/Debug/SerializedDebugDataEntry.cs
@@ -43,17 +43,22 @@ public SerializedDebugDataEntry(
         /// <inheritdoc />
         protected override IDebugDataSegment? GetContents()
         {
-            if (_sizeOfData == 0)
-                return null;
+            BinaryStreamReader reader;
 
-            var reference = _context.File.GetReferenceToRva(_addressOfRawData);
-            if (!reference.CanRead)
+            if (_addressOfRawData == 0 || _context.File.MappingMode == File.PEMappingMode.Unmapped)
+            {
+                if (!_context.File.TryCreateReaderAtFileOffset(_pointerToRawData, out reader))
+                {
+                    _context.BadImage("Debug data entry contains an invalid file offset.");
+                    return null;
+                }
+            }
+            else if (!_context.File.TryCreateReaderAtRva(_addressOfRawData, out reader))
             {
                 _context.BadImage("Debug data entry contains an invalid RVA.");
                 return null;
             }
 
-            var reader = reference.CreateReader();
             if (_sizeOfData > reader.Length)
             {
                 _context.BadImage("Debug data entry contains a too large size.");