mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-12-22 18:17:52 -05:00
99 lines
2.5 KiB
Go
99 lines
2.5 KiB
Go
package internet
|
|
|
|
import (
|
|
"context"
|
|
"syscall"
|
|
"time"
|
|
|
|
"v2ray.com/core/common/net"
|
|
"v2ray.com/core/common/session"
|
|
)
|
|
|
|
var (
|
|
effectiveSystemDialer SystemDialer = DefaultSystemDialer{}
|
|
)
|
|
|
|
type SystemDialer interface {
|
|
Dial(ctx context.Context, source net.Address, destination net.Destination) (net.Conn, error)
|
|
}
|
|
|
|
type DefaultSystemDialer struct {
|
|
}
|
|
|
|
func getSocketSettings(ctx context.Context) *SocketConfig {
|
|
streamSettings := StreamSettingsFromContext(ctx)
|
|
if streamSettings != nil && streamSettings.SocketSettings != nil {
|
|
return streamSettings.SocketSettings
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (DefaultSystemDialer) Dial(ctx context.Context, src net.Address, dest net.Destination) (net.Conn, error) {
|
|
dialer := &net.Dialer{
|
|
Timeout: time.Second * 60,
|
|
DualStack: true,
|
|
}
|
|
|
|
sockopts := getSocketSettings(ctx)
|
|
if sockopts != nil {
|
|
bindAddress := BindAddressFromContext(ctx)
|
|
dialer.Control = func(network, address string, c syscall.RawConn) error {
|
|
return c.Control(func(fd uintptr) {
|
|
if err := applyOutboundSocketOptions(network, address, fd, sockopts); err != nil {
|
|
newError("failed to apply socket options").Base(err).WriteToLog(session.ExportIDToError(ctx))
|
|
}
|
|
if dest.Network == net.Network_UDP && bindAddress.IsValid() {
|
|
if err := bindAddr(fd, bindAddress.Address, bindAddress.Port); err != nil {
|
|
newError("failed to bind source address to ", bindAddress).Base(err).WriteToLog(session.ExportIDToError(ctx))
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
if src != nil && src != net.AnyIP {
|
|
var addr net.Addr
|
|
if dest.Network == net.Network_TCP {
|
|
addr = &net.TCPAddr{
|
|
IP: src.IP(),
|
|
Port: 0,
|
|
}
|
|
} else {
|
|
addr = &net.UDPAddr{
|
|
IP: src.IP(),
|
|
Port: 0,
|
|
}
|
|
}
|
|
dialer.LocalAddr = addr
|
|
}
|
|
return dialer.DialContext(ctx, dest.Network.SystemString(), dest.NetAddr())
|
|
}
|
|
|
|
type SystemDialerAdapter interface {
|
|
Dial(network string, address string) (net.Conn, error)
|
|
}
|
|
|
|
type SimpleSystemDialer struct {
|
|
adapter SystemDialerAdapter
|
|
}
|
|
|
|
func WithAdapter(dialer SystemDialerAdapter) SystemDialer {
|
|
return &SimpleSystemDialer{
|
|
adapter: dialer,
|
|
}
|
|
}
|
|
|
|
func (v *SimpleSystemDialer) Dial(ctx context.Context, src net.Address, dest net.Destination) (net.Conn, error) {
|
|
return v.adapter.Dial(dest.Network.SystemString(), dest.NetAddr())
|
|
}
|
|
|
|
// UseAlternativeSystemDialer replaces the current system dialer with a given one.
|
|
// Caller must ensure there is no race condition.
|
|
func UseAlternativeSystemDialer(dialer SystemDialer) {
|
|
if dialer == nil {
|
|
effectiveSystemDialer = DefaultSystemDialer{}
|
|
}
|
|
effectiveSystemDialer = dialer
|
|
}
|