noq/fuzz/fuzz_targets/streams.rs

#![no_main]

use arbitrary::{Arbitrary, Unstructured};
use libfuzzer_sys::fuzz_target;

extern crate proto;
use proto::fuzzing::{ResetStream, Streams, TransportParameters};
use proto::{Dir, Side, StreamId, VarInt};

#[derive(Arbitrary, Debug)]
struct StreamParams {
    side: Side,
    max_remote_uni: u16,
    max_remote_bi: u16,
    send_window: u16,
    receive_window: u16,
    stream_receive_window: u16,
    dir: Dir,
    transport_params: TransportParameters
}

#[derive(Arbitrary, Debug)]
enum Operation {
    Open,
    Accept(Dir),
    Finish(StreamId),
    ReceivedStopSending(StreamId, VarInt),
    ReceivedReset(ResetStream),
    Reset(StreamId),
}

fuzz_target!(|input: (StreamParams, Vec<Operation>)| {
    let (params, operations) = input;
    let mut stream = Streams::new(
        params.side,
        params.max_remote_uni.into(),
        params.max_remote_bi.into(),
        params.send_window.into(),
        params.receive_window.into(),
        params.stream_receive_window.into(),
    );

    for operation in operations {
        match operation {
            Operation::Open => {
                stream.open(&params.transport_params, params.dir);
            }
            Operation::Accept(dir) => {
                stream.accept(dir);
            }
            Operation::Finish(id) => {
                stream.finish(id);
            }
            Operation::ReceivedStopSending(sid, err_code) => {
                stream.received_stop_sending(sid, err_code);
            }
            Operation::ReceivedReset(rs) => {
                stream.received_reset(rs);
            }
            Operation::Reset(id) => {
                stream.reset(id);
            }
        }
    }
});