mirror of
https://github.com/v2ray/v2ray-core.git
synced 2026-11-13 22:45:13 +00:00
separate network and transport protocol
This commit is contained in:
41 files changed
+261
-321
No files matched your search
@@ -2,52 +2,58 @@ package internet
|
||||
|
||||
import (
|
||||
"v2ray.com/core/common/errors"
|
||||
"v2ray.com/core/common/log"
|
||||
v2net "v2ray.com/core/common/net"
|
||||
"v2ray.com/core/common/serial"
|
||||
)
|
||||
|
||||
type ConfigCreator func() interface{}
|
||||
|
||||
var (
|
||||
globalNetworkConfigCreatorCache = make(map[v2net.Network]ConfigCreator)
|
||||
|
||||
globalNetworkSettings []*NetworkSettings
|
||||
|
||||
ErrUnconfiguredNetwork = errors.New("Network config creator not set.")
|
||||
globalTransportConfigCreatorCache = make(map[TransportProtocol]ConfigCreator)
|
||||
globalTransportSettings []*TransportSettings
|
||||
)
|
||||
|
||||
func RegisterNetworkConfigCreator(network v2net.Network, creator ConfigCreator) error {
|
||||
func RegisterProtocolConfigCreator(protocol TransportProtocol, creator ConfigCreator) error {
|
||||
// TODO: check duplicate
|
||||
globalNetworkConfigCreatorCache[network] = creator
|
||||
globalTransportConfigCreatorCache[protocol] = creator
|
||||
return nil
|
||||
}
|
||||
|
||||
func CreateNetworkConfig(network v2net.Network) (interface{}, error) {
|
||||
creator, ok := globalNetworkConfigCreatorCache[network]
|
||||
func CreateTransportConfig(protocol TransportProtocol) (interface{}, error) {
|
||||
creator, ok := globalTransportConfigCreatorCache[protocol]
|
||||
if !ok {
|
||||
log.Warning("Internet: Network config creator not found: ", network)
|
||||
return nil, ErrUnconfiguredNetwork
|
||||
return nil, errors.New("Internet: Unknown transport protocol: ", protocol)
|
||||
}
|
||||
return creator(), nil
|
||||
}
|
||||
|
||||
func (v *NetworkSettings) GetTypedSettings() (interface{}, error) {
|
||||
func (v *TransportSettings) GetTypedSettings() (interface{}, error) {
|
||||
return v.Settings.GetInstance()
|
||||
}
|
||||
|
||||
func (v *StreamConfig) GetEffectiveNetworkSettings() (interface{}, error) {
|
||||
for _, settings := range v.NetworkSettings {
|
||||
if settings.Network == v.Network {
|
||||
func (v *StreamConfig) GetEffectiveProtocol() TransportProtocol {
|
||||
if v == nil {
|
||||
return TransportProtocol_TCP
|
||||
}
|
||||
return v.Protocol
|
||||
}
|
||||
|
||||
func (v *StreamConfig) GetEffectiveTransportSettings() (interface{}, error) {
|
||||
protocol := v.GetEffectiveProtocol()
|
||||
|
||||
if v != nil {
|
||||
for _, settings := range v.TransportSettings {
|
||||
if settings.Protocol == protocol {
|
||||
return settings.GetTypedSettings()
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
for _, settings := range globalTransportSettings {
|
||||
if settings.Protocol == protocol {
|
||||
return settings.GetTypedSettings()
|
||||
}
|
||||
}
|
||||
for _, settings := range globalNetworkSettings {
|
||||
if settings.Network == v.Network {
|
||||
return settings.GetTypedSettings()
|
||||
}
|
||||
}
|
||||
return CreateNetworkConfig(v.Network)
|
||||
return CreateTransportConfig(protocol)
|
||||
}
|
||||
|
||||
func (v *StreamConfig) GetEffectiveSecuritySettings() (interface{}, error) {
|
||||
@@ -63,8 +69,8 @@ func (v *StreamConfig) HasSecuritySettings() bool {
|
||||
return len(v.SecurityType) > 0
|
||||
}
|
||||
|
||||
func ApplyGlobalNetworkSettings(settings []*NetworkSettings) error {
|
||||
globalNetworkSettings = settings
|
||||
func ApplyGlobalTransportSettings(settings []*TransportSettings) error {
|
||||
globalTransportSettings = settings
|
||||
return nil
|
||||
}
|
||||
|
||||
|
||||
@@ -3,7 +3,6 @@ package internet
|
||||
import proto "github.com/golang/protobuf/proto"
|
||||
import fmt "fmt"
|
||||
import math "math"
|
||||
import v2ray_core_common_net "v2ray.com/core/common/net"
|
||||
import v2ray_core_common_serial "v2ray.com/core/common/serial"
|
||||
|
||||
// Reference imports to suppress errors if they are not otherwise used.
|
||||
@@ -17,26 +16,53 @@ var _ = math.Inf
|
||||
// proto package needs to be updated.
|
||||
const _ = proto.ProtoPackageIsVersion2 // please upgrade the proto package
|
||||
|
||||
type NetworkSettings struct {
|
||||
type TransportProtocol int32
|
||||
|
||||
const (
|
||||
TransportProtocol_TCP TransportProtocol = 0
|
||||
TransportProtocol_UDP TransportProtocol = 1
|
||||
TransportProtocol_MKCP TransportProtocol = 2
|
||||
TransportProtocol_WebSocket TransportProtocol = 3
|
||||
)
|
||||
|
||||
var TransportProtocol_name = map[int32]string{
|
||||
0: "TCP",
|
||||
1: "UDP",
|
||||
2: "MKCP",
|
||||
3: "WebSocket",
|
||||
}
|
||||
var TransportProtocol_value = map[string]int32{
|
||||
"TCP": 0,
|
||||
"UDP": 1,
|
||||
"MKCP": 2,
|
||||
"WebSocket": 3,
|
||||
}
|
||||
|
||||
func (x TransportProtocol) String() string {
|
||||
return proto.EnumName(TransportProtocol_name, int32(x))
|
||||
}
|
||||
func (TransportProtocol) EnumDescriptor() ([]byte, []int) { return fileDescriptor0, []int{0} }
|
||||
|
||||
type TransportSettings struct {
|
||||
// Type of network that this settings supports.
|
||||
Network v2ray_core_common_net.Network `protobuf:"varint,1,opt,name=network,enum=v2ray.core.common.net.Network" json:"network,omitempty"`
|
||||
Protocol TransportProtocol `protobuf:"varint,1,opt,name=protocol,enum=v2ray.core.transport.internet.TransportProtocol" json:"protocol,omitempty"`
|
||||
// Specific settings.
|
||||
Settings *v2ray_core_common_serial.TypedMessage `protobuf:"bytes,2,opt,name=settings" json:"settings,omitempty"`
|
||||
}
|
||||
|
||||
func (m *NetworkSettings) Reset() { *m = NetworkSettings{} }
|
||||
func (m *NetworkSettings) String() string { return proto.CompactTextString(m) }
|
||||
func (*NetworkSettings) ProtoMessage() {}
|
||||
func (*NetworkSettings) Descriptor() ([]byte, []int) { return fileDescriptor0, []int{0} }
|
||||
func (m *TransportSettings) Reset() { *m = TransportSettings{} }
|
||||
func (m *TransportSettings) String() string { return proto.CompactTextString(m) }
|
||||
func (*TransportSettings) ProtoMessage() {}
|
||||
func (*TransportSettings) Descriptor() ([]byte, []int) { return fileDescriptor0, []int{0} }
|
||||
|
||||
func (m *NetworkSettings) GetNetwork() v2ray_core_common_net.Network {
|
||||
func (m *TransportSettings) GetProtocol() TransportProtocol {
|
||||
if m != nil {
|
||||
return m.Network
|
||||
return m.Protocol
|
||||
}
|
||||
return v2ray_core_common_net.Network_Unknown
|
||||
return TransportProtocol_TCP
|
||||
}
|
||||
|
||||
func (m *NetworkSettings) GetSettings() *v2ray_core_common_serial.TypedMessage {
|
||||
func (m *TransportSettings) GetSettings() *v2ray_core_common_serial.TypedMessage {
|
||||
if m != nil {
|
||||
return m.Settings
|
||||
}
|
||||
@@ -45,8 +71,8 @@ func (m *NetworkSettings) GetSettings() *v2ray_core_common_serial.TypedMessage {
|
||||
|
||||
type StreamConfig struct {
|
||||
// Effective network.
|
||||
Network v2ray_core_common_net.Network `protobuf:"varint,1,opt,name=network,enum=v2ray.core.common.net.Network" json:"network,omitempty"`
|
||||
NetworkSettings []*NetworkSettings `protobuf:"bytes,2,rep,name=network_settings,json=networkSettings" json:"network_settings,omitempty"`
|
||||
Protocol TransportProtocol `protobuf:"varint,1,opt,name=protocol,enum=v2ray.core.transport.internet.TransportProtocol" json:"protocol,omitempty"`
|
||||
TransportSettings []*TransportSettings `protobuf:"bytes,2,rep,name=transport_settings,json=transportSettings" json:"transport_settings,omitempty"`
|
||||
// Type of security. Must be a message name of the settings proto.
|
||||
SecurityType string `protobuf:"bytes,3,opt,name=security_type,json=securityType" json:"security_type,omitempty"`
|
||||
SecuritySettings []*v2ray_core_common_serial.TypedMessage `protobuf:"bytes,4,rep,name=security_settings,json=securitySettings" json:"security_settings,omitempty"`
|
||||
@@ -57,16 +83,16 @@ func (m *StreamConfig) String() string { return proto.CompactTextStri
|
||||
func (*StreamConfig) ProtoMessage() {}
|
||||
func (*StreamConfig) Descriptor() ([]byte, []int) { return fileDescriptor0, []int{1} }
|
||||
|
||||
func (m *StreamConfig) GetNetwork() v2ray_core_common_net.Network {
|
||||
func (m *StreamConfig) GetProtocol() TransportProtocol {
|
||||
if m != nil {
|
||||
return m.Network
|
||||
return m.Protocol
|
||||
}
|
||||
return v2ray_core_common_net.Network_Unknown
|
||||
return TransportProtocol_TCP
|
||||
}
|
||||
|
||||
func (m *StreamConfig) GetNetworkSettings() []*NetworkSettings {
|
||||
func (m *StreamConfig) GetTransportSettings() []*TransportSettings {
|
||||
if m != nil {
|
||||
return m.NetworkSettings
|
||||
return m.TransportSettings
|
||||
}
|
||||
return nil
|
||||
}
|
||||
@@ -102,35 +128,38 @@ func (m *ProxyConfig) GetTag() string {
|
||||
}
|
||||
|
||||
func init() {
|
||||
proto.RegisterType((*NetworkSettings)(nil), "v2ray.core.transport.internet.NetworkSettings")
|
||||
proto.RegisterType((*TransportSettings)(nil), "v2ray.core.transport.internet.TransportSettings")
|
||||
proto.RegisterType((*StreamConfig)(nil), "v2ray.core.transport.internet.StreamConfig")
|
||||
proto.RegisterType((*ProxyConfig)(nil), "v2ray.core.transport.internet.ProxyConfig")
|
||||
proto.RegisterEnum("v2ray.core.transport.internet.TransportProtocol", TransportProtocol_name, TransportProtocol_value)
|
||||
}
|
||||
|
||||
func init() { proto.RegisterFile("v2ray.com/core/transport/internet/config.proto", fileDescriptor0) }
|
||||
|
||||
var fileDescriptor0 = []byte{
|
||||
// 344 bytes of a gzipped FileDescriptorProto
|
||||
0x1f, 0x8b, 0x08, 0x00, 0x00, 0x09, 0x6e, 0x88, 0x02, 0xff, 0xa4, 0x92, 0xcf, 0x4a, 0xc3, 0x40,
|
||||
0x10, 0xc6, 0x49, 0x2b, 0xda, 0x6e, 0xab, 0xad, 0x39, 0x95, 0x42, 0xb5, 0x56, 0xd0, 0x9c, 0x26,
|
||||
0x12, 0x2f, 0x9e, 0xdb, 0x93, 0x87, 0x8a, 0xa4, 0x45, 0xd0, 0x4b, 0x59, 0xe3, 0x18, 0x82, 0x66,
|
||||
0xb7, 0xec, 0x8e, 0x7f, 0xf2, 0x14, 0x3e, 0x81, 0x2f, 0xe0, 0x53, 0x4a, 0x92, 0xdd, 0x50, 0x4a,
|
||||
0x29, 0x82, 0xb7, 0x21, 0xf9, 0xe6, 0xcb, 0x6f, 0x7e, 0x84, 0xc1, 0x7b, 0xa0, 0x78, 0x06, 0x91,
|
||||
0x4c, 0xfd, 0x48, 0x2a, 0xf4, 0x49, 0x71, 0xa1, 0x97, 0x52, 0x91, 0x9f, 0x08, 0x42, 0x25, 0x90,
|
||||
0xfc, 0x48, 0x8a, 0xe7, 0x24, 0x86, 0xa5, 0x92, 0x24, 0xdd, 0x81, 0xcd, 0x2b, 0x84, 0x2a, 0x0b,
|
||||
0x36, 0xdb, 0x3f, 0x5f, 0xab, 0x8b, 0x64, 0x9a, 0x4a, 0xe1, 0xe7, 0x35, 0x02, 0xe9, 0x43, 0xaa,
|
||||
0x97, 0xb2, 0xa7, 0x7f, 0xb1, 0x39, 0xa8, 0x51, 0x25, 0xfc, 0xd5, 0xa7, 0x6c, 0x89, 0x4f, 0x8b,
|
||||
0x14, 0xb5, 0xe6, 0x31, 0x96, 0x1b, 0xa3, 0x2f, 0x87, 0x75, 0x6e, 0xca, 0x8e, 0x19, 0x12, 0x25,
|
||||
0x22, 0xd6, 0xee, 0x15, 0xdb, 0x33, 0xb5, 0x3d, 0x67, 0xe8, 0x78, 0x07, 0xc1, 0x11, 0xac, 0xf0,
|
||||
0x95, 0x9d, 0x20, 0x90, 0xc0, 0x2c, 0x86, 0x36, 0xee, 0x8e, 0x59, 0x43, 0x9b, 0x96, 0x5e, 0x6d,
|
||||
0xe8, 0x78, 0xad, 0xe0, 0x6c, 0xc3, 0x6a, 0x89, 0x03, 0xf3, 0x1c, 0x67, 0x5a, 0xd2, 0x84, 0xd5,
|
||||
0xde, 0xe8, 0xbb, 0xc6, 0xda, 0x33, 0x52, 0xc8, 0xd3, 0x49, 0xa1, 0xe8, 0x1f, 0x38, 0xf7, 0xac,
|
||||
0x6b, 0xc6, 0xc5, 0x0a, 0x56, 0xdd, 0x6b, 0x05, 0x00, 0x5b, 0x8d, 0xc3, 0x9a, 0x92, 0xb0, 0x23,
|
||||
0xd6, 0x1c, 0x9d, 0xb2, 0x7d, 0x8d, 0xd1, 0x9b, 0x4a, 0x28, 0x5b, 0xe4, 0x5e, 0x7b, 0xf5, 0xa1,
|
||||
0xe3, 0x35, 0xc3, 0xb6, 0x7d, 0x98, 0x1f, 0xe7, 0xce, 0xd8, 0x61, 0x15, 0xaa, 0x00, 0x76, 0x0a,
|
||||
0x80, 0xbf, 0x7a, 0xe9, 0xda, 0x02, 0xfb, 0xe5, 0xd1, 0x31, 0x6b, 0xdd, 0x2a, 0xf9, 0x99, 0x19,
|
||||
0x3b, 0x5d, 0x56, 0x27, 0x1e, 0x17, 0x66, 0x9a, 0x61, 0x3e, 0x8e, 0xa7, 0xec, 0x24, 0x92, 0xe9,
|
||||
0xf6, 0x03, 0x1f, 0x1a, 0x76, 0xfa, 0xa9, 0x0d, 0xee, 0x82, 0x90, 0x67, 0x30, 0xc9, 0x93, 0xf3,
|
||||
0x2a, 0x79, 0x6d, 0xde, 0x3f, 0xee, 0x16, 0x3f, 0xca, 0xe5, 0x6f, 0x00, 0x00, 0x00, 0xff, 0xff,
|
||||
0x5e, 0x2e, 0x73, 0x09, 0xd4, 0x02, 0x00, 0x00,
|
||||
// 372 bytes of a gzipped FileDescriptorProto
|
||||
0x1f, 0x8b, 0x08, 0x00, 0x00, 0x09, 0x6e, 0x88, 0x02, 0xff, 0xac, 0x92, 0x4f, 0x4b, 0xf3, 0x40,
|
||||
0x10, 0xc6, 0xdf, 0x24, 0xe5, 0x35, 0x9d, 0xb6, 0x92, 0xee, 0xa9, 0x08, 0xc5, 0x5a, 0x41, 0x82,
|
||||
0x87, 0x4d, 0x89, 0x77, 0x0f, 0x8d, 0x17, 0xd1, 0x42, 0x48, 0xaa, 0x82, 0x97, 0x90, 0xae, 0x6b,
|
||||
0x08, 0x36, 0xd9, 0xb2, 0x59, 0xc5, 0x7c, 0x20, 0x2f, 0x7e, 0x01, 0xbf, 0x9e, 0xe4, 0xcf, 0x2e,
|
||||
0x45, 0xa1, 0xf4, 0xe0, 0x6d, 0xc8, 0x3c, 0xf3, 0x9b, 0x27, 0xcf, 0x2c, 0xe0, 0x37, 0x97, 0xc7,
|
||||
0x25, 0x26, 0x2c, 0x73, 0x08, 0xe3, 0xd4, 0x11, 0x3c, 0xce, 0x8b, 0x0d, 0xe3, 0xc2, 0x49, 0x73,
|
||||
0x41, 0x79, 0x4e, 0x85, 0x43, 0x58, 0xfe, 0x9c, 0x26, 0x78, 0xc3, 0x99, 0x60, 0x68, 0x2c, 0xf5,
|
||||
0x9c, 0x62, 0xa5, 0xc5, 0x52, 0x7b, 0x34, 0xfb, 0x81, 0x23, 0x2c, 0xcb, 0x58, 0xee, 0x14, 0x94,
|
||||
0xa7, 0xf1, 0xda, 0x11, 0xe5, 0x86, 0x3e, 0x45, 0x19, 0x2d, 0x8a, 0x38, 0xa1, 0x0d, 0x70, 0xfa,
|
||||
0xa1, 0xc1, 0x70, 0x29, 0x41, 0x21, 0x15, 0x22, 0xcd, 0x93, 0x02, 0xdd, 0x82, 0x59, 0xb7, 0x09,
|
||||
0x5b, 0x8f, 0xb4, 0x89, 0x66, 0x1f, 0xba, 0x33, 0xbc, 0x73, 0x33, 0x56, 0x0c, 0xbf, 0x9d, 0x0b,
|
||||
0x14, 0x01, 0xcd, 0xc1, 0x2c, 0x5a, 0xf2, 0x48, 0x9f, 0x68, 0x76, 0xcf, 0x3d, 0xdb, 0xa6, 0x35,
|
||||
0x26, 0x71, 0x63, 0x12, 0x2f, 0x2b, 0x93, 0x8b, 0xc6, 0x63, 0xa0, 0xe6, 0xa6, 0x5f, 0x3a, 0xf4,
|
||||
0x43, 0xc1, 0x69, 0x9c, 0x79, 0x75, 0x1e, 0x7f, 0x6c, 0x31, 0x02, 0xa4, 0x26, 0xa2, 0x2d, 0xb3,
|
||||
0x86, 0xdd, 0xdb, 0x9f, 0x2b, 0xe3, 0x0b, 0x86, 0xe2, 0x57, 0xa2, 0xa7, 0x30, 0x28, 0x28, 0x79,
|
||||
0xe5, 0xa9, 0x28, 0xa3, 0xea, 0x0e, 0x23, 0x63, 0xa2, 0xd9, 0xdd, 0xa0, 0x2f, 0x3f, 0x56, 0xbf,
|
||||
0x8d, 0x42, 0x18, 0x2a, 0x91, 0x32, 0xd1, 0xa9, 0x4d, 0xec, 0x9b, 0x98, 0x25, 0x01, 0x72, 0xf3,
|
||||
0xf4, 0x18, 0x7a, 0x3e, 0x67, 0xef, 0x65, 0x9b, 0x9b, 0x05, 0x86, 0x88, 0x93, 0x3a, 0xb2, 0x6e,
|
||||
0x50, 0x95, 0xe7, 0x97, 0x5b, 0x2f, 0x40, 0x46, 0x83, 0x0e, 0xc0, 0x58, 0x7a, 0xbe, 0xf5, 0xaf,
|
||||
0x2a, 0xee, 0xae, 0x7c, 0x4b, 0x43, 0x26, 0x74, 0x16, 0x37, 0x9e, 0x6f, 0xe9, 0x68, 0x00, 0xdd,
|
||||
0x07, 0xba, 0x0a, 0x19, 0x79, 0xa1, 0xc2, 0x32, 0xe6, 0x0b, 0x38, 0x21, 0x2c, 0xdb, 0x1d, 0xd2,
|
||||
0xa3, 0x29, 0xab, 0x4f, 0x7d, 0x7c, 0xef, 0x06, 0x71, 0x89, 0xbd, 0x4a, 0xa9, 0x16, 0xe3, 0xeb,
|
||||
0xb6, 0xbf, 0xfa, 0x5f, 0x1f, 0xe5, 0xe2, 0x3b, 0x00, 0x00, 0xff, 0xff, 0xca, 0x28, 0x5d, 0x77,
|
||||
0x1b, 0x03, 0x00, 0x00,
|
||||
}
|
||||
@@ -5,12 +5,18 @@ option csharp_namespace = "V2Ray.Core.Transport.Internet";
|
||||
option go_package = "internet";
|
||||
option java_package = "com.v2ray.core.transport.internet";
|
||||
|
||||
import "v2ray.com/core/common/net/network.proto";
|
||||
import "v2ray.com/core/common/serial/typed_message.proto";
|
||||
|
||||
message NetworkSettings {
|
||||
enum TransportProtocol {
|
||||
TCP = 0;
|
||||
UDP = 1;
|
||||
MKCP = 2;
|
||||
WebSocket = 3;
|
||||
}
|
||||
|
||||
message TransportSettings {
|
||||
// Type of network that this settings supports.
|
||||
v2ray.core.common.net.Network network = 1;
|
||||
TransportProtocol protocol = 1;
|
||||
|
||||
// Specific settings.
|
||||
v2ray.core.common.serial.TypedMessage settings = 2;
|
||||
@@ -18,9 +24,9 @@ message NetworkSettings {
|
||||
|
||||
message StreamConfig {
|
||||
// Effective network.
|
||||
v2ray.core.common.net.Network network = 1;
|
||||
TransportProtocol protocol = 1;
|
||||
|
||||
repeated NetworkSettings network_settings = 2;
|
||||
repeated TransportSettings transport_settings = 2;
|
||||
|
||||
// Type of security. Must be a message name of the settings proto.
|
||||
string security_type = 3;
|
||||
@@ -30,4 +36,4 @@ message StreamConfig {
|
||||
|
||||
message ProxyConfig {
|
||||
string tag = 1;
|
||||
}
|
||||
}
|
||||
@@ -8,10 +8,6 @@ import (
|
||||
v2net "v2ray.com/core/common/net"
|
||||
)
|
||||
|
||||
var (
|
||||
ErrUnsupportedStreamType = errors.New("Unsupported stream type.")
|
||||
)
|
||||
|
||||
type DialerOptions struct {
|
||||
Stream *StreamConfig
|
||||
Proxy *ProxyConfig
|
||||
@@ -20,16 +16,16 @@ type DialerOptions struct {
|
||||
type Dialer func(src v2net.Address, dest v2net.Destination, options DialerOptions) (Connection, error)
|
||||
|
||||
var (
|
||||
networkDialerCache = make(map[v2net.Network]Dialer)
|
||||
transportDialerCache = make(map[TransportProtocol]Dialer)
|
||||
|
||||
ProxyDialer Dialer
|
||||
)
|
||||
|
||||
func RegisterNetworkDialer(network v2net.Network, dialer Dialer) error {
|
||||
if _, found := networkDialerCache[network]; found {
|
||||
return errors.New("Internet|Dialer: ", network, " dialer already registered.")
|
||||
func RegisterTransportDialer(protocol TransportProtocol, dialer Dialer) error {
|
||||
if _, found := transportDialerCache[protocol]; found {
|
||||
return errors.New("Internet|Dialer: ", protocol, " dialer already registered.")
|
||||
}
|
||||
networkDialerCache[network] = dialer
|
||||
transportDialerCache[protocol] = dialer
|
||||
return nil
|
||||
}
|
||||
|
||||
@@ -40,14 +36,15 @@ func Dial(src v2net.Address, dest v2net.Destination, options DialerOptions) (Con
|
||||
}
|
||||
|
||||
if dest.Network == v2net.Network_TCP {
|
||||
dialer := networkDialerCache[options.Stream.Network]
|
||||
protocol := options.Stream.GetEffectiveProtocol()
|
||||
dialer := transportDialerCache[protocol]
|
||||
if dialer == nil {
|
||||
return nil, errors.New("Internet|Dialer: ", options.Stream.Network, " dialer not registered.")
|
||||
return nil, errors.New("Internet|Dialer: ", options.Stream.Protocol, " dialer not registered.")
|
||||
}
|
||||
return dialer(src, dest, options)
|
||||
}
|
||||
|
||||
udpDialer := networkDialerCache[v2net.Network_UDP]
|
||||
udpDialer := transportDialerCache[TransportProtocol_UDP]
|
||||
if udpDialer == nil {
|
||||
return nil, errors.New("Internet|Dialer: UDP dialer not registered.")
|
||||
}
|
||||
|
||||
@@ -3,7 +3,7 @@ package kcp
|
||||
import (
|
||||
"crypto/cipher"
|
||||
|
||||
v2net "v2ray.com/core/common/net"
|
||||
"v2ray.com/core/common"
|
||||
"v2ray.com/core/transport/internet"
|
||||
)
|
||||
|
||||
@@ -104,7 +104,7 @@ func (v *Config) IsConnectionReuse() bool {
|
||||
}
|
||||
|
||||
func init() {
|
||||
internet.RegisterNetworkConfigCreator(v2net.Network_KCP, func() interface{} {
|
||||
common.Must(internet.RegisterProtocolConfigCreator(internet.TransportProtocol_MKCP, func() interface{} {
|
||||
return new(Config)
|
||||
})
|
||||
}))
|
||||
}
|
||||
@@ -129,7 +129,7 @@ func DialKCP(src v2net.Address, dest v2net.Destination, options internet.DialerO
|
||||
conn = c
|
||||
}
|
||||
|
||||
networkSettings, err := options.Stream.GetEffectiveNetworkSettings()
|
||||
networkSettings, err := options.Stream.GetEffectiveTransportSettings()
|
||||
if err != nil {
|
||||
log.Error("KCP|Dialer: Failed to get KCP settings: ", err)
|
||||
return nil, err
|
||||
@@ -173,5 +173,5 @@ func DialKCP(src v2net.Address, dest v2net.Destination, options internet.DialerO
|
||||
}
|
||||
|
||||
func init() {
|
||||
common.Must(internet.RegisterNetworkDialer(v2net.Network_KCP, DialKCP))
|
||||
common.Must(internet.RegisterTransportDialer(internet.TransportProtocol_MKCP, DialKCP))
|
||||
}
|
||||
@@ -20,10 +20,10 @@ func TestDialAndListen(t *testing.T) {
|
||||
|
||||
listerner, err := NewListener(v2net.LocalHostIP, v2net.Port(0), internet.ListenOptions{
|
||||
Stream: &internet.StreamConfig{
|
||||
Network: v2net.Network_KCP,
|
||||
NetworkSettings: []*internet.NetworkSettings{
|
||||
Protocol: internet.TransportProtocol_MKCP,
|
||||
TransportSettings: []*internet.TransportSettings{
|
||||
{
|
||||
Network: v2net.Network_KCP,
|
||||
Protocol: internet.TransportProtocol_MKCP,
|
||||
Settings: serial.ToTypedMessage(&Config{}),
|
||||
},
|
||||
},
|
||||
@@ -59,10 +59,10 @@ func TestDialAndListen(t *testing.T) {
|
||||
for i := 0; i < 10; i++ {
|
||||
clientConn, err := DialKCP(v2net.LocalHostIP, v2net.UDPDestination(v2net.LocalHostIP, port), internet.DialerOptions{
|
||||
Stream: &internet.StreamConfig{
|
||||
Network: v2net.Network_KCP,
|
||||
NetworkSettings: []*internet.NetworkSettings{
|
||||
Protocol: internet.TransportProtocol_MKCP,
|
||||
TransportSettings: []*internet.TransportSettings{
|
||||
{
|
||||
Network: v2net.Network_KCP,
|
||||
Protocol: internet.TransportProtocol_MKCP,
|
||||
Settings: serial.ToTypedMessage(&Config{}),
|
||||
},
|
||||
},
|
||||
|
||||
@@ -93,7 +93,7 @@ type Listener struct {
|
||||
}
|
||||
|
||||
func NewListener(address v2net.Address, port v2net.Port, options internet.ListenOptions) (*Listener, error) {
|
||||
networkSettings, err := options.Stream.GetEffectiveNetworkSettings()
|
||||
networkSettings, err := options.Stream.GetEffectiveTransportSettings()
|
||||
if err != nil {
|
||||
log.Error("KCP|Listener: Failed to get KCP settings: ", err)
|
||||
return nil, err
|
||||
@@ -298,5 +298,5 @@ func ListenKCP(address v2net.Address, port v2net.Port, options internet.ListenOp
|
||||
}
|
||||
|
||||
func init() {
|
||||
common.Must(internet.RegisterNetworkListener(v2net.Network_KCP, ListenKCP))
|
||||
common.Must(internet.RegisterTransportListener(internet.TransportProtocol_MKCP, ListenKCP))
|
||||
}
|
||||
@@ -1,7 +1,7 @@
|
||||
package tcp
|
||||
|
||||
import (
|
||||
v2net "v2ray.com/core/common/net"
|
||||
"v2ray.com/core/common"
|
||||
"v2ray.com/core/transport/internet"
|
||||
)
|
||||
|
||||
@@ -13,7 +13,7 @@ func (v *Config) IsConnectionReuse() bool {
|
||||
}
|
||||
|
||||
func init() {
|
||||
internet.RegisterNetworkConfigCreator(v2net.Network_TCP, func() interface{} {
|
||||
common.Must(internet.RegisterProtocolConfigCreator(internet.TransportProtocol_TCP, func() interface{} {
|
||||
return new(Config)
|
||||
})
|
||||
}))
|
||||
}
|
||||
@@ -22,7 +22,7 @@ func Dial(src v2net.Address, dest v2net.Destination, options internet.DialerOpti
|
||||
if src == nil {
|
||||
src = v2net.AnyIP
|
||||
}
|
||||
networkSettings, err := options.Stream.GetEffectiveNetworkSettings()
|
||||
networkSettings, err := options.Stream.GetEffectiveTransportSettings()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
@@ -70,5 +70,5 @@ func Dial(src v2net.Address, dest v2net.Destination, options internet.DialerOpti
|
||||
}
|
||||
|
||||
func init() {
|
||||
common.Must(internet.RegisterNetworkDialer(v2net.Network_TCP, Dial))
|
||||
common.Must(internet.RegisterTransportDialer(internet.TransportProtocol_TCP, Dial))
|
||||
}
|
||||
@@ -42,7 +42,7 @@ func ListenTCP(address v2net.Address, port v2net.Port, options internet.ListenOp
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
networkSettings, err := options.Stream.GetEffectiveNetworkSettings()
|
||||
networkSettings, err := options.Stream.GetEffectiveTransportSettings()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
@@ -159,5 +159,5 @@ func (v *TCPListener) Close() error {
|
||||
}
|
||||
|
||||
func init() {
|
||||
common.Must(internet.RegisterNetworkListener(v2net.Network_TCP, ListenTCP))
|
||||
common.Must(internet.RegisterTransportListener(internet.TransportProtocol_TCP, ListenTCP))
|
||||
}
|
||||
@@ -11,16 +11,14 @@ import (
|
||||
)
|
||||
|
||||
var (
|
||||
ErrClosedConnection = errors.New("Connection already closed.")
|
||||
|
||||
networkListenerCache = make(map[v2net.Network]ListenFunc)
|
||||
transportListenerCache = make(map[TransportProtocol]ListenFunc)
|
||||
)
|
||||
|
||||
func RegisterNetworkListener(network v2net.Network, listener ListenFunc) error {
|
||||
if _, found := networkListenerCache[network]; found {
|
||||
return errors.New("Internet|TCPHub: ", network, " listener already registered.")
|
||||
func RegisterTransportListener(protocol TransportProtocol, listener ListenFunc) error {
|
||||
if _, found := transportListenerCache[protocol]; found {
|
||||
return errors.New("Internet|TCPHub: ", protocol, " listener already registered.")
|
||||
}
|
||||
networkListenerCache[network] = listener
|
||||
transportListenerCache[protocol] = listener
|
||||
return nil
|
||||
}
|
||||
|
||||
@@ -46,13 +44,14 @@ func ListenTCP(address v2net.Address, port v2net.Port, callback ConnectionHandle
|
||||
options := ListenOptions{
|
||||
Stream: settings,
|
||||
}
|
||||
listenFunc := networkListenerCache[settings.Network]
|
||||
protocol := settings.GetEffectiveProtocol()
|
||||
listenFunc := transportListenerCache[protocol]
|
||||
if listenFunc == nil {
|
||||
return nil, errors.New("Internet|TCPHub: ", settings.Network, " listener not registered.")
|
||||
return nil, errors.New("Internet|TCPHub: ", protocol, " listener not registered.")
|
||||
}
|
||||
listener, err := listenFunc(address, port, options)
|
||||
if err != nil {
|
||||
return nil, errors.Base(err).Message("Interent|TCPHub: Failed to listen: ")
|
||||
return nil, errors.Base(err).Message("Interent|TCPHub: Failed to listen on address: ", address, ":", port)
|
||||
}
|
||||
|
||||
hub := &TCPHub{
|
||||
|
||||
@@ -8,7 +8,7 @@ import (
|
||||
)
|
||||
|
||||
func init() {
|
||||
common.Must(internet.RegisterNetworkDialer(v2net.Network_UDP,
|
||||
common.Must(internet.RegisterTransportDialer(internet.TransportProtocol_UDP,
|
||||
func(src v2net.Address, dest v2net.Destination, options internet.DialerOptions) (internet.Connection, error) {
|
||||
conn, err := internet.DialSystem(src, dest)
|
||||
if err != nil {
|
||||
|
||||
@@ -1,7 +1,7 @@
|
||||
package websocket
|
||||
|
||||
import (
|
||||
v2net "v2ray.com/core/common/net"
|
||||
"v2ray.com/core/common"
|
||||
"v2ray.com/core/transport/internet"
|
||||
)
|
||||
|
||||
@@ -13,7 +13,7 @@ func (c *Config) IsConnectionReuse() bool {
|
||||
}
|
||||
|
||||
func init() {
|
||||
internet.RegisterNetworkConfigCreator(v2net.Network_WebSocket, func() interface{} {
|
||||
common.Must(internet.RegisterProtocolConfigCreator(internet.TransportProtocol_WebSocket, func() interface{} {
|
||||
return new(Config)
|
||||
})
|
||||
}))
|
||||
}
|
||||
@@ -22,7 +22,7 @@ func Dial(src v2net.Address, dest v2net.Destination, options internet.DialerOpti
|
||||
if src == nil {
|
||||
src = v2net.AnyIP
|
||||
}
|
||||
networkSettings, err := options.Stream.GetEffectiveNetworkSettings()
|
||||
networkSettings, err := options.Stream.GetEffectiveTransportSettings()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
@@ -48,11 +48,11 @@ func Dial(src v2net.Address, dest v2net.Destination, options internet.DialerOpti
|
||||
}
|
||||
|
||||
func init() {
|
||||
common.Must(internet.RegisterNetworkDialer(v2net.Network_WebSocket, Dial))
|
||||
common.Must(internet.RegisterTransportDialer(internet.TransportProtocol_WebSocket, Dial))
|
||||
}
|
||||
|
||||
func wsDial(src v2net.Address, dest v2net.Destination, options internet.DialerOptions) (*wsconn, error) {
|
||||
networkSettings, err := options.Stream.GetEffectiveNetworkSettings()
|
||||
networkSettings, err := options.Stream.GetEffectiveTransportSettings()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
@@ -38,7 +38,7 @@ type WSListener struct {
|
||||
}
|
||||
|
||||
func ListenWS(address v2net.Address, port v2net.Port, options internet.ListenOptions) (internet.Listener, error) {
|
||||
networkSettings, err := options.Stream.GetEffectiveNetworkSettings()
|
||||
networkSettings, err := options.Stream.GetEffectiveTransportSettings()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
@@ -199,5 +199,5 @@ func (v *WSListener) Close() error {
|
||||
}
|
||||
|
||||
func init() {
|
||||
common.Must(internet.RegisterNetworkListener(v2net.Network_WebSocket, ListenWS))
|
||||
common.Must(internet.RegisterTransportListener(internet.TransportProtocol_WebSocket, ListenWS))
|
||||
}
|
||||
@@ -21,10 +21,10 @@ func Test_listenWSAndDial(t *testing.T) {
|
||||
assert := assert.On(t)
|
||||
listen, err := ListenWS(v2net.DomainAddress("localhost"), 13146, internet.ListenOptions{
|
||||
Stream: &internet.StreamConfig{
|
||||
Network: v2net.Network_WebSocket,
|
||||
NetworkSettings: []*internet.NetworkSettings{
|
||||
Protocol: internet.TransportProtocol_WebSocket,
|
||||
TransportSettings: []*internet.TransportSettings{
|
||||
{
|
||||
Network: v2net.Network_WebSocket,
|
||||
Protocol: internet.TransportProtocol_WebSocket,
|
||||
Settings: serial.ToTypedMessage(&Config{
|
||||
Path: "ws",
|
||||
}),
|
||||
@@ -58,10 +58,10 @@ func Test_listenWSAndDial(t *testing.T) {
|
||||
}()
|
||||
conn, err := Dial(v2net.AnyIP, v2net.TCPDestination(v2net.DomainAddress("localhost"), 13146), internet.DialerOptions{
|
||||
Stream: &internet.StreamConfig{
|
||||
Network: v2net.Network_WebSocket,
|
||||
NetworkSettings: []*internet.NetworkSettings{
|
||||
Protocol: internet.TransportProtocol_WebSocket,
|
||||
TransportSettings: []*internet.TransportSettings{
|
||||
{
|
||||
Network: v2net.Network_WebSocket,
|
||||
Protocol: internet.TransportProtocol_WebSocket,
|
||||
Settings: serial.ToTypedMessage(&Config{
|
||||
Path: "ws",
|
||||
}),
|
||||
@@ -82,10 +82,10 @@ func Test_listenWSAndDial(t *testing.T) {
|
||||
<-time.After(time.Second * 5)
|
||||
conn, err = Dial(v2net.AnyIP, v2net.TCPDestination(v2net.DomainAddress("localhost"), 13146), internet.DialerOptions{
|
||||
Stream: &internet.StreamConfig{
|
||||
Network: v2net.Network_WebSocket,
|
||||
NetworkSettings: []*internet.NetworkSettings{
|
||||
Protocol: internet.TransportProtocol_WebSocket,
|
||||
TransportSettings: []*internet.TransportSettings{
|
||||
{
|
||||
Network: v2net.Network_WebSocket,
|
||||
Protocol: internet.TransportProtocol_WebSocket,
|
||||
Settings: serial.ToTypedMessage(&Config{
|
||||
Path: "ws",
|
||||
}),
|
||||
@@ -103,10 +103,10 @@ func Test_listenWSAndDial(t *testing.T) {
|
||||
<-time.After(time.Second * 15)
|
||||
conn, err = Dial(v2net.AnyIP, v2net.TCPDestination(v2net.DomainAddress("localhost"), 13146), internet.DialerOptions{
|
||||
Stream: &internet.StreamConfig{
|
||||
Network: v2net.Network_WebSocket,
|
||||
NetworkSettings: []*internet.NetworkSettings{
|
||||
Protocol: internet.TransportProtocol_WebSocket,
|
||||
TransportSettings: []*internet.TransportSettings{
|
||||
{
|
||||
Network: v2net.Network_WebSocket,
|
||||
Protocol: internet.TransportProtocol_WebSocket,
|
||||
Settings: serial.ToTypedMessage(&Config{
|
||||
Path: "ws",
|
||||
}),
|
||||
@@ -145,10 +145,10 @@ func Test_listenWSAndDial_TLS(t *testing.T) {
|
||||
Stream: &internet.StreamConfig{
|
||||
SecurityType: serial.GetMessageType(new(v2tls.Config)),
|
||||
SecuritySettings: []*serial.TypedMessage{serial.ToTypedMessage(tlsSettings)},
|
||||
Network: v2net.Network_WebSocket,
|
||||
NetworkSettings: []*internet.NetworkSettings{
|
||||
Protocol: internet.TransportProtocol_WebSocket,
|
||||
TransportSettings: []*internet.TransportSettings{
|
||||
{
|
||||
Network: v2net.Network_WebSocket,
|
||||
Protocol: internet.TransportProtocol_WebSocket,
|
||||
Settings: serial.ToTypedMessage(&Config{
|
||||
Path: "wss",
|
||||
ConnectionReuse: &ConnectionReuse{
|
||||
@@ -170,10 +170,10 @@ func Test_listenWSAndDial_TLS(t *testing.T) {
|
||||
Stream: &internet.StreamConfig{
|
||||
SecurityType: serial.GetMessageType(new(v2tls.Config)),
|
||||
SecuritySettings: []*serial.TypedMessage{serial.ToTypedMessage(tlsSettings)},
|
||||
Network: v2net.Network_WebSocket,
|
||||
NetworkSettings: []*internet.NetworkSettings{
|
||||
Protocol: internet.TransportProtocol_WebSocket,
|
||||
TransportSettings: []*internet.TransportSettings{
|
||||
{
|
||||
Network: v2net.Network_WebSocket,
|
||||
Protocol: internet.TransportProtocol_WebSocket,
|
||||
Settings: serial.ToTypedMessage(&Config{
|
||||
Path: "wss",
|
||||
ConnectionReuse: &ConnectionReuse{
|
||||
|
||||
Reference in new issue
Block a user