improve ss

pkg/dialer/dialer.go

@@ -0,0 +1,22 @@
+package dialer
+
+import (
+	"context"
+	"net"
+
+	"github.com/go-gost/gost/pkg/metadata"
+)
+
+// Transporter is responsible for dialing to the proxy server.
+type Dialer interface {
+	Init(metadata.Metadata) error
+	Dial(ctx context.Context, addr string, opts ...DialOption) (net.Conn, error)
+}
+
+type Handshaker interface {
+	Handshake(ctx context.Context, conn net.Conn) (net.Conn, error)
+}
+
+type Multiplexer interface {
+	IsMultiplex() bool
+}

pkg/dialer/option.go

@@ -0,0 +1,32 @@
+package dialer
+
+import (
+	"context"
+	"net"
+
+	"github.com/go-gost/gost/pkg/logger"
+)
+
+type Options struct {
+	Logger logger.Logger
+}
+
+type Option func(opts *Options)
+
+func LoggerOption(logger logger.Logger) Option {
+	return func(opts *Options) {
+		opts.Logger = logger
+	}
+}
+
+type DialOptions struct {
+	DialFunc func(ctx context.Context, addr string) (net.Conn, error)
+}
+
+type DialOption func(opts *DialOptions)
+
+func DialFuncDialOption(dialf func(ctx context.Context, addr string) (net.Conn, error)) DialOption {
+	return func(opts *DialOptions) {
+		opts.DialFunc = dialf
+	}
+}

pkg/dialer/tcp/dialer.go

@@ -0,0 +1,76 @@
+package tcp
+
+import (
+	"context"
+	"net"
+
+	"github.com/go-gost/gost/pkg/dialer"
+	"github.com/go-gost/gost/pkg/logger"
+	md "github.com/go-gost/gost/pkg/metadata"
+	"github.com/go-gost/gost/pkg/registry"
+)
+
+func init() {
+	registry.RegisterDialer("tcp", NewDialer)
+}
+
+type tcpDialer struct {
+	md     metadata
+	logger logger.Logger
+}
+
+func NewDialer(opts ...dialer.Option) dialer.Dialer {
+	options := &dialer.Options{}
+	for _, opt := range opts {
+		opt(options)
+	}
+
+	return &tcpDialer{
+		logger: options.Logger,
+	}
+}
+
+func (d *tcpDialer) Init(md md.Metadata) (err error) {
+	return d.parseMetadata(md)
+}
+
+func (d *tcpDialer) Dial(ctx context.Context, addr string, opts ...dialer.DialOption) (net.Conn, error) {
+	var options dialer.DialOptions
+	for _, opt := range opts {
+		opt(&options)
+	}
+
+	dial := options.DialFunc
+	if dial != nil {
+		conn, err := dial(ctx, addr)
+		if err != nil {
+			d.logger.Error(err)
+		} else {
+			if d.logger.IsLevelEnabled(logger.DebugLevel) {
+				d.logger.WithFields(map[string]interface{}{
+					"src": conn.LocalAddr().String(),
+					"dst": addr,
+				}).Debug("dial with dial func")
+			}
+		}
+		return conn, err
+	}
+
+	var netd net.Dialer
+	conn, err := netd.DialContext(ctx, "tcp", addr)
+	if err != nil {
+		d.logger.Error(err)
+	} else {
+		if d.logger.IsLevelEnabled(logger.DebugLevel) {
+			d.logger.WithFields(map[string]interface{}{
+				"src": conn.LocalAddr().String(),
+				"dst": addr,
+			}).Debug("dial direct")
+		}
+	}
+	return conn, err
+}
+
+func (d *tcpDialer) parseMetadata(md md.Metadata) (err error) {
+	return
+}

pkg/dialer/tcp/metadata.go

@@ -0,0 +1,15 @@
+package tcp
+
+import "time"
+
+const (
+	dialTimeout = "dialTimeout"
+)
+
+const (
+	defaultDialTimeout = 5 * time.Second
+)
+
+type metadata struct {
+	dialTimeout time.Duration
+}