fumiama/WireGold
1
0
Fork 0
mirror of https://github.com/fumiama/WireGold.git synced 2026-06-29 23:30:37 +08:00
Code Activity

optimize(head): packet encapsuling

Browse Source
This commit is contained in:
源文雨
2024-07-13 00:22:20 +09:00
parent 8215abb695
commit 6ede65bf28
6 changed files with 136 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
gold/link/recv.go
View File

@@ -3,6 +3,7 @@ package link
import (
"encoding/binary"
"encoding/hex"
"strconv"
"unsafe"
"github.com/fumiama/WireGold/gold/head"
@@ -27,17 +28,18 @@ func (m *Me) wait(data []byte) *head.Packet {
logrus.Debugln("[recv] data bytes", hex.EncodeToString(data[:bound]), endl)
data = m.xordec(data)
logrus.Debugln("[recv] data xored", hex.EncodeToString(data[:bound]), endl)
flags := binary.LittleEndian.Uint16(data[10:12])
if flags&0x8000 != 0 { // not a valid packet
flags := head.Flags(data)
if !flags.IsValid() {
logrus.Debugln("[recv] drop invalid flags packet:", hex.EncodeToString(data[11:12]), hex.EncodeToString(data[10:11]))
return nil
}
crc := binary.LittleEndian.Uint64(data[52:60])
if m.recved.Get(crc) { // 是重放攻击
logrus.Warnln("[recv] ignore duplicated crc packet", strconv.FormatUint(crc, 16))
return nil
}
logrus.Debugln("[recv]", len(data), "bytes data with flag", hex.EncodeToString(data[11:12]), hex.EncodeToString(data[10:11]))
if flags == 0 || flags == 0x4000 {
if flags.IsSingle() || flags.NoFrag() {
h := head.SelectPacket()
_, err := h.Unmarshal(data)
if err != nil {