优化架构

2026-02-17 20:42:25 +08:00 · 2022-11-23 18:05:11 +08:00
parent 3efed3defe
commit 43403202b5
6760 changed files with 33748 additions and 554768 deletions
--- a/gate/kcp/readloop.go
+++ b/gate/kcp/readloop.go
@@ -0,0 +1,126 @@
+package kcp
+
+import (
+	"bytes"
+	"encoding/binary"
+	"github.com/pkg/errors"
+)
+
+func (s *UDPSession) defaultReadLoop() {
+	buf := make([]byte, mtuLimit)
+	var src string
+	for {
+		if n, addr, err := s.conn.ReadFrom(buf); err == nil {
+			udpPayload := buf[:n]
+
+			// make sure the packet is from the same source
+			if src == "" { // set source address
+				src = addr.String()
+			} else if addr.String() != src {
+				//atomic.AddUint64(&DefaultSnmp.InErrs, 1)
+				//continue
+				s.remote = addr
+				src = addr.String()
+			}
+
+			s.packetInput(udpPayload)
+		} else {
+			s.notifyReadError(errors.WithStack(err))
+			return
+		}
+	}
+}
+
+func (l *Listener) defaultMonitor() {
+	buf := make([]byte, mtuLimit)
+	for {
+		if n, from, err := l.conn.ReadFrom(buf); err == nil {
+			udpPayload := buf[:n]
+			var convId uint64 = 0
+			if n == 20 {
+				// 原神KCP的Enet协议
+				// 提取convId
+				convId += uint64(udpPayload[4]) << 24
+				convId += uint64(udpPayload[5]) << 16
+				convId += uint64(udpPayload[6]) << 8
+				convId += uint64(udpPayload[7]) << 0
+				convId += uint64(udpPayload[8]) << 56
+				convId += uint64(udpPayload[9]) << 48
+				convId += uint64(udpPayload[10]) << 40
+				convId += uint64(udpPayload[11]) << 32
+				// 提取Enet协议头部和尾部幻数
+				udpPayloadEnetHead := udpPayload[:4]
+				udpPayloadEnetTail := udpPayload[len(udpPayload)-4:]
+				// 提取Enet协议类型
+				enetTypeData := udpPayload[12:16]
+				enetTypeDataBuffer := bytes.NewBuffer(enetTypeData)
+				var enetType uint32
+				_ = binary.Read(enetTypeDataBuffer, binary.BigEndian, &enetType)
+				equalHead := bytes.Compare(udpPayloadEnetHead, MagicEnetSynHead)
+				equalTail := bytes.Compare(udpPayloadEnetTail, MagicEnetSynTail)
+				if equalHead == 0 && equalTail == 0 {
+					// 客户端前置握手获取conv
+					l.EnetNotify <- &Enet{
+						Addr:     from.String(),
+						ConvId:   convId,
+						ConnType: ConnEnetSyn,
+						EnetType: enetType,
+					}
+					continue
+				}
+				equalHead = bytes.Compare(udpPayloadEnetHead, MagicEnetEstHead)
+				equalTail = bytes.Compare(udpPayloadEnetTail, MagicEnetEstTail)
+				if equalHead == 0 && equalTail == 0 {
+					// 连接建立
+					l.EnetNotify <- &Enet{
+						Addr:     from.String(),
+						ConvId:   convId,
+						ConnType: ConnEnetEst,
+						EnetType: enetType,
+					}
+					continue
+				}
+				equalHead = bytes.Compare(udpPayloadEnetHead, MagicEnetFinHead)
+				equalTail = bytes.Compare(udpPayloadEnetTail, MagicEnetFinTail)
+				if equalHead == 0 && equalTail == 0 {
+					// 连接断开
+					l.EnetNotify <- &Enet{
+						Addr:     from.String(),
+						ConvId:   convId,
+						ConnType: ConnEnetFin,
+						EnetType: enetType,
+					}
+					continue
+				}
+			} else {
+				// 正常KCP包
+				convId += uint64(udpPayload[0]) << 0
+				convId += uint64(udpPayload[1]) << 8
+				convId += uint64(udpPayload[2]) << 16
+				convId += uint64(udpPayload[3]) << 24
+				convId += uint64(udpPayload[4]) << 32
+				convId += uint64(udpPayload[5]) << 40
+				convId += uint64(udpPayload[6]) << 48
+				convId += uint64(udpPayload[7]) << 56
+			}
+			l.sessionLock.RLock()
+			conn, exist := l.sessions[convId]
+			l.sessionLock.RUnlock()
+			if exist {
+				if conn.remote.String() != from.String() {
+					conn.remote = from
+					// 连接地址改变
+					l.EnetNotify <- &Enet{
+						Addr:     conn.remote.String(),
+						ConvId:   convId,
+						ConnType: ConnEnetAddrChange,
+					}
+				}
+			}
+			l.packetInput(udpPayload, from, convId)
+		} else {
+			l.notifyReadError(errors.WithStack(err))
+			return
+		}
+	}
+}