forked/h2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Augment the fuzzer to open multiple concurrent streams. (#328)

Browse Source 
This is how I found #319 and #320.
This commit is contained in:
Geoffry Song
2018-10-29 14:07:08 -07:00
committed by Carl Lerche
parent fc5efe73d6
commit 4321caf6b3
1 changed files with 13 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
tests/h2-fuzz/src/main.rs
View File

@@ -1,4 +1,3 @@
#[macro_use]
extern crate futures; extern crate futures;
extern crate tokio_io; extern crate tokio_io;
#[macro_use] #[macro_use]
@@ -14,6 +13,7 @@ use std::cell::Cell;
use std::io::{self, Read, Write}; use std::io::{self, Read, Write};
use std::sync::Arc; use std::sync::Arc;
use tokio_io::{AsyncRead, AsyncWrite}; use tokio_io::{AsyncRead, AsyncWrite};
use futures::stream::futures_unordered::FuturesUnordered;
struct MockIo<'a> { struct MockIo<'a> {
input: &'a [u8], input: &'a [u8],
@@ -114,13 +114,15 @@ fn run(script: &[u8]) -> Result<(), h2::Error> {
let notify_handle: executor::NotifyHandle = notify.clone().into(); let notify_handle: executor::NotifyHandle = notify.clone().into();
let io = MockIo { input: script }; let io = MockIo { input: script };
let (mut h2, mut connection) = h2::client::handshake(io).wait()?; let (mut h2, mut connection) = h2::client::handshake(io).wait()?;
let mut in_progress = None; let mut futs = FuturesUnordered::new();
let future = future::poll_fn(|| { let future = future::poll_fn(|| {
if let Async::Ready(()) = connection.poll()? { if let Async::Ready(()) = connection.poll()? {
return Ok(Async::Ready(())); return Ok(Async::Ready(()));
} }
if in_progress.is_none() { while futs.len() < 128 {
try_ready!(h2.poll_ready()); if h2.poll_ready()?.is_not_ready() {
break;
}
let request = Request::builder() let request = Request::builder()
.method(Method::POST) .method(Method::POST)
.uri("https://example.com/") .uri("https://example.com/")
@@ -129,14 +131,15 @@ fn run(script: &[u8]) -> Result<(), h2::Error> {
let (resp, mut send) = h2.send_request(request, false)?; let (resp, mut send) = h2.send_request(request, false)?;
send.send_data(vec![0u8; 32769].into(), true).unwrap(); send.send_data(vec![0u8; 32769].into(), true).unwrap();
drop(send); drop(send);
in_progress = Some(resp); futs.push(resp);
} }
match in_progress.as_mut().unwrap().poll() { loop {
match futs.poll() {
Ok(Async::NotReady) | Ok(Async::Ready(None)) => break,
r @ Ok(Async::Ready(_)) | r @ Err(_) => { r @ Ok(Async::Ready(_)) | r @ Err(_) => {
eprintln!("{:?}", r); eprintln!("{:?}", r);
in_progress = None;
} }
Ok(Async::NotReady) => (), }
} }
Ok::<_, h2::Error>(Async::NotReady) Ok::<_, h2::Error>(Async::NotReady)
}); });