sing-box/transport/v2ray/transport.go

64 lines
2.5 KiB
Go
Raw Normal View History

2022-08-22 10:53:47 +00:00
package v2ray
import (
"context"
"github.com/sagernet/sing-box/adapter"
2022-09-09 10:45:10 +00:00
"github.com/sagernet/sing-box/common/tls"
2022-08-22 10:53:47 +00:00
C "github.com/sagernet/sing-box/constant"
"github.com/sagernet/sing-box/option"
2022-08-22 14:19:25 +00:00
"github.com/sagernet/sing-box/transport/v2rayhttp"
2022-08-22 12:20:56 +00:00
"github.com/sagernet/sing-box/transport/v2raywebsocket"
2022-08-22 10:53:47 +00:00
E "github.com/sagernet/sing/common/exceptions"
M "github.com/sagernet/sing/common/metadata"
N "github.com/sagernet/sing/common/network"
)
2022-09-26 11:37:06 +00:00
type (
ServerConstructor[O any] func(ctx context.Context, options O, tlsConfig tls.ServerConfig, handler adapter.V2RayServerTransportHandler) (adapter.V2RayServerTransport, error)
2022-09-26 11:37:06 +00:00
ClientConstructor[O any] func(ctx context.Context, dialer N.Dialer, serverAddr M.Socksaddr, options O, tlsConfig tls.Config) (adapter.V2RayClientTransport, error)
)
func NewServerTransport(ctx context.Context, options option.V2RayTransportOptions, tlsConfig tls.ServerConfig, handler adapter.V2RayServerTransportHandler) (adapter.V2RayServerTransport, error) {
2022-08-22 10:53:47 +00:00
if options.Type == "" {
return nil, nil
}
switch options.Type {
2022-08-22 14:19:25 +00:00
case C.V2RayTransportTypeHTTP:
return v2rayhttp.NewServer(ctx, options.HTTPOptions, tlsConfig, handler)
2022-08-22 12:20:56 +00:00
case C.V2RayTransportTypeWebsocket:
return v2raywebsocket.NewServer(ctx, options.WebsocketOptions, tlsConfig, handler)
2022-08-22 13:20:05 +00:00
case C.V2RayTransportTypeQUIC:
2022-08-22 14:19:25 +00:00
if tlsConfig == nil {
return nil, C.ErrTLSRequired
}
return NewQUICServer(ctx, options.QUICOptions, tlsConfig, handler)
2022-08-22 14:19:25 +00:00
case C.V2RayTransportTypeGRPC:
return NewGRPCServer(ctx, options.GRPCOptions, tlsConfig, handler)
2022-08-22 10:53:47 +00:00
default:
return nil, E.New("unknown transport type: " + options.Type)
}
}
2022-09-09 10:45:10 +00:00
func NewClientTransport(ctx context.Context, dialer N.Dialer, serverAddr M.Socksaddr, options option.V2RayTransportOptions, tlsConfig tls.Config) (adapter.V2RayClientTransport, error) {
2022-08-22 10:53:47 +00:00
if options.Type == "" {
return nil, nil
}
switch options.Type {
2022-08-22 14:19:25 +00:00
case C.V2RayTransportTypeHTTP:
2023-04-12 07:14:55 +00:00
return v2rayhttp.NewClient(ctx, dialer, serverAddr, options.HTTPOptions, tlsConfig)
2022-08-22 10:53:47 +00:00
case C.V2RayTransportTypeGRPC:
2022-08-22 12:20:56 +00:00
return NewGRPCClient(ctx, dialer, serverAddr, options.GRPCOptions, tlsConfig)
case C.V2RayTransportTypeWebsocket:
return v2raywebsocket.NewClient(ctx, dialer, serverAddr, options.WebsocketOptions, tlsConfig), nil
2022-08-22 13:20:05 +00:00
case C.V2RayTransportTypeQUIC:
2022-08-22 14:19:25 +00:00
if tlsConfig == nil {
return nil, C.ErrTLSRequired
}
2022-08-22 13:20:05 +00:00
return NewQUICClient(ctx, dialer, serverAddr, options.QUICOptions, tlsConfig)
2022-08-22 14:19:25 +00:00
2022-08-22 10:53:47 +00:00
default:
return nil, E.New("unknown transport type: " + options.Type)
}
}