Skip to content

Commit

Permalink
test(server): add fuzzing harness for conn
Browse files Browse the repository at this point in the history
  • Loading branch information
5225225 committed Dec 10, 2021
1 parent 84b78b6 commit 37d2909
Show file tree
Hide file tree
Showing 3 changed files with 96 additions and 0 deletions.
4 changes: 4 additions & 0 deletions fuzz/.gitignore
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@

target
corpus
artifacts
29 changes: 29 additions & 0 deletions fuzz/Cargo.toml
Original file line number Diff line number Diff line change
@@ -0,0 +1,29 @@

[package]
name = "hyper-fuzz"
version = "0.0.0"
authors = ["Automatically generated"]
publish = false
edition = "2018"

[package.metadata]
cargo-fuzz = true

[dependencies]
libfuzzer-sys = "0.4"
tokio = "1.14.0"
async-io = "1.6.0"

[dependencies.hyper]
path = ".."
features = ["full"]

# Prevent this from interfering with workspaces
[workspace]
members = ["."]

[[bin]]
name = "fuzz_request"
path = "fuzz_targets/fuzz_request.rs"
test = false
doc = false
63 changes: 63 additions & 0 deletions fuzz/fuzz_targets/fuzz_request.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,63 @@
#![feature(never_type)]
#![no_main]
use libfuzzer_sys::fuzz_target;

use std::pin::Pin;

use std::task::{Context, Poll};

use hyper::{Body, Request, Response};

struct WriteVoidReadData(std::io::Cursor<Vec<u8>>);

impl tokio::io::AsyncRead for WriteVoidReadData {
fn poll_read(
mut self: core::pin::Pin<&mut Self>,
cx: &mut Context<'_>,
buf: &mut tokio::io::ReadBuf<'_>,
) -> Poll<std::io::Result<()>> {
tokio::io::AsyncRead::poll_read(Pin::new(&mut self.0), cx, buf)
}
}

impl tokio::io::AsyncWrite for WriteVoidReadData {
fn poll_write(
self: Pin<&mut Self>,
_cx: &mut Context<'_>,
buf: &[u8],
) -> Poll<Result<usize, std::io::Error>> {
Poll::Ready(Ok(buf.len()))
}
fn poll_flush(self: Pin<&mut Self>, _cx: &mut Context<'_>) -> Poll<Result<(), std::io::Error>> {
Poll::Ready(Ok(()))
}

fn poll_shutdown(
self: Pin<&mut Self>,
_cx: &mut Context<'_>,
) -> Poll<Result<(), std::io::Error>> {
Poll::Ready(Ok(()))
}
}

struct VoidService;

impl hyper::service::Service<Request<Body>> for VoidService {
type Response = Response<hyper::body::Body>;
type Error = !;
type Future = core::future::Ready<Result<Self::Response, !>>;

fn poll_ready(&mut self, _cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> {
Poll::Ready(Ok(()))
}

fn call(&mut self, _req: Request<Body>) -> Self::Future {
core::future::ready(Ok(Response::new(hyper::body::Body::empty())))
}
}

fuzz_target!(|data: Vec<u8>| {
let s = hyper::server::conn::Http::new();
let svc = s.serve_connection(WriteVoidReadData(std::io::Cursor::new(data)), VoidService);
drop(async_io::block_on(svc));
});

0 comments on commit 37d2909

Please sign in to comment.