Fix overflow in ByteBlockPool, #1003 (#1055)

apache · Dec 4, 2024 · d597a5f · d597a5f
1 parent 5b7d0ac
commit d597a5f
Show file tree

Hide file tree

Showing 2 changed files with 57 additions and 2 deletions.
diff --git a/src/Lucene.Net.Tests/Util/TestByteBlockPool.cs b/src/Lucene.Net.Tests/Util/TestByteBlockPool.cs
@@ -1,5 +1,7 @@
-using NUnit.Framework;
+using Lucene.Net.Attributes;
+using NUnit.Framework;
 using RandomizedTesting.Generators;
+using System;
 using System.Collections.Generic;
 using JCG = J2N.Collections.Generic;
 using Assert = Lucene.Net.TestFramework.Assert;
@@ -68,5 +70,54 @@ public virtual void TestReadAndWrite()
                 }
             }
         }
+
+        [Test]
+        [LuceneNetSpecific] // LUCENENET issue #1003
+        public void TestTooManyAllocs()
+        {
+            // Use a mock allocator that doesn't waste memory
+            ByteBlockPool pool = new ByteBlockPool(new MockAllocator(0));
+            pool.NextBuffer();
+
+            bool throwsException = false;
+            int maxIterations = int.MaxValue / ByteBlockPool.BYTE_BLOCK_SIZE + 1;
+
+            for (int i = 0; i < maxIterations; i++)
+            {
+                try
+                {
+                    pool.NextBuffer();
+                }
+                catch (OverflowException)
+                {
+                    // The offset overflows on the last attempt to call NextBuffer()
+                    throwsException = true;
+                    break;
+                }
+            }
+
+            Assert.IsTrue(throwsException);
+            Assert.IsTrue(pool.ByteOffset + ByteBlockPool.BYTE_BLOCK_SIZE < pool.ByteOffset);
+        }
+
+        private class MockAllocator : ByteBlockPool.Allocator
+        {
+            private readonly byte[] buffer;
+
+            public MockAllocator(int blockSize) : base(blockSize)
+            {
+                buffer = Array.Empty<byte>();
+            }
+
+            public override void RecycleByteBlocks(byte[][] blocks, int start, int end)
+            {
+                // No-op
+            }
+
+            public override byte[] GetByteBlock()
+            {
+                return buffer;
+            }
+        }
     }
 }
diff --git a/src/Lucene.Net/Util/ByteBlockPool.cs b/src/Lucene.Net/Util/ByteBlockPool.cs
@@ -262,7 +262,11 @@ public void NextBuffer()
             bufferUpto++;
 
             ByteUpto = 0;
-            ByteOffset += BYTE_BLOCK_SIZE;
+
+            checked // LUCENENET specific - added checked to prevent overflow, issue #1003
+            {
+                ByteOffset += BYTE_BLOCK_SIZE;
+            }
         }
 
         /// <summary>