public void CopyFromRepeatedlyAndCopyToRepeatedly_LargeCopies_Success()
{
ReadOnlySpan <byte> source = Enumerable.Range(0, 64 * 1024 - 1).Select(x => (byte)x).ToArray().AsSpan();
const int RepeatCount = 13;
MultiArrayBuffer buffer = new MultiArrayBuffer(0);
for (int i = 0; i < RepeatCount; i++)
{
buffer.EnsureAvailableSpace(source.Length);
buffer.AvailableMemory.CopyFrom(source);
buffer.Commit(source.Length);
}
Span <byte> destination = new byte[source.Length].AsSpan();
for (int i = 0; i < RepeatCount; i++)
{
buffer.ActiveMemory.Slice(0, source.Length).CopyTo(destination);
Assert.True(source.SequenceEqual(destination));
buffer.Discard(source.Length);
}
Assert.Equal(0, buffer.ActiveMemory.Length);
Assert.True(buffer.IsEmpty);
Assert.True(buffer.ActiveMemory.IsEmpty);
}
public StreamBuffer(int initialBufferSize = DefaultInitialBufferSize, int maxBufferSize = DefaultMaxBufferSize)
{
_buffer = new MultiArrayBuffer(initialBufferSize);
_maxBufferSize = maxBufferSize;
_readTaskSource = new ResettableValueTaskSource();
_writeTaskSource = new ResettableValueTaskSource();
}
public void CopyFromRepeatedlyAndCopyToRepeatedly_Success()
{
ReadOnlySpan <byte> source = new byte[] { 1, 2, 3, 4, 5, 6, 7 }.AsSpan();
const int RepeatCount = 8 * 1024; // enough to ensure we cross several block boundaries
MultiArrayBuffer buffer = new MultiArrayBuffer(0);
for (int i = 0; i < RepeatCount; i++)
{
buffer.EnsureAvailableSpace(source.Length);
buffer.AvailableMemory.CopyFrom(source);
buffer.Commit(source.Length);
}
Span <byte> destination = new byte[source.Length].AsSpan();
for (int i = 0; i < RepeatCount; i++)
{
buffer.ActiveMemory.Slice(0, source.Length).CopyTo(destination);
Assert.True(source.SequenceEqual(destination));
buffer.Discard(source.Length);
}
Assert.Equal(0, buffer.ActiveMemory.Length);
Assert.True(buffer.IsEmpty);
Assert.True(buffer.ActiveMemory.IsEmpty);
}
public void AddSeveralBytesRepeatedlyAndConsumeSeveralBytesRepeatedly_UsingSliceWithLength_Success()
{
const int ByteCount = 7;
const int RepeatCount = 8 * 1024; // enough to ensure we cross several block boundaries
MultiArrayBuffer buffer = new MultiArrayBuffer(0);
for (int i = 0; i < RepeatCount; i++)
{
buffer.EnsureAvailableSpace(ByteCount);
for (int j = 0; j < ByteCount; j++)
{
buffer.AvailableMemory.Slice(j, ByteCount - j)[0] = (byte)(j + 1);
}
buffer.Commit(ByteCount);
}
for (int i = 0; i < RepeatCount; i++)
{
for (int j = 0; j < ByteCount; j++)
{
Assert.Equal(j + 1, buffer.ActiveMemory.Slice(j, ByteCount - j)[0]);
}
buffer.Discard(ByteCount);
}
Assert.Equal(0, buffer.ActiveMemory.Length);
Assert.True(buffer.IsEmpty);
Assert.True(buffer.ActiveMemory.IsEmpty);
}
public void EnsureAvailableSpaceUpToLimitTest()
{
MultiArrayBuffer buffer = new MultiArrayBuffer(0);
Assert.Equal(0, buffer.ActiveMemory.Length);
Assert.Equal(0, buffer.AvailableMemory.Length);
buffer.EnsureAvailableSpaceUpToLimit(0, 0);
Assert.Equal(0, buffer.AvailableMemory.Length);
buffer.EnsureAvailableSpaceUpToLimit(0, 1);
Assert.Equal(0, buffer.AvailableMemory.Length);
buffer.EnsureAvailableSpaceUpToLimit(1, 0);
Assert.Equal(0, buffer.AvailableMemory.Length);
buffer.EnsureAvailableSpaceUpToLimit(1, 1);
Assert.Equal(BlockSize, buffer.AvailableMemory.Length);
buffer.EnsureAvailableSpaceUpToLimit(1, 2);
Assert.Equal(BlockSize, buffer.AvailableMemory.Length);
buffer.EnsureAvailableSpaceUpToLimit(BlockSize, 0);
Assert.Equal(BlockSize, buffer.AvailableMemory.Length);
buffer.EnsureAvailableSpaceUpToLimit(BlockSize + 1, 0);
Assert.Equal(BlockSize, buffer.AvailableMemory.Length);
buffer.EnsureAvailableSpaceUpToLimit(BlockSize, BlockSize);
Assert.Equal(BlockSize, buffer.AvailableMemory.Length);
buffer.EnsureAvailableSpaceUpToLimit(BlockSize + 1, BlockSize);
Assert.Equal(BlockSize, buffer.AvailableMemory.Length);
buffer.EnsureAvailableSpaceUpToLimit(BlockSize + 1, BlockSize + 1);
Assert.Equal(BlockSize * 2, buffer.AvailableMemory.Length);
buffer.Commit(2);
buffer.Discard(1);
Assert.Equal(1, buffer.ActiveMemory.Length);
Assert.Equal(BlockSize * 2 - 2, buffer.AvailableMemory.Length);
buffer.EnsureAvailableSpaceUpToLimit(BlockSize * 2 - 2, BlockSize * 2 - 3);
Assert.Equal(BlockSize * 2 - 2, buffer.AvailableMemory.Length);
buffer.EnsureAvailableSpaceUpToLimit(BlockSize * 2 - 2, BlockSize * 2 - 2);
Assert.Equal(BlockSize * 2 - 2, buffer.AvailableMemory.Length);
buffer.EnsureAvailableSpaceUpToLimit(BlockSize * 2 - 2, BlockSize * 2 - 1);
Assert.Equal(BlockSize * 2 - 2, buffer.AvailableMemory.Length);
buffer.EnsureAvailableSpaceUpToLimit(BlockSize * 2 - 1, BlockSize * 2 - 1);
Assert.Equal(BlockSize * 2 - 2, buffer.AvailableMemory.Length);
buffer.EnsureAvailableSpaceUpToLimit(BlockSize * 2 - 1, BlockSize * 2);
Assert.Equal(BlockSize * 3 - 2, buffer.AvailableMemory.Length);
}
public void AddByteByByteAndConsumeByteByByte_Success()
{
const int Size = 64 * 1024 + 1;
MultiArrayBuffer buffer = new MultiArrayBuffer(0);
for (int i = 0; i < Size; i++)
{
buffer.EnsureAvailableSpace(1);
buffer.AvailableMemory[0] = (byte)i;
buffer.Commit(1);
}
for (int i = 0; i < Size; i++)
{
Assert.Equal((byte)i, buffer.ActiveMemory[0]);
buffer.Discard(1);
}
Assert.Equal(0, buffer.ActiveMemory.Length);
Assert.True(buffer.IsEmpty);
Assert.True(buffer.ActiveMemory.IsEmpty);
}
public ReceiveBuffers()
{
_syncRoot = new object();
_buffer = default;
_final = default;
}
public void EnsureAvailableSpaceTest()
{
MultiArrayBuffer buffer = new MultiArrayBuffer(0);
Assert.Equal(0, buffer.ActiveMemory.Length);
Assert.Equal(0, buffer.AvailableMemory.Length);
buffer.EnsureAvailableSpace(0);
Assert.Equal(0, buffer.AvailableMemory.Length);
buffer.EnsureAvailableSpace(1);
Assert.Equal(BlockSize, buffer.AvailableMemory.Length);
buffer.EnsureAvailableSpace(2);
Assert.Equal(BlockSize, buffer.AvailableMemory.Length);
buffer.EnsureAvailableSpace(BlockSize - 1);
Assert.Equal(BlockSize, buffer.AvailableMemory.Length);
buffer.EnsureAvailableSpace(BlockSize);
Assert.Equal(BlockSize, buffer.AvailableMemory.Length);
buffer.EnsureAvailableSpace(BlockSize + 1);
Assert.Equal(BlockSize * 2, buffer.AvailableMemory.Length);
buffer.Commit(BlockSize - 1);
Assert.Equal(BlockSize - 1, buffer.ActiveMemory.Length);
Assert.Equal(BlockSize + 1, buffer.AvailableMemory.Length);
buffer.Commit(BlockSize);
Assert.Equal(BlockSize * 2 - 1, buffer.ActiveMemory.Length);
Assert.Equal(1, buffer.AvailableMemory.Length);
buffer.EnsureAvailableSpace(0);
Assert.Equal(1, buffer.AvailableMemory.Length);
buffer.EnsureAvailableSpace(1);
Assert.Equal(1, buffer.AvailableMemory.Length);
buffer.EnsureAvailableSpace(2);
Assert.Equal(BlockSize + 1, buffer.AvailableMemory.Length);
buffer.Commit(2);
Assert.Equal(BlockSize * 2 + 1, buffer.ActiveMemory.Length);
Assert.Equal(BlockSize - 1, buffer.AvailableMemory.Length);
buffer.Discard(1);
Assert.Equal(BlockSize * 2, buffer.ActiveMemory.Length);
Assert.Equal(BlockSize - 1, buffer.AvailableMemory.Length);
buffer.Discard(1);
Assert.Equal(BlockSize * 2 - 1, buffer.ActiveMemory.Length);
Assert.Equal(BlockSize - 1, buffer.AvailableMemory.Length);
// This should not free the first block
buffer.Discard(BlockSize - 3);
Assert.Equal(BlockSize + 2, buffer.ActiveMemory.Length);
Assert.Equal(BlockSize - 1, buffer.AvailableMemory.Length);
// This should free the first block
buffer.Discard(1);
Assert.Equal(BlockSize + 1, buffer.ActiveMemory.Length);
Assert.Equal(BlockSize - 1, buffer.AvailableMemory.Length);
buffer.EnsureAvailableSpace(BlockSize - 1);
Assert.Equal(BlockSize - 1, buffer.AvailableMemory.Length);
buffer.EnsureAvailableSpace(BlockSize);
Assert.Equal(BlockSize * 2 - 1, buffer.AvailableMemory.Length);
buffer.Discard(BlockSize - 1);
Assert.Equal(2, buffer.ActiveMemory.Length);
Assert.Equal(BlockSize * 2 - 1, buffer.AvailableMemory.Length);
// This will cause shifting the block array down, but not reallocating
buffer.EnsureAvailableSpace(BlockSize * 2);
Assert.Equal(BlockSize * 3 - 1, buffer.AvailableMemory.Length);
buffer.Commit(BlockSize - 2);
Assert.Equal(BlockSize, buffer.ActiveMemory.Length);
Assert.Equal(BlockSize * 2 + 1, buffer.AvailableMemory.Length);
buffer.Commit(1);
Assert.Equal(BlockSize + 1, buffer.ActiveMemory.Length);
Assert.Equal(BlockSize * 2, buffer.AvailableMemory.Length);
buffer.Commit(1);
Assert.Equal(BlockSize + 2, buffer.ActiveMemory.Length);
Assert.Equal(BlockSize * 2 - 1, buffer.AvailableMemory.Length);
buffer.Discard(1);
Assert.Equal(BlockSize + 1, buffer.ActiveMemory.Length);
Assert.Equal(BlockSize * 2 - 1, buffer.AvailableMemory.Length);
// This will cause reallocating the block array, and dealing with an unused block in the first slot
buffer.EnsureAvailableSpace(BlockSize * 4);
Assert.Equal(BlockSize + 1, buffer.ActiveMemory.Length);
Assert.Equal(BlockSize * 5 - 1, buffer.AvailableMemory.Length);
buffer.Discard(2);
Assert.Equal(BlockSize - 1, buffer.ActiveMemory.Length);
Assert.Equal(BlockSize * 5 - 1, buffer.AvailableMemory.Length);
buffer.Commit(1);
Assert.Equal(BlockSize, buffer.ActiveMemory.Length);
Assert.Equal(BlockSize * 5 - 2, buffer.AvailableMemory.Length);
// This will discard all active bytes, which will reset the buffer
buffer.Discard(BlockSize);
Assert.Equal(0, buffer.ActiveMemory.Length);
Assert.Equal(0, buffer.AvailableMemory.Length);
buffer.EnsureAvailableSpace(2);
buffer.Commit(2);
Assert.Equal(2, buffer.ActiveMemory.Length);
Assert.Equal(BlockSize - 2, buffer.AvailableMemory.Length);
buffer.Discard(1);
Assert.Equal(1, buffer.ActiveMemory.Length);
Assert.Equal(BlockSize - 2, buffer.AvailableMemory.Length);
// Request a very large amount of available space.
buffer.EnsureAvailableSpace(BlockSize * 64 + 1);
Assert.Equal(1, buffer.ActiveMemory.Length);
Assert.Equal(BlockSize * 65 - 2, buffer.AvailableMemory.Length);
buffer.DiscardAll();
Assert.Equal(0, buffer.ActiveMemory.Length);
Assert.Equal(0, buffer.AvailableMemory.Length);
}
public void BasicTest()
{
MultiArrayBuffer buffer = new MultiArrayBuffer(0);
Assert.True(buffer.IsEmpty);
Assert.True(buffer.ActiveMemory.IsEmpty);
Assert.Equal(0, buffer.ActiveMemory.Length);
Assert.Equal(0, buffer.ActiveMemory.BlockCount);
Assert.True(buffer.AvailableMemory.IsEmpty);
Assert.Equal(0, buffer.AvailableMemory.Length);
Assert.Equal(0, buffer.AvailableMemory.BlockCount);
buffer.EnsureAvailableSpace(3);
Assert.True(buffer.IsEmpty);
Assert.True(buffer.ActiveMemory.IsEmpty);
Assert.Equal(0, buffer.ActiveMemory.Length);
Assert.Equal(0, buffer.ActiveMemory.BlockCount);
Assert.False(buffer.AvailableMemory.IsEmpty);
Assert.NotEqual(0, buffer.AvailableMemory.Length);
Assert.NotEqual(0, buffer.AvailableMemory.BlockCount);
int available = buffer.AvailableMemory.Length;
Assert.True(available >= 3);
buffer.AvailableMemory[0] = 10;
buffer.Commit(1);
Assert.False(buffer.IsEmpty);
Assert.False(buffer.ActiveMemory.IsEmpty);
Assert.Equal(1, buffer.ActiveMemory.Length);
Assert.Equal(10, buffer.ActiveMemory[0]);
Assert.Equal(available - 1, buffer.AvailableMemory.Length);
Assert.Equal(1, buffer.ActiveMemory.BlockCount);
Assert.Equal(10, buffer.ActiveMemory.GetBlock(0).Span[0]);
buffer.AvailableMemory[0] = 20;
buffer.Commit(1);
Assert.False(buffer.IsEmpty);
Assert.False(buffer.ActiveMemory.IsEmpty);
Assert.Equal(2, buffer.ActiveMemory.Length);
Assert.Equal(20, buffer.ActiveMemory[1]);
Assert.Equal(available - 2, buffer.AvailableMemory.Length);
Assert.InRange(buffer.ActiveMemory.BlockCount, 1, 2);
buffer.AvailableMemory[0] = 30;
buffer.Commit(1);
Assert.False(buffer.IsEmpty);
Assert.False(buffer.ActiveMemory.IsEmpty);
Assert.Equal(3, buffer.ActiveMemory.Length);
Assert.Equal(30, buffer.ActiveMemory[2]);
Assert.Equal(available - 3, buffer.AvailableMemory.Length);
Assert.InRange(buffer.ActiveMemory.BlockCount, 1, 2);
buffer.Discard(1);
Assert.False(buffer.IsEmpty);
Assert.False(buffer.ActiveMemory.IsEmpty);
Assert.Equal(2, buffer.ActiveMemory.Length);
Assert.Equal(20, buffer.ActiveMemory[0]);
Assert.InRange(buffer.ActiveMemory.BlockCount, 1, 2);
Assert.Equal(20, buffer.ActiveMemory.GetBlock(0).Span[0]);
buffer.Discard(1);
Assert.False(buffer.IsEmpty);
Assert.False(buffer.ActiveMemory.IsEmpty);
Assert.Equal(1, buffer.ActiveMemory.Length);
Assert.Equal(30, buffer.ActiveMemory[0]);
Assert.Equal(1, buffer.ActiveMemory.BlockCount);
Assert.Equal(30, buffer.ActiveMemory.GetBlock(0).Span[0]);
buffer.Discard(1);
Assert.True(buffer.IsEmpty);
Assert.True(buffer.ActiveMemory.IsEmpty);
Assert.Equal(0, buffer.ActiveMemory.Length);
Assert.Equal(0, buffer.ActiveMemory.BlockCount);
}
