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

Update network handler usages

Browse source
This commit is contained in:
世界 2024-10-20 13:42:53 +08:00
parent a8ae222eeb
commit 99eea00432
No known key found for this signature in database
GPG key ID: CD109927C34A63C4
12 changed files with 51 additions and 78 deletions
Download patch file Download diff file Expand all files Collapse all files

2
monitor_windows.go
View file

@ -4,7 +4,6 @@ import (
"sync" "sync"
"github.com/sagernet/sing-tun/internal/winipcfg" "github.com/sagernet/sing-tun/internal/winipcfg"
E "github.com/sagernet/sing/common/exceptions"
"github.com/sagernet/sing/common/logger" "github.com/sagernet/sing/common/logger"
"github.com/sagernet/sing/common/x/list" "github.com/sagernet/sing/common/x/list"
@ -14,7 +13,6 @@ import (
type networkUpdateMonitor struct { type networkUpdateMonitor struct {
routeListener *winipcfg.RouteChangeCallback routeListener *winipcfg.RouteChangeCallback
interfaceListener *winipcfg.InterfaceChangeCallback interfaceListener *winipcfg.InterfaceChangeCallback
errorHandler E.Handler
access sync.Mutex access sync.Mutex
callbacks list.List[NetworkUpdateCallback] callbacks list.List[NetworkUpdateCallback]

3
redirect.go
View file

@ -5,6 +5,7 @@ import (
"github.com/sagernet/sing/common/control" "github.com/sagernet/sing/common/control"
"github.com/sagernet/sing/common/logger" "github.com/sagernet/sing/common/logger"
N "github.com/sagernet/sing/common/network"
"go4.org/netipx" "go4.org/netipx"
) )
@ -23,7 +24,7 @@ type AutoRedirect interface {
type AutoRedirectOptions struct { type AutoRedirectOptions struct {
TunOptions *Options TunOptions *Options
Context context.Context Context context.Context
Handler Handler Handler N.TCPConnectionHandlerEx
Logger logger.Logger Logger logger.Logger
NetworkMonitor NetworkUpdateMonitor NetworkMonitor NetworkUpdateMonitor
InterfaceFinder control.InterfaceFinder InterfaceFinder control.InterfaceFinder

3
redirect_linux.go
View file

@ -13,6 +13,7 @@ import (
E "github.com/sagernet/sing/common/exceptions" E "github.com/sagernet/sing/common/exceptions"
"github.com/sagernet/sing/common/logger" "github.com/sagernet/sing/common/logger"
M "github.com/sagernet/sing/common/metadata" M "github.com/sagernet/sing/common/metadata"
N "github.com/sagernet/sing/common/network"
"github.com/sagernet/sing/common/x/list" "github.com/sagernet/sing/common/x/list"
"go4.org/netipx" "go4.org/netipx"
@ -21,7 +22,7 @@ import (
type autoRedirect struct { type autoRedirect struct {
tunOptions *Options tunOptions *Options
ctx context.Context ctx context.Context
handler Handler handler N.TCPConnectionHandlerEx
logger logger.Logger logger logger.Logger
tableName string tableName string
networkMonitor NetworkUpdateMonitor networkMonitor NetworkUpdateMonitor

17
redirect_server.go
View file

@ -14,20 +14,19 @@ import (
E "github.com/sagernet/sing/common/exceptions" E "github.com/sagernet/sing/common/exceptions"
"github.com/sagernet/sing/common/logger" "github.com/sagernet/sing/common/logger"
M "github.com/sagernet/sing/common/metadata" M "github.com/sagernet/sing/common/metadata"
N "github.com/sagernet/sing/common/network"
) )
const ProtocolRedirect = "redirect"
type redirectServer struct { type redirectServer struct {
ctx context.Context ctx context.Context
handler Handler handler N.TCPConnectionHandlerEx
logger logger.Logger logger logger.Logger
listenAddr netip.Addr listenAddr netip.Addr
listener *net.TCPListener listener *net.TCPListener
inShutdown atomic.Bool inShutdown atomic.Bool
} }
func newRedirectServer(ctx context.Context, handler Handler, logger logger.Logger, listenAddr netip.Addr) *redirectServer { func newRedirectServer(ctx context.Context, handler N.TCPConnectionHandlerEx, logger logger.Logger, listenAddr netip.Addr) *redirectServer {
return &redirectServer{ return &redirectServer{
ctx: ctx, ctx: ctx,
handler: handler, handler: handler,
@ -59,7 +58,6 @@ func (s *redirectServer) loopIn() {
conn, err := s.listener.AcceptTCP() conn, err := s.listener.AcceptTCP()
if err != nil { if err != nil {
var netError net.Error var netError net.Error
//goland:noinspection GoDeprecation
//nolint:staticcheck //nolint:staticcheck
if errors.As(err, &netError) && netError.Temporary() { if errors.As(err, &netError) && netError.Temporary() {
s.logger.Error(err) s.logger.Error(err)
@ -72,17 +70,14 @@ func (s *redirectServer) loopIn() {
s.logger.Error("serve error: ", err) s.logger.Error("serve error: ", err)
continue continue
} }
var metadata M.Metadata source := M.SocksaddrFromNet(conn.RemoteAddr()).Unwrap()
metadata.Protocol = ProtocolRedirect
metadata.Source = M.SocksaddrFromNet(conn.RemoteAddr()).Unwrap()
destination, err := control.GetOriginalDestination(conn) destination, err := control.GetOriginalDestination(conn)
if err != nil { if err != nil {
_ = conn.SetLinger(0) _ = conn.SetLinger(0)
_ = conn.Close() _ = conn.Close()
s.logger.Error("process connection from ", metadata.Source, ": invalid connection: ", err) s.logger.Error("process redirect connection from ", source, ": invalid connection: ", err)
continue continue
} }
metadata.Destination = M.SocksaddrFromNetIP(destination).Unwrap() go s.handler.NewConnectionEx(s.ctx, conn, source, M.SocksaddrFromNetIP(destination).Unwrap(), nil)
go s.handler.NewConnection(s.ctx, conn, metadata)
} }
} }

2
stack.go
View file

@ -11,6 +11,8 @@ import (
"github.com/sagernet/sing/common/logger" "github.com/sagernet/sing/common/logger"
) )
var ErrDrop = E.New("drop connections by rule")
type Stack interface { type Stack interface {
Start() error Start() error
Close() error Close() error

23
stack_gvisor.go
View file

@ -76,17 +76,16 @@ func (t *GVisor) Start() error {
return err return err
} }
tcpForwarder := tcp.NewForwarder(ipStack, 0, 1024, func(r *tcp.ForwarderRequest) { tcpForwarder := tcp.NewForwarder(ipStack, 0, 1024, func(r *tcp.ForwarderRequest) {
var metadata M.Metadata source := M.SocksaddrFrom(AddrFromAddress(r.ID().RemoteAddress), r.ID().RemotePort)
metadata.Source = M.SocksaddrFrom(AddrFromAddress(r.ID().RemoteAddress), r.ID().RemotePort) destination := M.SocksaddrFrom(AddrFromAddress(r.ID().LocalAddress), r.ID().LocalPort)
metadata.Destination = M.SocksaddrFrom(AddrFromAddress(r.ID().LocalAddress), r.ID().LocalPort)
conn := &gLazyConn{ conn := &gLazyConn{
parentCtx: t.ctx, parentCtx: t.ctx,
stack: t.stack, stack: t.stack,
request: r, request: r,
localAddr: metadata.Source.TCPAddr(), localAddr: source.TCPAddr(),
remoteAddr: metadata.Destination.TCPAddr(), remoteAddr: destination.TCPAddr(),
} }
_ = t.handler.NewConnection(t.ctx, conn, metadata) go t.handler.NewConnectionEx(t.ctx, conn, source, destination, nil)
}) })
ipStack.SetTransportProtocolHandler(tcp.ProtocolNumber, tcpForwarder.HandlePacket) ipStack.SetTransportProtocolHandler(tcp.ProtocolNumber, tcpForwarder.HandlePacket)
if !t.endpointIndependentNat { if !t.endpointIndependentNat {
@ -104,14 +103,10 @@ func (t *GVisor) Start() error {
return return
} }
go func() { go func() {
var metadata M.Metadata source := M.SocksaddrFromNet(lAddr)
metadata.Source = M.SocksaddrFromNet(lAddr) destination := M.SocksaddrFromNet(rAddr)
metadata.Destination = M.SocksaddrFromNet(rAddr) ctx, conn := canceler.NewPacketConn(t.ctx, bufio.NewUnbindPacketConnWithAddr(udpConn, destination), time.Duration(t.udpTimeout)*time.Second)
ctx, conn := canceler.NewPacketConn(t.ctx, bufio.NewUnbindPacketConnWithAddr(udpConn, metadata.Destination), time.Duration(t.udpTimeout)*time.Second) t.handler.NewPacketConnectionEx(ctx, conn, source, destination, nil)
hErr := t.handler.NewPacketConnection(ctx, conn, metadata)
if hErr != nil {
endpoint.Abort()
}
}() }()
}) })
ipStack.SetTransportProtocolHandler(udp.ProtocolNumber, udpForwarder.HandlePacket) ipStack.SetTransportProtocolHandler(udp.ProtocolNumber, udpForwarder.HandlePacket)

9
stack_gvisor_lazy.go
View file

@ -71,8 +71,7 @@ func (c *gLazyConn) HandshakeFailure(err error) error {
if c.handshakeDone { if c.handshakeDone {
return nil return nil
} }
wErr := gWriteUnreachable(c.stack, c.request.Packet(), err) c.request.Complete(gWriteUnreachable(c.stack, c.request.Packet(), err) == os.ErrInvalid)
c.request.Complete(wErr == os.ErrInvalid)
c.handshakeDone = true c.handshakeDone = true
c.handshakeErr = err c.handshakeErr = err
return nil return nil
@ -196,9 +195,11 @@ func (c *gLazyConn) Upstream() any {
} }
func gWriteUnreachable(gStack *stack.Stack, packet *stack.PacketBuffer, err error) error { func gWriteUnreachable(gStack *stack.Stack, packet *stack.PacketBuffer, err error) error {
if errors.Is(err, syscall.ENETUNREACH) { if errors.Is(err, ErrDrop) {
return nil
} else if errors.Is(err, syscall.ENETUNREACH) {
if packet.NetworkProtocolNumber == header.IPv4ProtocolNumber { if packet.NetworkProtocolNumber == header.IPv4ProtocolNumber {
return gWriteUnreachable4(gStack, packet, stack.RejectIPv4WithICMPPortUnreachable) return gWriteUnreachable4(gStack, packet, stack.RejectIPv4WithICMPNetProhibited)
} else { } else {
return gWriteUnreachable6(gStack, packet, stack.RejectIPv6WithICMPNoRoute) return gWriteUnreachable6(gStack, packet, stack.RejectIPv6WithICMPNoRoute)
} }

16
stack_gvisor_udp.go
View file

@ -36,15 +36,14 @@ func NewUDPForwarder(ctx context.Context, stack *stack.Stack, handler Handler, u
return &UDPForwarder{ return &UDPForwarder{
ctx: ctx, ctx: ctx,
stack: stack, stack: stack,
udpNat: udpnat.New[netip.AddrPort](udpTimeout, handler), udpNat: udpnat.NewEx[netip.AddrPort](udpTimeout, handler),
} }
} }
func (f *UDPForwarder) HandlePacket(id stack.TransportEndpointID, pkt *stack.PacketBuffer) bool { func (f *UDPForwarder) HandlePacket(id stack.TransportEndpointID, pkt *stack.PacketBuffer) bool {
var upstreamMetadata M.Metadata source := M.SocksaddrFrom(AddrFromAddress(id.RemoteAddress), id.RemotePort)
upstreamMetadata.Source = M.SocksaddrFrom(AddrFromAddress(id.RemoteAddress), id.RemotePort) destination := M.SocksaddrFrom(AddrFromAddress(id.LocalAddress), id.LocalPort)
upstreamMetadata.Destination = M.SocksaddrFrom(AddrFromAddress(id.LocalAddress), id.LocalPort) if source.IsIPv4() {
if upstreamMetadata.Source.IsIPv4() {
f.cacheProto = header.IPv4ProtocolNumber f.cacheProto = header.IPv4ProtocolNumber
} else { } else {
f.cacheProto = header.IPv6ProtocolNumber f.cacheProto = header.IPv6ProtocolNumber
@ -55,11 +54,12 @@ func (f *UDPForwarder) HandlePacket(id stack.TransportEndpointID, pkt *stack.Pac
sBuffer.Write(view.AsSlice()) sBuffer.Write(view.AsSlice())
}) })
f.cacheID = id f.cacheID = id
f.udpNat.NewPacket( f.udpNat.NewPacketEx(
f.ctx, f.ctx,
upstreamMetadata.Source.AddrPort(), source.AddrPort(),
sBuffer, sBuffer,
upstreamMetadata, source,
destination,
f.newUDPConn, f.newUDPConn,
) )
return true return true

12
stack_mixed.go
View file

@ -64,14 +64,10 @@ func (m *Mixed) Start() error {
return return
} }
go func() { go func() {
var metadata M.Metadata source := M.SocksaddrFromNet(lAddr)
metadata.Source = M.SocksaddrFromNet(lAddr) destination := M.SocksaddrFromNet(rAddr)
metadata.Destination = M.SocksaddrFromNet(rAddr) ctx, conn := canceler.NewPacketConn(m.ctx, bufio.NewUnbindPacketConnWithAddr(udpConn, destination), time.Duration(m.udpTimeout)*time.Second)
ctx, conn := canceler.NewPacketConn(m.ctx, bufio.NewUnbindPacketConnWithAddr(udpConn, metadata.Destination), time.Duration(m.udpTimeout)*time.Second) m.handler.NewPacketConnectionEx(ctx, conn, source, destination, nil)
hErr := m.handler.NewPacketConnection(ctx, conn, metadata)
if hErr != nil {
endpoint.Abort()
}
}() }()
}) })
ipStack.SetTransportProtocolHandler(udp.ProtocolNumber, udpForwarder.HandlePacket) ipStack.SetTransportProtocolHandler(udp.ProtocolNumber, udpForwarder.HandlePacket)

25
stack_system.go
View file

@ -152,7 +152,7 @@ func (s *System) start() error {
go s.acceptLoop(tcpListener) go s.acceptLoop(tcpListener)
} }
s.tcpNat = NewNat(s.ctx, time.Second*time.Duration(s.udpTimeout)) s.tcpNat = NewNat(s.ctx, time.Second*time.Duration(s.udpTimeout))
s.udpNat = udpnat.New[netip.AddrPort](s.udpTimeout, s.handler) s.udpNat = udpnat.NewEx[netip.AddrPort](s.udpTimeout, s.handler)
return nil return nil
} }
@ -300,16 +300,7 @@ func (s *System) acceptLoop(listener net.Listener) {
} }
} }
} }
go func() { go s.handler.NewConnectionEx(s.ctx, conn, M.SocksaddrFromNet(conn.RemoteAddr()), destination, nil)
_ = s.handler.NewConnection(s.ctx, conn, M.Metadata{
Source: M.SocksaddrFromNetIP(session.Source),
Destination: destination,
})
if tcpConn, isTCPConn := conn.(*net.TCPConn); isTCPConn {
_ = tcpConn.SetLinger(0)
}
_ = conn.Close()
}()
} }
} }
@ -427,11 +418,7 @@ func (s *System) processIPv4UDP(packet clashtcpip.IPv4Packet, header clashtcpip.
if data.Len() == 0 { if data.Len() == 0 {
return nil return nil
} }
metadata := M.Metadata{ s.udpNat.NewPacketEx(s.ctx, source, data.ToOwned(), M.SocksaddrFromNetIP(source), M.SocksaddrFromNetIP(destination), func(natConn N.PacketConn) N.PacketWriter {
Source: M.SocksaddrFromNetIP(source),
Destination: M.SocksaddrFromNetIP(destination),
}
s.udpNat.NewPacket(s.ctx, source, data.ToOwned(), metadata, func(natConn N.PacketConn) N.PacketWriter {
headerLen := packet.HeaderLen() + clashtcpip.UDPHeaderSize headerLen := packet.HeaderLen() + clashtcpip.UDPHeaderSize
headerCopy := make([]byte, headerLen) headerCopy := make([]byte, headerLen)
copy(headerCopy, packet[:headerLen]) copy(headerCopy, packet[:headerLen])
@ -459,11 +446,7 @@ func (s *System) processIPv6UDP(packet clashtcpip.IPv6Packet, header clashtcpip.
if data.Len() == 0 { if data.Len() == 0 {
return nil return nil
} }
metadata := M.Metadata{ s.udpNat.NewPacketEx(s.ctx, source, data.ToOwned(), M.SocksaddrFromNetIP(source), M.SocksaddrFromNetIP(destination), func(natConn N.PacketConn) N.PacketWriter {
Source: M.SocksaddrFromNetIP(source),
Destination: M.SocksaddrFromNetIP(destination),
}
s.udpNat.NewPacket(s.ctx, source, data.ToOwned(), metadata, func(natConn N.PacketConn) N.PacketWriter {
headerLen := len(packet) - int(header.Length()) + clashtcpip.UDPHeaderSize headerLen := len(packet) - int(header.Length()) + clashtcpip.UDPHeaderSize
headerCopy := make([]byte, headerLen) headerCopy := make([]byte, headerLen)
copy(headerCopy, packet[:headerLen]) copy(headerCopy, packet[:headerLen])

6
tun.go
View file

@ -8,7 +8,6 @@ import (
"strconv" "strconv"
"strings" "strings"
E "github.com/sagernet/sing/common/exceptions"
F "github.com/sagernet/sing/common/format" F "github.com/sagernet/sing/common/format"
"github.com/sagernet/sing/common/logger" "github.com/sagernet/sing/common/logger"
N "github.com/sagernet/sing/common/network" N "github.com/sagernet/sing/common/network"
@ -16,9 +15,8 @@ import (
) )
type Handler interface { type Handler interface {
N.TCPConnectionHandler N.TCPConnectionHandlerEx
N.UDPConnectionHandler N.UDPConnectionHandlerEx
E.Handler
} }
type Tun interface { type Tun interface {

11
tun_rules.go
View file

@ -1,7 +1,6 @@
package tun package tun
import ( import (
"context"
"net/netip" "net/netip"
"os" "os"
"runtime" "runtime"
@ -20,7 +19,7 @@ const (
userEnd uint32 = 0xFFFFFFFF - 1 userEnd uint32 = 0xFFFFFFFF - 1
) )
func (o *Options) BuildAndroidRules(packageManager PackageManager, errorHandler E.Handler) { func (o *Options) BuildAndroidRules(packageManager PackageManager) {
var includeUser []uint32 var includeUser []uint32
if len(o.IncludeAndroidUser) > 0 { if len(o.IncludeAndroidUser) > 0 {
o.IncludeAndroidUser = common.Uniq(o.IncludeAndroidUser) o.IncludeAndroidUser = common.Uniq(o.IncludeAndroidUser)
@ -64,7 +63,9 @@ func (o *Options) BuildAndroidRules(packageManager PackageManager, errorHandler
} }
continue continue
} }
errorHandler.NewError(context.Background(), E.New("package to include not found: ", packageName)) if o.Logger != nil {
o.Logger.Debug("package to include not found: ", packageName)
}
} }
} }
if len(o.ExcludePackage) > 0 { if len(o.ExcludePackage) > 0 {
@ -81,7 +82,9 @@ func (o *Options) BuildAndroidRules(packageManager PackageManager, errorHandler
} }
continue continue
} }
errorHandler.NewError(context.Background(), E.New("package to exclude not found: ", packageName)) if o.Logger != nil {
o.Logger.Debug("package to exclude not found: ", packageName)
}
} }
} }
} }