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
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
// copyright 2017 Kaz Wesley

//! Portable JH with optimizations for x86-64

#![cfg_attr(not(feature = "std"), no_std)]

extern crate crypto_simd;
pub extern crate digest;
#[macro_use]
extern crate hex_literal;
#[cfg(all(feature = "std", feature = "simd"))]
#[macro_use]
extern crate lazy_static;
#[cfg(feature = "packed_simd")]
extern crate packed_simd_crate;
#[cfg(not(any(feature = "simd", feature = "packed_simd")))]
extern crate ppv_null;
#[cfg(all(feature = "simd", not(feature = "packed_simd")))]
extern crate simd;

mod compressor;
mod consts;

pub use digest::Digest;

use crate::compressor::Compressor;
use block_buffer::byteorder::{BigEndian, ByteOrder};
use block_buffer::generic_array::GenericArray as BBGenericArray;
use block_buffer::BlockBuffer;
use core::fmt::{Debug, Formatter, Result};
use digest::generic_array::typenum::{Unsigned, U28, U32, U48, U64};
use digest::generic_array::GenericArray as DGenericArray;

macro_rules! define_hasher {
    ($name:ident, $init:path, $OutputBytes:ident) => {
        #[derive(Clone)]
        pub struct $name {
            state: Compressor,
            buffer: BlockBuffer<U64>,
            datalen: usize,
        }

        impl Debug for $name {
            fn fmt(&self, f: &mut Formatter) -> Result {
                f.debug_struct("Jh")
                    .field("state", &"(state)")
                    .field("buffer", &"(BlockBuffer<U64>)")
                    .field("datalen", &self.datalen)
                    .finish()
            }
        }

        impl Default for $name {
            fn default() -> Self {
                Self {
                    state: Compressor::new($init),
                    buffer: BlockBuffer::default(),
                    datalen: 0,
                }
            }
        }

        impl digest::BlockInput for $name {
            type BlockSize = U64;
        }

        impl digest::Input for $name {
            fn input<T: AsRef<[u8]>>(&mut self, data: T) {
                let data = data.as_ref();
                self.datalen += data.len();
                let state = &mut self.state;
                self.buffer.input(data, |b| state.input(b))
            }
        }

        impl digest::FixedOutput for $name {
            type OutputSize = $OutputBytes;

            fn fixed_result(mut self) -> DGenericArray<u8, $OutputBytes> {
                let state = &mut self.state;
                let buffer = &mut self.buffer;
                let len = self.datalen as u64 * 8;
                if buffer.position() == 0 {
                    buffer.len64_padding::<BigEndian, _>(len, |b| state.input(b));
                } else {
                    use block_buffer::block_padding::Iso7816;
                    state.input(buffer.pad_with::<Iso7816>().unwrap());
                    let mut last = BBGenericArray::default();
                    BigEndian::write_u64(&mut last[56..], len);
                    state.input(&last);
                }
                let finalized = self.state.finalize();
                DGenericArray::clone_from_slice(&finalized[(128 - $OutputBytes::to_usize())..])
            }
        }

        impl digest::Reset for $name {
            fn reset(&mut self) {
                *self = Self::default();
            }
        }
    };
}

define_hasher!(Jh224, consts::JH224_H0, U28);
define_hasher!(Jh256, consts::JH256_H0, U32);
define_hasher!(Jh384, consts::JH384_H0, U48);
define_hasher!(Jh512, consts::JH512_H0, U64);