#![no_std]
#![feature(test)]
extern crate threefish_cipher;

extern crate block_cipher_trait;
extern crate generic_array;
extern crate test;

use block_cipher_trait::BlockCipher;
use generic_array::GenericArray;
use test::Bencher;

#[bench]
pub fn encrypt_1_256(bh: &mut Bencher) {
    let key = Default::default();
    let state = threefish_cipher::Threefish256::new(&key);
    let mut input = [1u8; 32];
    let input = GenericArray::from_mut_slice(&mut input);
    bh.iter(|| state.encrypt_block(input));
    bh.bytes = 32u64;
}

#[bench]
pub fn encrypt_2_512(bh: &mut Bencher) {
    let key = Default::default();
    let state = threefish_cipher::Threefish512::new(&key);
    let mut input = [1u8; 64];
    let input = GenericArray::from_mut_slice(&mut input);
    bh.iter(|| state.encrypt_block(input));
    bh.bytes = 64u64;
}

#[bench]
pub fn encrypt_3_1024(bh: &mut Bencher) {
    let key = Default::default();
    let state = threefish_cipher::Threefish1024::new(&key);
    let mut input = [1u8; 128];
    let input = GenericArray::from_mut_slice(&mut input);
    bh.iter(|| state.encrypt_block(input));
    bh.bytes = 128u64;
}