fix: sing-based listener panic

This commit is contained in:
wwqgtxx
2023-05-12 09:14:27 +08:00
parent f1be9b3f4a
commit a22b1cd69e
8 changed files with 57 additions and 16 deletions

View File

@@ -15,20 +15,24 @@ var NewDeadlinePacketConn = deadline.NewPacketConn
var NewDeadlineEnhancePacketConn = deadline.NewEnhancePacketConn
type threadSafePacketConn struct {
net.PacketConn
EnhancePacketConn
access sync.Mutex
}
func (c *threadSafePacketConn) WriteTo(b []byte, addr net.Addr) (int, error) {
c.access.Lock()
defer c.access.Unlock()
return c.PacketConn.WriteTo(b, addr)
return c.EnhancePacketConn.WriteTo(b, addr)
}
func (c *threadSafePacketConn) Upstream() any {
return c.PacketConn
return c.EnhancePacketConn
}
func (c *threadSafePacketConn) ReaderReplaceable() bool {
return true
}
func NewThreadSafePacketConn(pc net.PacketConn) net.PacketConn {
return &threadSafePacketConn{PacketConn: pc}
return &threadSafePacketConn{EnhancePacketConn: NewEnhancePacketConn(pc)}
}