Test+fix for allocating int.MaxValue stream size.

microsoft · Dec 10, 2016 · ef151ba · ef151ba
1 parent dac6386
commit ef151ba
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/UnitTests/Tests.cs b/UnitTests/Tests.cs
@@ -444,6 +444,9 @@ public void GiantAllocationSucceeds()
                 var stream = mgr.GetStream(null, requestedSize);
                 Assert.IsTrue(stream.Capacity >= requestedSize);
             }
+
+            var maxStream = mgr.GetStream(null, int.MaxValue);
+            Assert.IsTrue(maxStream.Capacity == int.MaxValue);
         }
         #endregion
 

diff --git a/src/RecyclableMemoryStream.cs b/src/RecyclableMemoryStream.cs
@@ -314,7 +314,7 @@ public override int Capacity
                     return this.largeBuffer.Length;
                 }
 
-                long size = this.blocks.Count * this.memoryManager.BlockSize;
+                long size = (long)this.blocks.Count * this.memoryManager.BlockSize;
                 return (int)Math.Min(int.MaxValue, size);
             }
             set