init

go-gost/x/dialer/dtls/dialer.go

@@ -0,0 +1,74 @@
+package dtls
+
+import (
+	"context"
+	"crypto/tls"
+	"net"
+
+	"github.com/go-gost/core/dialer"
+	"github.com/go-gost/core/logger"
+	md "github.com/go-gost/core/metadata"
+	xdtls "github.com/go-gost/x/internal/util/dtls"
+	"github.com/go-gost/x/registry"
+	"github.com/pion/dtls/v2"
+)
+
+func init() {
+	registry.DialerRegistry().Register("dtls", NewDialer)
+}
+
+type dtlsDialer 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 &dtlsDialer{
+		logger:  options.Logger,
+		options: options,
+	}
+}
+
+func (d *dtlsDialer) Init(md md.Metadata) (err error) {
+	return d.parseMetadata(md)
+}
+
+func (d *dtlsDialer) 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.Dialer.Dial(ctx, "udp", addr)
+	if err != nil {
+		return nil, err
+	}
+
+	tlsCfg := d.options.TLSConfig
+	if tlsCfg == nil {
+		tlsCfg = &tls.Config{
+			InsecureSkipVerify: true,
+		}
+	}
+	config := dtls.Config{
+		Certificates:         tlsCfg.Certificates,
+		InsecureSkipVerify:   tlsCfg.InsecureSkipVerify,
+		ExtendedMasterSecret: dtls.RequireExtendedMasterSecret,
+		ServerName:           tlsCfg.ServerName,
+		RootCAs:              tlsCfg.RootCAs,
+		FlightInterval:       d.md.flightInterval,
+		MTU:                  d.md.mtu,
+	}
+
+	c, err := dtls.ClientWithContext(ctx, conn, &config)
+	if err != nil {
+		return nil, err
+	}
+	return xdtls.Conn(c, d.md.bufferSize), nil
+}

go-gost/x/dialer/dtls/metadata.go

@@ -0,0 +1,28 @@
+package dtls
+
+import (
+	"time"
+
+	mdata "github.com/go-gost/core/metadata"
+	mdutil "github.com/go-gost/x/metadata/util"
+)
+
+const (
+	defaultBufferSize = 1200
+)
+
+type metadata struct {
+	mtu            int
+	bufferSize     int
+	flightInterval time.Duration
+}
+
+func (d *dtlsDialer) parseMetadata(md mdata.Metadata) (err error) {
+	d.md.mtu = mdutil.GetInt(md, "dtls.mtu", "mtu")
+	d.md.bufferSize = mdutil.GetInt(md, "dtls.bufferSize", "bufferSize")
+	if d.md.bufferSize <= 0 {
+		d.md.bufferSize = defaultBufferSize
+	}
+	d.md.flightInterval = mdutil.GetDuration(md, "dtls.flightInterval", "flightInterval")
+	return
+}