Exemple #1
0
        public override void Update()
        {
            this.TimeNow = (uint)(TimeHelper.CurrentLocalMilliseconds() - this.StartTime);

            this.Recv();

            this.TimerOut();

            foreach (long id in updateChannels)
            {
                KChannel kChannel = this.GetKChannel(id);
                if (kChannel == null)
                {
                    continue;
                }
                if (kChannel.Id == 0)
                {
                    continue;
                }
                kChannel.Update();
            }
            this.updateChannels.Clear();

            while (this.removedChannels.Count > 0)
            {
                long     id = this.removedChannels.Dequeue();
                KChannel channel;
                if (!this.localConnChannels.TryGetValue(id, out channel))
                {
                    continue;
                }
                this.localConnChannels.Remove(id);
                channel.Dispose();
            }
        }
Exemple #2
0
        public override AChannel ConnectChannel(IPEndPoint remoteEndPoint)
        {
            uint     localConn = (uint)RandomHelper.RandomNumber(1000, int.MaxValue);
            KChannel oldChannel;

            if (this.localConnChannels.TryGetValue(localConn, out oldChannel))
            {
                this.localConnChannels.Remove(oldChannel.LocalConn);
                oldChannel.Dispose();
            }

            KChannel channel = new KChannel(localConn, this.socket, remoteEndPoint, this);

            this.localConnChannels[channel.LocalConn] = channel;
            return(channel);
        }
Exemple #3
0
        public static void Output(IntPtr bytes, int count, IntPtr user)
        {
            if (Instance == null)
            {
                return;
            }
            AChannel aChannel = Instance.GetChannel((uint)user);

            if (aChannel == null)
            {
                Log.Error($"not found kchannel, {(uint)user}");
                return;
            }

            KChannel kChannel = aChannel as KChannel;

            kChannel.Output(bytes, count);
        }
Exemple #4
0
        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.Exception(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返回
                    // 长度!=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);

                    this.waitConnectChannels.Remove(remoteConn);

                    kChannel = this.GetKChannel(localConn);
                    if (kChannel != null)
                    {
                        // 校验remoteConn,防止第三方攻击
                        if (kChannel.RemoteConn == remoteConn)
                        {
                            kChannel.HandleRecv(this.cache, 5, messageLength - 5);
                        }
                    }
                    break;
                }
            }
        }