diff options
Diffstat (limited to 'src/encoding/binary')
| -rw-r--r-- | src/encoding/binary/varint.go | 9 | ||||
| -rw-r--r-- | src/encoding/binary/varint_test.go | 6 |
2 files changed, 14 insertions, 1 deletions
diff --git a/src/encoding/binary/varint.go b/src/encoding/binary/varint.go index c807d15f44..18e1ff1511 100644 --- a/src/encoding/binary/varint.go +++ b/src/encoding/binary/varint.go @@ -126,12 +126,18 @@ func Varint(buf []byte) (int64, int) { var overflow = errors.New("binary: varint overflows a 64-bit integer") // ReadUvarint reads an encoded unsigned integer from r and returns it as a uint64. +// The error is EOF only if no bytes were read. +// If an EOF happens after reading some but not all the bytes, +// ReadUvarint returns io.ErrUnexpectedEOF. func ReadUvarint(r io.ByteReader) (uint64, error) { var x uint64 var s uint for i := 0; i < MaxVarintLen64; i++ { b, err := r.ReadByte() if err != nil { + if i > 0 && err == io.EOF { + err = io.ErrUnexpectedEOF + } return x, err } if b < 0x80 { @@ -147,6 +153,9 @@ func ReadUvarint(r io.ByteReader) (uint64, error) { } // ReadVarint reads an encoded signed integer from r and returns it as an int64. +// The error is EOF only if no bytes were read. +// If an EOF happens after reading some but not all the bytes, +// ReadVarint returns io.ErrUnexpectedEOF. func ReadVarint(r io.ByteReader) (int64, error) { ux, err := ReadUvarint(r) // ok to continue in presence of error x := int64(ux >> 1) diff --git a/src/encoding/binary/varint_test.go b/src/encoding/binary/varint_test.go index 080a2148f0..a3caea8a43 100644 --- a/src/encoding/binary/varint_test.go +++ b/src/encoding/binary/varint_test.go @@ -128,7 +128,11 @@ func TestBufferTooSmall(t *testing.T) { } x, err := ReadUvarint(bytes.NewReader(buf)) - if x != 0 || err != io.EOF { + wantErr := io.EOF + if i > 0 { + wantErr = io.ErrUnexpectedEOF + } + if x != 0 || err != wantErr { t.Errorf("ReadUvarint(%v): got x = %d, err = %s", buf, x, err) } } |
