public void Recv() { if (this.socket == null) { return; } while (socket != null && this.socket.Available > 0) { int messageLength = 0; try { messageLength = this.socket.ReceiveFrom(this.cache, ref this.ipEndPoint); } catch (Exception e) { Log.Error(e); continue; } // 长度小于1,不是正常的消息 if (messageLength < 1) { continue; } // accept byte flag = this.cache[0]; // conn从1000开始,如果为1,2,3则是特殊包 uint remoteConn = 0; uint localConn = 0; KChannel kChannel = null; switch (flag) { case KcpProtocalType.SYN: // accept // 长度!=5,不是accpet消息 if (messageLength != 5) { break; } IPEndPoint acceptIpEndPoint = (IPEndPoint)this.ipEndPoint; this.ipEndPoint = new IPEndPoint(0, 0); remoteConn = BitConverter.ToUInt32(this.cache, 1); // 如果等待连接状态,则重新响应请求 if (this.waitConnectChannels.TryGetValue(remoteConn, out kChannel)) { kChannel.HandleAccept(remoteConn); break; } localConn = ++this.IdGenerater; kChannel = new KChannel(localConn, remoteConn, this.socket, acceptIpEndPoint, this); this.localConnChannels[kChannel.LocalConn] = kChannel; this.waitConnectChannels[remoteConn] = kChannel; kChannel.HandleAccept(remoteConn); this.OnAccept(kChannel); break; case KcpProtocalType.ACK: // connect返回 // 长度!=9,不是connect消息 if (messageLength != 9) { break; } remoteConn = BitConverter.ToUInt32(this.cache, 1); localConn = BitConverter.ToUInt32(this.cache, 5); kChannel = this.GetKChannel(localConn); if (kChannel != null) { kChannel.HandleConnnect(remoteConn); } break; case KcpProtocalType.FIN: // 断开 // 长度!=13,不是DisConnect消息 if (messageLength != 13) { break; } remoteConn = BitConverter.ToUInt32(this.cache, 1); localConn = BitConverter.ToUInt32(this.cache, 5); // 处理chanel kChannel = this.GetKChannel(localConn); if (kChannel != null) { // 校验remoteConn,防止第三方攻击 if (kChannel.RemoteConn == remoteConn) { kChannel.Disconnect(ErrorCode.ERR_PeerDisconnect); } } break; case KcpProtocalType.MSG: // 断开 // 长度<9,不是Msg消息 if (messageLength < 9) { break; } // 处理chanel remoteConn = BitConverter.ToUInt32(this.cache, 1); localConn = BitConverter.ToUInt32(this.cache, 5); kChannel = this.GetKChannel(localConn); if (kChannel != null) { // 校验remoteConn,防止第三方攻击 if (kChannel.RemoteConn == remoteConn) { kChannel.HandleRecv(this.cache, 5, messageLength - 5); } } break; } } }
public void Recv() { //如果当前使用的是非阻止 Socket,一种较好的做法是在调用 Receive 之前使用 Available //来确定数据是否排队等待读取。可用的数据即网络缓冲区中排队等待读取的全部数据。 //如果在网络缓冲区中没有排队的数据,则 Available 返回 0。 if (this.socket == null) { return; } while (socket != null && this.socket.Available > 0) { int messageLength = 0; try { //接收到通讯数据 ref this.ipEndPoint 保存远程端IP messageLength = this.socket.ReceiveFrom(this.cache, ref this.ipEndPoint); } catch (Exception e) { Log.Error(e); continue; } // 长度小于1,不是正常的消息 if (messageLength < 1) { continue; } // accept byte flag = this.cache[0]; // conn从1000开始,如果为1,2,3则是特殊包 uint remoteConn = 0; uint localConn = 0; KChannel kChannel = null; switch (flag) { case KcpProtocalType.SYN: // accept //收到连接请求 // 长度!=5,不是accpet消息 if (messageLength != 5) { break; } IPEndPoint acceptIpEndPoint = (IPEndPoint)this.ipEndPoint; this.ipEndPoint = new IPEndPoint(0, 0); remoteConn = BitConverter.ToUInt32(this.cache, 1); // 如果已经收到连接,则忽略 if (this.waitConnectChannels.TryGetValue(remoteConn, out kChannel)) { break; } localConn = ++this.IdGenerater; kChannel = new KChannel(localConn, remoteConn, this.socket, acceptIpEndPoint, this); this.localConnChannels[kChannel.LocalConn] = kChannel; this.waitConnectChannels[remoteConn] = kChannel; this.OnAccept(kChannel); break; case KcpProtocalType.ACK: // connect返回 //2次握手 // 长度!=9,不是connect消息 if (messageLength != 9) { break; } remoteConn = BitConverter.ToUInt32(this.cache, 1); //请求的KChannel ID localConn = BitConverter.ToUInt32(this.cache, 5); kChannel = this.GetKChannel(localConn); //如果字典里面没有 说明没有连接过 没必要2次握手 退出不处理 if (kChannel != null) { kChannel.HandleConnnect(remoteConn); } break; case KcpProtocalType.FIN: // 断开 //收到断开请求 // 长度!=13,不是DisConnect消息 if (messageLength != 13) { break; } remoteConn = BitConverter.ToUInt32(this.cache, 1); localConn = BitConverter.ToUInt32(this.cache, 5); // 处理chanel kChannel = this.GetKChannel(localConn); if (kChannel != null) { // 校验remoteConn,防止第三方攻击 if (kChannel.RemoteConn == remoteConn) { kChannel.Disconnect(ErrorCode.ERR_PeerDisconnect); } } break; case KcpProtocalType.MSG: // 断开? // 长度<9,不是Msg消息 if (messageLength < 9) { break; } // 处理chanel remoteConn = BitConverter.ToUInt32(this.cache, 1); localConn = BitConverter.ToUInt32(this.cache, 5); this.waitConnectChannels.Remove(remoteConn); kChannel = this.GetKChannel(localConn); if (kChannel != null) { // 校验remoteConn,防止第三方攻击 if (kChannel.RemoteConn == remoteConn) { //处理普通消息 kChannel.HandleRecv(this.cache, 5, messageLength - 5); } } break; } } }