tokio-rs/tokio
1
0
Fork 0
mirror of https://github.com/tokio-rs/tokio.git synced 2026-09-07 00:00:08 +02:00
Code Issues Packages Projects Releases Wiki Activity

stream: update StreamMap fuzz test (#5600)

Browse Source 
This fuzz test touches some quadratic time code paths. This changes the
test to limit the test size so that the test doesn't time out.
This commit is contained in:
Alice Ryhl
2023-04-11 10:40:10 +02:00
committed by GitHub
parent b02c550c52
commit 3b45e8614d
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tokio-stream/fuzz/fuzz_targets/fuzz_stream_map.rs
+4 -3
View File
@@ -28,7 +28,7 @@ fn pin_box<T: Stream<Item = U> + 'static, U>(s: T) -> Pin<Box<dyn Stream<Item =
Box::pin(s) Box::pin(s)
} }
fuzz_target!(|data: &[u8]| { fuzz_target!(|data: [bool; 64]| {
use std::task::{Context, Poll}; use std::task::{Context, Poll};
struct DidPoll<T> { struct DidPoll<T> {
@@ -45,11 +45,12 @@ fuzz_target!(|data: &[u8]| {
} }
} }
for _ in 0..10 { // Try the test with each possible length.
for len in 0..data.len() {
let mut map = task::spawn(StreamMap::new()); let mut map = task::spawn(StreamMap::new());
let mut expect = 0; let mut expect = 0;
for (i, is_empty) in data.iter().map(|x| *x != 0).enumerate() { for (i, is_empty) in data[..len].iter().copied().enumerate() {
let inner = if is_empty { let inner = if is_empty {
pin_box(stream::empty::<()>()) pin_box(stream::empty::<()>())
} else { } else {