Fixed regression

dotnet · Jan 12, 2024 · 0d6f47f · 0d6f47f
1 parent 40b6f54
commit 0d6f47f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/src/DotNext.Tests/Buffers/UnmanagedMemoryPoolTests.cs b/src/DotNext.Tests/Buffers/UnmanagedMemoryPoolTests.cs
@@ -113,7 +113,7 @@ public static void SliceTest()
     [Fact]
     public static void Allocation()
     {
-        using var manager = UnmanagedMemory.Allocate<long>(2);
+        using var manager = UnmanagedMemory.AllocateZeroed<long>(2);
         Equal(2, manager.Length);
 
         Equal(sizeof(long) * 2U, manager.Size);
@@ -130,7 +130,7 @@ public static void Allocation()
     [Fact]
     public static void Pooling()
     {
-        using var pool = new UnmanagedMemoryPool<long>(10) { TrackAllocations = true };
+        using var pool = new UnmanagedMemoryPool<long>(10) { TrackAllocations = true, AllocateZeroedMemory = true };
         using var manager = pool.Rent(2);
         Equal(2, manager.Memory.Length);
 

diff --git a/src/DotNext.Unsafe/Buffers/UnmanagedMemoryPool.cs b/src/DotNext.Unsafe/Buffers/UnmanagedMemoryPool.cs
@@ -91,12 +91,16 @@ private void AddTracking(IUnmanagedMemory<T> owner)
     public override IMemoryOwner<T> Rent(int length = -1)
     {
         ArgumentOutOfRangeException.ThrowIfGreaterThan(length, MaxBufferSize);
+        ArgumentOutOfRangeException.ThrowIfZero(length);
+
+        if (length < 0)
+            length = defaultBufferSize;
 
         PoolingUnmanagedMemoryOwner<T> result;
 
         unsafe
         {
-            result = new PoolingUnmanagedMemoryOwner<T>(Math.Max(length, defaultBufferSize), allocator)
+            result = new PoolingUnmanagedMemoryOwner<T>(length, allocator)
             {
                 OnDisposed = removeMemory,
             };