Fix trivial DoS when deserializing messages from the network

This commit is contained in:
Matt Corallo 2019-05-20 20:16:18 -04:00
parent 08c756d20e
commit 98796576d2
1 changed files with 6 additions and 0 deletions

View File

@ -659,6 +659,12 @@ impl<D: Decoder> Decodable<D> for CheckedData {
#[inline]
fn consensus_decode(d: &mut D) -> Result<CheckedData, self::Error> {
let len: u32 = Decodable::consensus_decode(d)?;
if len > MAX_VEC_SIZE as u32 {
return Err(self::Error::OversizedVectorAllocation {
requested: len as usize,
max: MAX_VEC_SIZE
});
}
let checksum: [u8; 4] = Decodable::consensus_decode(d)?;
let mut ret = Vec::with_capacity(len as usize);
ret.resize(len as usize, 0);