kanidm/server/lib/benches/image_benches.rs

/// This file contains benchmarks for the image module so we can work out the best order to run things in
use std::time::Duration;

use criterion::{black_box, criterion_group, criterion_main, Criterion};
use kanidmd_lib::valueset::image::jpg;
use kanidmd_lib::valueset::image::png;

pub fn bench_png_lodepng_validate(c: &mut Criterion) {
    let mut group = c.benchmark_group("png_lodepng_validate");
    group.bench_function("png_lodepng_validate_oversize", |b| {
        let filename = black_box(format!(
            "{}/src/valueset/image/test_images/oversize_dimensions.png",
            env!("CARGO_MANIFEST_DIR")
        ));
        let contents = black_box(std::fs::read(filename).unwrap());
        b.iter(|| png::png_lodepng_validate(&contents, black_box("oversize_dimensions.png")))
    });
    group.bench_function("png_lodepng_validate_ok", |b| {
        let filename = black_box(format!(
            "{}/src/valueset/image/test_images/oversize_dimensions.png",
            env!("CARGO_MANIFEST_DIR")
        ));
        let contents = black_box(std::fs::read(filename).unwrap());
        b.iter(|| png::png_lodepng_validate(&contents, black_box("oversize_dimensions.png")))
    });
    group.finish();
}

pub fn bench_png_has_trailer(c: &mut Criterion) {
    let mut group = c.benchmark_group("png_has_trailer");
    group.bench_function("png_has_trailer_oversize", |b| {
        let filename = black_box(format!(
            "{}/src/valueset/image/test_images/oversize_dimensions.png",
            env!("CARGO_MANIFEST_DIR")
        ));
        let contents = black_box(std::fs::read(filename).unwrap());
        b.iter(|| png::png_has_trailer(&contents));
    });
    group.bench_function("png_has_trailer_ok", |b| {
        let filename = black_box(format!(
            "{}/src/valueset/image/test_images/ok.png",
            env!("CARGO_MANIFEST_DIR")
        ));
        let contents = black_box(std::fs::read(filename).unwrap());
        b.iter(|| png::png_has_trailer(&contents));
    });
    group.finish();
}

pub fn bench_jpg(c: &mut Criterion) {
    let mut group = c.benchmark_group("jpg");
    group.bench_function("check_jpg_header", |b| {
        let filename = black_box(format!(
            "{}/src/valueset/image/test_images/oversize_dimensions.jpg",
            env!("CARGO_MANIFEST_DIR")
        ));
        let contents = black_box(std::fs::read(filename).unwrap());
        b.iter(|| jpg::check_jpg_header(&contents));
    });
    group.bench_function("has_trailer", |b| {
        let filename = black_box(format!(
            "{}/src/valueset/image/test_images/oversize_dimensions.jpg",
            env!("CARGO_MANIFEST_DIR")
        ));
        let contents = black_box(std::fs::read(filename).unwrap());
        b.iter(|| jpg::has_trailer(&contents));
    });
    group.bench_function("use_decoder", |b| {
        let filename = black_box(format!(
            "{}/src/valueset/image/test_images/oversize_dimensions.jpg",
            env!("CARGO_MANIFEST_DIR")
        ));
        let contents = black_box(std::fs::read(&filename).unwrap());
        b.iter(|| jpg::validate_decoding(&filename, &contents, image::io::Limits::default()));
    });
    group.finish();
}

pub fn compare_jpg(c: &mut Criterion) {
    let mut group = c.benchmark_group("compare_jpg");
    group.bench_function("header, trailer, decoder", |b| {
        let filename = black_box(format!(
            "{}/src/valueset/image/test_images/oversize_dimensions.jpg",
            env!("CARGO_MANIFEST_DIR")
        ));
        let contents = black_box(std::fs::read(&filename).unwrap());
        b.iter(|| {
            assert!(jpg::check_jpg_header(&contents).is_ok());
            assert!(jpg::has_trailer(&contents).is_ok());
            assert!(
                jpg::validate_decoding(&filename, &contents, image::io::Limits::default()).is_ok()
            );
        });
    });
    group.bench_function("trailer, header, decoder", |b| {
        let filename = black_box(format!(
            "{}/src/valueset/image/test_images/oversize_dimensions.jpg",
            env!("CARGO_MANIFEST_DIR")
        ));
        let contents = black_box(std::fs::read(&filename).unwrap());
        b.iter(|| {
            assert!(jpg::has_trailer(&contents).is_ok());
            assert!(jpg::check_jpg_header(&contents).is_ok());
            assert!(
                jpg::validate_decoding(&filename, &contents, image::io::Limits::default()).is_ok()
            );
        });
    });
    group.bench_function("decoder, trailer, header", |b| {
        let filename = black_box(format!(
            "{}/src/valueset/image/test_images/oversize_dimensions.jpg",
            env!("CARGO_MANIFEST_DIR")
        ));
        let contents = black_box(std::fs::read(&filename).unwrap());
        b.iter(|| {
            assert!(
                jpg::validate_decoding(&filename, &contents, image::io::Limits::default()).is_ok()
            );
            assert!(jpg::has_trailer(&contents).is_ok());
            assert!(jpg::check_jpg_header(&contents).is_ok());
        });
    });

    group.finish();
}

criterion_group!(
    name = png_tests;
    config = Criterion::default()
        .measurement_time(Duration::from_secs(15))
        .with_plots();
    targets = bench_png_lodepng_validate, bench_png_has_trailer, bench_jpg, compare_jpg
);
criterion_main!(png_tests);