Alireza Ahmadi
30 lines
464 B
Go
30 lines
464 B
Go
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()
|
|
}
|