142 lines
3.7 KiB
Rust
142 lines
3.7 KiB
Rust
#![allow(clippy::unreadable_literal)]
|
|
#![cfg(target_pointer_width = "64")]
|
|
|
|
use nom::bytes::streaming::take;
|
|
#[cfg(feature = "alloc")]
|
|
use nom::multi::{length_data, many};
|
|
#[cfg(feature = "alloc")]
|
|
use nom::number::streaming::be_u64;
|
|
use nom::{Err, IResult, Needed, Parser};
|
|
|
|
// Parser definition
|
|
|
|
// We request a length that would trigger an overflow if computing consumed + requested
|
|
fn parser02(i: &[u8]) -> IResult<&[u8], (&[u8], &[u8])> {
|
|
(take(1_usize), take(18446744073709551615_usize)).parse(i)
|
|
}
|
|
|
|
#[test]
|
|
fn overflow_incomplete_tuple() {
|
|
assert_eq!(
|
|
parser02(&b"3"[..]),
|
|
Err(Err::Incomplete(Needed::new(18446744073709551615)))
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
#[cfg(feature = "alloc")]
|
|
fn overflow_incomplete_length_bytes() {
|
|
fn multi(i: &[u8]) -> IResult<&[u8], Vec<&[u8]>> {
|
|
many(0.., length_data(be_u64)).parse(i)
|
|
}
|
|
|
|
// Trigger an overflow in length_data
|
|
assert_eq!(
|
|
multi(&b"\x00\x00\x00\x00\x00\x00\x00\x01\xaa\xff\xff\xff\xff\xff\xff\xff\xff"[..]),
|
|
Err(Err::Incomplete(Needed::new(18446744073709551615)))
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
#[cfg(feature = "alloc")]
|
|
fn overflow_incomplete_many0() {
|
|
fn multi(i: &[u8]) -> IResult<&[u8], Vec<&[u8]>> {
|
|
many(0.., length_data(be_u64)).parse(i)
|
|
}
|
|
|
|
// Trigger an overflow in many0
|
|
assert_eq!(
|
|
multi(&b"\x00\x00\x00\x00\x00\x00\x00\x01\xaa\xff\xff\xff\xff\xff\xff\xff\xef"[..]),
|
|
Err(Err::Incomplete(Needed::new(18446744073709551599)))
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
#[cfg(feature = "alloc")]
|
|
fn overflow_incomplete_many1() {
|
|
fn multi(i: &[u8]) -> IResult<&[u8], Vec<&[u8]>> {
|
|
many(1.., length_data(be_u64)).parse(i)
|
|
}
|
|
|
|
// Trigger an overflow in many1
|
|
assert_eq!(
|
|
multi(&b"\x00\x00\x00\x00\x00\x00\x00\x01\xaa\xff\xff\xff\xff\xff\xff\xff\xef"[..]),
|
|
Err(Err::Incomplete(Needed::new(18446744073709551599)))
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
#[cfg(feature = "alloc")]
|
|
fn overflow_incomplete_many_till() {
|
|
use nom::{bytes::complete::tag, multi::many_till};
|
|
|
|
#[allow(clippy::type_complexity)]
|
|
fn multi(i: &[u8]) -> IResult<&[u8], (Vec<&[u8]>, &[u8])> {
|
|
many_till(length_data(be_u64), tag("abc")).parse(i)
|
|
}
|
|
|
|
// Trigger an overflow in many_till
|
|
assert_eq!(
|
|
multi(&b"\x00\x00\x00\x00\x00\x00\x00\x01\xaa\xff\xff\xff\xff\xff\xff\xff\xef"[..]),
|
|
Err(Err::Incomplete(Needed::new(18446744073709551599)))
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
#[cfg(feature = "alloc")]
|
|
fn overflow_incomplete_many_m_n() {
|
|
fn multi(i: &[u8]) -> IResult<&[u8], Vec<&[u8]>> {
|
|
many(2..=4, length_data(be_u64)).parse(i)
|
|
}
|
|
|
|
// Trigger an overflow in many_m_n
|
|
assert_eq!(
|
|
multi(&b"\x00\x00\x00\x00\x00\x00\x00\x01\xaa\xff\xff\xff\xff\xff\xff\xff\xef"[..]),
|
|
Err(Err::Incomplete(Needed::new(18446744073709551599)))
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
#[cfg(feature = "alloc")]
|
|
fn overflow_incomplete_count() {
|
|
use nom::multi::count;
|
|
|
|
fn counter(i: &[u8]) -> IResult<&[u8], Vec<&[u8]>> {
|
|
count(length_data(be_u64), 2).parse(i)
|
|
}
|
|
|
|
assert_eq!(
|
|
counter(&b"\x00\x00\x00\x00\x00\x00\x00\x01\xaa\xff\xff\xff\xff\xff\xff\xff\xef"[..]),
|
|
Err(Err::Incomplete(Needed::new(18446744073709551599)))
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
#[cfg(feature = "alloc")]
|
|
fn overflow_incomplete_length_count() {
|
|
use nom::multi::length_count;
|
|
use nom::number::streaming::be_u8;
|
|
|
|
fn multi(i: &[u8]) -> IResult<&[u8], Vec<&[u8]>> {
|
|
length_count(be_u8, length_data(be_u64)).parse(i)
|
|
}
|
|
|
|
assert_eq!(
|
|
multi(&b"\x04\x00\x00\x00\x00\x00\x00\x00\x01\xaa\xff\xff\xff\xff\xff\xff\xff\xee"[..]),
|
|
Err(Err::Incomplete(Needed::new(18446744073709551598)))
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
#[cfg(feature = "alloc")]
|
|
fn overflow_incomplete_length_data() {
|
|
fn multi(i: &[u8]) -> IResult<&[u8], Vec<&[u8]>> {
|
|
many(0.., length_data(be_u64)).parse(i)
|
|
}
|
|
|
|
assert_eq!(
|
|
multi(&b"\x00\x00\x00\x00\x00\x00\x00\x01\xaa\xff\xff\xff\xff\xff\xff\xff\xff"[..]),
|
|
Err(Err::Incomplete(Needed::new(18446744073709551615)))
|
|
);
|
|
}
|