Streaming compression parameters validation

ZstdNet.Tests/SteamingCompressionTests.cs

@@ -70,7 +70,6 @@ public void StreamingCompressionSingleWrite(byte[] data, int offset, int count)
 		[TestCase(3)]
 		[TestCase(5)]
 		[TestCase(10)]
-		[TestCase(11)]
 		public void StreamingDecompressionSingleRead(int readCount)
 		{
 			var data = new byte[] {0, 1, 2, 3, 4, 5, 6, 7, 8, 9};
@@ -81,12 +80,12 @@ public void StreamingDecompressionSingleRead(int readCount)
 
 			tempStream.Seek(0, SeekOrigin.Begin);
 
-			var buffer = new byte[10];
+			var buffer = new byte[data.Length];
 			using(var decompressionStream = new DecompressionStream(tempStream))
 			{
 				int bytesRead;
 				int totalBytesRead = 0;
-				while((bytesRead = decompressionStream.Read(buffer, totalBytesRead, readCount)) > 0)
+				while((bytesRead = decompressionStream.Read(buffer, totalBytesRead, Math.Min(readCount, buffer.Length - totalBytesRead))) > 0)
 				{
 					Assert.LessOrEqual(bytesRead, readCount);
 					totalBytesRead += bytesRead;

ZstdNet/CompressionStream.cs

@@ -23,6 +23,9 @@ public CompressionStream(Stream stream, int bufferSize)
 
 		public CompressionStream(Stream stream, CompressionOptions options, int bufferSize = 0)
 		{
+			if(bufferSize < 0)
+				throw new ArgumentOutOfRangeException(nameof(bufferSize));
+
 			innerStream = stream;
 
 			cStream = ZSTD_createCStream();
@@ -43,6 +46,13 @@ private static ArraySegmentPtr CreateOutputBuffer(int bufferSize)
 
 		public override void Write(byte[] buffer, int offset, int count)
 		{
+			if(offset < 0)
+				throw new ArgumentOutOfRangeException(nameof(offset));
+			if(count < 0)
+				throw new ArgumentOutOfRangeException(nameof(count));
+			if(offset + count > buffer.Length)
+				throw new ArgumentException("The sum of offset and count is greater than the buffer length");
+
 			if(count == 0)
 				return;
 

ZstdNet/DecompressionStream.cs

@@ -23,6 +23,9 @@ public DecompressionStream(Stream stream, int bufferSize)
 
 		public DecompressionStream(Stream stream, DecompressionOptions options, int bufferSize = 0)
 		{
+			if(bufferSize < 0)
+				throw new ArgumentOutOfRangeException(nameof(bufferSize));
+
 			innerStream = stream;
 
 			dStream = ZSTD_createDStream();
@@ -44,6 +47,13 @@ private static ArraySegmentPtr CreateInputBuffer(int bufferSize)
 
 		public override int Read(byte[] buffer, int offset, int count)
 		{
+			if(offset < 0)
+				throw new ArgumentOutOfRangeException(nameof(offset));
+			if(count < 0)
+				throw new ArgumentOutOfRangeException(nameof(count));
+			if(offset + count > buffer.Length)
+				throw new ArgumentException("The sum of offset and count is greater than the buffer length");
+
 			if(count == 0)
 				return 0;