package tls import ( "context" "crypto/tls" "net" "time" "github.com/go-gost/gost/v3/pkg/dialer" "github.com/go-gost/gost/v3/pkg/logger" md "github.com/go-gost/gost/v3/pkg/metadata" "github.com/go-gost/gost/v3/pkg/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 }