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

Fix h2mux request context

Browse source
This commit is contained in:
世界 2023-12-31 15:51:48 +08:00
parent f99a0dfe65
commit 6be79e969e
No known key found for this signature in database
GPG key ID: CD109927C34A63C4
2 changed files with 20 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

15
h2mux.go
View file

@ -204,10 +204,21 @@ func (s *h2MuxClientSession) OpenContext(ctx context.Context) (net.Conn, error)
Body: pipeInReader,
URL: &url.URL{Scheme: "https", Host: "localhost"},
}
request = request.WithContext(ctx)
conn := newLateHTTPConn(pipeInWriter)
connCtx, cancel := context.WithCancel(context.Background())
request = request.WithContext(connCtx)
conn := newLateHTTPConn(pipeInWriter, cancel)
requestDone := make(chan struct{})
go func() {
select {
case <-requestDone:
return
case <-ctx.Done():
cancel()
}
}()
go func() {
response, err := s.transport.RoundTrip(request)
close(requestDone)
if err != nil {
conn.setup(nil, err)
} else if response.StatusCode != 200 {