prevent sending more data in passive closing

xtaci · Aug 30, 2024 · d245826 · d245826
1 parent a9a927f
commit d245826
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/stream.go b/stream.go
@@ -346,6 +346,8 @@ func (s *Stream) Write(b []byte) (n int, err error) {
 
 	// check if stream has closed
 	select {
+	case <-s.chFinEvent: // passive closing
+		return 0, io.EOF
 	case <-s.die:
 		return 0, io.ErrClosedPipe
 	default:
@@ -382,6 +384,8 @@ func (s *Stream) writeV2(b []byte) (n int, err error) {
 
 	// check if stream has closed
 	select {
+	case <-s.chFinEvent:
+		return 0, io.EOF
 	case <-s.die:
 		return 0, io.ErrClosedPipe
 	default: