SagerNet/sing
1
0
Fork 0
mirror of https://github.com/SagerNet/sing.git synced 2025-04-03 03:47:38 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

udpnat2: Add timeout check

Browse source
This commit is contained in:
世界 2024-11-26 19:08:35 +08:00
parent 30e9d91b57
commit a8f5bf4eb0
No known key found for this signature in database
GPG key ID: CD109927C34A63C4
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
common/udpnat2/service.go
View file

@ -35,6 +35,9 @@ type Metrics struct {
} }
func New(handler N.UDPConnectionHandlerEx, prepare PrepareFunc, timeout time.Duration, shared bool) *Service { 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] var cache freelru.Cache[netip.AddrPort, *natConn]
if !shared { if !shared {
cache = common.Must1(freelru.New[netip.AddrPort, *natConn](1024, maphash.NewHasher[netip.AddrPort]().Hash32)) 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 return nil
} }
func (s *Service) Close() error { func (s *Service) Close() {
s.closeOnce.Do(func() { s.closeOnce.Do(func() {
close(s.doneChan) close(s.doneChan)
}) })
return nil
} }
func (s *Service) NewPacket(bufferSlices [][]byte, source M.Socksaddr, destination M.Socksaddr, userData any) { func (s *Service) NewPacket(bufferSlices [][]byte, source M.Socksaddr, destination M.Socksaddr, userData any) {