fix BrotliCompressor

Cysharp · Apr 12, 2024 · c55e631 · c55e631
1 parent f30eb95
commit c55e631
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/MemoryPack.Core/Compression/BrotliCompressor.cs b/src/MemoryPack.Core/Compression/BrotliCompressor.cs
@@ -262,9 +262,11 @@ static int CompressCore<TBufferWriter>(ref BrotliEncoder encoder, ReadOnlySpan<b
         var totalWritten = 0;
 
         var lastResult = OperationStatus.DestinationTooSmall;
+
+        var destLength = initialLength ?? BrotliUtils.BrotliEncoderMaxCompressedSize(source.Length);
         while (lastResult == OperationStatus.DestinationTooSmall)
         {
-            ref var spanRef = ref destBufferWriter.GetSpanReference(initialLength ?? source.Length);
+            ref var spanRef = ref destBufferWriter.GetSpanReference(destLength);
             var dest = MemoryMarshal.CreateSpan(ref spanRef, destBufferWriter.BufferLength);
 
             lastResult = encoder.Compress(source, dest, out int bytesConsumed, out int bytesWritten, isFinalBlock: isFinalBlock);