xmengnet/mihomo
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: don't close tuic when read timeout

Browse source
This commit is contained in:
wwqgtxx 2022-11-25 11:04:28 +08:00
parent 30ca59dab7
commit 142d17ebad
1 changed files with 19 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
transport/tuic/client.go
View file

@ -214,22 +214,26 @@ func (t *Client) DialContext(ctx context.Context, metadata *C.Metadata, dialFn f
if err != nil {
return nil, err
}
defer func() {
t.deferQuicConn(quicConn, err)
stream, err := func() (quic.Stream, error) {
defer func() {
t.deferQuicConn(quicConn, err)
}()
buf := &bytes.Buffer{}
err = NewConnect(NewAddress(metadata)).WriteTo(buf)
if err != nil {
return nil, err
}
stream, err := quicConn.OpenStream()
if err != nil {
return nil, err
}
_, err = buf.WriteTo(stream)
if err != nil {
return nil, err
}
return stream, err
}()
buf := &bytes.Buffer{}
err = NewConnect(NewAddress(metadata)).WriteTo(buf)
if err != nil {
return nil, err
}
stream, err := quicConn.OpenStream()
if err != nil {
return nil, err
}
_, err = buf.WriteTo(stream)
if err != nil {
return nil, err
}
if t.RequestTimeout > 0 {
_ = stream.SetReadDeadline(time.Now().Add(time.Duration(t.RequestTimeout) * time.Millisecond))
}