You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
88 lines
2.2 KiB
88 lines
2.2 KiB
using IllusionInjector.Logging;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Diagnostics;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace IllusionInjector.Utilities
|
|
{
|
|
class BlockingStream : Stream
|
|
{
|
|
public BlockingStream(Stream bstr)
|
|
{
|
|
BaseStream = bstr;
|
|
}
|
|
|
|
public Stream BaseStream { get; set; }
|
|
|
|
private bool _open = true;
|
|
public bool Open {
|
|
get
|
|
{
|
|
return CanWrite;
|
|
}
|
|
set
|
|
{
|
|
if (!_open)
|
|
throw new InvalidOperationException("Blocking stream has already been closed!");
|
|
else
|
|
_open = value;
|
|
}
|
|
}
|
|
|
|
private bool canReadOverride = true;
|
|
public override bool CanRead => BaseStream.CanRead && canReadOverride;
|
|
|
|
public override bool CanSeek => BaseStream.CanSeek;
|
|
|
|
public override bool CanWrite => BaseStream.CanWrite && _open;
|
|
|
|
public override long Length => BaseStream.Length;
|
|
|
|
public override long Position { get => BaseStream.Position; set => BaseStream.Position = value; }
|
|
|
|
public override void Flush()
|
|
{
|
|
BaseStream.Flush();
|
|
}
|
|
|
|
public override int Read(byte[] buffer, int offset, int count)
|
|
{
|
|
var read = 0;
|
|
while (read < count && Open)
|
|
{
|
|
read += BaseStream.Read(buffer, read, count-read);
|
|
}
|
|
|
|
if (read == 0)
|
|
{
|
|
canReadOverride = false;
|
|
}
|
|
|
|
return read;
|
|
}
|
|
|
|
public override long Seek(long offset, SeekOrigin origin)
|
|
{
|
|
return BaseStream.Seek(offset, origin);
|
|
}
|
|
|
|
public override void SetLength(long value)
|
|
{
|
|
BaseStream.SetLength(value);
|
|
}
|
|
|
|
public override void Write(byte[] buffer, int offset, int count)
|
|
{
|
|
BaseStream.Write(buffer, offset, count);
|
|
}
|
|
|
|
public override string ToString()
|
|
{
|
|
return $"{base.ToString()} ({BaseStream?.ToString()})";
|
|
}
|
|
}
|
|
}
|