yume-post-smtp-proto/fuzz/fuzz_targets/smtp_proto.rs
2023-12-31 10:36:54 +01:00

67 lines
3 KiB
Rust

/*
* Copyright (c) 2020-2024, Stalwart Labs Ltd.
*
* Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
* https://www.apache.org/licenses/LICENSE-2.0> or the MIT license
* <LICENSE-MIT or https://opensource.org/licenses/MIT>, at your
* option. This file may not be copied, modified, or distributed
* except according to those terms.
*/
#![no_main]
use libfuzzer_sys::fuzz_target;
use smtp_proto::{
request::{
parser::Rfc5321Parser,
receiver::{
BdatReceiver, DataReceiver, DummyDataReceiver, DummyLineReceiver, LineReceiver,
RequestReceiver,
},
},
response::parser::ResponseReceiver,
EhloResponse, Request,
};
static RFC5321_ALPHABET: &[u8] = b"0123456789abcdefghijklmnopqrstuvwxyz:=-<>,; \r\n";
fuzz_target!(|data: &[u8]| {
let data_rfc5321 = into_alphabet(data, RFC5321_ALPHABET);
for bytes in [data, &data_rfc5321] {
let _ = Request::parse(&mut bytes.iter());
let _ = RequestReceiver::default().ingest(&mut bytes.iter(), &[]);
let _ = DataReceiver::new().ingest(&mut bytes.iter(), &mut vec![]);
let _ = BdatReceiver::new(bytes.len(), true).ingest(&mut bytes.iter(), &mut vec![]);
let _ = BdatReceiver::new(bytes.len(), false).ingest(&mut bytes.iter(), &mut vec![]);
let _ = DummyDataReceiver::new_bdat(bytes.len()).ingest(&mut bytes.iter());
let _ = DummyDataReceiver::new_data(&DataReceiver::new()).ingest(&mut bytes.iter());
let _ = LineReceiver::new(()).ingest(&mut bytes.iter());
let _ = DummyLineReceiver::default().ingest(&mut bytes.iter());
let _ = ResponseReceiver::default().parse(&mut bytes.iter());
let _ = EhloResponse::<String>::parse(&mut bytes.iter());
let _ = Rfc5321Parser::new(&mut bytes.iter()).hashed_value();
let _ = Rfc5321Parser::new(&mut bytes.iter()).hashed_value_long();
let _ = Rfc5321Parser::new(&mut bytes.iter()).address();
let _ = Rfc5321Parser::new(&mut bytes.iter()).string();
let _ = Rfc5321Parser::new(&mut bytes.iter()).text();
let _ = Rfc5321Parser::new(&mut bytes.iter()).xtext();
let _ = Rfc5321Parser::new(&mut bytes.iter()).seek_char(0);
let _ = Rfc5321Parser::new(&mut bytes.iter()).seek_lf();
let _ = Rfc5321Parser::new(&mut bytes.iter()).next_char();
let _ = Rfc5321Parser::new(&mut bytes.iter()).read_char();
let _ = Rfc5321Parser::new(&mut bytes.iter()).size();
let _ = Rfc5321Parser::new(&mut bytes.iter()).integer();
let _ = Rfc5321Parser::new(&mut bytes.iter()).timestamp();
let _ = Rfc5321Parser::new(&mut bytes.iter()).mail_from_parameters(String::new());
let _ = Rfc5321Parser::new(&mut bytes.iter()).rcpt_to_parameters(String::new());
let _ = Rfc5321Parser::new(&mut bytes.iter()).mechanism();
}
});
fn into_alphabet(data: &[u8], alphabet: &[u8]) -> Vec<u8> {
data.iter()
.map(|&byte| alphabet[byte as usize % alphabet.len()])
.collect()
}