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; }