kirillius
/
Cloak
mirror of https://github.com/cbeuw/Cloak
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Make WriteTo return the correct errors

This commit is contained in:
Andy Wang 2020-04-12 16:34:49 +01:00
parent f0e8b4556e
commit 248389b3f6
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
internal/multiplex/stream.go
View File

@ -83,7 +83,12 @@ func (s *Stream) Read(buf []byte) (n int, err error) {
func (s *Stream) WriteTo(w io.Writer) (int64, error) { func (s *Stream) WriteTo(w io.Writer) (int64, error) {
// will keep writing until the underlying buffer is closed // will keep writing until the underlying buffer is closed
return s.recvBuf.WriteTo(w) n, err := s.recvBuf.WriteTo(w)
if err == io.EOF {
return n, ErrBrokenStream
}
log.Tracef("%v read from stream %v with err %v", n, s.id, err)
return n, nil
} }
// Write implements io.Write // Write implements io.Write