rlp: fix integer overflow in list element size validation

It is not safe to add anything to s.size.
This commit is contained in:
Felix Lange
2015-04-14 00:54:12 +02:00
parent 56a48101dc
commit 2750ec47b7
2 changed files with 5 additions and 2 deletions

View File

@ -751,7 +751,7 @@ func (s *Stream) Kind() (kind Kind, size uint64, err error) {
tos = &s.stack[len(s.stack)-1]
}
if s.kind < 0 {
// don't read further if we're at the end of the
// Don't read further if we're at the end of the
// innermost list.
if tos != nil && tos.pos == tos.size {
return 0, 0, EOL
@ -772,7 +772,7 @@ func (s *Stream) Kind() (kind Kind, size uint64, err error) {
}
} else {
// Inside a list, check that the value doesn't overflow the list.
if tos.pos+s.size > tos.size {
if s.size > tos.size-tos.pos {
return 0, 0, ErrElemTooLarge
}
}