Skip to content

Commit e81a438

Browse files
author
Brian Ollenberger
committed
Revert "Remove spurious Peek. A partial Peek can ruin the header read."
This reverts commit ea0c0f7.
1 parent 71ef340 commit e81a438

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

v2.go

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -108,6 +108,10 @@ func parseVersion2(reader *bufio.Reader) (header *Header, err error) {
108108
return header, nil
109109
}
110110

111+
if _, err := reader.Peek(int(length)); err != nil {
112+
return nil, fmt.Errorf("%w: %w", ErrInvalidLength, err)
113+
}
114+
111115
// Length-limited reader for payload section
112116
payloadReader := io.LimitReader(reader, int64(length)).(*io.LimitedReader)
113117

0 commit comments

Comments
 (0)