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

Fix udp nat

Browse source
This commit is contained in:
世界 2022-04-29 17:43:09 +08:00
parent 8d95ae4cff
commit b4b6c838d1
No known key found for this signature in database
GPG key ID: CD109927C34A63C4
11 changed files with 144 additions and 331 deletions
Download patch file Download diff file Expand all files Collapse all files

12
common/udpnat/service.go
View file

@ -9,8 +9,8 @@ import (
"github.com/sagernet/sing/common"
"github.com/sagernet/sing/common/buf"
"github.com/sagernet/sing/common/cache"
E "github.com/sagernet/sing/common/exceptions"
"github.com/sagernet/sing/common/gsync"
M "github.com/sagernet/sing/common/metadata"
"github.com/sagernet/sing/protocol/socks"
)
@ -21,17 +21,18 @@ type Handler interface {
}
type Service[K comparable] struct {
nat gsync.Map[K, *conn]
nat cache.LruCache[K, *conn]
handler Handler
}
func New[T comparable](handler Handler) *Service[T] {
return &Service[T]{
func New[K comparable](maxAge int64, handler Handler) Service[K] {
return Service[K]{
nat: cache.NewLRU[K, *conn](maxAge, true),
handler: handler,
}
}
func (s *Service[T]) NewPacket(key T, writer func() socks.PacketWriter, buffer *buf.Buffer, metadata M.Metadata) error {
func (s *Service[T]) NewPacket(key T, writer func() socks.PacketWriter, buffer *buf.Buffer, metadata M.Metadata) {
c, loaded := s.nat.LoadOrStore(key, func() *conn {
c := &conn{
data: make(chan packet),
@ -57,7 +58,6 @@ func (s *Service[T]) NewPacket(key T, writer func() socks.PacketWriter, buffer *
}
c.data <- p
<-ctx.Done()
return nil
}
type packet struct {