add check to only read stream maximum

cfi2017 · Apr 18, 2020 · c11977a · c11977a
1 parent c274f1d
commit c11977a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/internal/item/bits.go b/internal/item/bits.go
@@ -12,6 +12,9 @@ type Reader struct {
 }
 
 func (r *Reader) ReadInt(n int) (uint64, error) {
+	if len(r.stream)-n < 0 {
+		n = len(r.stream)
+	}
 	val, err := strconv.ParseUint(r.stream[len(r.stream)-n:], 2, 64)
 	r.stream = r.stream[:len(r.stream)-n]
 	return val, err