@@ -0,0 +1,29 @@
|
||||
package core
|
||||
|
||||
import (
|
||||
"net"
|
||||
|
||||
"github.com/sagernet/sing/common/network"
|
||||
)
|
||||
|
||||
type wrappedConn struct {
|
||||
net.Conn
|
||||
tracker *ConnTracker
|
||||
connID string
|
||||
}
|
||||
|
||||
func (w *wrappedConn) Close() error {
|
||||
w.tracker.untrackConnection(w.connID)
|
||||
return w.Conn.Close()
|
||||
}
|
||||
|
||||
type wrappedPacketConn struct {
|
||||
network.PacketConn
|
||||
tracker *ConnTracker
|
||||
connID string
|
||||
}
|
||||
|
||||
func (w *wrappedPacketConn) Close() error {
|
||||
w.tracker.untrackConnection(w.connID)
|
||||
return w.PacketConn.Close()
|
||||
}
|
||||
Reference in New Issue
Block a user