fumiama/WireGold
1
0
Fork 0
mirror of https://github.com/fumiama/WireGold.git synced 2026-06-13 05:31:08 +08:00
Code Activity

fix: flag

Browse Source
This commit is contained in:
源文雨
2022-05-14 00:12:42 +08:00
parent e91d7a33a8
commit 998a16ec56
1 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
gold/head/packet.go
View File

@@ -110,13 +110,13 @@ func (p *Packet) Marshal(src net.IP, teatype uint8, datasz uint32, offset uint16
if src != nil { if src != nil {
p.TeaTypeDataSZ = uint32(teatype)<<24 | datasz p.TeaTypeDataSZ = uint32(teatype)<<24 | datasz
p.Src = src p.Src = src
p.Flags = offset & 0x1fff
if dontfrag { if dontfrag {
offset |= 0x4000 offset |= 0x4000
} }
if hasmore { if hasmore {
offset |= 0x2000 offset |= 0x2000
} }
p.Flags = offset
} }
return helper.OpenWriterF(func(w *helper.Writer) { return helper.OpenWriterF(func(w *helper.Writer) {
@@ -140,8 +140,7 @@ func (p *Packet) FillHash() {
logrus.Error("[packet] err when fill hash:", err) logrus.Error("[packet] err when fill hash:", err)
return return
} }
_ = h.Sum(p.Hash[:0]) logrus.Debugln("[packet] sum calulated:", hex.EncodeToString(h.Sum(p.Hash[:0])))
logrus.Debugln("[packet] sum calulated:", hex.EncodeToString(p.Hash[:]))
} }
// IsVaildHash 验证 packet 合法性 // IsVaildHash 验证 packet 合法性
@@ -153,8 +152,7 @@ func (p *Packet) IsVaildHash() bool {
return false return false
} }
var sum [32]byte var sum [32]byte
_ = h.Sum(sum[:0]) logrus.Debugln("[packet] sum calulated:", hex.EncodeToString(h.Sum(sum[:0])))
logrus.Debugln("[packet] sum calulated:", hex.EncodeToString(sum[:]))
logrus.Debugln("[packet] sum in packet:", hex.EncodeToString(p.Hash[:])) logrus.Debugln("[packet] sum in packet:", hex.EncodeToString(p.Hash[:]))
return sum == p.Hash return sum == p.Hash
} }