From 78e50cd7b58775e9b4280424e5cf26707683a53f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E4=B8=96=E7=95=8C?= <i@sekai.icu>
Date: Tue, 21 Feb 2023 17:33:48 +0800
Subject: [PATCH] Make tls handshake func optional

---
 client.go | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/client.go b/client.go
index fe44c8a..9825f03 100644
--- a/client.go
+++ b/client.go
@@ -42,7 +42,7 @@ func NewClient(config ClientConfig) (*Client, error) {
 		tlsHandshake: config.TLSHandshake,
 		logger:       config.Logger,
 	}
-	if !client.server.IsValid() || client.dialer == nil || client.tlsHandshake == nil {
+	if !client.server.IsValid() || client.dialer == nil {
 		return nil, os.ErrInvalid
 	}
 	switch client.version {
@@ -56,11 +56,18 @@ func NewClient(config ClientConfig) (*Client, error) {
 	return client, nil
 }
 
+func (c *Client) SetHandshakeFunc(handshakeFunc TLSHandshakeFunc) {
+	c.tlsHandshake = handshakeFunc
+}
+
 func (c *Client) DialContext(ctx context.Context) (net.Conn, error) {
 	conn, err := c.dialer.DialContext(ctx, N.NetworkTCP, c.server)
 	if err != nil {
 		return nil, err
 	}
+	if c.tlsHandshake == nil {
+		return nil, os.ErrInvalid
+	}
 	switch c.version {
 	default:
 		fallthrough