|
@@ -6,16 +6,27 @@ import (
|
|
|
"net/netip"
|
|
|
"os"
|
|
|
|
|
|
+ "github.com/sagernet/sing-box/adapter"
|
|
|
"github.com/sagernet/sing/common/buf"
|
|
|
"github.com/sagernet/sing/common/control"
|
|
|
E "github.com/sagernet/sing/common/exceptions"
|
|
|
M "github.com/sagernet/sing/common/metadata"
|
|
|
N "github.com/sagernet/sing/common/network"
|
|
|
+ "github.com/sagernet/sing/service"
|
|
|
)
|
|
|
|
|
|
func (l *Listener) ListenUDP() (net.PacketConn, error) {
|
|
|
bindAddr := M.SocksaddrFrom(l.listenOptions.Listen.Build(netip.AddrFrom4([4]byte{127, 0, 0, 1})), l.listenOptions.ListenPort)
|
|
|
- var lc net.ListenConfig
|
|
|
+ var listenConfig net.ListenConfig
|
|
|
+ if l.listenOptions.BindInterface != "" {
|
|
|
+ listenConfig.Control = control.Append(listenConfig.Control, control.BindToInterface(service.FromContext[adapter.NetworkManager](l.ctx).InterfaceFinder(), l.listenOptions.BindInterface, -1))
|
|
|
+ }
|
|
|
+ if l.listenOptions.RoutingMark != 0 {
|
|
|
+ listenConfig.Control = control.Append(listenConfig.Control, control.RoutingMark(uint32(l.listenOptions.RoutingMark)))
|
|
|
+ }
|
|
|
+ if l.listenOptions.ReuseAddr {
|
|
|
+ listenConfig.Control = control.Append(listenConfig.Control, control.ReuseAddr())
|
|
|
+ }
|
|
|
var udpFragment bool
|
|
|
if l.listenOptions.UDPFragment != nil {
|
|
|
udpFragment = *l.listenOptions.UDPFragment
|
|
@@ -23,10 +34,10 @@ func (l *Listener) ListenUDP() (net.PacketConn, error) {
|
|
|
udpFragment = l.listenOptions.UDPFragmentDefault
|
|
|
}
|
|
|
if !udpFragment {
|
|
|
- lc.Control = control.Append(lc.Control, control.DisableUDPFragment())
|
|
|
+ listenConfig.Control = control.Append(listenConfig.Control, control.DisableUDPFragment())
|
|
|
}
|
|
|
udpConn, err := ListenNetworkNamespace[net.PacketConn](l.listenOptions.NetNs, func() (net.PacketConn, error) {
|
|
|
- return lc.ListenPacket(l.ctx, M.NetworkFromNetAddr(N.NetworkUDP, bindAddr.Addr), bindAddr.String())
|
|
|
+ return listenConfig.ListenPacket(l.ctx, M.NetworkFromNetAddr(N.NetworkUDP, bindAddr.Addr), bindAddr.String())
|
|
|
})
|
|
|
if err != nil {
|
|
|
return nil, err
|
|
@@ -39,12 +50,30 @@ func (l *Listener) ListenUDP() (net.PacketConn, error) {
|
|
|
|
|
|
func (l *Listener) DialContext(dialer net.Dialer, ctx context.Context, network string, address string) (net.Conn, error) {
|
|
|
return ListenNetworkNamespace[net.Conn](l.listenOptions.NetNs, func() (net.Conn, error) {
|
|
|
+ if l.listenOptions.BindInterface != "" {
|
|
|
+ dialer.Control = control.Append(dialer.Control, control.BindToInterface(service.FromContext[adapter.NetworkManager](l.ctx).InterfaceFinder(), l.listenOptions.BindInterface, -1))
|
|
|
+ }
|
|
|
+ if l.listenOptions.RoutingMark != 0 {
|
|
|
+ dialer.Control = control.Append(dialer.Control, control.RoutingMark(uint32(l.listenOptions.RoutingMark)))
|
|
|
+ }
|
|
|
+ if l.listenOptions.ReuseAddr {
|
|
|
+ dialer.Control = control.Append(dialer.Control, control.ReuseAddr())
|
|
|
+ }
|
|
|
return dialer.DialContext(ctx, network, address)
|
|
|
})
|
|
|
}
|
|
|
|
|
|
func (l *Listener) ListenPacket(listenConfig net.ListenConfig, ctx context.Context, network string, address string) (net.PacketConn, error) {
|
|
|
return ListenNetworkNamespace[net.PacketConn](l.listenOptions.NetNs, func() (net.PacketConn, error) {
|
|
|
+ if l.listenOptions.BindInterface != "" {
|
|
|
+ listenConfig.Control = control.Append(listenConfig.Control, control.BindToInterface(service.FromContext[adapter.NetworkManager](l.ctx).InterfaceFinder(), l.listenOptions.BindInterface, -1))
|
|
|
+ }
|
|
|
+ if l.listenOptions.RoutingMark != 0 {
|
|
|
+ listenConfig.Control = control.Append(listenConfig.Control, control.RoutingMark(uint32(l.listenOptions.RoutingMark)))
|
|
|
+ }
|
|
|
+ if l.listenOptions.ReuseAddr {
|
|
|
+ listenConfig.Control = control.Append(listenConfig.Control, control.ReuseAddr())
|
|
|
+ }
|
|
|
return listenConfig.ListenPacket(ctx, network, address)
|
|
|
})
|
|
|
}
|