vendor/tiff/tests/fuzz_tests.rs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

extern crate tiff;

use tiff::decoder::Decoder;
use tiff::TiffResult;

use std::fs::File;

fn test_directory<F: Fn(File) -> bool>(path: &str, f: F) {
    for entry in std::fs::read_dir(path).unwrap() {
        let file = File::open(entry.unwrap().path()).unwrap();
        assert!(f(file));
    }
}

fn decode_tiff(file: File) -> TiffResult<()> {
    let mut decoder = Decoder::new(file)?;
    decoder.read_image()?;
    Ok(())
}

#[test]
fn oor_panic() {
    test_directory("./tests/fuzz_images/oor_panic", |file| {
        let _ = decode_tiff(file);
        true
    });
}

#[test]
fn oom_crash() {
    test_directory("./tests/fuzz_images/oom_crash", |file| {
        decode_tiff(file).is_err()
    });
}

#[test]
fn inf_loop() {
    test_directory("./tests/fuzz_images/inf_loop", |file| {
        let _ = decode_tiff(file);
        true
    });
}

// https://github.com/image-rs/image-tiff/issues/33
#[test]
fn divide_by_zero() {
    test_directory("./tests/fuzz_images/divide_by_zero", |file| {
        let _ = decode_tiff(file);
        true
    });
}