Add recover() to H2 server's flushWriter.Write()

Fixes https://github.com/XTLS/Xray-core/issues/1748
This commit is contained in:
RPRX 2023-03-08 14:06:20 +00:00 committed by GitHub
parent 4a0b45d1ff
commit 836e84b851
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 10 additions and 5 deletions

View file

@ -204,7 +204,7 @@ type WaitReadCloser struct {
func (w *WaitReadCloser) Set(rc io.ReadCloser) {
w.ReadCloser = rc
defer func() {
if err := recover(); err != nil {
if recover() != nil {
rc.Close()
}
}()
@ -225,10 +225,8 @@ func (w *WaitReadCloser) Close() error {
return w.ReadCloser.Close()
}
defer func() {
if err := recover(); err != nil {
if w.ReadCloser != nil {
w.ReadCloser.Close()
}
if recover() != nil && w.ReadCloser != nil {
w.ReadCloser.Close()
}
}()
close(w.Wait)