public override IWebSocketFragment Filter(byte[] readBuffer, int offset, int length, bool isReusableBuffer, out int rest)
{
rest = 0;
int prevMatched = m_SearchState.Matched;
var result = readBuffer.SearchMark(offset, length, m_SearchState);
if (result < 0)
{
this.AddArraySegment(readBuffer, offset, length, isReusableBuffer);
return null;
}
int findLen = result - offset;
string header = string.Empty;
if (this.BufferSegments.Count > 0)
{
if (findLen > 0)
{
this.AddArraySegment(readBuffer, offset, findLen, false);
header = this.BufferSegments.Decode(Encoding.UTF8);
}
else
{
header = this.BufferSegments.Decode(Encoding.UTF8, 0, this.BufferSegments.Count - prevMatched);
}
}
else
{
header = Encoding.UTF8.GetString(readBuffer, offset, findLen);
}
var webSocketSession = Session;
try
{
WebSocketServer.ParseHandshake(webSocketSession, new StringReader(header));
}
catch (Exception e)
{
webSocketSession.Logger.Error("Failed to parse handshake!" + Environment.NewLine + header, e);
webSocketSession.Close(CloseReason.ProtocolError);
return null;
}
var secWebSocketKey1 = webSocketSession.Items.GetValue<string>(WebSocketConstant.SecWebSocketKey1, string.Empty);
var secWebSocketKey2 = webSocketSession.Items.GetValue<string>(WebSocketConstant.SecWebSocketKey2, string.Empty);
var secWebSocketVersion = webSocketSession.SecWebSocketVersion;
rest = length - findLen - (m_HeaderTerminator.Length - prevMatched);
this.ClearBufferSegments();
if (string.IsNullOrEmpty(secWebSocketKey1) && string.IsNullOrEmpty(secWebSocketKey2))
{
//draft-hixie-thewebsocketprotocol-75
if(Handshake(webSocketSession.AppServer.WebSocketProtocolProcessor, webSocketSession))
return HandshakeRequestInfo;
}
else if ("6".Equals(secWebSocketVersion)) //draft-ietf-hybi-thewebsocketprotocol-06
{
if(Handshake(webSocketSession.AppServer.WebSocketProtocolProcessor, webSocketSession))
return HandshakeRequestInfo;
}
else
{
//draft-hixie-thewebsocketprotocol-76/draft-ietf-hybi-thewebsocketprotocol-00
//Read SecWebSocketKey3(8 bytes)
if (rest == SecKey3Len)
{
webSocketSession.Items[WebSocketConstant.SecWebSocketKey3] = readBuffer.CloneRange(offset + length - rest, rest);
rest = 0;
if(Handshake(webSocketSession.AppServer.WebSocketProtocolProcessor, webSocketSession))
return HandshakeRequestInfo;
}
else if (rest > SecKey3Len)
{
webSocketSession.Items[WebSocketConstant.SecWebSocketKey3] = readBuffer.CloneRange(offset + length - rest, 8);
rest -= 8;
if(Handshake(webSocketSession.AppServer.WebSocketProtocolProcessor, webSocketSession))
return HandshakeRequestInfo;
}
else
{
//rest < 8
if (rest > 0)
{
AddArraySegment(readBuffer, offset + length - rest, rest, isReusableBuffer);
rest = 0;
}
NextReceiveFilter = new WebSocketSecKey3ReceiveFilter(this);
return null;
}
}
return null;
}
public override IWebSocketFragment Filter(byte[] readBuffer, int offset, int length, bool isReusableBuffer, out int rest)
{
rest = 0;
int prevMatched = m_SearchState.Matched;
var result = readBuffer.SearchMark(offset, length, m_SearchState);
if (result < 0)
{
this.AddArraySegment(readBuffer, offset, length, isReusableBuffer);
return(null);
}
int findLen = result - offset;
string header = string.Empty;
if (this.BufferSegments.Count > 0)
{
if (findLen > 0)
{
this.AddArraySegment(readBuffer, offset, findLen, false);
header = this.BufferSegments.Decode(Encoding.UTF8);
}
else
{
header = this.BufferSegments.Decode(Encoding.UTF8, 0, this.BufferSegments.Count - prevMatched);
}
}
else
{
header = Encoding.UTF8.GetString(readBuffer, offset, findLen);
}
var webSocketSession = Session;
try
{
WebSocketServer.ParseHandshake(webSocketSession, new StringReader(header));
}
catch (Exception e)
{
webSocketSession.Logger.Error("Failed to parse handshake!" + Environment.NewLine + header, e);
webSocketSession.Close(CloseReason.ProtocolError);
return(null);
}
var secWebSocketKey1 = webSocketSession.Items.GetValue <string>(WebSocketConstant.SecWebSocketKey1, string.Empty);
var secWebSocketKey2 = webSocketSession.Items.GetValue <string>(WebSocketConstant.SecWebSocketKey2, string.Empty);
var secWebSocketVersion = webSocketSession.SecWebSocketVersion;
rest = length - findLen - (m_HeaderTerminator.Length - prevMatched);
this.ClearBufferSegments();
if (string.IsNullOrEmpty(secWebSocketKey1) && string.IsNullOrEmpty(secWebSocketKey2))
{
//draft-hixie-thewebsocketprotocol-75
if (Handshake(webSocketSession.AppServer.WebSocketProtocolProcessor, webSocketSession))
{
return(HandshakeRequestInfo);
}
}
else if ("6".Equals(secWebSocketVersion)) //draft-ietf-hybi-thewebsocketprotocol-06
{
if (Handshake(webSocketSession.AppServer.WebSocketProtocolProcessor, webSocketSession))
{
return(HandshakeRequestInfo);
}
}
else
{
//draft-hixie-thewebsocketprotocol-76/draft-ietf-hybi-thewebsocketprotocol-00
//Read SecWebSocketKey3(8 bytes)
if (rest == SecKey3Len)
{
webSocketSession.Items[WebSocketConstant.SecWebSocketKey3] = readBuffer.CloneRange(offset + length - rest, rest);
rest = 0;
if (Handshake(webSocketSession.AppServer.WebSocketProtocolProcessor, webSocketSession))
{
return(HandshakeRequestInfo);
}
}
else if (rest > SecKey3Len)
{
webSocketSession.Items[WebSocketConstant.SecWebSocketKey3] = readBuffer.CloneRange(offset + length - rest, 8);
rest -= 8;
if (Handshake(webSocketSession.AppServer.WebSocketProtocolProcessor, webSocketSession))
{
return(HandshakeRequestInfo);
}
}
else
{
//rest < 8
if (rest > 0)
{
AddArraySegment(readBuffer, offset + length - rest, rest, isReusableBuffer);
rest = 0;
}
NextReceiveFilter = new WebSocketSecKey3ReceiveFilter(this);
return(null);
}
}
return(null);
}