tuxmain/qoi
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Update fuzz targets

Browse source
This commit is contained in:
Ivan Smirnov 2022-01-03 12:02:27 +03:00
parent 4e43c8ba01
commit e6e156f974
2 changed files with 15 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
fuzz/fuzz_targets/qoi_decode.rs
View file

@ -3,7 +3,7 @@ use libfuzzer_sys::fuzz_target;
fuzz_target!(|input: (u16, u16, bool, &[u8])| { fuzz_target!(|input: (u16, u16, bool, &[u8])| {
let (w, h, is_4, data) = input; let (w, h, is_4, data) = input;
let (w, h) = (w % 300, h % 300); let (w, h) = (1 + w % 260, 1 + h % 260);
let channels = if is_4 { 4 } else { 3 }; let channels = if is_4 { 4 } else { 3 };
let mut vec = vec![ let mut vec = vec![
@ -23,15 +23,18 @@ fuzz_target!(|input: (u16, u16, bool, &[u8])| {
0, 0,
]; ];
vec.extend(&*data); vec.extend(&*data);
vec.extend(&[0, 0, 0, 0]); vec.extend(&[0, 0, 0, 0, 0, 0, 0, 1]);
let out = qoi_fast::qoi_decode_to_vec(&vec, channels); let header_expected = qoi_fast::Header {
if let Ok((header, out)) = out { width: w as u32,
assert_eq!(header.magic, qoi_fast::consts::QOI_MAGIC); height: h as u32,
assert_eq!(header.width, w as u32); channels: qoi_fast::Channels::try_from(channels).unwrap(),
assert_eq!(header.height, h as u32); colorspace: qoi_fast::ColorSpace::try_from(0).unwrap(),
assert_eq!(header.channels, channels); };
assert_eq!(header.colorspace.to_u8(), 0); assert_eq!(qoi_fast::qoi_decode_header(&vec).unwrap(), header_expected);
assert_eq!(out.len(), w as usize * h as usize * channels as usize);
if let Ok((header, out)) = qoi_fast::qoi_decode_to_vec(&vec) {
assert_eq!(header, header_expected);
assert_eq!(out.len(), header.n_bytes());
} }
}); });

10
fuzz/fuzz_targets/qoi_encode.rs
View file

@ -13,16 +13,10 @@ fuzz_target!(|input: (bool, u8, &[u8])| {
let h = n_pixels / w; let h = n_pixels / w;
(w, h) (w, h)
}; };
let out = qoi_fast::qoi_encode_to_vec( let out = qoi_fast::qoi_encode_to_vec(&data[..(w * h * channels as usize)], w as u32, h as u32);
&data[..(w * h * channels as usize)],
w as u32,
h as u32,
channels,
0,
);
if w * h != 0 { if w * h != 0 {
let out = out.unwrap(); let out = out.unwrap();
assert!(out.len() <= qoi_fast::encode_size_required(w as u32, h as u32, channels)); assert!(out.len() <= qoi_fast::encoded_size_limit(w as u32, h as u32, channels));
} else { } else {
assert!(out.is_err()); assert!(out.is_err());
} }