From a8f5bf4eb026563845e2d0886b4dcef5af534dd2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E4=B8=96=E7=95=8C?= <i@sekai.icu>
Date: Tue, 26 Nov 2024 19:08:35 +0800
Subject: [PATCH] udpnat2: Add timeout check

---
 common/udpnat2/service.go | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/common/udpnat2/service.go b/common/udpnat2/service.go
index 492a04e..b04d233 100644
--- a/common/udpnat2/service.go
+++ b/common/udpnat2/service.go
@@ -35,6 +35,9 @@ type Metrics struct {
 }
 
 func New(handler N.UDPConnectionHandlerEx, prepare PrepareFunc, timeout time.Duration, shared bool) *Service {
+	if timeout == 0 {
+		panic("invalid timeout")
+	}
 	var cache freelru.Cache[netip.AddrPort, *natConn]
 	if !shared {
 		cache = common.Must1(freelru.New[netip.AddrPort, *natConn](1024, maphash.NewHasher[netip.AddrPort]().Hash32))
@@ -80,11 +83,10 @@ func (s *Service) Start() error {
 	return nil
 }
 
-func (s *Service) Close() error {
+func (s *Service) Close() {
 	s.closeOnce.Do(func() {
 		close(s.doneChan)
 	})
-	return nil
 }
 
 func (s *Service) NewPacket(bufferSlices [][]byte, source M.Socksaddr, destination M.Socksaddr, userData any) {