package internet import ( "context" "v2ray.com/core/common/net" ) type Dialer func(ctx context.Context, dest net.Destination) (Connection, error) var ( transportDialerCache = make(map[TransportProtocol]Dialer) ) func RegisterTransportDialer(protocol TransportProtocol, dialer Dialer) error { if _, found := transportDialerCache[protocol]; found { return newError(protocol, " dialer already registered").AtError() } transportDialerCache[protocol] = dialer return nil } // Dial dials a internet connection towards the given destination. func Dial(ctx context.Context, dest net.Destination) (Connection, error) { if dest.Network == net.Network_TCP { streamSettings := StreamSettingsFromContext(ctx) protocol := streamSettings.GetEffectiveProtocol() transportSettings, err := streamSettings.GetEffectiveTransportSettings() if err != nil { return nil, err } ctx = ContextWithTransportSettings(ctx, transportSettings) if streamSettings != nil && streamSettings.HasSecuritySettings() { securitySettings, err := streamSettings.GetEffectiveSecuritySettings() if err != nil { return nil, err } ctx = ContextWithSecuritySettings(ctx, securitySettings) } dialer := transportDialerCache[protocol] if dialer == nil { return nil, newError(protocol, " dialer not registered").AtError() } return dialer(ctx, dest) } udpDialer := transportDialerCache[TransportProtocol_UDP] if udpDialer == nil { return nil, newError("UDP dialer not registered").AtError() } return udpDialer(ctx, dest) } // DialSystem calls system dialer to create a network connection. func DialSystem(ctx context.Context, src net.Address, dest net.Destination) (net.Conn, error) { return effectiveSystemDialer.Dial(ctx, src, dest) }