From 282cb387cb3422febd3500d9ba5da8a332b68038 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=BA=90=E6=96=87=E9=9B=A8?=
 <41315874+fumiama@users.noreply.github.com>
Date: Sat, 13 Jul 2024 02:00:51 +0900
Subject: [PATCH] feat(listen): add src check

---
 gold/link/listen.go | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/gold/link/listen.go b/gold/link/listen.go
index eb2fa11..0b0faff 100644
--- a/gold/link/listen.go
+++ b/gold/link/listen.go
@@ -109,6 +109,11 @@ func (m *Me) listenthread(packet *head.Packet, addr *net.UDPAddr, index int, fin
 	}
 	switch {
 	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()
 		var err error
 		packet.Data, err = p.Decode(packet.CipherIndex(), addt, packet.Data)