lzhr/v2fly
1
0
Fork 0
mirror of https://github.com/v2fly/v2ray-core.git synced 2026-04-29 17:19:07 -04:00
Code Issues Releases Wiki Activity

make sure writer not buffered

Browse Source
This commit is contained in:
Darien Raymond
2017-11-15 11:23:25 +01:00
parent 5a6022d1ed
commit f3c5df8798
2 changed files with 5 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
common/buf/writer.go
View File

@@ -121,6 +121,7 @@ func (w *BufferedWriter) SetBuffered(f bool) error {
return nil
}
// ReadFrom implements io.ReaderFrom.
func (w *BufferedWriter) ReadFrom(reader io.Reader) (int64, error) {
var sc SizeCounter
if !w.buffer.IsEmpty() {
@@ -130,11 +131,12 @@ func (w *BufferedWriter) ReadFrom(reader io.Reader) (int64, error) {
}
}
w.buffered = false
if readerFrom, ok := w.writer.(io.ReaderFrom); ok {
return readerFrom.ReadFrom(reader)
}
w.buffered = false
err := Copy(NewReader(reader), w, CountSize(&sc))
return sc.Size, err
}