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

fix(recv): panic on short data len

Browse Source
This commit is contained in:
源文雨
2025-02-21 22:41:16 +09:00
parent ac1d325bf0
commit 3377d87d7a
1 changed files with 18 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
gold/link/recv.go
View File

@@ -19,6 +19,9 @@ func (l *Link) Read() *head.Packet {
func (m *Me) wait(data []byte) *head.Packet { func (m *Me) wait(data []byte) *head.Packet {
if len(data) < head.PacketHeadLen { // not a valid packet if len(data) < head.PacketHeadLen { // not a valid packet
if config.ShowDebugLog {
logrus.Debugln("[recv] invalid data len", len(data))
}
return nil return nil
} }
bound := 64 bound := 64
@@ -28,7 +31,7 @@ func (m *Me) wait(data []byte) *head.Packet {
endl = "." endl = "."
} }
if config.ShowDebugLog { if config.ShowDebugLog {
logrus.Debugln("[recv] data bytes", hex.EncodeToString(data[:bound]), endl) logrus.Debugln("[recv] data bytes, len", len(data), "val", hex.EncodeToString(data[:bound]), endl)
} }
if m.base14 { if m.base14 {
data = base14.Decode(data) data = base14.Decode(data)
@@ -37,7 +40,13 @@ func (m *Me) wait(data []byte) *head.Packet {
endl = "." endl = "."
} }
if config.ShowDebugLog { if config.ShowDebugLog {
logrus.Debugln("[recv] data b14ed", hex.EncodeToString(data[:bound]), endl) logrus.Debugln("[recv] data b14ed, len", len(data), "val", hex.EncodeToString(data[:bound]), endl)
}
if len(data) < head.PacketHeadLen { // not a valid packet
if config.ShowDebugLog {
logrus.Debugln("[recv] invalid data len", len(data))
}
return nil
} }
} }
seq, data := m.xordec(data) seq, data := m.xordec(data)
@@ -46,7 +55,13 @@ func (m *Me) wait(data []byte) *head.Packet {
endl = "." endl = "."
} }
if config.ShowDebugLog { if config.ShowDebugLog {
logrus.Debugln("[recv] data xored", hex.EncodeToString(data[:bound]), endl) logrus.Debugln("[recv] data xored, len", len(data), "val", hex.EncodeToString(data[:bound]), endl)
}
if len(data) < head.PacketHeadLen { // not a valid packet
if config.ShowDebugLog {
logrus.Debugln("[recv] invalid data len", len(data))
}
return nil
} }
flags := head.Flags(data) flags := head.Flags(data)
if !flags.IsValid() { if !flags.IsValid() {