mirror of
https://github.com/adamhathcock/sharpcompress.git
synced 2026-02-12 13:35:11 +00:00
885 lines
25 KiB
C#
885 lines
25 KiB
C#
using System;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using SharpCompress.IO;
|
|
using SharpCompress.Test.Mocks;
|
|
using Xunit;
|
|
|
|
namespace SharpCompress.Test.Streams;
|
|
|
|
public class SharpCompressStreamTest
|
|
{
|
|
[Fact]
|
|
public void TestRewind()
|
|
{
|
|
var ms = new MemoryStream();
|
|
var bw = new BinaryWriter(ms);
|
|
bw.Write(1);
|
|
bw.Write(2);
|
|
bw.Write(3);
|
|
bw.Write(4);
|
|
bw.Write(5);
|
|
bw.Write(6);
|
|
bw.Write(7);
|
|
bw.Flush();
|
|
ms.Position = 0;
|
|
var stream = new SharpCompressStream(ms);
|
|
stream.StartRecording();
|
|
var br = new BinaryReader(stream);
|
|
Assert.Equal(1, br.ReadInt32());
|
|
Assert.Equal(2, br.ReadInt32());
|
|
Assert.Equal(3, br.ReadInt32());
|
|
Assert.Equal(4, br.ReadInt32());
|
|
stream.Rewind(true);
|
|
Assert.Equal(1, br.ReadInt32());
|
|
Assert.Equal(2, br.ReadInt32());
|
|
Assert.Equal(3, br.ReadInt32());
|
|
Assert.Equal(4, br.ReadInt32());
|
|
Assert.Equal(5, br.ReadInt32());
|
|
Assert.Equal(6, br.ReadInt32());
|
|
Assert.Equal(7, br.ReadInt32());
|
|
}
|
|
|
|
[Fact]
|
|
public void TestIncompleteRewind()
|
|
{
|
|
var ms = new MemoryStream();
|
|
var bw = new BinaryWriter(ms);
|
|
bw.Write(1);
|
|
bw.Write(2);
|
|
bw.Write(3);
|
|
bw.Write(4);
|
|
bw.Write(5);
|
|
bw.Write(6);
|
|
bw.Write(7);
|
|
bw.Flush();
|
|
ms.Position = 0;
|
|
var stream = new SharpCompressStream(ms);
|
|
stream.StartRecording();
|
|
var br = new BinaryReader(stream);
|
|
Assert.Equal(1, br.ReadInt32());
|
|
Assert.Equal(2, br.ReadInt32());
|
|
Assert.Equal(3, br.ReadInt32());
|
|
Assert.Equal(4, br.ReadInt32());
|
|
stream.Rewind(true);
|
|
Assert.Equal(1, br.ReadInt32());
|
|
Assert.Equal(2, br.ReadInt32());
|
|
Assert.Equal(3, br.ReadInt32());
|
|
Assert.Equal(4, br.ReadInt32());
|
|
Assert.Equal(5, br.ReadInt32());
|
|
Assert.Equal(6, br.ReadInt32());
|
|
Assert.Equal(7, br.ReadInt32());
|
|
}
|
|
|
|
[Fact]
|
|
public void TestRecording()
|
|
{
|
|
var ms = new MemoryStream();
|
|
var bw = new BinaryWriter(ms);
|
|
bw.Write(1);
|
|
bw.Write(2);
|
|
bw.Write(3);
|
|
bw.Write(4);
|
|
bw.Flush();
|
|
ms.Position = 0;
|
|
var stream = new SharpCompressStream(ms);
|
|
stream.StartRecording();
|
|
var br = new BinaryReader(stream);
|
|
Assert.Equal(1, br.ReadInt32());
|
|
Assert.Equal(2, br.ReadInt32());
|
|
stream.Rewind(false);
|
|
Assert.Equal(1, br.ReadInt32());
|
|
Assert.Equal(2, br.ReadInt32());
|
|
Assert.Equal(3, br.ReadInt32());
|
|
Assert.Equal(4, br.ReadInt32());
|
|
}
|
|
|
|
[Fact]
|
|
public void TestPosition()
|
|
{
|
|
var ms = new MemoryStream();
|
|
var bw = new BinaryWriter(ms);
|
|
for (int i = 0; i < 10; i++)
|
|
{
|
|
bw.Write(i);
|
|
}
|
|
bw.Flush();
|
|
ms.Position = 0;
|
|
var stream = new SharpCompressStream(new ForwardOnlyStream(ms));
|
|
Assert.Equal(0, stream.Position);
|
|
|
|
var buffer = new byte[4];
|
|
stream.Read(buffer, 0, 4);
|
|
Assert.Equal(4, stream.Position);
|
|
|
|
stream.StartRecording();
|
|
stream.Read(buffer, 0, 4);
|
|
Assert.Equal(8, stream.Position);
|
|
|
|
stream.Rewind();
|
|
Assert.Equal(4, stream.Position);
|
|
}
|
|
|
|
[Fact]
|
|
public void TestPositionSeek()
|
|
{
|
|
var ms = new MemoryStream();
|
|
var bw = new BinaryWriter(ms);
|
|
for (int i = 0; i < 10; i++)
|
|
{
|
|
bw.Write(i);
|
|
}
|
|
bw.Flush();
|
|
ms.Position = 0;
|
|
var stream = new SharpCompressStream(new ForwardOnlyStream(ms));
|
|
stream.StartRecording();
|
|
var br = new BinaryReader(stream);
|
|
|
|
Assert.Equal(0, br.ReadInt32());
|
|
Assert.Equal(1, br.ReadInt32());
|
|
Assert.Equal(2, br.ReadInt32());
|
|
|
|
stream.Position = 4;
|
|
Assert.Equal(1, br.ReadInt32());
|
|
}
|
|
|
|
[Fact]
|
|
public void TestDispose()
|
|
{
|
|
var ms = new MemoryStream();
|
|
var bw = new BinaryWriter(ms);
|
|
bw.Write(1);
|
|
bw.Write(2);
|
|
bw.Flush();
|
|
ms.Position = 0;
|
|
var stream = new SharpCompressStream(new ForwardOnlyStream(ms));
|
|
stream.Dispose();
|
|
Assert.Throws<ObjectDisposedException>(() => stream.Read(new byte[4], 0, 4));
|
|
}
|
|
|
|
[Fact]
|
|
public void TestStopRecordingBasic()
|
|
{
|
|
var ms = new MemoryStream();
|
|
var bw = new BinaryWriter(ms);
|
|
bw.Write(1);
|
|
bw.Write(2);
|
|
bw.Write(3);
|
|
bw.Write(4);
|
|
bw.Write(5);
|
|
bw.Write(6);
|
|
bw.Write(7);
|
|
bw.Flush();
|
|
ms.Position = 0;
|
|
|
|
var stream = new SharpCompressStream(new ForwardOnlyStream(ms));
|
|
stream.StartRecording();
|
|
var br = new BinaryReader(stream);
|
|
|
|
Assert.Equal(1, br.ReadInt32());
|
|
Assert.Equal(2, br.ReadInt32());
|
|
Assert.Equal(3, br.ReadInt32());
|
|
Assert.Equal(4, br.ReadInt32());
|
|
|
|
stream.StopRecording();
|
|
|
|
Assert.Equal(1, br.ReadInt32());
|
|
Assert.Equal(2, br.ReadInt32());
|
|
Assert.Equal(3, br.ReadInt32());
|
|
Assert.Equal(4, br.ReadInt32());
|
|
Assert.Equal(5, br.ReadInt32());
|
|
Assert.Equal(6, br.ReadInt32());
|
|
Assert.Equal(7, br.ReadInt32());
|
|
|
|
Assert.False(stream.IsRecording);
|
|
}
|
|
|
|
[Fact]
|
|
public void TestStopRecordingNoFurtherBuffering()
|
|
{
|
|
var ms = new MemoryStream();
|
|
var bw = new BinaryWriter(ms);
|
|
bw.Write(1);
|
|
bw.Write(2);
|
|
bw.Write(3);
|
|
bw.Write(4);
|
|
bw.Flush();
|
|
ms.Position = 0;
|
|
|
|
var stream = new SharpCompressStream(new ForwardOnlyStream(ms));
|
|
stream.StartRecording();
|
|
|
|
var buffer = new byte[8];
|
|
stream.Read(buffer, 0, 8);
|
|
|
|
stream.StopRecording();
|
|
|
|
stream.Read(buffer, 0, 8);
|
|
Assert.Equal(BitConverter.GetBytes(1), buffer.Take(4).ToArray());
|
|
Assert.Equal(BitConverter.GetBytes(2), buffer.Skip(4).Take(4).ToArray());
|
|
|
|
int bytesRead = stream.Read(buffer, 0, 8);
|
|
Assert.Equal(8, bytesRead);
|
|
|
|
Assert.False(stream.IsRecording);
|
|
|
|
bytesRead = stream.Read(buffer, 0, 8);
|
|
Assert.Equal(0, bytesRead);
|
|
}
|
|
|
|
#if !LEGACY_DOTNET
|
|
[Fact]
|
|
public void TestStopRecordingWithSpan()
|
|
{
|
|
var ms = new MemoryStream();
|
|
var bw = new BinaryWriter(ms);
|
|
bw.Write(1);
|
|
bw.Write(2);
|
|
bw.Write(3);
|
|
bw.Write(4);
|
|
bw.Flush();
|
|
ms.Position = 0;
|
|
|
|
var stream = new SharpCompressStream(ms);
|
|
stream.StartRecording();
|
|
|
|
var buffer = new byte[8];
|
|
stream.Read(buffer);
|
|
|
|
stream.StopRecording();
|
|
|
|
stream.Read(buffer);
|
|
Assert.Equal(BitConverter.GetBytes(1), buffer.Take(4).ToArray());
|
|
Assert.Equal(BitConverter.GetBytes(2), buffer.Skip(4).Take(4).ToArray());
|
|
|
|
int bytesRead = stream.Read(buffer);
|
|
Assert.Equal(8, bytesRead);
|
|
Assert.Equal(BitConverter.GetBytes(3), buffer.Take(4).ToArray());
|
|
Assert.Equal(BitConverter.GetBytes(4), buffer.Skip(4).Take(4).ToArray());
|
|
}
|
|
#endif
|
|
|
|
[Fact]
|
|
public void TestNonSeekableStream_Rewind()
|
|
{
|
|
var ms = new MemoryStream();
|
|
var bw = new BinaryWriter(ms);
|
|
bw.Write(1);
|
|
bw.Write(2);
|
|
bw.Write(3);
|
|
bw.Write(4);
|
|
bw.Write(5);
|
|
bw.Write(6);
|
|
bw.Write(7);
|
|
bw.Flush();
|
|
ms.Position = 0;
|
|
|
|
var nonSeekableStream = new NonSeekableStreamWrapper(ms);
|
|
var stream = new SharpCompressStream(nonSeekableStream);
|
|
stream.StartRecording();
|
|
var br = new BinaryReader(stream);
|
|
|
|
Assert.Equal(1, br.ReadInt32());
|
|
Assert.Equal(2, br.ReadInt32());
|
|
Assert.Equal(3, br.ReadInt32());
|
|
Assert.Equal(4, br.ReadInt32());
|
|
stream.Rewind(true);
|
|
Assert.Equal(1, br.ReadInt32());
|
|
Assert.Equal(2, br.ReadInt32());
|
|
Assert.Equal(3, br.ReadInt32());
|
|
Assert.Equal(4, br.ReadInt32());
|
|
Assert.Equal(5, br.ReadInt32());
|
|
Assert.Equal(6, br.ReadInt32());
|
|
Assert.Equal(7, br.ReadInt32());
|
|
}
|
|
|
|
[Fact]
|
|
public void TestNonSeekableStream_Recording()
|
|
{
|
|
var ms = new MemoryStream();
|
|
var bw = new BinaryWriter(ms);
|
|
bw.Write(1);
|
|
bw.Write(2);
|
|
bw.Write(3);
|
|
bw.Write(4);
|
|
bw.Flush();
|
|
ms.Position = 0;
|
|
|
|
var nonSeekableStream = new NonSeekableStreamWrapper(ms);
|
|
var stream = new SharpCompressStream(nonSeekableStream);
|
|
stream.StartRecording();
|
|
var br = new BinaryReader(stream);
|
|
|
|
Assert.Equal(1, br.ReadInt32());
|
|
Assert.Equal(2, br.ReadInt32());
|
|
stream.Rewind(false);
|
|
Assert.Equal(1, br.ReadInt32());
|
|
Assert.Equal(2, br.ReadInt32());
|
|
Assert.Equal(3, br.ReadInt32());
|
|
Assert.Equal(4, br.ReadInt32());
|
|
}
|
|
|
|
[Fact]
|
|
public void TestNonSeekableStream_Position()
|
|
{
|
|
var ms = new MemoryStream();
|
|
var bw = new BinaryWriter(ms);
|
|
for (int i = 0; i < 10; i++)
|
|
{
|
|
bw.Write(i);
|
|
}
|
|
bw.Flush();
|
|
ms.Position = 0;
|
|
|
|
var nonSeekableStream = new NonSeekableStreamWrapper(ms);
|
|
var stream = new SharpCompressStream(nonSeekableStream);
|
|
Assert.Equal(0, stream.Position);
|
|
Assert.True(stream.CanSeek);
|
|
|
|
var buffer = new byte[4];
|
|
stream.Read(buffer, 0, 4);
|
|
Assert.Equal(4, stream.Position);
|
|
|
|
stream.StartRecording();
|
|
stream.Read(buffer, 0, 4);
|
|
Assert.Equal(8, stream.Position);
|
|
|
|
stream.Rewind();
|
|
Assert.Equal(4, stream.Position);
|
|
}
|
|
|
|
[Fact]
|
|
public void TestNonSeekableStream_PositionSet_WithinBuffer()
|
|
{
|
|
var ms = new MemoryStream();
|
|
var bw = new BinaryWriter(ms);
|
|
bw.Write(1);
|
|
bw.Write(2);
|
|
bw.Write(3);
|
|
bw.Write(4);
|
|
bw.Flush();
|
|
ms.Position = 0;
|
|
|
|
var nonSeekableStream = new NonSeekableStreamWrapper(ms);
|
|
var stream = new SharpCompressStream(nonSeekableStream);
|
|
stream.StartRecording();
|
|
|
|
var buffer = new byte[4];
|
|
stream.Read(buffer, 0, 4);
|
|
Assert.Equal(1, BitConverter.ToInt32(buffer, 0));
|
|
|
|
stream.Read(buffer, 0, 4);
|
|
Assert.Equal(2, BitConverter.ToInt32(buffer, 0));
|
|
|
|
stream.Position = 0;
|
|
Assert.Equal(0, stream.Position);
|
|
|
|
stream.Read(buffer, 0, 4);
|
|
Assert.Equal(1, BitConverter.ToInt32(buffer, 0));
|
|
|
|
stream.Read(buffer, 0, 4);
|
|
Assert.Equal(2, BitConverter.ToInt32(buffer, 0));
|
|
}
|
|
|
|
[Fact]
|
|
public void TestNonSeekableStream_PositionSet_OutsideBuffer_Throws()
|
|
{
|
|
var ms = new MemoryStream();
|
|
var bw = new BinaryWriter(ms);
|
|
bw.Write(1);
|
|
bw.Write(2);
|
|
bw.Write(3);
|
|
bw.Write(4);
|
|
bw.Flush();
|
|
ms.Position = 0;
|
|
|
|
var nonSeekableStream = new NonSeekableStreamWrapper(ms);
|
|
var stream = new SharpCompressStream(nonSeekableStream);
|
|
stream.StartRecording();
|
|
|
|
var buffer = new byte[4];
|
|
stream.Read(buffer, 0, 4);
|
|
|
|
Assert.Throws<NotSupportedException>(() => stream.Position = 100);
|
|
}
|
|
|
|
[Fact]
|
|
public void TestNonSeekableStream_StopRecordingBasic()
|
|
{
|
|
var ms = new MemoryStream();
|
|
var bw = new BinaryWriter(ms);
|
|
bw.Write(1);
|
|
bw.Write(2);
|
|
bw.Write(3);
|
|
bw.Write(4);
|
|
bw.Write(5);
|
|
bw.Write(6);
|
|
bw.Write(7);
|
|
bw.Flush();
|
|
ms.Position = 0;
|
|
|
|
var nonSeekableStream = new NonSeekableStreamWrapper(ms);
|
|
var stream = new SharpCompressStream(nonSeekableStream);
|
|
stream.StartRecording();
|
|
var br = new BinaryReader(stream);
|
|
|
|
Assert.Equal(1, br.ReadInt32());
|
|
Assert.Equal(2, br.ReadInt32());
|
|
Assert.Equal(3, br.ReadInt32());
|
|
Assert.Equal(4, br.ReadInt32());
|
|
|
|
stream.StopRecording();
|
|
|
|
Assert.Equal(1, br.ReadInt32());
|
|
Assert.Equal(2, br.ReadInt32());
|
|
Assert.Equal(3, br.ReadInt32());
|
|
Assert.Equal(4, br.ReadInt32());
|
|
Assert.Equal(5, br.ReadInt32());
|
|
Assert.Equal(6, br.ReadInt32());
|
|
Assert.Equal(7, br.ReadInt32());
|
|
|
|
Assert.False(stream.IsRecording);
|
|
}
|
|
|
|
[Fact]
|
|
public void TestStopRecordingThenRewind()
|
|
{
|
|
var ms = new MemoryStream();
|
|
var bw = new BinaryWriter(ms);
|
|
bw.Write(1);
|
|
bw.Write(2);
|
|
bw.Write(3);
|
|
bw.Write(4);
|
|
bw.Write(5);
|
|
bw.Write(6);
|
|
bw.Write(7);
|
|
bw.Write(8);
|
|
bw.Flush();
|
|
ms.Position = 0;
|
|
|
|
var stream = new SharpCompressStream(ms);
|
|
stream.StartRecording();
|
|
var br = new BinaryReader(new ForwardOnlyStream(stream));
|
|
|
|
// Read first 4 values (gets buffered)
|
|
Assert.Equal(1, br.ReadInt32());
|
|
Assert.Equal(2, br.ReadInt32());
|
|
Assert.Equal(3, br.ReadInt32());
|
|
Assert.Equal(4, br.ReadInt32());
|
|
|
|
// Stop recording
|
|
stream.StopRecording();
|
|
Assert.False(stream.IsRecording);
|
|
|
|
// Rewind to start of buffer
|
|
stream.Rewind(true);
|
|
|
|
// Should be able to read from buffer again
|
|
Assert.Equal(1, br.ReadInt32());
|
|
Assert.Equal(2, br.ReadInt32());
|
|
Assert.Equal(3, br.ReadInt32());
|
|
Assert.Equal(4, br.ReadInt32());
|
|
// Rewind to start of buffer
|
|
stream.Rewind();
|
|
// Should be able to read from buffer again
|
|
Assert.Equal(1, br.ReadInt32());
|
|
Assert.Equal(2, br.ReadInt32());
|
|
Assert.Equal(3, br.ReadInt32());
|
|
Assert.Equal(4, br.ReadInt32());
|
|
|
|
// Continue reading remaining data from underlying stream
|
|
Assert.Equal(5, br.ReadInt32());
|
|
Assert.Equal(6, br.ReadInt32());
|
|
Assert.Equal(7, br.ReadInt32());
|
|
Assert.Equal(8, br.ReadInt32());
|
|
}
|
|
|
|
[Fact]
|
|
public void TestNonSeekableStream_StopRecordingThenRewind()
|
|
{
|
|
var ms = new MemoryStream();
|
|
var bw = new BinaryWriter(ms);
|
|
bw.Write(1);
|
|
bw.Write(2);
|
|
bw.Write(3);
|
|
bw.Write(4);
|
|
bw.Write(5);
|
|
bw.Write(6);
|
|
bw.Write(7);
|
|
bw.Write(8);
|
|
bw.Flush();
|
|
ms.Position = 0;
|
|
|
|
var nonSeekableStream = new NonSeekableStreamWrapper(ms);
|
|
var stream = new SharpCompressStream(nonSeekableStream);
|
|
stream.StartRecording();
|
|
var br = new BinaryReader(stream);
|
|
|
|
// Read first 4 values (gets buffered)
|
|
Assert.Equal(1, br.ReadInt32());
|
|
Assert.Equal(2, br.ReadInt32());
|
|
Assert.Equal(3, br.ReadInt32());
|
|
Assert.Equal(4, br.ReadInt32());
|
|
|
|
// Stop recording
|
|
stream.StopRecording();
|
|
Assert.False(stream.IsRecording);
|
|
|
|
// Rewind to start of buffer
|
|
stream.Rewind(true);
|
|
|
|
// Should be able to read from buffer again
|
|
Assert.Equal(1, br.ReadInt32());
|
|
Assert.Equal(2, br.ReadInt32());
|
|
Assert.Equal(3, br.ReadInt32());
|
|
Assert.Equal(4, br.ReadInt32());
|
|
|
|
// Continue reading remaining data from underlying stream
|
|
Assert.Equal(5, br.ReadInt32());
|
|
Assert.Equal(6, br.ReadInt32());
|
|
Assert.Equal(7, br.ReadInt32());
|
|
Assert.Equal(8, br.ReadInt32());
|
|
}
|
|
|
|
[Fact]
|
|
public void TestMultipleRewindsAfterStopRecording()
|
|
{
|
|
var ms = new MemoryStream();
|
|
var bw = new BinaryWriter(ms);
|
|
bw.Write(1);
|
|
bw.Write(2);
|
|
bw.Write(3);
|
|
bw.Write(4);
|
|
bw.Write(5);
|
|
bw.Write(6);
|
|
bw.Write(7);
|
|
bw.Write(8);
|
|
bw.Flush();
|
|
ms.Position = 0;
|
|
|
|
var stream = new SharpCompressStream(ms);
|
|
stream.StartRecording();
|
|
var br = new BinaryReader(stream);
|
|
|
|
// Read first 4 values (gets buffered)
|
|
Assert.Equal(1, br.ReadInt32());
|
|
Assert.Equal(2, br.ReadInt32());
|
|
Assert.Equal(3, br.ReadInt32());
|
|
Assert.Equal(4, br.ReadInt32());
|
|
|
|
// Stop recording
|
|
stream.StopRecording();
|
|
Assert.False(stream.IsRecording);
|
|
|
|
// First rewind - read all buffered data, then continue with underlying stream
|
|
stream.Rewind();
|
|
Assert.Equal(1, br.ReadInt32());
|
|
Assert.Equal(2, br.ReadInt32());
|
|
Assert.Equal(3, br.ReadInt32());
|
|
Assert.Equal(4, br.ReadInt32());
|
|
Assert.Equal(5, br.ReadInt32());
|
|
Assert.Equal(6, br.ReadInt32());
|
|
|
|
// Second rewind - should still be able to read from buffer
|
|
stream.Rewind();
|
|
Assert.Equal(1, br.ReadInt32());
|
|
Assert.Equal(2, br.ReadInt32());
|
|
Assert.Equal(3, br.ReadInt32());
|
|
Assert.Equal(4, br.ReadInt32());
|
|
|
|
// Third rewind - still works
|
|
stream.Rewind();
|
|
Assert.Equal(1, br.ReadInt32());
|
|
Assert.Equal(2, br.ReadInt32());
|
|
Assert.Equal(3, br.ReadInt32());
|
|
Assert.Equal(4, br.ReadInt32());
|
|
Assert.Equal(5, br.ReadInt32());
|
|
Assert.Equal(6, br.ReadInt32());
|
|
Assert.Equal(7, br.ReadInt32());
|
|
Assert.Equal(8, br.ReadInt32());
|
|
}
|
|
|
|
[Fact]
|
|
public void TestStopRecordingTwiceThrows()
|
|
{
|
|
var ms = new MemoryStream();
|
|
var bw = new BinaryWriter(ms);
|
|
bw.Write(1);
|
|
bw.Write(2);
|
|
bw.Flush();
|
|
ms.Position = 0;
|
|
|
|
var stream = new SharpCompressStream(ms);
|
|
stream.StartRecording();
|
|
|
|
var br = new BinaryReader(stream);
|
|
Assert.Equal(1, br.ReadInt32());
|
|
|
|
// First StopRecording should succeed
|
|
stream.StopRecording();
|
|
Assert.False(stream.IsRecording);
|
|
|
|
// Second StopRecording should throw
|
|
Assert.Throws<InvalidOperationException>(() => stream.StopRecording());
|
|
}
|
|
|
|
[Fact]
|
|
public void TestReadMoreThanBufferSizeAfterRewind()
|
|
{
|
|
// This test verifies the fix for the bug where reading more bytes than
|
|
// are in the buffer after a rewind would only return the buffered bytes
|
|
// instead of continuing to read from the underlying stream.
|
|
var ms = new MemoryStream();
|
|
var bw = new BinaryWriter(ms);
|
|
|
|
// Write 29 bytes (simulating Arc header)
|
|
for (int i = 0; i < 29; i++)
|
|
{
|
|
bw.Write((byte)i);
|
|
}
|
|
|
|
// Write 5252 bytes (simulating Arc compressed data)
|
|
for (int i = 0; i < 5252; i++)
|
|
{
|
|
bw.Write((byte)(i % 256));
|
|
}
|
|
|
|
bw.Flush();
|
|
ms.Position = 0;
|
|
|
|
var stream = new SharpCompressStream(new ForwardOnlyStream(ms));
|
|
|
|
// Simulate factory detection: record first 512 bytes
|
|
stream.StartRecording();
|
|
var probeBuffer = new byte[512];
|
|
int probeRead = stream.Read(probeBuffer, 0, 512);
|
|
Assert.Equal(512, probeRead);
|
|
|
|
// Stop recording and rewind (simulates what ReaderFactory does)
|
|
stream.Rewind(true);
|
|
|
|
// Read header (29 bytes) - should come from buffer
|
|
var headerBuffer = new byte[29];
|
|
int headerRead = stream.Read(headerBuffer, 0, 29);
|
|
Assert.Equal(29, headerRead);
|
|
|
|
// Read compressed data (5252 bytes) - buffer has 483 bytes left,
|
|
// but we need 5252 bytes. This should read all 5252 bytes, not just 483.
|
|
var dataBuffer = new byte[5252];
|
|
int dataRead = stream.Read(dataBuffer, 0, 5252);
|
|
Assert.Equal(5252, dataRead);
|
|
|
|
// Verify we read the correct data
|
|
for (int i = 0; i < 5252; i++)
|
|
{
|
|
Assert.Equal((byte)(i % 256), dataBuffer[i]);
|
|
}
|
|
|
|
// Verify stream position is correct (29 + 5252 = 5281)
|
|
Assert.Equal(5281, stream.Position);
|
|
}
|
|
|
|
[Fact]
|
|
public void TestReadExactlyBufferSizeAfterRewind()
|
|
{
|
|
var ms = new MemoryStream();
|
|
var bw = new BinaryWriter(ms);
|
|
|
|
// Write 1024 bytes
|
|
for (int i = 0; i < 1024; i++)
|
|
{
|
|
bw.Write((byte)(i % 256));
|
|
}
|
|
|
|
bw.Flush();
|
|
ms.Position = 0;
|
|
|
|
var stream = new SharpCompressStream(new ForwardOnlyStream(ms));
|
|
|
|
// Record first 512 bytes
|
|
stream.StartRecording();
|
|
var probeBuffer = new byte[512];
|
|
stream.Read(probeBuffer, 0, 512);
|
|
stream.Rewind(true);
|
|
|
|
// Read exactly the buffer size (512 bytes)
|
|
var buffer = new byte[512];
|
|
int bytesRead = stream.Read(buffer, 0, 512);
|
|
Assert.Equal(512, bytesRead);
|
|
|
|
// Verify we read the correct data
|
|
for (int i = 0; i < 512; i++)
|
|
{
|
|
Assert.Equal((byte)(i % 256), buffer[i]);
|
|
}
|
|
}
|
|
|
|
[Fact]
|
|
public void TestReadLessThanBufferSizeAfterRewind()
|
|
{
|
|
var ms = new MemoryStream();
|
|
var bw = new BinaryWriter(ms);
|
|
|
|
// Write 1024 bytes
|
|
for (int i = 0; i < 1024; i++)
|
|
{
|
|
bw.Write((byte)(i % 256));
|
|
}
|
|
|
|
bw.Flush();
|
|
ms.Position = 0;
|
|
|
|
var stream = new SharpCompressStream(new ForwardOnlyStream(ms));
|
|
|
|
// Record first 512 bytes
|
|
stream.StartRecording();
|
|
var probeBuffer = new byte[512];
|
|
stream.Read(probeBuffer, 0, 512);
|
|
stream.Rewind(true);
|
|
|
|
// Read less than buffer size (256 bytes)
|
|
var buffer = new byte[256];
|
|
int bytesRead = stream.Read(buffer, 0, 256);
|
|
Assert.Equal(256, bytesRead);
|
|
|
|
// Verify we read the correct data
|
|
for (int i = 0; i < 256; i++)
|
|
{
|
|
Assert.Equal((byte)(i % 256), buffer[i]);
|
|
}
|
|
}
|
|
|
|
[Fact]
|
|
public void TestMultipleReadsExceedingBufferAfterRewind()
|
|
{
|
|
var ms = new MemoryStream();
|
|
var bw = new BinaryWriter(ms);
|
|
|
|
// Write 2048 bytes
|
|
for (int i = 0; i < 2048; i++)
|
|
{
|
|
bw.Write((byte)(i % 256));
|
|
}
|
|
|
|
bw.Flush();
|
|
ms.Position = 0;
|
|
|
|
var stream = new SharpCompressStream(new ForwardOnlyStream(ms));
|
|
|
|
// Record first 512 bytes
|
|
stream.StartRecording();
|
|
var probeBuffer = new byte[512];
|
|
stream.Read(probeBuffer, 0, 512);
|
|
stream.Rewind(true);
|
|
|
|
// Read in chunks that will exceed the buffer
|
|
var buffer = new byte[800];
|
|
|
|
// First read: 800 bytes (512 from buffer + 288 from underlying stream)
|
|
int bytesRead1 = stream.Read(buffer, 0, 800);
|
|
Assert.Equal(800, bytesRead1);
|
|
|
|
// Second read: 800 bytes (all from underlying stream)
|
|
int bytesRead2 = stream.Read(buffer, 0, 800);
|
|
Assert.Equal(800, bytesRead2);
|
|
|
|
// Third read: remaining 448 bytes
|
|
int bytesRead3 = stream.Read(buffer, 0, 800);
|
|
Assert.Equal(448, bytesRead3);
|
|
|
|
// Verify stream position
|
|
Assert.Equal(2048, stream.Position);
|
|
}
|
|
|
|
[Fact]
|
|
public void TestReadPartiallyFromBufferThenUnderlyingStream()
|
|
{
|
|
var ms = new MemoryStream();
|
|
var bw = new BinaryWriter(ms);
|
|
|
|
// Write 1000 bytes with specific pattern
|
|
for (int i = 0; i < 1000; i++)
|
|
{
|
|
bw.Write((byte)i);
|
|
}
|
|
|
|
bw.Flush();
|
|
ms.Position = 0;
|
|
|
|
var stream = new SharpCompressStream(new ForwardOnlyStream(ms));
|
|
|
|
// Record first 100 bytes
|
|
stream.StartRecording();
|
|
var probeBuffer = new byte[100];
|
|
stream.Read(probeBuffer, 0, 100);
|
|
stream.Rewind(true);
|
|
|
|
// Read 50 bytes (from buffer)
|
|
var buffer1 = new byte[50];
|
|
int read1 = stream.Read(buffer1, 0, 50);
|
|
Assert.Equal(50, read1);
|
|
for (int i = 0; i < 50; i++)
|
|
{
|
|
Assert.Equal((byte)i, buffer1[i]);
|
|
}
|
|
|
|
// Read 150 bytes (50 from buffer + 100 from underlying stream)
|
|
var buffer2 = new byte[150];
|
|
int read2 = stream.Read(buffer2, 0, 150);
|
|
Assert.Equal(150, read2);
|
|
for (int i = 0; i < 150; i++)
|
|
{
|
|
Assert.Equal((byte)(i + 50), buffer2[i]);
|
|
}
|
|
|
|
// Verify position
|
|
Assert.Equal(200, stream.Position);
|
|
}
|
|
|
|
private class NonSeekableStreamWrapper : Stream
|
|
{
|
|
private readonly Stream _baseStream;
|
|
|
|
public NonSeekableStreamWrapper(Stream baseStream)
|
|
{
|
|
_baseStream = baseStream;
|
|
}
|
|
|
|
public override bool CanRead => _baseStream.CanRead;
|
|
|
|
public override bool CanSeek => false;
|
|
|
|
public override bool CanWrite => _baseStream.CanWrite;
|
|
|
|
public override long Length => _baseStream.Length;
|
|
|
|
public override long Position
|
|
{
|
|
get => throw new NotSupportedException();
|
|
set => throw new NotSupportedException();
|
|
}
|
|
|
|
public override void Flush() => _baseStream.Flush();
|
|
|
|
public override int Read(byte[] buffer, int offset, int count) =>
|
|
_baseStream.Read(buffer, offset, count);
|
|
|
|
public override long Seek(long offset, SeekOrigin origin) =>
|
|
throw new NotSupportedException();
|
|
|
|
public override void SetLength(long value) => _baseStream.SetLength(value);
|
|
|
|
public override void Write(byte[] buffer, int offset, int count) =>
|
|
_baseStream.Write(buffer, offset, count);
|
|
|
|
protected override void Dispose(bool disposing)
|
|
{
|
|
if (disposing)
|
|
{
|
|
_baseStream.Dispose();
|
|
}
|
|
base.Dispose(disposing);
|
|
}
|
|
}
|
|
}
|