public override async Task WriteAsync(byte[] buffer, int offset, int length, CancellationToken cancellationToken)
{
CheckNotDisposed();
ValidateBufferArgs(buffer, offset, length);
if (!IsOpen)
{
throw new TTransportException(TTransportException.ExceptionType.NotOpen);
}
// enough space left in buffer?
var free = WriteBuffer.Capacity - WriteBuffer.Length;
if (length > free)
{
ArraySegment <byte> bufSegment;
WriteBuffer.TryGetBuffer(out bufSegment);
await InnerTransport.WriteAsync(bufSegment.Array, 0, bufSegment.Count, cancellationToken);
WriteBuffer.SetLength(0);
}
// do the data even fit into the buffer?
// Note we test for < instead of <= to avoid nonsense buffering
if (length < WriteBuffer.Capacity)
{
await WriteBuffer.WriteAsync(buffer, offset, length, cancellationToken);
return;
}
// write thru
await InnerTransport.WriteAsync(buffer, offset, length, cancellationToken);
}
public override async Task FlushAsync(CancellationToken cancellationToken)
{
CheckNotDisposed();
if (!IsOpen)
{
throw new TTransportException(TTransportException.ExceptionType.NotOpen);
}
ArraySegment <byte> bufSegment;
WriteBuffer.TryGetBuffer(out bufSegment);
int dataLen = bufSegment.Count - HeaderSize;
if (dataLen < 0)
{
throw new InvalidOperationException(); // logic error actually
}
// Inject message header into the reserved buffer space
//BinaryPrimitives.WriteInt32BigEndian(bufSegment.Array, dataLen);
Binary.BigEndian.Set(dataLen, bufSegment.Array);
// Send the entire message at once
await InnerTransport.WriteAsync(bufSegment.Array, 0, bufSegment.Count, cancellationToken);
InitWriteBuffer();
await InnerTransport.FlushAsync(cancellationToken);
}
private async ValueTask ReadFrameAsync(CancellationToken cancellationToken)
{
await InnerTransport.ReadAllAsync(HeaderBuf, 0, HeaderSize, cancellationToken);
var size = DecodeFrameSize(HeaderBuf);
ReadBuffer.SetLength(size);
ReadBuffer.Seek(0, SeekOrigin.Begin);
ArraySegment <byte> bufSegment;
ReadBuffer.TryGetBuffer(out bufSegment);
await InnerTransport.ReadAllAsync(bufSegment.Array, 0, size, cancellationToken);
}
private async ValueTask ReadFrameAsync(CancellationToken cancellationToken)
{
UpdateKnownMessageSize(-1);
await InnerTransport.ReadAllAsync(HeaderBuf, 0, HeaderSize, cancellationToken);
int size = BinaryPrimitives.ReadInt32BigEndian(HeaderBuf);
if ((0 > size) || (size > Configuration.MaxFrameSize)) // size must be in the range 0 to allowed max
{
throw new TTransportException(TTransportException.ExceptionType.Unknown, $"Maximum frame size exceeded ({size} bytes)");
}
UpdateKnownMessageSize(size + HeaderSize);
ReadBuffer.SetLength(size);
ReadBuffer.Seek(0, SeekOrigin.Begin);
ReadBuffer.TryGetBuffer(out ArraySegment <byte> bufSegment);
await InnerTransport.ReadAllAsync(bufSegment.Array, 0, size, cancellationToken);
}
public override async ValueTask <int> ReadAsync(byte[] buffer, int offset, int length, CancellationToken cancellationToken)
{
CheckNotDisposed();
ValidateBufferArgs(buffer, offset, length);
if (!IsOpen)
{
throw new TTransportException(TTransportException.ExceptionType.NotOpen);
}
// do we have something buffered?
var count = ReadBuffer.Length - ReadBuffer.Position;
if (count > 0)
{
return(await ReadBuffer.ReadAsync(buffer, offset, length, cancellationToken));
}
// does the request even fit into the buffer?
// Note we test for >= instead of > to avoid nonsense buffering
if (length >= ReadBuffer.Capacity)
{
return(await InnerTransport.ReadAsync(buffer, offset, length, cancellationToken));
}
// buffer a new chunk of bytes from the underlying transport
ReadBuffer.Length = ReadBuffer.Capacity;
ArraySegment <byte> bufSegment;
ReadBuffer.TryGetBuffer(out bufSegment);
ReadBuffer.Length = await InnerTransport.ReadAsync(bufSegment.Array, 0, bufSegment.Count, cancellationToken);
ReadBuffer.Position = 0;
// deliver the bytes
return(await ReadBuffer.ReadAsync(buffer, offset, length, cancellationToken));
}
