mirror of
https://github.com/aaru-dps/Aaru.git
synced 2025-12-16 19:24:25 +00:00
Add stream for splitted files.
This commit is contained in:
@@ -58,6 +58,7 @@
|
|||||||
<ItemGroup>
|
<ItemGroup>
|
||||||
<Compile Include="OffsetStream.cs" />
|
<Compile Include="OffsetStream.cs" />
|
||||||
<Compile Include="Register.cs" />
|
<Compile Include="Register.cs" />
|
||||||
|
<Compile Include="SplitJoinStream.cs" />
|
||||||
<Compile Include="ZZZNoFilter.cs" />
|
<Compile Include="ZZZNoFilter.cs" />
|
||||||
<Compile Include="ForcedSeekStream.cs" />
|
<Compile Include="ForcedSeekStream.cs" />
|
||||||
<Compile Include="GZip.cs" />
|
<Compile Include="GZip.cs" />
|
||||||
@@ -87,7 +88,7 @@
|
|||||||
</ItemGroup>
|
</ItemGroup>
|
||||||
<ItemGroup>
|
<ItemGroup>
|
||||||
<PackageReference Include="SharpCompress" Version="0.25.1" />
|
<PackageReference Include="SharpCompress" Version="0.25.1" />
|
||||||
<PackageReference Include="Unclassified.NetRevisionTask" Version="0.3.0" PrivateAssets="all"/>
|
<PackageReference Include="Unclassified.NetRevisionTask" Version="0.3.0" PrivateAssets="all" />
|
||||||
</ItemGroup>
|
</ItemGroup>
|
||||||
<ProjectExtensions>
|
<ProjectExtensions>
|
||||||
<MonoDevelop>
|
<MonoDevelop>
|
||||||
|
|||||||
242
Aaru.Filters/SplitJoinStream.cs
Normal file
242
Aaru.Filters/SplitJoinStream.cs
Normal file
@@ -0,0 +1,242 @@
|
|||||||
|
using System;
|
||||||
|
using System.Collections.Generic;
|
||||||
|
using System.Data;
|
||||||
|
using System.IO;
|
||||||
|
using System.Linq;
|
||||||
|
using Aaru.CommonTypes.Interfaces;
|
||||||
|
using Microsoft.Win32.SafeHandles;
|
||||||
|
|
||||||
|
namespace Aaru.Filters
|
||||||
|
{
|
||||||
|
public class SplitJoinStream : Stream
|
||||||
|
{
|
||||||
|
readonly Dictionary<long, Stream> _baseStreams;
|
||||||
|
long _position;
|
||||||
|
long _streamLength;
|
||||||
|
|
||||||
|
public SplitJoinStream()
|
||||||
|
{
|
||||||
|
_baseStreams = new Dictionary<long, Stream>();
|
||||||
|
_streamLength = 0;
|
||||||
|
_position = 0;
|
||||||
|
Filter = new ZZZNoFilter();
|
||||||
|
Filter.Open(this);
|
||||||
|
}
|
||||||
|
|
||||||
|
public override bool CanRead => true;
|
||||||
|
|
||||||
|
public override bool CanSeek => true;
|
||||||
|
|
||||||
|
public override bool CanWrite => false;
|
||||||
|
|
||||||
|
public override long Length => _streamLength;
|
||||||
|
|
||||||
|
public override long Position
|
||||||
|
{
|
||||||
|
get => _position;
|
||||||
|
|
||||||
|
set
|
||||||
|
{
|
||||||
|
if(value >= _streamLength)
|
||||||
|
throw new IOException("Cannot set position past stream end.");
|
||||||
|
|
||||||
|
_position = value;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public IFilter Filter { get; }
|
||||||
|
|
||||||
|
public void Add(Stream stream)
|
||||||
|
{
|
||||||
|
if(!stream.CanSeek)
|
||||||
|
throw new ArgumentException("Non-seekable streams are not supported");
|
||||||
|
|
||||||
|
if(!stream.CanRead)
|
||||||
|
throw new ArgumentException("Non-readable streams are not supported");
|
||||||
|
|
||||||
|
_baseStreams[_streamLength] = stream;
|
||||||
|
_streamLength += stream.Length;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void Add(string path, FileMode mode, FileAccess access, FileShare share, int bufferSize,
|
||||||
|
FileOptions options) => Add(new FileStream(path, mode, access, share, bufferSize, options));
|
||||||
|
|
||||||
|
public void Add(SafeFileHandle handle, FileAccess access) => Add(new FileStream(handle, access));
|
||||||
|
|
||||||
|
public void Add(SafeFileHandle handle, FileAccess access, int bufferSize) =>
|
||||||
|
Add(new FileStream(handle, access, bufferSize));
|
||||||
|
|
||||||
|
public void Add(SafeFileHandle handle, FileAccess access, int bufferSize, bool isAsync) =>
|
||||||
|
Add(new FileStream(handle, access, bufferSize, isAsync));
|
||||||
|
|
||||||
|
public void Add(string path, FileMode mode, FileAccess access, FileShare share, int bufferSize,
|
||||||
|
bool useAsync) => Add(new FileStream(path, mode, access, share, bufferSize, useAsync));
|
||||||
|
|
||||||
|
public void Add(string path, FileMode mode, FileAccess access, FileShare share, int bufferSize) =>
|
||||||
|
Add(new FileStream(path, mode, access, share, bufferSize));
|
||||||
|
|
||||||
|
public void Add(string path, FileMode mode, FileAccess access, FileShare share) =>
|
||||||
|
Add(new FileStream(path, mode, access, share));
|
||||||
|
|
||||||
|
public void Add(string path, FileMode mode, FileAccess access) => Add(new FileStream(path, mode, access));
|
||||||
|
|
||||||
|
public void Add(string path, FileMode mode) => Add(new FileStream(path, mode));
|
||||||
|
|
||||||
|
public void Add(byte[] buffer, int index, int count, bool writable, bool publiclyVisible) =>
|
||||||
|
Add(new MemoryStream(buffer, index, count, writable, publiclyVisible));
|
||||||
|
|
||||||
|
public void Add(byte[] buffer, int index, int count, bool writable, long start, long end) =>
|
||||||
|
Add(new MemoryStream(buffer, index, count, writable));
|
||||||
|
|
||||||
|
public void Add(byte[] buffer, int index, int count) => Add(new MemoryStream(buffer, index, count));
|
||||||
|
|
||||||
|
public void Add(byte[] buffer, bool writable) => Add(new MemoryStream(buffer, writable));
|
||||||
|
|
||||||
|
public void Add(byte[] buffer) => Add(new MemoryStream(buffer));
|
||||||
|
|
||||||
|
public void Add(IFilter filter) => Add(filter.GetDataForkStream());
|
||||||
|
|
||||||
|
public void AddRange(string basePath, string counterFormat = "{0:D3}", int counterStart = 0)
|
||||||
|
{
|
||||||
|
while(true)
|
||||||
|
{
|
||||||
|
string filePath = Path.Combine(basePath, string.Format(counterFormat, counterStart));
|
||||||
|
|
||||||
|
if(!File.Exists(filePath))
|
||||||
|
break;
|
||||||
|
|
||||||
|
Add(filePath, FileMode.Open);
|
||||||
|
|
||||||
|
counterStart++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
~SplitJoinStream()
|
||||||
|
{
|
||||||
|
foreach(Stream stream in _baseStreams.Values)
|
||||||
|
{
|
||||||
|
stream.Close();
|
||||||
|
stream.Dispose();
|
||||||
|
}
|
||||||
|
|
||||||
|
_baseStreams.Clear();
|
||||||
|
_position = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
public override IAsyncResult BeginRead(byte[] buffer, int offset, int count, AsyncCallback callback,
|
||||||
|
object state) =>
|
||||||
|
throw new NotSupportedException("Asynchronous I/O is not supported.");
|
||||||
|
|
||||||
|
public override IAsyncResult BeginWrite(byte[] buffer, int offset, int count, AsyncCallback callback,
|
||||||
|
object state) =>
|
||||||
|
throw new NotSupportedException("Asynchronous I/O is not supported.");
|
||||||
|
|
||||||
|
public override void Close()
|
||||||
|
{
|
||||||
|
foreach(Stream stream in _baseStreams.Values)
|
||||||
|
stream.Close();
|
||||||
|
|
||||||
|
_baseStreams.Clear();
|
||||||
|
_position = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
new void Dispose()
|
||||||
|
{
|
||||||
|
foreach(Stream stream in _baseStreams.Values)
|
||||||
|
stream.Dispose();
|
||||||
|
|
||||||
|
_baseStreams.Clear();
|
||||||
|
_position = 0;
|
||||||
|
|
||||||
|
base.Dispose();
|
||||||
|
}
|
||||||
|
|
||||||
|
public override int EndRead(IAsyncResult asyncResult) =>
|
||||||
|
throw new NotSupportedException("Asynchronous I/O is not supported.");
|
||||||
|
|
||||||
|
public override void EndWrite(IAsyncResult asyncResult) =>
|
||||||
|
throw new NotSupportedException("Asynchronous I/O is not supported.");
|
||||||
|
|
||||||
|
public override int ReadByte()
|
||||||
|
{
|
||||||
|
if(_position >= _streamLength)
|
||||||
|
return -1;
|
||||||
|
|
||||||
|
KeyValuePair<long, Stream> baseStream = _baseStreams.FirstOrDefault(s => s.Key >= _position);
|
||||||
|
|
||||||
|
if(baseStream.Value == null)
|
||||||
|
return -1;
|
||||||
|
|
||||||
|
baseStream.Value.Position = _position - baseStream.Key;
|
||||||
|
_position++;
|
||||||
|
|
||||||
|
return baseStream.Value.ReadByte();
|
||||||
|
}
|
||||||
|
|
||||||
|
public override void WriteByte(byte value) => throw new ReadOnlyException("This stream is read-only");
|
||||||
|
|
||||||
|
public override void Flush() {}
|
||||||
|
|
||||||
|
public override int Read(byte[] buffer, int offset, int count)
|
||||||
|
{
|
||||||
|
int read = 0;
|
||||||
|
|
||||||
|
while(count > 0)
|
||||||
|
{
|
||||||
|
KeyValuePair<long, Stream> baseStream = _baseStreams.LastOrDefault(s => s.Key <= _position);
|
||||||
|
|
||||||
|
if(baseStream.Value == null)
|
||||||
|
break;
|
||||||
|
|
||||||
|
baseStream.Value.Position = _position - baseStream.Key;
|
||||||
|
|
||||||
|
int currentCount = count;
|
||||||
|
|
||||||
|
if(baseStream.Value.Position + currentCount > baseStream.Value.Length)
|
||||||
|
currentCount = (int)(baseStream.Value.Length - baseStream.Value.Position);
|
||||||
|
|
||||||
|
read += baseStream.Value.Read(buffer, offset, currentCount);
|
||||||
|
|
||||||
|
count -= currentCount;
|
||||||
|
offset += currentCount;
|
||||||
|
}
|
||||||
|
|
||||||
|
return read;
|
||||||
|
}
|
||||||
|
|
||||||
|
public override long Seek(long offset, SeekOrigin origin)
|
||||||
|
{
|
||||||
|
switch(origin)
|
||||||
|
{
|
||||||
|
case SeekOrigin.Begin:
|
||||||
|
if(offset >= _streamLength)
|
||||||
|
throw new IOException("Cannot seek past stream end.");
|
||||||
|
|
||||||
|
_position = offset;
|
||||||
|
|
||||||
|
break;
|
||||||
|
case SeekOrigin.End:
|
||||||
|
if(_position - offset < 0)
|
||||||
|
throw new IOException("Cannot seek before stream start.");
|
||||||
|
|
||||||
|
_position -= offset;
|
||||||
|
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
if(_position + offset >= _streamLength)
|
||||||
|
throw new IOException("Cannot seek past stream end.");
|
||||||
|
|
||||||
|
_position += offset;
|
||||||
|
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
return _position;
|
||||||
|
}
|
||||||
|
|
||||||
|
public override void SetLength(long value) => throw new ReadOnlyException("This stream is read-only");
|
||||||
|
|
||||||
|
public override void Write(byte[] buffer, int offset, int count) =>
|
||||||
|
throw new ReadOnlyException("This stream is read-only");
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user