From 524a6bd0d1c63737ab4353681fa3db200b863188 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E4=B8=96=E7=95=8C?= <i@sekai.icu>
Date: Fri, 8 Nov 2024 10:22:04 +0800
Subject: [PATCH] udpnat2: Set upstream to writer

---
 common/udpnat2/conn.go    | 4 ++++
 common/udpnat2/service.go | 1 -
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/common/udpnat2/conn.go b/common/udpnat2/conn.go
index a96f4c8..5d474e6 100644
--- a/common/udpnat2/conn.go
+++ b/common/udpnat2/conn.go
@@ -104,3 +104,7 @@ func (c *Conn) SetReadDeadline(t time.Time) error {
 func (c *Conn) SetWriteDeadline(t time.Time) error {
 	return os.ErrInvalid
 }
+
+func (c *Conn) Upstream() any {
+	return c.writer
+}
diff --git a/common/udpnat2/service.go b/common/udpnat2/service.go
index 8c8afc9..e2a1482 100644
--- a/common/udpnat2/service.go
+++ b/common/udpnat2/service.go
@@ -55,7 +55,6 @@ func (s *Service) NewPacket(bufferSlices [][]byte, source M.Socksaddr, destinati
 	if !loaded {
 		ok, ctx, writer, onClose := s.prepare(source, destination, userData)
 		if !ok {
-			println(2)
 			s.metrics.Rejects++
 			return
 		}