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

fix: h2 close panic

Browse source
This commit is contained in:
wwqgtxx 2023-04-17 23:42:15 +08:00
parent f100ce6a04
commit 47df97322d
1 changed files with 10 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
transport/vmess/h2.go
View file

@ -1,6 +1,7 @@
package vmess
import (
"context"
"io"
"net"
"net/http"
@ -84,10 +85,16 @@ func (hc *h2Conn) Write(b []byte) (int, error) {
}
func (hc *h2Conn) Close() error {
if err := hc.pwriter.Close(); err != nil {
return err
if hc.pwriter != nil {
if err := hc.pwriter.Close(); err != nil {
return err
}
}
if err := hc.ClientConn.Shutdown(hc.res.Request.Context()); err != nil {
ctx := context.Background()
if hc.res != nil {
ctx = hc.res.Request.Context()
}
if err := hc.ClientConn.Shutdown(ctx); err != nil {
return err
}
return hc.Conn.Close()