/// <summary>
/// Client has been disconnected. Reset state
/// </summary>
public void Reset()
{
_messages.Clear();
_packet = null;
_bytesLeft = Packet.HeaderLength;
_parserMethod = ReadHeaderBytes;
}
/// <summary>
/// Read all header bytes from the incoming buffer
/// </summary>
/// <param name="stream">The stream.</param>
/// <returns><c>false</c> if we have not received all bytes yet; otherwise <c>true</c>.</returns>
protected virtual bool ReadHeaderBytes(IBufferReader stream)
{
var bytesLeftInStream = stream.Count - stream.Position;
var bytesToCopy = bytesLeftInStream < _bytesLeft
? bytesLeftInStream
: _bytesLeft;
stream.Read(_header, 0, bytesToCopy);
_bytesLeft -= bytesToCopy;
if (_bytesLeft > 0)
return false;
_packet = CreatePacket(_header);
_bytesLeft = _packet.ContentLength;
_parserMethod = ReadBodyBytes;
return true;
}
/// <summary>
/// Create a new packet from the header bytes.
/// </summary>
/// <param name="header">Header bytes</param>
/// <returns>Created packet (filled with info)</returns>
/// <remarks>The packet bytes are described in the <see cref="Packet"/> class doc.</remarks>
protected virtual Packet CreatePacket(byte[] header)
{
var message = new Packet
{
Version = _header[0],
ContentLength = BitConverter.ToInt32(header, 1)
};
if (message.Version <= 0)
throw new InvalidDataException(string.Format(
"Received '{0}' as version. Must be larger or equal to 1.", message.Version));
if (message.ContentLength <= 0)
throw new InvalidDataException(string.Format(
"Got invalid content length: '{0}', expected 1 or larger.", message.ContentLength));
message.Message = new MemoryStream(message.ContentLength);
return message;
}
/// <summary>
/// Read all body bytes from the incoming buffer
/// </summary>
/// <param name="reader">Contains received bytes</param>
/// <returns><c>false</c> if we have not received all bytes yet; otherwise <c>true</c>.</returns>
protected virtual bool ReadBodyBytes(IBufferReader reader)
{
var bytesLeftInStream = reader.Count - reader.Position;
var bytesToCopy = bytesLeftInStream < _bytesLeft
? bytesLeftInStream
: _bytesLeft;
reader.CopyTo(_packet.Message, bytesToCopy);
_bytesLeft -= bytesToCopy;
if (_bytesLeft > 0)
return false;
_packet.Message.Position = 0;
_messages.Enqueue(_packet);
_packet = null;
_bytesLeft = Packet.HeaderLength;
_parserMethod = ReadHeaderBytes;
return true;
}
