initial commit

dialer/dialer.go

@@ -0,0 +1,22 @@
+package dialer
+
+import (
+	"context"
+	"net"
+
+	"github.com/go-gost/core/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, opts ...HandshakeOption) (net.Conn, error)
+}
+
+type Multiplexer interface {
+	Multiplex() bool
+}

dialer/option.go

@@ -0,0 +1,66 @@
+package dialer
+
+import (
+	"crypto/tls"
+	"net/url"
+
+	"github.com/go-gost/core/common/net/dialer"
+	"github.com/go-gost/core/logger"
+)
+
+type Options struct {
+	Auth      *url.Userinfo
+	TLSConfig *tls.Config
+	Logger    logger.Logger
+}
+
+type Option func(opts *Options)
+
+func AuthOption(auth *url.Userinfo) Option {
+	return func(opts *Options) {
+		opts.Auth = auth
+	}
+}
+
+func TLSConfigOption(tlsConfig *tls.Config) Option {
+	return func(opts *Options) {
+		opts.TLSConfig = tlsConfig
+	}
+}
+
+func LoggerOption(logger logger.Logger) Option {
+	return func(opts *Options) {
+		opts.Logger = logger
+	}
+}
+
+type DialOptions struct {
+	Host      string
+	NetDialer *dialer.NetDialer
+}
+
+type DialOption func(opts *DialOptions)
+
+func HostDialOption(host string) DialOption {
+	return func(opts *DialOptions) {
+		opts.Host = host
+	}
+}
+
+func NetDialerDialOption(netd *dialer.NetDialer) DialOption {
+	return func(opts *DialOptions) {
+		opts.NetDialer = netd
+	}
+}
+
+type HandshakeOptions struct {
+	Addr string
+}
+
+type HandshakeOption func(opts *HandshakeOptions)
+
+func AddrHandshakeOption(addr string) HandshakeOption {
+	return func(opts *HandshakeOptions) {
+		opts.Addr = addr
+	}
+}

dialer/tcp/dialer.go

@@ -0,0 +1,48 @@
+package tcp
+
+import (
+	"context"
+	"net"
+
+	"github.com/go-gost/core/dialer"
+	"github.com/go-gost/core/logger"
+	md "github.com/go-gost/core/metadata"
+	"github.com/go-gost/core/registry"
+)
+
+func init() {
+	registry.DialerRegistry().Register("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)
+	}
+
+	conn, err := options.NetDialer.Dial(ctx, "tcp", addr)
+	if err != nil {
+		d.logger.Error(err)
+	}
+	return conn, err
+}

dialer/tcp/metadata.go

@@ -0,0 +1,23 @@
+package tcp
+
+import (
+	"time"
+
+	md "github.com/go-gost/core/metadata"
+)
+
+const (
+	dialTimeout = "dialTimeout"
+)
+
+const (
+	defaultDialTimeout = 5 * time.Second
+)
+
+type metadata struct {
+	dialTimeout time.Duration
+}
+
+func (d *tcpDialer) parseMetadata(md md.Metadata) (err error) {
+	return
+}

dialer/tls/dialer.go

