qoi/fuzz/fuzz_targets/decode.rs

43 lines
1 KiB
Rust
Raw Normal View History

2021-12-01 00:05:46 +00:00
#![no_main]
use libfuzzer_sys::fuzz_target;
use qoi_fast::{decode_header, decode_to_vec, Channels, ColorSpace, Header};
2021-12-01 00:05:46 +00:00
fuzz_target!(|input: (u16, u16, bool, &[u8])| {
let (w, h, is_4, data) = input;
2022-01-03 09:02:27 +00:00
let (w, h) = (1 + w % 260, 1 + h % 260);
2021-12-01 00:05:46 +00:00
let channels = if is_4 { 4 } else { 3 };
let mut vec = vec![
b'q',
b'o',
b'i',
b'f',
0,
0,
(w >> 8) as u8,
(w & 0xff) as u8,
0,
0,
(h >> 8) as u8,
(h & 0xff) as u8,
channels,
0,
];
vec.extend(&*data);
2022-01-03 09:02:27 +00:00
vec.extend(&[0, 0, 0, 0, 0, 0, 0, 1]);
2021-12-01 00:05:46 +00:00
let header_expected = Header {
2022-01-03 09:02:27 +00:00
width: w as u32,
height: h as u32,
channels: Channels::try_from(channels).unwrap(),
colorspace: ColorSpace::try_from(0).unwrap(),
2022-01-03 09:02:27 +00:00
};
assert_eq!(decode_header(&vec).unwrap(), header_expected);
2022-01-03 09:02:27 +00:00
if let Ok((header, out)) = decode_to_vec(&vec) {
2022-01-03 09:02:27 +00:00
assert_eq!(header, header_expected);
assert_eq!(out.len(), header.n_bytes());
2021-12-01 00:05:46 +00:00
}
});