bug fix for #82

xtaci · Nov 25, 2020 · fa45f32 · fa45f32
1 parent f718802
commit fa45f32
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/stream.go b/stream.go
@@ -272,6 +272,12 @@ func (s *Stream) waitRead() error {
 	case <-s.chReadEvent:
 		return nil
 	case <-s.chFinEvent:
+		// BUG(xtaci): Fix for https://github.com/xtaci/smux/issues/82
+		s.bufferLock.Lock()
+		defer s.bufferLock.Unlock()
+		if len(s.buffers) > 0 {
+			return nil
+		}
 		return io.EOF
 	case <-s.sess.chSocketReadError:
 		return s.sess.socketReadError.Load().(error)