image/tests/truncate_images.rs

//! Ensure truncated images are read without panics.

use std::fs;
use std::io::Read;
use std::path::PathBuf;

extern crate glob;
extern crate image;

const BASE_PATH: [&str; 2] = [".", "tests"];
const IMAGE_DIR: &str = "images";

fn process_images<F>(dir: &str, input_decoder: Option<&str>, func: F)
where
    F: Fn(PathBuf),
{
    let base: PathBuf = BASE_PATH.iter().collect();
    let decoders = &[
        "tga", "tiff", "png", "gif", "bmp", "ico", "jpg", "hdr", "farbfeld", "exr",
    ];
    for decoder in decoders {
        let mut path = base.clone();
        path.push(dir);
        path.push(decoder);
        path.push("*");
        path.push(
            "*.".to_string()
                + match input_decoder {
                    Some(val) => val,
                    None => decoder,
                },
        );
        let pattern = &*format!("{}", path.display());
        for path in glob::glob(pattern).unwrap().filter_map(Result::ok) {
            func(path)
        }
    }
}

fn truncate_images(decoder: &str) {
    process_images(IMAGE_DIR, Some(decoder), |path| {
        println!("{:?}", path);
        let fin = fs::File::open(&path).unwrap();
        let max_length = 1000;
        let mut buf = Vec::with_capacity(max_length);
        fin.take(max_length as u64).read_to_end(&mut buf).unwrap();
        for i in (0..buf.len()).step_by(37) {
            image::load_from_memory(&buf[..i + 1]).ok();
        }
    })
}

#[test]
fn truncate_tga() {
    truncate_images("tga")
}

#[test]
fn truncate_tiff() {
    truncate_images("tiff")
}

#[test]
fn truncate_png() {
    truncate_images("png")
}

#[test]
fn truncate_gif() {
    truncate_images("gif")
}

#[test]
fn truncate_bmp() {
    truncate_images("bmp")
}

#[test]
fn truncate_ico() {
    truncate_images("ico")
}

#[test]
fn truncate_jpg() {
    truncate_images("jpg")
}

#[test]
fn truncate_hdr() {
    truncate_images("hdr");
}

#[test]
fn truncate_farbfeld() {
    truncate_images("farbfeld");
}

#[test]
fn truncate_exr() {
    truncate_images("exr");
}