Validate proper implementation of bufferWriter to prevent overflows

brantburnett · Dec 20, 2024 · e21e70d · e21e70d
1 parent c0e5d27
commit e21e70d
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/Snappier/Internal/SnappyCompressor.cs b/Snappier/Internal/SnappyCompressor.cs
@@ -149,6 +149,13 @@ private void CompressFragment(ReadOnlySpan<byte> fragment, IBufferWriter<byte> b
             int maxOutput = Helpers.MaxCompressedLength(fragment.Length);
 
             Span<byte> fragmentBuffer = bufferWriter.GetSpan(maxOutput);
+
+            // Validate proper implementation of bufferWriter to prevent buffer overflows
+            if (fragmentBuffer.Length < maxOutput)
+            {
+                ThrowHelper.ThrowInvalidOperationException("IBufferWriter<byte> did not return a sufficient span");
+            }
+
             int bytesWritten = CompressFragment(fragment, fragmentBuffer, hashTable);
             bufferWriter.Advance(bytesWritten);
         }