SagerNet/sing-shadowtls
1
0
Fork 0
mirror of https://github.com/SagerNet/sing-shadowtls.git synced 2025-04-03 04:07:35 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

Make tls handshake func optional

Browse source
This commit is contained in:
世界 2023-02-21 17:33:48 +08:00
parent 574313aaae
commit 78e50cd7b5
No known key found for this signature in database
GPG key ID: CD109927C34A63C4
1 changed files with 8 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

9
client.go
View file

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