mirror
/
websocket
mirror of https://github.com/gorilla/websocket.git
1
0
Fork 1
Code Issues Packages Projects Releases Wiki Activity

Merge 873ca6009f into ce903f6d1d

This commit is contained in:
好为 2024-06-14 19:14:13 -07:00 committed by GitHub
parent ce903f6d1d 873ca6009f
commit 71e79a20b9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
client.go
View File

@ -186,6 +186,10 @@ func (d *Dialer) DialContext(ctx context.Context, urlStr string, requestHeader h
return nil, nil, errMalformedURL
}
host := requestHeader.Get("Host")
if host == "" {
host = u.Host
}
req := &http.Request{
Method: http.MethodGet,
URL: u,
@ -193,7 +197,7 @@ func (d *Dialer) DialContext(ctx context.Context, urlStr string, requestHeader h
ProtoMajor: 1,
ProtoMinor: 1,
Header: make(http.Header),
Host: u.Host,
Host: host,
}
req = req.WithContext(ctx)
@ -339,7 +343,11 @@ func (d *Dialer) DialContext(ctx context.Context, urlStr string, requestHeader h
cfg := cloneTLSConfig(d.TLSClientConfig)
if cfg.ServerName == "" {
cfg.ServerName = hostNoPort
if host != "" {
cfg.ServerName = host
} else {
cfg.ServerName = hostNoPort
}
}
tlsConn := tls.Client(netConn, cfg)
netConn = tlsConn