Merge pull request #536 from CreateAndInject/CheckOverlaps

Check overlaps
0xd4d · Jan 15, 2024 · 83a0ac6 · 83a0ac6
2 parents 8f7edaf + a45d40c
commit 83a0ac6
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/DotNet/Writer/NativeModuleWriter.cs b/src/DotNet/Writer/NativeModuleWriter.cs
@@ -354,6 +354,12 @@ public sealed class OrigSection : IDisposable {
 			if (((uint)origRva & (requiredAlignment - 1)) != 0)
 				return;
 
+			var origEnd = origRva + origSize;
+			foreach (var reusedChunk in reusedChunks) {
+				if (origRva < reusedChunk.RVA + reusedChunk.Chunk.GetVirtualSize() && origEnd > reusedChunk.RVA)
+					return;
+			}
+
 			if (section.Remove(chunk) is null)
 				throw new InvalidOperationException();
 			reusedChunks.Add(new ReusedChunkInfo(chunk, origRva));