hyperium / http

Rust HTTP types
Apache License 2.0
1.12k stars 283 forks source link

Arithmetic overflow found on `with_capacity()` #627

Open HeeillWang opened 9 months ago

HeeillWang commented 9 months ago

I executed fuzz testing on http-0.2.9, and found some arithmetic overflow. Please note that overflow condition is different with #626 .

    pub fn with_capacity(capacity: usize) -> HeaderMap<T> {
        if capacity == 0 {
            HeaderMap {
                mask: 0,
                indices: Box::new([]), // as a ZST, this doesn't actually allocate anything
                entries: Vec::new(),
                extra_values: Vec::new(),
                danger: Danger::Green,
            }
        } else {
            let raw_cap = to_raw_capacity(capacity).next_power_of_two();   // overflow!
            assert!(raw_cap <= MAX_SIZE, "requested capacity too large");
            debug_assert!(raw_cap > 0);

            HeaderMap {
                mask: (raw_cap - 1) as Size,
                indices: vec![Pos::none(); raw_cap].into_boxed_slice(),
                entries: Vec::with_capacity(raw_cap),
                extra_values: Vec::new(),
                danger: Danger::Green,
            }
        }
    }

reproduce with :

HeaderMap::<u32>::with_capacity(12538021362599493900);  // put some big number here

If you input TOO big number on with_capacity(), #626 occurs before reaching to next_power_of_two().

hawkw commented 9 months ago

It seems like there should be an assertion prior to the next_power_of_two call, in addition to the one after it?

HeeillWang commented 9 months ago

That would resolve #626 as well. Plus, consider to use checked_next_power_of_two

seanmonstar commented 9 months ago

Thanks for detecting those! Would you like to submit a PR that uses an assertion or checked math?

HeeillWang commented 9 months ago

submitted on #628