@@ -0,0 +1,68 @@
+package tls
+
+import (
+	"context"
+	"crypto/tls"
+	"net"
+	"time"
+
+	"github.com/go-gost/core/dialer"
+	"github.com/go-gost/core/logger"
+	md "github.com/go-gost/core/metadata"
+	"github.com/go-gost/core/registry"
+)
+
+func init() {
+	registry.DialerRegistry().Register("tls", NewDialer)
+}
+
+type tlsDialer struct {
+	md      metadata
+	logger  logger.Logger
+	options dialer.Options
+}
+
+func NewDialer(opts ...dialer.Option) dialer.Dialer {
+	options := dialer.Options{}
+	for _, opt := range opts {
+		opt(&options)
+	}
+
+	return &tlsDialer{
+		logger:  options.Logger,
+		options: options,
+	}
+}
+
+func (d *tlsDialer) Init(md md.Metadata) (err error) {
+	return d.parseMetadata(md)
+}
+
+func (d *tlsDialer) Dial(ctx context.Context, addr string, opts ...dialer.DialOption) (net.Conn, error) {
+	var options dialer.DialOptions
+	for _, opt := range opts {
+		opt(&options)
+	}
+
+	conn, err := options.NetDialer.Dial(ctx, "tcp", addr)
+	if err != nil {
+		d.logger.Error(err)
+	}
+	return conn, err
+}
+
+// Handshake implements dialer.Handshaker
+func (d *tlsDialer) Handshake(ctx context.Context, conn net.Conn, options ...dialer.HandshakeOption) (net.Conn, error) {
+	if d.md.handshakeTimeout > 0 {
+		conn.SetDeadline(time.Now().Add(d.md.handshakeTimeout))
+		defer conn.SetDeadline(time.Time{})
+	}
+
+	tlsConn := tls.Client(conn, d.options.TLSConfig)
+	if err := tlsConn.HandshakeContext(ctx); err != nil {
+		conn.Close()
+		return nil, err
+	}
+
+	return tlsConn, nil
+}

dialer/tls/metadata.go

@@ -0,0 +1,21 @@
+package tls
+
+import (
+	"time"
+
+	mdata "github.com/go-gost/core/metadata"
+)
+
+type metadata struct {
+	handshakeTimeout time.Duration
+}
+
+func (d *tlsDialer) parseMetadata(md mdata.Metadata) (err error) {
+	const (
+		handshakeTimeout = "handshakeTimeout"
+	)
+
+	d.md.handshakeTimeout = mdata.GetDuration(md, handshakeTimeout)
+
+	return
+}

dialer/udp/conn.go

@@ -0,0 +1,17 @@
+package udp
+
+import "net"
+
+type conn struct {
+	*net.UDPConn
+}
+
+func (c *conn) WriteTo(b []byte, addr net.Addr) (int, error) {
+	return c.UDPConn.Write(b)
+}
+
+func (c *conn) ReadFrom(b []byte) (n int, addr net.Addr, err error) {
+	n, err = c.UDPConn.Read(b)
+	addr = c.RemoteAddr()
+	return
+}

dialer/udp/dialer.go

@@ -0,0 +1,50 @@
+package udp
+
+import (
+	"context"
+	"net"
+
+	"github.com/go-gost/core/dialer"
+	"github.com/go-gost/core/logger"
+	md "github.com/go-gost/core/metadata"
+	"github.com/go-gost/core/registry"
+)
+
+func init() {
+	registry.DialerRegistry().Register("udp", NewDialer)
+}
+
+type udpDialer struct {
+	md     metadata
+	logger logger.Logger
+}
+
+func NewDialer(opts ...dialer.Option) dialer.Dialer {
+	options := &dialer.Options{}
+	for _, opt := range opts {
+		opt(options)
+	}
+
+	return &udpDialer{
+		logger: options.Logger,
+	}
+}
+
+func (d *udpDialer) Init(md md.Metadata) (err error) {
+	return d.parseMetadata(md)
+}
+
+func (d *udpDialer) Dial(ctx context.Context, addr string, opts ...dialer.DialOption) (net.Conn, error) {
+	var options dialer.DialOptions
+	for _, opt := range opts {
+		opt(&options)
+	}
+
+	c, err := options.NetDialer.Dial(ctx, "udp", addr)
+	if err != nil {
+		return nil, err
+	}
+	return &conn{
+		UDPConn: c.(*net.UDPConn),
+	}, nil
+}

dialer/udp/metadata.go

@@ -0,0 +1,23 @@
+package udp
+
+import (
+	"time"
+
+	md "github.com/go-gost/core/metadata"
+)
+
+const (
+	dialTimeout = "dialTimeout"
+)
+
+const (
+	defaultDialTimeout = 5 * time.Second
+)
+
+type metadata struct {
+	dialTimeout time.Duration
+}
+
+func (d *udpDialer) parseMetadata(md md.Metadata) (err error) {
+	return
+}