public override WebSocketCommandInfo FindCommand(SocketContext context, byte[] readBuffer, int offset, int length, bool isReusableBuffer)
{
AddArraySegment(readBuffer, offset, length, isReusableBuffer);
int? result = BufferSegments.SearchMark(m_HeaderTerminator);
if (!result.HasValue || result.Value <= 0)
{
NextCommandReader = this;
return null;
}
string header = Encoding.UTF8.GetString(BufferSegments.ToArrayData(0, result.Value));
var socketContext = context as WebSocketContext;
WebSocketServer.ParseHandshake(socketContext, new StringReader(header));
var secWebSocketKey1 = socketContext.SecWebSocketKey1;
var secWebSocketKey2 = socketContext.SecWebSocketKey2;
int left = BufferSegments.Count - result.Value - m_HeaderTerminator.Length;
BufferSegments.ClearSegements();
if (string.IsNullOrEmpty(secWebSocketKey1) && string.IsNullOrEmpty(secWebSocketKey2))
{
//v.75
if (left > 0)
AddArraySegment(readBuffer, offset + length - left, left, isReusableBuffer);
NextCommandReader = new WebSocketDataReader(this);
return CreateHeadCommandInfo();
}
else
{
//v.76
//Read SecWebSocketKey3(8 bytes)
if (left == 8)
{
socketContext.SecWebSocketKey3 = readBuffer.Skip(offset + length - left).Take(left).ToArray();
NextCommandReader = new WebSocketDataReader(this);
return CreateHeadCommandInfo();
}
else if (left > 8)
{
socketContext.SecWebSocketKey3 = readBuffer.Skip(offset + length - left).Take(8).ToArray();
AddArraySegment(readBuffer, offset + length - left + 8, left - 8, isReusableBuffer);
NextCommandReader = new WebSocketDataReader(this);
return CreateHeadCommandInfo();
}
else
{
//left < 8
if (left > 0)
AddArraySegment(readBuffer, offset + length - left, left, isReusableBuffer);
NextCommandReader = new WebSocketSecKey3Reader(this);
return null;
}
}
}
public override IWebSocketFragment FindCommandInfo(IAppSession session, byte[] readBuffer, int offset, int length, bool isReusableBuffer, out int left)
{
left = 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 as IWebSocketSession;
try
{
WebSocketServer.ParseHandshake(webSocketSession, new StringReader(header));
}
catch (Exception e)
{
session.Logger.LogError("Failed to parse handshake!" + Environment.NewLine + header, e);
session.Close(CloseReason.Unknown);
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;
left = 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(HandshakeCommandInfo);
}
}
else if ("6".Equals(secWebSocketVersion)) //draft-ietf-hybi-thewebsocketprotocol-06
{
if (Handshake(webSocketSession.AppServer.WebSocketProtocolProcessor, webSocketSession))
{
return(HandshakeCommandInfo);
}
}
else
{
//draft-hixie-thewebsocketprotocol-76/draft-ietf-hybi-thewebsocketprotocol-00
//Read SecWebSocketKey3(8 bytes)
if (left == SecKey3Len)
{
webSocketSession.Items[WebSocketConstant.SecWebSocketKey3] = readBuffer.CloneRange(offset + length - left, left);
left = 0;
if (Handshake(webSocketSession.AppServer.WebSocketProtocolProcessor, webSocketSession))
{
return(HandshakeCommandInfo);
}
}
else if (left > SecKey3Len)
{
webSocketSession.Items[WebSocketConstant.SecWebSocketKey3] = readBuffer.CloneRange(offset + length - left, 8);
left -= 8;
if (Handshake(webSocketSession.AppServer.WebSocketProtocolProcessor, webSocketSession))
{
return(HandshakeCommandInfo);
}
}
else
{
//left < 8
if (left > 0)
{
AddArraySegment(readBuffer, offset + length - left, left, isReusableBuffer);
left = 0;
}
NextCommandReader = new WebSocketSecKey3Reader(this);
return(null);
}
}
return(null);
}
public override IWebSocketFragment FindCommandInfo(IAppSession session, byte[] readBuffer, int offset, int length, bool isReusableBuffer, out int left)
{
left = 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 as IWebSocketSession;
WebSocketServer.ParseHandshake(webSocketSession, new StringReader(header));
var secWebSocketKey1 = webSocketSession.Items.GetValue<string>(WebSocketConstant.SecWebSocketKey1, string.Empty);
var secWebSocketKey2 = webSocketSession.Items.GetValue<string>(WebSocketConstant.SecWebSocketKey2, string.Empty);
var secWebSocketVersion = webSocketSession.SecWebSocketVersion;
left = 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 HandshakeCommandInfo;
}
else if ("6".Equals(secWebSocketVersion)) //draft-ietf-hybi-thewebsocketprotocol-06
{
if(Handshake(webSocketSession.AppServer.WebSocketProtocolProcessor, webSocketSession))
return HandshakeCommandInfo;
}
else
{
//draft-hixie-thewebsocketprotocol-76/draft-ietf-hybi-thewebsocketprotocol-00
//Read SecWebSocketKey3(8 bytes)
if (left == SecKey3Len)
{
webSocketSession.Items[WebSocketConstant.SecWebSocketKey3] = readBuffer.CloneRange(offset + length - left, left);
left = 0;
if(Handshake(webSocketSession.AppServer.WebSocketProtocolProcessor, webSocketSession))
return HandshakeCommandInfo;
}
else if (left > SecKey3Len)
{
webSocketSession.Items[WebSocketConstant.SecWebSocketKey3] = readBuffer.CloneRange(offset + length - left, 8);
left -= 8;
if(Handshake(webSocketSession.AppServer.WebSocketProtocolProcessor, webSocketSession))
return HandshakeCommandInfo;
}
else
{
//left < 8
if (left > 0)
{
AddArraySegment(readBuffer, offset + length - left, left, isReusableBuffer);
left = 0;
}
NextCommandReader = new WebSocketSecKey3Reader(this);
return null;
}
}
return null;
}
