fumiama/WireGold
1
0
Fork 0
mirror of https://github.com/fumiama/WireGold.git synced 2026-06-20 02:16:17 +08:00
Code Activity

feat(listen): add src check

Browse Source
This commit is contained in:
源文雨
2024-07-13 02:00:51 +09:00
parent af3aa02105
commit 282cb387cb
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
gold/link/listen.go
View File

@@ -109,6 +109,11 @@ func (m *Me) listenthread(packet *head.Packet, addr *net.UDPAddr, index int, fin
} }
switch { switch {
case p.IsToMe(packet.Dst): case p.IsToMe(packet.Dst):
if !p.Accept(packet.Src) {
logrus.Warnln("[listen] @", index, "refused packet from", packet.Src.String()+":"+strconv.Itoa(int(packet.SrcPort)))
packet.Put()
return
}
addt := packet.AdditionalData() addt := packet.AdditionalData()
var err error var err error
packet.Data, err = p.Decode(packet.CipherIndex(), addt, packet.Data) packet.Data, err = p.Decode(packet.CipherIndex(), addt, packet.Data)