// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at https://mozilla.org/MPL/2.0/.

use criterion::{black_box, criterion_group, criterion_main, Criterion};
use dominion_parser::DnsPacket;

use pprof::criterion::{Output, PProfProfiler};

const REQ: &[u8; 33] = include_bytes!("../assets/dns_request.bin");
const RES: &[u8; 49] = include_bytes!("../assets/dns_response.bin");

const LONG_REQ: &[u8; 270] = include_bytes!("../assets/dns_longreq.bin");

pub fn parse_long_request(c: &mut Criterion) {
    c.bench_function("parse_longreq", |b| {
        b.iter(|| DnsPacket::try_from(black_box(&LONG_REQ[..])).unwrap())
    });
}

pub fn parse_request(c: &mut Criterion) {
    c.bench_function("parse_req", |b| {
        b.iter(|| DnsPacket::try_from(black_box(&REQ[..])).unwrap())
    });
}

pub fn parse_response(c: &mut Criterion) {
    c.bench_function("parse_res", |b| {
        b.iter(|| DnsPacket::try_from(black_box(&RES[..])).unwrap())
    });
}

criterion_group!(
    name = parse;
    config = Criterion::default()
            .with_profiler(
                PProfProfiler::new(100, Output::Flamegraph(None))
            );
    targets = parse_request, parse_long_request, parse_response
);
criterion_main!(parse);