|
|
@@ -0,0 +1,132 @@
|
|
|
+package outbound
|
|
|
+
|
|
|
+import (
|
|
|
+ "context"
|
|
|
+ "io"
|
|
|
+
|
|
|
+ "v2ray.com/core"
|
|
|
+ "v2ray.com/core/app/proxyman"
|
|
|
+ "v2ray.com/core/app/proxyman/mux"
|
|
|
+ "v2ray.com/core/common"
|
|
|
+ "v2ray.com/core/common/errors"
|
|
|
+ "v2ray.com/core/common/net"
|
|
|
+ "v2ray.com/core/proxy"
|
|
|
+ "v2ray.com/core/transport/internet"
|
|
|
+ "v2ray.com/core/transport/internet/domainsocket"
|
|
|
+ "v2ray.com/core/transport/ray"
|
|
|
+)
|
|
|
+
|
|
|
+type UnixHandler struct {
|
|
|
+ config *core.OutboundHandlerConfig
|
|
|
+ senderSettings *proxyman.UnixSenderConfig
|
|
|
+ proxy proxy.Outbound
|
|
|
+ outboundManager core.OutboundHandlerManager
|
|
|
+ mux *mux.ClientManager
|
|
|
+}
|
|
|
+
|
|
|
+func NewUnixHandler(ctx context.Context, config *core.OutboundHandlerConfig) (core.OutboundHandler, error) {
|
|
|
+ v := core.FromContext(ctx)
|
|
|
+ if v == nil {
|
|
|
+ return nil, newError("V is not in context")
|
|
|
+ }
|
|
|
+ h := &UnixHandler{
|
|
|
+ config: config,
|
|
|
+ outboundManager: v.OutboundHandlerManager(),
|
|
|
+ }
|
|
|
+
|
|
|
+ if config.SenderSettings != nil {
|
|
|
+ senderSettings, err := config.SenderSettings.GetInstance()
|
|
|
+ if err != nil {
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+ switch s := senderSettings.(type) {
|
|
|
+ case *proxyman.UnixSenderConfig:
|
|
|
+ h.senderSettings = s
|
|
|
+ default:
|
|
|
+ return nil, newError("settings is not SenderConfig")
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ proxyConfig, err := config.ProxySettings.GetInstance()
|
|
|
+ if err != nil {
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+
|
|
|
+ rawProxyHandler, err := common.CreateObject(ctx, proxyConfig)
|
|
|
+ if err != nil {
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+
|
|
|
+ proxyHandler, ok := rawProxyHandler.(proxy.Outbound)
|
|
|
+ if !ok {
|
|
|
+ return nil, newError("not an outbound handler")
|
|
|
+ }
|
|
|
+
|
|
|
+ if h.senderSettings != nil && h.senderSettings.MultiplexSettings != nil && h.senderSettings.MultiplexSettings.Enabled {
|
|
|
+ config := h.senderSettings.MultiplexSettings
|
|
|
+ if config.Concurrency < 1 || config.Concurrency > 1024 {
|
|
|
+ return nil, newError("invalid mux concurrency: ", config.Concurrency).AtWarning()
|
|
|
+ }
|
|
|
+ h.mux = mux.NewClientManager(proxyHandler, h, config)
|
|
|
+ }
|
|
|
+
|
|
|
+ h.proxy = proxyHandler
|
|
|
+ return h, nil
|
|
|
+}
|
|
|
+
|
|
|
+// Tag implements core.OutboundHandler.
|
|
|
+func (h *UnixHandler) Tag() string {
|
|
|
+ return h.config.Tag
|
|
|
+}
|
|
|
+
|
|
|
+// Dispatch implements proxy.Outbound.Dispatch.
|
|
|
+func (h *UnixHandler) Dispatch(ctx context.Context, outboundRay ray.OutboundRay) {
|
|
|
+ if h.mux != nil {
|
|
|
+ err := h.mux.Dispatch(ctx, outboundRay)
|
|
|
+ if err != nil {
|
|
|
+ newError("failed to process outbound traffic").Base(err).WriteToLog()
|
|
|
+ outboundRay.OutboundOutput().CloseError()
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+ err := h.proxy.Process(ctx, outboundRay, h)
|
|
|
+ // Ensure outbound ray is properly closed.
|
|
|
+ if err != nil && errors.Cause(err) != io.EOF {
|
|
|
+ newError("failed to process outbound traffic").Base(err).WriteToLog()
|
|
|
+ outboundRay.OutboundOutput().CloseError()
|
|
|
+ } else {
|
|
|
+ outboundRay.OutboundOutput().Close()
|
|
|
+ }
|
|
|
+ outboundRay.OutboundInput().CloseError()
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+// Dial implements proxy.Dialer.Dial().
|
|
|
+func (h *UnixHandler) Dial(ctx context.Context, dest net.Destination) (internet.Connection, error) {
|
|
|
+ if h.senderSettings != nil {
|
|
|
+ if h.senderSettings.ProxySettings.HasTag() {
|
|
|
+ newError("Unix domain socket does not support redirect").AtWarning().WriteToLog()
|
|
|
+ }
|
|
|
+
|
|
|
+ if h.senderSettings.StreamSettings != nil {
|
|
|
+ newError("Unix domain socket does not support stream setting").AtWarning().WriteToLog()
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ return domainsocket.DialDS(ctx, h.senderSettings.GetVia().GetPath())
|
|
|
+}
|
|
|
+
|
|
|
+// GetOutbound implements proxy.GetOutbound.
|
|
|
+func (h *UnixHandler) GetOutbound() proxy.Outbound {
|
|
|
+ return h.proxy
|
|
|
+}
|
|
|
+
|
|
|
+// Start implements common.Runnable.
|
|
|
+func (h *UnixHandler) Start() error {
|
|
|
+ return nil
|
|
|
+}
|
|
|
+
|
|
|
+// Close implements common.Runnable.
|
|
|
+func (h *UnixHandler) Close() error {
|
|
|
+ common.Close(h.mux)
|
|
|
+ return nil
|
|
|
+}
|