GdfxFileSystem: Truncate file if the stream ends before the declared …

…size
emoose · Mar 2, 2023 · d57bb9a · d57bb9a
1 parent e9d61cb
commit d57bb9a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/GdfxFileSystem.cs b/GdfxFileSystem.cs
@@ -249,6 +249,12 @@ public bool Read(Stream stream)
                 stream.Read(fileNameBytes, 0, DirEntry.FileNameLength);
                 fileName = Encoding.ASCII.GetString(fileNameBytes);
 
+                if(FileSystem.GdfSectorToAddress(DirEntry.FirstSector) + DirEntry.FileSize > (ulong)stream.Length)
+                {
+                    // The stream ends before the file, truncate it
+                    DirEntry.FileSize = (uint)((ulong)stream.Length - FileSystem.GdfSectorToAddress(DirEntry.FirstSector));
+                }
+
                 // Align code from Velocity, seems to work great
                 stream.Position = (long)((ulong)(DirEntryAddr + DirEntry.FileNameLength + 0x11) & 0xFFFFFFFFFFFFFFFC);
                 return true;