add data size check (neo-project#1381)

Tommo-L · Jan 10, 2020 · 38c0d6b · 38c0d6b
1 parent 2a22d22
commit 38c0d6b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/neo/IO/Helper.cs b/src/neo/IO/Helper.cs
@@ -79,7 +79,8 @@ public static byte[] CompressLz4(this byte[] data)
 
         public static byte[] DecompressLz4(this byte[] data, int maxOutput)
         {
-            maxOutput = Math.Min(maxOutput, data.Length * 255);
+            var maxDecompressDataLength = data.Length * 255;
+            if (maxDecompressDataLength > 0) maxOutput = Math.Min(maxOutput, maxDecompressDataLength);
             using var buffer = MemoryPool<byte>.Shared.Rent(maxOutput);
             int length = LZ4Codec.Decode(data, buffer.Memory.Span);
             if (length < 0 || length > maxOutput) throw new FormatException();