148 lines
3.0 KiB
C#
148 lines
3.0 KiB
C#
using System;
|
|
using System.IO;
|
|
|
|
namespace Org.BouncyCastle.Crypto.Tls;
|
|
|
|
public class ByteQueue
|
|
{
|
|
private const int DefaultCapacity = 1024;
|
|
|
|
private byte[] databuf;
|
|
|
|
private int skipped = 0;
|
|
|
|
private int available = 0;
|
|
|
|
private bool readOnlyBuf = false;
|
|
|
|
public int Available => available;
|
|
|
|
public static int NextTwoPow(int i)
|
|
{
|
|
i |= i >> 1;
|
|
i |= i >> 2;
|
|
i |= i >> 4;
|
|
i |= i >> 8;
|
|
i |= i >> 16;
|
|
return i + 1;
|
|
}
|
|
|
|
public ByteQueue()
|
|
: this(1024)
|
|
{
|
|
}
|
|
|
|
public ByteQueue(int capacity)
|
|
{
|
|
databuf = ((capacity == 0) ? TlsUtilities.EmptyBytes : new byte[capacity]);
|
|
}
|
|
|
|
public ByteQueue(byte[] buf, int off, int len)
|
|
{
|
|
databuf = buf;
|
|
skipped = off;
|
|
available = len;
|
|
readOnlyBuf = true;
|
|
}
|
|
|
|
public void AddData(byte[] data, int offset, int len)
|
|
{
|
|
if (readOnlyBuf)
|
|
{
|
|
throw new InvalidOperationException("Cannot add data to read-only buffer");
|
|
}
|
|
if (skipped + available + len > databuf.Length)
|
|
{
|
|
int num = NextTwoPow(available + len);
|
|
if (num > databuf.Length)
|
|
{
|
|
byte[] destinationArray = new byte[num];
|
|
Array.Copy(databuf, skipped, destinationArray, 0, available);
|
|
databuf = destinationArray;
|
|
}
|
|
else
|
|
{
|
|
Array.Copy(databuf, skipped, databuf, 0, available);
|
|
}
|
|
skipped = 0;
|
|
}
|
|
Array.Copy(data, offset, databuf, skipped + available, len);
|
|
available += len;
|
|
}
|
|
|
|
public void CopyTo(Stream output, int length)
|
|
{
|
|
if (length > available)
|
|
{
|
|
throw new InvalidOperationException("Cannot copy " + length + " bytes, only got " + available);
|
|
}
|
|
output.Write(databuf, skipped, length);
|
|
}
|
|
|
|
public void Read(byte[] buf, int offset, int len, int skip)
|
|
{
|
|
if (buf.Length - offset < len)
|
|
{
|
|
throw new ArgumentException("Buffer size of " + buf.Length + " is too small for a read of " + len + " bytes");
|
|
}
|
|
if (available - skip < len)
|
|
{
|
|
throw new InvalidOperationException("Not enough data to read");
|
|
}
|
|
Array.Copy(databuf, skipped + skip, buf, offset, len);
|
|
}
|
|
|
|
public MemoryStream ReadFrom(int length)
|
|
{
|
|
if (length > available)
|
|
{
|
|
throw new InvalidOperationException("Cannot read " + length + " bytes, only got " + available);
|
|
}
|
|
int index = skipped;
|
|
available -= length;
|
|
skipped += length;
|
|
return new MemoryStream(databuf, index, length, writable: false);
|
|
}
|
|
|
|
public void RemoveData(int i)
|
|
{
|
|
if (i > available)
|
|
{
|
|
throw new InvalidOperationException("Cannot remove " + i + " bytes, only got " + available);
|
|
}
|
|
available -= i;
|
|
skipped += i;
|
|
}
|
|
|
|
public void RemoveData(byte[] buf, int off, int len, int skip)
|
|
{
|
|
Read(buf, off, len, skip);
|
|
RemoveData(skip + len);
|
|
}
|
|
|
|
public byte[] RemoveData(int len, int skip)
|
|
{
|
|
byte[] array = new byte[len];
|
|
RemoveData(array, 0, len, skip);
|
|
return array;
|
|
}
|
|
|
|
public void Shrink()
|
|
{
|
|
if (available == 0)
|
|
{
|
|
databuf = TlsUtilities.EmptyBytes;
|
|
skipped = 0;
|
|
return;
|
|
}
|
|
int num = NextTwoPow(available);
|
|
if (num < databuf.Length)
|
|
{
|
|
byte[] destinationArray = new byte[num];
|
|
Array.Copy(databuf, skipped, destinationArray, 0, available);
|
|
databuf = destinationArray;
|
|
skipped = 0;
|
|
}
|
|
}
|
|
}
|