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

Fix h2 conn race

Browse source
This commit is contained in:
世界 2023-05-17 21:46:06 +08:00
parent 9836fc9b05
commit 1ebe6bb266
No known key found for this signature in database
GPG key ID: CD109927C34A63C4
1 changed files with 10 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

18
h2mux.go
View file

@ -106,8 +106,9 @@ func (s *h2MuxServerSession) CanTakeNewRequest() bool {
type h2MuxConnWrapper struct {
N.ExtendedConn
flusher http.Flusher
done chan struct{}
access sync.Mutex
closed bool
done chan struct{}
}
func newHTTP2Wrapper(conn net.Conn, flusher http.Flusher) *h2MuxConnWrapper {
@ -119,10 +120,10 @@ func newHTTP2Wrapper(conn net.Conn, flusher http.Flusher) *h2MuxConnWrapper {
}
func (w *h2MuxConnWrapper) Write(p []byte) (n int, err error) {
select {
case <-w.done:
w.access.Lock()
defer w.access.Unlock()
if w.closed {
return 0, net.ErrClosed
default:
}
n, err = w.ExtendedConn.Write(p)
if err == nil {
@ -132,10 +133,10 @@ func (w *h2MuxConnWrapper) Write(p []byte) (n int, err error) {
}
func (w *h2MuxConnWrapper) WriteBuffer(buffer *buf.Buffer) error {
select {
case <-w.done:
w.access.Lock()
defer w.access.Unlock()
if w.closed {
return net.ErrClosed
default:
}
err := w.ExtendedConn.WriteBuffer(buffer)
if err == nil {
@ -146,12 +147,13 @@ func (w *h2MuxConnWrapper) WriteBuffer(buffer *buf.Buffer) error {
func (w *h2MuxConnWrapper) Close() error {
w.access.Lock()
defer w.access.Unlock()
select {
case <-w.done:
default:
close(w.done)
}
w.closed = true
w.access.Unlock()
return w.ExtendedConn.Close()
}