1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-11-10 06:16:53 -05:00
v2fly/common/environment/systemnetworkimpl/systemnetwork.go

85 lines
2.8 KiB
Go

package systemnetworkimpl
import (
"context"
"github.com/v2fly/v2ray-core/v5/common/environment"
"github.com/v2fly/v2ray-core/v5/common/net"
"github.com/v2fly/v2ray-core/v5/transport/internet"
)
func NewSystemNetworkImpl(listener internet.SystemListener, dialer internet.SystemDialer) environment.SystemNetworkCapabilitySet {
return &systemNetworkImpl{dialer: dialer, listener: listener}
}
type systemDefaultDialer struct{}
func (s systemDefaultDialer) Listen(ctx context.Context, addr net.Addr, sockopt *internet.SocketConfig) (net.Listener, error) {
return internet.ListenSystem(ctx, addr, sockopt)
}
func (s systemDefaultDialer) ListenPacket(ctx context.Context, addr net.Addr, sockopt *internet.SocketConfig) (net.PacketConn, error) {
return internet.ListenSystemPacket(ctx, addr, sockopt)
}
func (s systemDefaultDialer) Dial(ctx context.Context, source net.Address, destination net.Destination, sockopt *internet.SocketConfig) (net.Conn, error) {
return internet.DialSystem(ctx, destination, sockopt)
}
func NewSystemNetworkDefault() environment.SystemNetworkCapabilitySet {
systemDefault := systemDefaultDialer{}
return &systemNetworkImpl{dialer: systemDefault, listener: systemDefault}
}
type systemNetworkImpl struct {
listener internet.SystemListener
dialer internet.SystemDialer
}
func (s systemNetworkImpl) Dialer() internet.SystemDialer {
return s.dialer
}
func (s systemNetworkImpl) Listener() internet.SystemListener {
return s.listener
}
func NewSystemListenerWithDefaultOpt(listener internet.SystemListener, opt *internet.SocketConfig) internet.SystemListener {
return systemListenerWithDefaultOpt{SystemListener: listener, opt: opt}
}
type systemListenerWithDefaultOpt struct {
internet.SystemListener
opt *internet.SocketConfig
}
func (s systemListenerWithDefaultOpt) Listen(ctx context.Context, addr net.Addr, sockopt *internet.SocketConfig) (net.Listener, error) {
if sockopt == nil {
return s.Listen(ctx, addr, s.opt)
}
return s.Listen(ctx, addr, sockopt)
}
func (s systemListenerWithDefaultOpt) ListenPacket(ctx context.Context, addr net.Addr, sockopt *internet.SocketConfig) (net.PacketConn, error) {
if sockopt == nil {
return s.ListenPacket(ctx, addr, s.opt)
}
return s.ListenPacket(ctx, addr, sockopt)
}
func NewSystemDialerWithDefaultOpt(listener internet.SystemDialer, opt *internet.SocketConfig) internet.SystemDialer {
return systemDialerWithDefaultOpt{SystemDialer: listener, opt: opt}
}
type systemDialerWithDefaultOpt struct {
internet.SystemDialer
opt *internet.SocketConfig
}
func (s systemDialerWithDefaultOpt) Dial(ctx context.Context, source net.Address, destination net.Destination, sockopt *internet.SocketConfig) (net.Conn, error) {
if sockopt == nil {
return s.Dial(ctx, source, destination, s.opt)
}
return s.Dial(ctx, source, destination, sockopt)
}