Преглед на файлове

return correct length in ReadFrom

v2ray преди 9 години
родител
ревизия
2c82f65189
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      common/io/buffered_writer.go

+ 1 - 1
common/io/buffered_writer.go

@@ -33,13 +33,13 @@ func (this *BufferedWriter) ReadFrom(reader io.Reader) (int64, error) {
 	totalBytes := int64(0)
 	for {
 		nBytes, err := this.buffer.FillFrom(reader)
+		totalBytes += int64(nBytes)
 		if err != nil {
 			if err == io.EOF {
 				return totalBytes, nil
 			}
 			return totalBytes, err
 		}
-		totalBytes += int64(nBytes)
 		this.FlushWithoutLock()
 	}
 }